1
0
Fork 0
mirror of synced 2024-11-30 04:35:35 -05:00
ultimate-vim/sources_non_forked/vim-go/autoload/go/fmt.vim

215 lines
6.3 KiB
VimL
Raw Normal View History

2014-10-31 17:30:24 -04:00
" Copyright 2011 The Go Authors. All rights reserved.
" Use of this source code is governed by a BSD-style
" license that can be found in the LICENSE file.
"
2017-02-11 08:01:38 -05:00
" fmt.vim: Vim command to format Go files with gofmt (and gofmt compatible
" toorls, such as goimports).
2014-10-31 17:30:24 -04:00
2018-12-17 06:28:27 -05:00
" don't spam the user when Vim is started in Vi compatibility mode
let s:cpo_save = &cpo
set cpo&vim
2016-08-02 08:48:32 -04:00
" we have those problems :
2014-10-31 17:30:24 -04:00
" http://stackoverflow.com/questions/12741977/prevent-vim-from-updating-its-undo-tree
" http://stackoverflow.com/questions/18532692/golang-formatter-and-vim-how-to-destroy-history-record?rq=1
"
" The below function is an improved version that aims to fix all problems.
" it doesn't undo changes and break undo history. If you are here reading
" this and have VimL experience, please look at the function for
" improvements, patches are welcome :)
2016-12-27 09:46:49 -05:00
function! go#fmt#Format(withGoimport) abort
2018-06-14 06:31:12 -04:00
if go#config#FmtExperimental()
2016-06-26 07:12:36 -04:00
" Using winsaveview to save/restore cursor state has the problem of
" closing folds on save:
" https://github.com/fatih/vim-go/issues/502
" One fix is to use mkview instead. Unfortunately, this sometimes causes
" other bad side effects:
" https://github.com/fatih/vim-go/issues/728
" and still closes all folds if foldlevel>0:
" https://github.com/fatih/vim-go/issues/732
let l:curw = {}
try
mkview!
catch
let l:curw = winsaveview()
endtry
2017-02-11 08:01:38 -05:00
" save our undo file to be restored after we are done. This is needed to
" prevent an additional undo jump due to BufWritePre auto command and also
" restore 'redo' history because it's getting being destroyed every
" BufWritePre
let tmpundofile = tempname()
exe 'wundo! ' . tmpundofile
2016-06-26 07:12:36 -04:00
else
" Save cursor position and many other things.
let l:curw = winsaveview()
endif
" Write current unsaved buffer to a temp file
let l:tmpname = tempname() . '.go'
2017-02-11 08:01:38 -05:00
call writefile(go#util#GetLines(), l:tmpname)
if go#util#IsWin()
let l:tmpname = tr(l:tmpname, '\', '/')
endif
2018-06-14 06:31:12 -04:00
let bin_name = go#config#FmtCommand()
2017-02-11 08:01:38 -05:00
if a:withGoimport == 1
2017-07-06 08:57:35 -04:00
let bin_name = "goimports"
2017-02-11 08:01:38 -05:00
endif
2017-09-02 06:43:18 -04:00
let current_col = col('.')
2018-06-14 06:31:12 -04:00
let [l:out, l:err] = go#fmt#run(bin_name, l:tmpname, expand('%'))
2017-09-02 06:43:18 -04:00
let diff_offset = len(readfile(l:tmpname)) - line('$')
2018-06-14 06:31:12 -04:00
if l:err == 0
2017-02-11 08:01:38 -05:00
call go#fmt#update_file(l:tmpname, expand('%'))
2018-06-14 06:31:12 -04:00
elseif !go#config#FmtFailSilently()
2017-03-07 12:04:28 -05:00
let errors = s:parse_errors(expand('%'), out)
2017-02-11 08:01:38 -05:00
call s:show_errors(errors)
endif
" We didn't use the temp file, so clean up
call delete(l:tmpname)
2016-06-26 07:12:36 -04:00
2018-06-14 06:31:12 -04:00
if go#config#FmtExperimental()
2017-02-11 08:01:38 -05:00
" restore our undo history
silent! exe 'rundo ' . tmpundofile
call delete(tmpundofile)
" Restore our cursor/windows positions, folds, etc.
if empty(l:curw)
silent! loadview
else
call winrestview(l:curw)
endif
else
" Restore our cursor/windows positions.
call winrestview(l:curw)
2016-06-26 07:12:36 -04:00
endif
2017-09-02 06:43:18 -04:00
" be smart and jump to the line the new statement was added/removed
call cursor(line('.') + diff_offset, current_col)
" Syntax highlighting breaks less often.
syntax sync fromstart
2017-02-11 08:01:38 -05:00
endfunction
2016-06-26 07:12:36 -04:00
2017-02-11 08:01:38 -05:00
" update_file updates the target file with the given formatted source
2017-03-07 12:04:28 -05:00
function! go#fmt#update_file(source, target)
2017-02-11 08:01:38 -05:00
" remove undo point caused via BufWritePre
try | silent undojoin | catch | endtry
let old_fileformat = &fileformat
if exists("*getfperm")
" save file permissions
let original_fperm = getfperm(a:target)
2016-06-26 07:12:36 -04:00
endif
2017-02-11 08:01:38 -05:00
call rename(a:source, a:target)
" restore file permissions
if exists("*setfperm") && original_fperm != ''
call setfperm(a:target , original_fperm)
2016-06-26 07:12:36 -04:00
endif
2017-02-11 08:01:38 -05:00
" reload buffer to reflect latest changes
2017-09-02 06:43:18 -04:00
silent edit!
2017-02-11 08:01:38 -05:00
let &fileformat = old_fileformat
let &syntax = &syntax
let l:listtype = go#list#Type("GoFmt")
2017-09-02 06:43:18 -04:00
2019-11-16 10:28:42 -05:00
" clean up previous list
if l:listtype == "quickfix"
let l:list_title = getqflist({'title': 1})
else
2019-11-16 10:28:42 -05:00
let l:list_title = getloclist(0, {'title': 1})
2017-09-02 06:43:18 -04:00
endif
if has_key(l:list_title, "title") && l:list_title['title'] == "Format"
call go#list#Clean(l:listtype)
endif
2017-02-11 08:01:38 -05:00
endfunction
" run runs the gofmt/goimport command for the given source file and returns
2018-02-04 06:35:08 -05:00
" the output of the executed command. Target is the real file to be formatted.
2017-03-07 12:04:28 -05:00
function! go#fmt#run(bin_name, source, target)
2018-06-14 06:31:12 -04:00
let l:cmd = s:fmt_cmd(a:bin_name, a:source, a:target)
if empty(l:cmd)
2017-04-01 07:22:06 -04:00
return
endif
2018-06-14 06:31:12 -04:00
return go#util#Exec(l:cmd)
2017-02-11 08:01:38 -05:00
endfunction
2018-06-14 06:31:12 -04:00
" fmt_cmd returns the command to run as a list.
2017-02-11 08:01:38 -05:00
function! s:fmt_cmd(bin_name, source, target)
2018-06-14 06:31:12 -04:00
let l:cmd = [a:bin_name, '-w']
2017-09-02 06:43:18 -04:00
2017-07-06 08:57:35 -04:00
" add the options for binary (if any). go_fmt_options was by default of type
" string, however to allow customization it's now a dictionary of binary
" name mapping to options.
2018-06-14 06:31:12 -04:00
let opts = go#config#FmtOptions()
if type(opts) == type({})
let opts = has_key(opts, a:bin_name) ? opts[a:bin_name] : ""
2017-07-06 08:57:35 -04:00
endif
call extend(cmd, split(opts, " "))
2018-06-14 06:31:12 -04:00
if a:bin_name is# 'goimports'
call extend(cmd, ["-srcdir", a:target])
2016-06-26 07:12:36 -04:00
endif
2017-02-11 08:01:38 -05:00
call add(cmd, a:source)
return cmd
endfunction
2016-06-26 07:12:36 -04:00
2017-02-11 08:01:38 -05:00
" parse_errors parses the given errors and returns a list of parsed errors
2017-03-07 12:04:28 -05:00
function! s:parse_errors(filename, content) abort
2017-02-11 08:01:38 -05:00
let splitted = split(a:content, '\n')
" list of errors to be put into location list
let errors = []
for line in splitted
let tokens = matchlist(line, '^\(.\{-}\):\(\d\+\):\(\d\+\)\s*\(.*\)')
if !empty(tokens)
call add(errors,{
2017-03-07 12:04:28 -05:00
\"filename": a:filename,
2017-02-11 08:01:38 -05:00
\"lnum": tokens[2],
\"col": tokens[3],
\"text": tokens[4],
\ })
2014-10-31 17:30:24 -04:00
endif
2017-02-11 08:01:38 -05:00
endfor
2014-10-31 17:30:24 -04:00
2017-02-11 08:01:38 -05:00
return errors
endfunction
2014-10-31 17:30:24 -04:00
2017-02-11 08:01:38 -05:00
" show_errors opens a location list and shows the given errors. If the given
" errors is empty, it closes the the location list
function! s:show_errors(errors) abort
let l:listtype = go#list#Type("GoFmt")
2017-02-11 08:01:38 -05:00
if !empty(a:errors)
call go#list#Populate(l:listtype, a:errors, 'Format')
2016-06-26 07:12:36 -04:00
endif
2016-03-14 06:04:57 -04:00
2017-03-07 12:04:28 -05:00
" this closes the window if there are no errors or it opens
2017-02-11 08:01:38 -05:00
" it if there is any
call go#list#Window(l:listtype, len(a:errors))
2014-10-31 17:30:24 -04:00
endfunction
2016-12-27 09:46:49 -05:00
function! go#fmt#ToggleFmtAutoSave() abort
2018-06-14 06:31:12 -04:00
if go#config#FmtAutosave()
call go#config#SetFmtAutosave(0)
2016-08-02 08:48:32 -04:00
call go#util#EchoProgress("auto fmt disabled")
return
end
2018-06-14 06:31:12 -04:00
call go#config#SetFmtAutosave(1)
2016-08-02 08:48:32 -04:00
call go#util#EchoProgress("auto fmt enabled")
endfunction
2017-02-11 08:01:38 -05:00
2018-12-17 06:28:27 -05:00
" restore Vi compatibility settings
let &cpo = s:cpo_save
unlet s:cpo_save
2016-06-26 07:12:36 -04:00
" vim: sw=2 ts=2 et