| Bram Moolenaar | 3c2881d | 2017-03-21 19:18:29 +0100 | [diff] [blame] | 1 | " Author: Stephen Sugden <stephen@stephensugden.com> |
| 2 | " |
| 3 | " Adapted from https://github.com/fatih/vim-go |
| 4 | " For bugs, patches and license go to https://github.com/rust-lang/rust.vim |
| 5 | |
| 6 | if !exists("g:rustfmt_autosave") |
| 7 | let g:rustfmt_autosave = 0 |
| 8 | endif |
| 9 | |
| 10 | if !exists("g:rustfmt_command") |
| 11 | let g:rustfmt_command = "rustfmt" |
| 12 | endif |
| 13 | |
| 14 | if !exists("g:rustfmt_options") |
| 15 | let g:rustfmt_options = "" |
| 16 | endif |
| 17 | |
| 18 | if !exists("g:rustfmt_fail_silently") |
| 19 | let g:rustfmt_fail_silently = 0 |
| 20 | endif |
| 21 | |
| 22 | let s:got_fmt_error = 0 |
| 23 | |
| 24 | function! s:RustfmtCommandRange(filename, line1, line2) |
| 25 | let l:arg = {"file": shellescape(a:filename), "range": [a:line1, a:line2]} |
| 26 | return printf("%s %s --write-mode=overwrite --file-lines '[%s]'", g:rustfmt_command, g:rustfmt_options, json_encode(l:arg)) |
| 27 | endfunction |
| 28 | |
| 29 | function! s:RustfmtCommand(filename) |
| 30 | return g:rustfmt_command . " --write-mode=overwrite " . g:rustfmt_options . " " . shellescape(a:filename) |
| 31 | endfunction |
| 32 | |
| 33 | function! s:RunRustfmt(command, curw, tmpname) |
| 34 | if exists("*systemlist") |
| 35 | let out = systemlist(a:command) |
| 36 | else |
| 37 | let out = split(system(a:command), '\r\?\n') |
| 38 | endif |
| 39 | |
| 40 | if v:shell_error == 0 || v:shell_error == 3 |
| 41 | " remove undo point caused via BufWritePre |
| 42 | try | silent undojoin | catch | endtry |
| 43 | |
| 44 | " Replace current file with temp file, then reload buffer |
| 45 | call rename(a:tmpname, expand('%')) |
| 46 | silent edit! |
| 47 | let &syntax = &syntax |
| 48 | |
| 49 | " only clear location list if it was previously filled to prevent |
| 50 | " clobbering other additions |
| 51 | if s:got_fmt_error |
| 52 | let s:got_fmt_error = 0 |
| 53 | call setloclist(0, []) |
| 54 | lwindow |
| 55 | endif |
| 56 | elseif g:rustfmt_fail_silently == 0 |
| 57 | " otherwise get the errors and put them in the location list |
| 58 | let errors = [] |
| 59 | |
| 60 | for line in out |
| 61 | " src/lib.rs:13:5: 13:10 error: expected `,`, or `}`, found `value` |
| 62 | let tokens = matchlist(line, '^\(.\{-}\):\(\d\+\):\(\d\+\):\s*\(\d\+:\d\+\s*\)\?\s*error: \(.*\)') |
| 63 | if !empty(tokens) |
| 64 | call add(errors, {"filename": @%, |
| 65 | \"lnum": tokens[2], |
| 66 | \"col": tokens[3], |
| 67 | \"text": tokens[5]}) |
| 68 | endif |
| 69 | endfor |
| 70 | |
| 71 | if empty(errors) |
| 72 | % | " Couldn't detect rustfmt error format, output errors |
| 73 | endif |
| 74 | |
| 75 | if !empty(errors) |
| 76 | call setloclist(0, errors, 'r') |
| 77 | echohl Error | echomsg "rustfmt returned error" | echohl None |
| 78 | endif |
| 79 | |
| 80 | let s:got_fmt_error = 1 |
| 81 | lwindow |
| 82 | " We didn't use the temp file, so clean up |
| 83 | call delete(a:tmpname) |
| 84 | endif |
| 85 | |
| 86 | call winrestview(a:curw) |
| 87 | endfunction |
| 88 | |
| 89 | function! rustfmt#FormatRange(line1, line2) |
| 90 | let l:curw = winsaveview() |
| 91 | let l:tmpname = expand("%:p:h") . "/." . expand("%:p:t") . ".rustfmt" |
| 92 | call writefile(getline(1, '$'), l:tmpname) |
| 93 | |
| 94 | let command = s:RustfmtCommandRange(l:tmpname, a:line1, a:line2) |
| 95 | |
| 96 | call s:RunRustfmt(command, l:curw, l:tmpname) |
| 97 | endfunction |
| 98 | |
| 99 | function! rustfmt#Format() |
| 100 | let l:curw = winsaveview() |
| 101 | let l:tmpname = expand("%:p:h") . "/." . expand("%:p:t") . ".rustfmt" |
| 102 | call writefile(getline(1, '$'), l:tmpname) |
| 103 | |
| 104 | let command = s:RustfmtCommand(l:tmpname) |
| 105 | |
| 106 | call s:RunRustfmt(command, l:curw, l:tmpname) |
| 107 | endfunction |