1
0
Fork 0
mirror of synced 2024-06-28 19:51:08 -04:00
ultimate-vim/sources_non_forked/vim-go/autoload/go/term.vim
2022-05-27 20:16:55 +08:00

275 lines
7.9 KiB
VimL

" don't spam the user when Vim is started in Vi compatibility mode
let s:cpo_save = &cpo
set cpo&vim
let s:bufnameprefix = 'goterm://'
" new creates a new terminal with the given command. Mode is set based on the
" global variable g:go_term_mode, which is by default set to :vsplit
function! go#term#new(bang, cmd, errorformat) abort
return go#term#newmode(a:bang, a:cmd, a:errorformat, go#config#TermMode())
endfunction
" go#term#newmode creates a new terminal with the given command and window mode.
function! go#term#newmode(bang, cmd, errorformat, mode) abort
let l:mode = a:mode
if empty(l:mode)
let l:mode = go#config#TermMode()
endif
if go#config#TermReuse()
call s:closeterm()
endif
let l:state = {
\ 'cmd': a:cmd,
\ 'bang' : a:bang,
\ 'winid': win_getid(winnr()),
\ 'stdout': [],
\ 'stdout_buf': '',
\ 'errorformat': a:errorformat,
\ }
" execute the command in the current file's directory
let l:dir = go#util#Chdir(expand('%:p:h'))
execute l:mode . ' __go_term__'
setlocal filetype=goterm
setlocal bufhidden=delete
setlocal winfixheight
" TODO(bc)?: setlocal winfixwidth
setlocal noswapfile
setlocal nobuflisted
" setup job for nvim
if has('nvim')
" explicitly bind callbacks to state so that within them, self will always
" refer to state. See :help Partial for more information.
"
" Don't set an on_stderr, because it will be passed the same data as
" on_stdout. See https://github.com/neovim/neovim/issues/2836
let l:job = {
\ 'on_stdout': function('s:on_stdout', [], state),
\ 'on_exit' : function('s:on_exit', [], state),
\ }
let l:state.id = termopen(a:cmd, l:job)
let l:state.termwinid = win_getid(winnr())
let s:lasttermwinid = l:state.termwinid
call go#util#Chdir(l:dir)
" resize new term if needed.
let l:height = go#config#TermHeight()
let l:width = go#config#TermWidth()
" Adjust the window width or height depending on whether it's a vertical or
" horizontal split.
if l:mode =~ "vertical" || l:mode =~ "vsplit" || l:mode =~ "vnew"
exe 'vertical resize ' . l:width
elseif mode =~ "split" || mode =~ "new"
exe 'resize ' . l:height
endif
" we also need to resize the pty, so there you go...
call jobresize(l:state.id, l:width, l:height)
" setup term for vim8
elseif has('terminal')
" Not great randomness, but "good enough" for our purpose here.
let l:rnd = sha256(printf('%s%s', reltimestr(reltime()), fnamemodify(bufname(''), ":p")))
let l:termname = printf("%s%s", s:bufnameprefix, l:rnd)
let l:term = {
\ 'out_cb': function('s:out_cb', [], state),
\ 'exit_cb' : function('s:exit_cb', [], state),
\ 'curwin': 1,
\ 'term_name': l:termname,
\ }
if l:mode =~ "vertical" || l:mode =~ "vsplit" || l:mode =~ "vnew"
let l:term["vertical"] = l:mode
endif
let l:state.id = term_start(a:cmd, l:term)
let l:state.termwinid = win_getid(bufwinnr(l:state.id))
let s:lasttermwinid = l:state.termwinid
call go#util#Chdir(l:dir)
" resize new term if needed.
let l:height = go#config#TermHeight()
let l:width = go#config#TermWidth()
" Adjust the window width or height depending on whether it's a vertical or
" horizontal split.
if l:mode =~ "vertical" || l:mode =~ "vsplit" || l:mode =~ "vnew"
exe 'vertical resize ' . l:width
elseif mode =~ "split" || mode =~ "new"
exe 'resize ' . l:height
endif
"if exists(*term_setsize)
"call term_setsize(l:state.id, l:height, l:width)
"endif
endif
call win_gotoid(l:state.winid)
return l:state.id
endfunction
" out_cb continually concat's the self.stdout_buf on recv of stdout
" and sets self.stdout to the new-lined split content in self.stdout_buf
func! s:out_cb(channel, msg) dict abort
let self.stdout_buf = self.stdout_buf . a:msg
let self.stdout = split(self.stdout_buf, '\n')
endfunction
function! s:on_stdout(job_id, data, event) dict abort
" A single empty string means EOF was reached. The first item will never be
" the empty string except for when it's the only item and is signaling that
" EOF was reached.
if len(a:data) == 1 && a:data[0] == ''
" when there's nothing buffered, return early so that an
" erroneous message will not be added.
if self.stdout_buf == ''
return
endif
let self.stdout = add(self.stdout, self.stdout_buf)
else
let l:data = copy(a:data)
let l:data[0] = self.stdout_buf . l:data[0]
" The last element may be a partial line; save it for next time.
let self.stdout_buf = l:data[-1]
let self.stdout = extend(self.stdout, l:data[:-2])
endif
endfunction
" vim8 exit callback
function! s:exit_cb(job_id, exit_status) dict abort
call s:handle_exit(a:job_id, a:exit_status, self)
endfunction
" nvim exit callback
function! s:on_exit(job_id, exit_status, event) dict abort
call s:handle_exit(a:job_id, a:exit_status, self)
endfunction
" handle_exit implements both vim8 and nvim exit callbacks
func s:handle_exit(job_id, exit_status, state) abort
let l:winid = win_getid(winnr())
call win_gotoid(a:state.winid)
let l:listtype = go#list#Type("_term")
if a:exit_status == 0
call go#list#Clean(l:listtype)
call win_gotoid(l:winid)
return
endif
let l:bufdir = expand('%:p:h')
if !isdirectory(l:bufdir)
call go#util#EchoWarning('terminal job failure not processed, because the job''s working directory no longer exists')
call win_gotoid(l:winid)
return
endif
" change to directory where the command was run. If we do not do this the
" quickfix items will have the incorrect paths.
" see: https://github.com/fatih/vim-go/issues/2400
let l:dir = go#util#Chdir(l:bufdir)
let l:title = a:state.cmd
if type(l:title) == v:t_list
let l:title = join(a:state.cmd)
endif
let l:i = 0
while l:i < len(a:state.stdout)
let a:state.stdout[l:i] = substitute(a:state.stdout[l:i], "\r$", '', 'g')
let l:i += 1
endwhile
call go#list#ParseFormat(l:listtype, a:state.errorformat, a:state.stdout, l:title, 0)
let l:errors = go#list#Get(l:listtype)
call go#list#Window(l:listtype, len(l:errors))
" close terminal; we don't need it anymore
if go#config#TermCloseOnExit()
call win_gotoid(a:state.termwinid)
close!
endif
if empty(l:errors)
call go#util#EchoError( '[' . l:title . '] ' . "FAIL")
call go#util#Chdir(l:dir)
call win_gotoid(l:winid)
return
endif
if a:state.bang
call go#util#Chdir(l:dir)
call win_gotoid(l:winid)
return
endif
call win_gotoid(a:state.winid)
call go#list#JumpToFirst(l:listtype)
" change back to original working directory
call go#util#Chdir(l:dir)
endfunction
function! go#term#ToggleCloseOnExit() abort
if go#config#TermCloseOnExit()
call go#config#SetTermCloseOnExit(0)
call go#util#EchoProgress("term close on exit disabled")
return
endif
call go#config#SetTermCloseOnExit(1)
call go#util#EchoProgress("term close on exit enabled")
return
endfunction
function! s:closeterm()
if !exists('s:lasttermwinid')
return
endif
try
let l:termwinid = s:lasttermwinid
unlet s:lasttermwinid
let l:info = getwininfo(l:termwinid)
if empty(l:info)
return
endif
let l:info = l:info[0]
if !get(l:info, 'terminal', 0) is 1
return
endif
if has('nvim')
if 'goterm' == nvim_buf_get_option(nvim_win_get_buf(l:termwinid), 'filetype')
call nvim_win_close(l:termwinid, v:true)
endif
return
endif
if stridx(bufname(winbufnr(l:termwinid)), s:bufnameprefix, 0) == 0
let l:winid = win_getid()
call win_gotoid(l:termwinid)
close!
call win_gotoid(l:winid)
endif
catch
call go#util#EchoError(printf("vim-go: %s", v:exception))
endtry
endfunction
" restore Vi compatibility settings
let &cpo = s:cpo_save
unlet s:cpo_save
" vim: sw=2 ts=2 et