fzf/plugin/fzf.vim

352 lines
8.6 KiB
VimL
Raw Normal View History

2015-01-06 16:14:35 +00:00
" Copyright (c) 2015 Junegunn Choi
2013-10-23 01:26:55 +00:00
"
" MIT License
"
" Permission is hereby granted, free of charge, to any person obtaining
" a copy of this software and associated documentation files (the
" "Software"), to deal in the Software without restriction, including
" without limitation the rights to use, copy, modify, merge, publish,
" distribute, sublicense, and/or sell copies of the Software, and to
" permit persons to whom the Software is furnished to do so, subject to
" the following conditions:
"
" The above copyright notice and this permission notice shall be
" included in all copies or substantial portions of the Software.
"
" THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
" EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
" MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
" NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
" LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
" OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
" WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
2014-03-28 08:15:32 +00:00
let s:default_tmux_height = '40%'
let s:launcher = 'xterm -e bash -ic %s'
2015-01-06 16:14:35 +00:00
let s:fzf_go = expand('<sfile>:h:h').'/bin/fzf'
let s:fzf_rb = expand('<sfile>:h:h').'/fzf'
let s:fzf_tmux = expand('<sfile>:h:h').'/bin/fzf-tmux'
let s:legacy = 0
2014-03-27 15:58:07 +00:00
2014-03-25 10:55:52 +00:00
let s:cpo_save = &cpo
set cpo&vim
function! s:fzf_exec()
if !exists('s:exec')
if executable(s:fzf_go)
let s:exec = s:fzf_go
else
let path = split(system('which fzf 2> /dev/null'), '\n')
if !v:shell_error && !empty(path)
let s:exec = path[0]
elseif executable(s:fzf_rb)
let s:exec = s:fzf_rb
let s:legacy = 1
else
call system('type fzf')
if v:shell_error
throw 'fzf executable not found'
else
let s:exec = 'fzf'
endif
endif
endif
return s:exec
else
return s:exec
endif
endfunction
2013-10-23 01:26:55 +00:00
function! s:tmux_enabled()
2014-06-14 18:04:15 +00:00
if has('gui_running')
return 0
endif
if exists('s:tmux')
return s:tmux
endif
let s:tmux = 0
if exists('$TMUX') && executable(s:fzf_tmux)
let output = system('tmux -V')
let s:tmux = !v:shell_error && output >= 'tmux 1.7'
endif
return s:tmux
endfunction
2014-03-27 15:58:07 +00:00
function! s:shellesc(arg)
return '"'.substitute(a:arg, '"', '\\"', 'g').'"'
endfunction
function! s:escape(path)
return substitute(a:path, ' ', '\\ ', 'g')
endfunction
" Upgrade legacy options
function! s:upgrade(dict)
let copy = copy(a:dict)
if has_key(copy, 'tmux')
let copy.down = remove(copy, 'tmux')
endif
if has_key(copy, 'tmux_height')
let copy.down = remove(copy, 'tmux_height')
endif
if has_key(copy, 'tmux_width')
let copy.right = remove(copy, 'tmux_width')
endif
return copy
endfunction
2014-03-25 10:55:52 +00:00
function! fzf#run(...) abort
let dict = exists('a:1') ? s:upgrade(a:1) : {}
2014-03-27 15:58:07 +00:00
let temps = { 'result': tempname() }
2014-03-25 10:55:52 +00:00
let optstr = get(dict, 'options', '')
try
let fzf_exec = s:fzf_exec()
catch
throw v:exception
endtry
2014-03-25 10:55:52 +00:00
if has_key(dict, 'source')
let source = dict.source
let type = type(source)
if type == 1
let prefix = source.'|'
elseif type == 3
2014-03-27 15:58:07 +00:00
let temps.input = tempname()
call writefile(source, temps.input)
let prefix = 'cat '.s:shellesc(temps.input).'|'
2014-03-25 10:55:52 +00:00
else
throw 'Invalid source type'
endif
else
let prefix = ''
endif
let tmux = !has('nvim') && s:tmux_enabled() && s:splittable(dict)
let command = prefix.(tmux ? s:fzf_tmux(dict) : fzf_exec).' '.optstr.' > '.temps.result
2014-03-25 10:55:52 +00:00
2015-04-04 02:55:57 +00:00
try
if tmux
2015-04-04 02:55:57 +00:00
return s:execute_tmux(dict, command, temps)
elseif has('nvim')
return s:execute_term(dict, command, temps)
2015-04-04 02:55:57 +00:00
else
return s:execute(dict, command, temps)
endif
finally
call s:popd(dict)
endtry
2014-03-27 15:58:07 +00:00
endfunction
function! s:present(dict, ...)
for key in a:000
if !empty(get(a:dict, key, ''))
return 1
endif
endfor
return 0
endfunction
function! s:fzf_tmux(dict)
let size = ''
for o in ['up', 'down', 'left', 'right']
if s:present(a:dict, o)
let size = '-'.o[0].(a:dict[o] == 1 ? '' : a:dict[o])
break
endif
endfor
return printf('LINES=%d COLUMNS=%d %s %s %s --',
\ &lines, &columns, s:fzf_tmux, size, (has_key(a:dict, 'source') ? '' : '-'))
endfunction
function! s:splittable(dict)
return s:present(a:dict, 'up', 'down', 'left', 'right')
endfunction
2014-03-27 15:58:07 +00:00
function! s:pushd(dict)
if s:present(a:dict, 'dir')
let cwd = getcwd()
if get(a:dict, 'prev_dir', '') ==# cwd
return 1
endif
let a:dict.prev_dir = cwd
2014-03-27 15:58:07 +00:00
execute 'chdir '.s:escape(a:dict.dir)
let a:dict.dir = getcwd()
return 1
2014-03-27 15:58:07 +00:00
endif
return 0
2014-03-27 15:58:07 +00:00
endfunction
function! s:popd(dict)
if has_key(a:dict, 'prev_dir')
execute 'chdir '.s:escape(remove(a:dict, 'prev_dir'))
endif
endfunction
function! s:execute(dict, command, temps)
call s:pushd(a:dict)
silent! !clear 2> /dev/null
2014-06-14 18:04:15 +00:00
if has('gui_running')
let launcher = get(a:dict, 'launcher', get(g:, 'fzf_launcher', s:launcher))
let command = printf(launcher, "'".substitute(a:command, "'", "'\"'\"'", 'g')."'")
2014-06-14 18:04:15 +00:00
else
let command = a:command
2014-06-14 18:04:15 +00:00
endif
execute 'silent !'.command
2014-03-27 15:58:07 +00:00
redraw!
if v:shell_error
" Do not print error message on exit status 1
if v:shell_error > 1
echohl ErrorMsg
echo 'Error running ' . command
endif
2014-03-27 15:58:07 +00:00
return []
else
return s:callback(a:dict, a:temps)
2014-03-27 15:58:07 +00:00
endif
endfunction
function! s:execute_tmux(dict, command, temps)
let command = a:command
if s:pushd(a:dict)
" -c '#{pane_current_path}' is only available on tmux 1.9 or above
let command = 'cd '.s:escape(a:dict.dir).' && '.command
2014-03-27 15:58:07 +00:00
endif
2014-03-28 08:15:32 +00:00
call system(command)
return s:callback(a:dict, a:temps)
2014-03-27 15:58:07 +00:00
endfunction
function! s:calc_size(max, val)
if a:val =~ '%$'
return a:max * str2nr(a:val[:-2]) / 100
else
return min([a:max, a:val])
endif
endfunction
function! s:split(dict)
let directions = {
\ 'up': ['topleft', &lines],
\ 'down': ['botright', &lines],
\ 'left': ['vertical topleft', &columns],
\ 'right': ['vertical botright', &columns] }
try
for [dir, pair] in items(directions)
let val = get(a:dict, dir, '')
if !empty(val)
let [cmd, max] = pair
execute cmd s:calc_size(max, val).'new'
return
endif
endfor
if s:present(a:dict, 'window')
execute a:dict.window
else
tabnew
endif
finally
setlocal winfixwidth winfixheight
endtry
endfunction
function! s:execute_term(dict, command, temps)
call s:pushd(a:dict)
call s:split(a:dict)
let fzf = { 'buf': bufnr('%'), 'dict': a:dict, 'temps': a:temps }
function! fzf.on_exit(id, code)
execute 'bd!' self.buf
call s:pushd(self.dict)
try
call s:callback(self.dict, self.temps)
finally
call s:popd(self.dict)
endtry
endfunction
call termopen(a:command, fzf)
silent file [FZF]
startinsert
return []
endfunction
function! s:callback(dict, temps)
2014-03-27 15:58:07 +00:00
if !filereadable(a:temps.result)
let lines = []
else
let lines = readfile(a:temps.result)
if has_key(a:dict, 'sink')
2014-03-25 10:55:52 +00:00
for line in lines
2014-03-27 15:58:07 +00:00
if type(a:dict.sink) == 2
call a:dict.sink(line)
2014-03-25 10:55:52 +00:00
else
execute a:dict.sink s:escape(line)
endif
endfor
2013-10-23 01:26:55 +00:00
endif
if has_key(a:dict, 'sink*')
call a:dict['sink*'](lines)
endif
2014-03-27 15:58:07 +00:00
endif
for tf in values(a:temps)
silent! call delete(tf)
endfor
return lines
2013-10-23 01:26:55 +00:00
endfunction
function! s:cmd_callback(lines) abort
if empty(a:lines)
return
endif
let key = remove(a:lines, 0)
if key == 'ctrl-t' | let cmd = 'tabedit'
elseif key == 'ctrl-x' | let cmd = 'split'
elseif key == 'ctrl-v' | let cmd = 'vsplit'
else | let cmd = 'e'
endif
call s:pushd(s:opts)
try
for item in a:lines
execute cmd s:escape(item)
endfor
finally
call s:popd(s:opts)
endtry
endfunction
2014-03-27 15:58:07 +00:00
function! s:cmd(bang, ...) abort
if bufexists('[FZF]')
echohl WarningMsg
echomsg 'FZF is already running!'
echohl NONE
return
endif
2014-03-25 10:55:52 +00:00
let args = copy(a:000)
if !s:legacy
2015-03-29 02:14:16 +00:00
let args = insert(args, '--expect=ctrl-t,ctrl-x,ctrl-v', 0)
endif
2014-03-25 10:55:52 +00:00
let opts = {}
if len(args) > 0 && isdirectory(expand(args[-1]))
let opts.dir = remove(args, -1)
endif
2014-03-27 15:58:07 +00:00
if !a:bang
let opts.down = get(g:, 'fzf_tmux_height', s:default_tmux_height)
2014-03-27 15:58:07 +00:00
endif
if s:legacy
call fzf#run(extend({ 'options': join(args), 'sink': 'e' }, opts))
else
let s:opts = opts
call fzf#run(extend({ 'options': join(args), 'sink*': function('<sid>cmd_callback') }, opts))
endif
2014-03-25 10:55:52 +00:00
endfunction
2014-03-27 15:58:07 +00:00
command! -nargs=* -complete=dir -bang FZF call s:cmd('<bang>' == '!', <f-args>)
2014-03-25 10:55:52 +00:00
let &cpo = s:cpo_save
unlet s:cpo_save
2013-10-23 01:26:55 +00:00