POC for fatih#3319 - add fzf wrap for GoImplements

autoload/fzf/decls.vim

@@ -25,37 +25,6 @@ function! s:color(str, group) abort
  return printf("\x1b[%sm%s\x1b[m", color, a:str)
 endfunction
 
-function! s:sink(str) abort
- if len(a:str) < 2
- return
- endif
- try
- " we jump to the file directory so we can get the fullpath via fnamemodify
- " below
- let l:dir = go#util#Chdir(s:current_dir)
-
- let vals = matchlist(a:str[1], '|\(.\{-}\):\(\d\+\):\(\d\+\)\s*\(.*\)|')
-
- " i.e: main.go
- let filename = vals[1]
- let line = vals[2]
- let col = vals[3]
-
- " i.e: /Users/fatih/vim-go/main.go
- let filepath = fnamemodify(filename, ":p")
-
- let cmd = get({'ctrl-x': 'split',
- \ 'ctrl-v': 'vertical split',
- \ 'ctrl-t': 'tabe'}, a:str[0], 'e')
- execute cmd fnameescape(filepath)
- call cursor(line, col)
- silent! norm! zvzz
- finally
- "jump back to old dir
- call go#util#Chdir(l:dir)
- endtry
-endfunction
-
 function! s:source(mode,...) abort
  let s:current_dir = expand('%:p:h')
  let ret_decls = []
@@ -144,7 +113,7 @@ function! fzf#decls#cmd(...) abort
  call fzf#run(fzf#wrap('GoDecls', {
  \ 'source': call('<sid>source', a:000),
  \ 'options': printf('-n 1 --with-nth 1,2 --delimiter=$''\t'' --preview "echo {3}" --preview-window "wrap" --ansi --prompt "GoDecls> " --expect=ctrl-t,ctrl-v,ctrl-x%s', colors),
- \ 'sink*': function('s:sink')
+ \ 'sink*': function('fzf#fzf#sink')
  \ }))
 endfunction
 

autoload/fzf/fzf.vim

@@ -0,0 +1,43 @@
+" don't spam the user when Vim is started in Vi compatibility mode
+let s:cpo_save = &cpo
+set cpo&vim
+
+function! fzf#fzf#sink(str) abort
+ if len(a:str) < 2
+ return
+ endif
+
+ let s:current_dir = expand('%:p:h')
+
+ try
+ " we jump to the file directory so we can get the fullpath via fnamemodify
+ " below
+ let l:dir = go#util#Chdir(s:current_dir)
+
+ let vals = matchlist(a:str[1], '|\(.\{-}\):\(\d\+\):\(\d\+\)\s*\(.*\)|')
+
+ " i.e: main.go
+ let filename = vals[1]
+ let line = vals[2]
+ let col = vals[3]
+
+ " i.e: /Users/fatih/vim-go/main.go
+ let filepath = fnamemodify(filename, ":p")
+
+ let cmd = get({'ctrl-x': 'split',
+ \ 'ctrl-v': 'vertical split',
+ \ 'ctrl-t': 'tabe'}, a:str[0], 'e')
+ execute cmd fnameescape(filepath)
+ call cursor(line, col)
+ silent! norm! zvzz
+ finally
+ "jump back to old dir
+ call go#util#Chdir(l:dir)
+ endtry
+endfunction
+
+" restore Vi compatibility settings
+let &cpo = s:cpo_save
+unlet s:cpo_save
+
+" vim: sw=2 ts=2 et

autoload/fzf/implements.vim

@@ -0,0 +1,62 @@
+" don't spam the user when Vim is started in Vi compatibility mode
+let s:cpo_save = &cpo
+set cpo&vim
+
+function! s:source(...) abort
+ let ret_files = []
+
+ let l:bin = "gopls"
+ let l:fname = expand('%:p')
+ let [l:line, l:col] = getpos('.')[1:2]
+ let pos = printf("%s:%s:%s",
+ \ l:fname,
+ \ l:line,
+ \ l:col
+ \)
+
+ let bin_path = go#path#CheckBinPath(l:bin)
+ if empty(bin_path)
+ return
+ endif
+
+ call go#cmd#autowrite()
+
+ let [l:out, l:err] = go#util#Exec([l:bin, 'implementation', l:pos])
+ if l:err
+ call go#util#EchoError(l:out)
+ return
+ endif
+
+ for line in split(out, '\n')
+ let vals = matchlist(line, '\(.\{-}\):\(\d\+\):\(\d\+\)-\(\d\+\)')
+ let filename = fnamemodify(expand(vals[1]), ":~:.")
+ let pos = printf("|%s:%s:%s|",
+ \ l:filename,
+ \ l:vals[2],
+ \ l:vals[3]
+ \)
+ call add(ret_files, printf("%s:%s\t%s",
+ \ l:filename,
+ \ l:vals[2],
+ \ pos 
+ \))
+ endfor
+ return sort(ret_files)
+endfunc
+
+
+function! fzf#implements#cmd(...) abort
+ let l:spec = {
+ \ 'source': call('<sid>source', [a:000]),
+ \ 'sink*': function('fzf#fzf#sink'),
+ \ 'options': printf('-n 1 --with-nth 1 --delimiter=$''\t'' --ansi --prompt "GoImplements> " --expect=ctrl-t,ctrl-v,ctrl-x')
+ \ }
+ call fzf#run(fzf#wrap("GoImplements", fzf#vim#with_preview(l:spec, 'right:50%:nohidden')))
+endfunc
+
+
+" restore Vi compatibility settings
+let &cpo = s:cpo_save
+unlet s:cpo_save
+
+" vim: sw=2 ts=2 et

autoload/go/implements.vim

@@ -16,6 +16,12 @@ function! go#implements#Implements(selected) abort
  let l:fname = expand('%:p')
  call go#lsp#Implements(l:fname, l:line, l:col, funcref('s:parse_output'))
  return
+ elseif l:mode == 'fzf'
+ if !go#config#GoplsEnabled()
+ call go#util#EchoError("go_implements_mode is 'fzf', but gopls is disabled")
+ endif
+ call fzf#implements#cmd() 
+ return
  else
  call go#util#EchoWarning('unknown value for g:go_implements_mode')
  endif