mirror of
https://github.com/junegunn/fzf
synced 2024-10-30 09:20:14 +00:00
46ee9ac41c
Just like it’s already done for `_fzf_compgen_path()` and `_fzf_compgen_dir()` allow a user to easily define his own version of `__fzf_list_hosts()`. Also add some documentation on the expected “interface” of such custom function. Signed-off-by: Christoph Anton Mitterer <mail@christoph.anton.mitterer.name>
536 lines
14 KiB
Bash
536 lines
14 KiB
Bash
# ____ ____
|
|
# / __/___ / __/
|
|
# / /_/_ / / /_
|
|
# / __/ / /_/ __/
|
|
# /_/ /___/_/ completion.bash
|
|
#
|
|
# - $FZF_TMUX (default: 0)
|
|
# - $FZF_TMUX_OPTS (default: empty)
|
|
# - $FZF_COMPLETION_TRIGGER (default: '**')
|
|
# - $FZF_COMPLETION_OPTS (default: empty)
|
|
|
|
[[ $- =~ i ]] || return 0
|
|
|
|
|
|
# To use custom commands instead of find, override _fzf_compgen_{path,dir}
|
|
if ! declare -F _fzf_compgen_path > /dev/null; then
|
|
_fzf_compgen_path() {
|
|
echo "$1"
|
|
command find -L "$1" \
|
|
-name .git -prune -o -name .hg -prune -o -name .svn -prune -o \( -type d -o -type f -o -type l \) \
|
|
-a -not -path "$1" -print 2> /dev/null | command sed 's@^\./@@'
|
|
}
|
|
fi
|
|
|
|
if ! declare -F _fzf_compgen_dir > /dev/null; then
|
|
_fzf_compgen_dir() {
|
|
command find -L "$1" \
|
|
-name .git -prune -o -name .hg -prune -o -name .svn -prune -o -type d \
|
|
-a -not -path "$1" -print 2> /dev/null | command sed 's@^\./@@'
|
|
}
|
|
fi
|
|
|
|
###########################################################
|
|
|
|
# To redraw line after fzf closes (printf '\e[5n')
|
|
bind '"\e[0n": redraw-current-line' 2> /dev/null
|
|
|
|
__fzf_comprun() {
|
|
if [[ "$(type -t _fzf_comprun 2>&1)" = function ]]; then
|
|
_fzf_comprun "$@"
|
|
elif [[ -n "${TMUX_PANE-}" ]] && { [[ "${FZF_TMUX:-0}" != 0 ]] || [[ -n "${FZF_TMUX_OPTS-}" ]]; }; then
|
|
shift
|
|
fzf-tmux ${FZF_TMUX_OPTS:--d${FZF_TMUX_HEIGHT:-40%}} -- "$@"
|
|
else
|
|
shift
|
|
fzf "$@"
|
|
fi
|
|
}
|
|
|
|
__fzf_orig_completion() {
|
|
local l comp f cmd
|
|
while read -r l; do
|
|
if [[ "$l" =~ ^(.*\ -F)\ *([^ ]*).*\ ([^ ]*)$ ]]; then
|
|
comp="${BASH_REMATCH[1]}"
|
|
f="${BASH_REMATCH[2]}"
|
|
cmd="${BASH_REMATCH[3]}"
|
|
[[ "$f" = _fzf_* ]] && continue
|
|
printf -v "_fzf_orig_completion_${cmd//[^A-Za-z0-9_]/_}" "%s" "${comp} %s ${cmd} #${f}"
|
|
if [[ "$l" = *" -o nospace "* ]] && [[ ! "${__fzf_nospace_commands-}" = *" $cmd "* ]]; then
|
|
__fzf_nospace_commands="${__fzf_nospace_commands-} $cmd "
|
|
fi
|
|
fi
|
|
done
|
|
}
|
|
|
|
_fzf_opts_completion() {
|
|
local cur prev opts
|
|
COMPREPLY=()
|
|
cur="${COMP_WORDS[COMP_CWORD]}"
|
|
prev="${COMP_WORDS[COMP_CWORD-1]}"
|
|
opts="
|
|
-h --help
|
|
-x --extended
|
|
-e --exact
|
|
--extended-exact
|
|
+x --no-extended
|
|
+e --no-exact
|
|
-q --query
|
|
-f --filter
|
|
--literal
|
|
--no-literal
|
|
--algo
|
|
--scheme
|
|
--expect
|
|
--no-expect
|
|
--enabled --no-phony
|
|
--disabled --phony
|
|
--tiebreak
|
|
--bind
|
|
--color
|
|
--toggle-sort
|
|
-d --delimiter
|
|
-n --nth
|
|
--with-nth
|
|
-s --sort
|
|
+s --no-sort
|
|
--track
|
|
--no-track
|
|
--tac
|
|
--no-tac
|
|
-i
|
|
+i
|
|
-m --multi
|
|
+m --no-multi
|
|
--ansi
|
|
--no-ansi
|
|
--no-mouse
|
|
+c --no-color
|
|
+2 --no-256
|
|
--black
|
|
--no-black
|
|
--bold
|
|
--no-bold
|
|
--layout
|
|
--reverse
|
|
--no-reverse
|
|
--cycle
|
|
--no-cycle
|
|
--keep-right
|
|
--no-keep-right
|
|
--hscroll
|
|
--no-hscroll
|
|
--hscroll-off
|
|
--scroll-off
|
|
--filepath-word
|
|
--no-filepath-word
|
|
--info
|
|
--no-info
|
|
--inline-info
|
|
--no-inline-info
|
|
--separator
|
|
--no-separator
|
|
--scrollbar
|
|
--no-scrollbar
|
|
--jump-labels
|
|
-1 --select-1
|
|
+1 --no-select-1
|
|
-0 --exit-0
|
|
+0 --no-exit-0
|
|
--read0
|
|
--no-read0
|
|
--print0
|
|
--no-print0
|
|
--print-query
|
|
--no-print-query
|
|
--prompt
|
|
--pointer
|
|
--marker
|
|
--sync
|
|
--no-sync
|
|
--async
|
|
--no-history
|
|
--history
|
|
--history-size
|
|
--no-header
|
|
--no-header-lines
|
|
--header
|
|
--header-lines
|
|
--header-first
|
|
--no-header-first
|
|
--ellipsis
|
|
--preview
|
|
--no-preview
|
|
--preview-window
|
|
--height
|
|
--min-height
|
|
--no-height
|
|
--no-margin
|
|
--no-padding
|
|
--no-border
|
|
--border
|
|
--no-border-label
|
|
--border-label
|
|
--border-label-pos
|
|
--no-preview-label
|
|
--preview-label
|
|
--preview-label-pos
|
|
--no-unicode
|
|
--unicode
|
|
--margin
|
|
--padding
|
|
--tabstop
|
|
--listen
|
|
--no-listen
|
|
--clear
|
|
--no-clear
|
|
--version
|
|
--"
|
|
|
|
case "${prev}" in
|
|
--algo)
|
|
COMPREPLY=( $(compgen -W "v1 v2" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--scheme)
|
|
COMPREPLY=( $(compgen -W "default path history" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--tiebreak)
|
|
COMPREPLY=( $(compgen -W "length chunk begin end index" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--color)
|
|
COMPREPLY=( $(compgen -W "dark light 16 bw no" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--layout)
|
|
COMPREPLY=( $(compgen -W "default reverse reverse-list" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--info)
|
|
COMPREPLY=( $(compgen -W "default right hidden inline inline-right" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--preview-window)
|
|
COMPREPLY=( $(compgen -W "
|
|
default
|
|
hidden
|
|
nohidden
|
|
wrap
|
|
nowrap
|
|
cycle
|
|
nocycle
|
|
up top
|
|
down bottom
|
|
left
|
|
right
|
|
rounded border border-rounded
|
|
sharp border-sharp
|
|
border-bold
|
|
border-block
|
|
border-thinblock
|
|
border-double
|
|
noborder border-none
|
|
border-horizontal
|
|
border-vertical
|
|
border-up border-top
|
|
border-down border-bottom
|
|
border-left
|
|
border-right
|
|
follow
|
|
nofollow" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--border)
|
|
COMPREPLY=( $(compgen -W "rounded sharp bold block thinblock double horizontal vertical top bottom left right none" -- "$cur") )
|
|
return 0
|
|
;;
|
|
--border-label-pos|--preview-label-pos)
|
|
COMPREPLY=( $(compgen -W "center bottom top" -- "$cur") )
|
|
return 0
|
|
;;
|
|
esac
|
|
|
|
if [[ "$cur" =~ ^-|\+ ]]; then
|
|
COMPREPLY=( $(compgen -W "${opts}" -- "$cur") )
|
|
return 0
|
|
fi
|
|
|
|
return 0
|
|
}
|
|
|
|
_fzf_handle_dynamic_completion() {
|
|
local cmd orig_var orig ret orig_cmd orig_complete
|
|
cmd="$1"
|
|
shift
|
|
orig_cmd="$1"
|
|
orig_var="_fzf_orig_completion_$cmd"
|
|
orig="${!orig_var-}"
|
|
orig="${orig##*#}"
|
|
if [[ -n "$orig" ]] && type "$orig" > /dev/null 2>&1; then
|
|
$orig "$@"
|
|
elif [[ -n "${_fzf_completion_loader-}" ]]; then
|
|
orig_complete=$(complete -p "$orig_cmd" 2> /dev/null)
|
|
_completion_loader "$@"
|
|
ret=$?
|
|
# _completion_loader may not have updated completion for the command
|
|
if [[ "$(complete -p "$orig_cmd" 2> /dev/null)" != "$orig_complete" ]]; then
|
|
__fzf_orig_completion < <(complete -p "$orig_cmd" 2> /dev/null)
|
|
if [[ "${__fzf_nospace_commands-}" = *" $orig_cmd "* ]]; then
|
|
eval "${orig_complete/ -F / -o nospace -F }"
|
|
else
|
|
eval "$orig_complete"
|
|
fi
|
|
fi
|
|
return $ret
|
|
fi
|
|
}
|
|
|
|
__fzf_generic_path_completion() {
|
|
local cur base dir leftover matches trigger cmd
|
|
cmd="${COMP_WORDS[0]}"
|
|
if [[ $cmd == \\* ]]; then
|
|
cmd="${cmd:1}"
|
|
fi
|
|
cmd="${cmd//[^A-Za-z0-9_=]/_}"
|
|
COMPREPLY=()
|
|
trigger=${FZF_COMPLETION_TRIGGER-'**'}
|
|
cur="${COMP_WORDS[COMP_CWORD]}"
|
|
if [[ "$cur" == *"$trigger" ]] && [[ $cur != *'$('* ]] && [[ $cur != *':='* ]] && [[ $cur != *'`'* ]]; then
|
|
base=${cur:0:${#cur}-${#trigger}}
|
|
eval "base=$base" 2> /dev/null || return
|
|
|
|
dir=
|
|
[[ $base = *"/"* ]] && dir="$base"
|
|
while true; do
|
|
if [[ -z "$dir" ]] || [[ -d "$dir" ]]; then
|
|
leftover=${base/#"$dir"}
|
|
leftover=${leftover/#\/}
|
|
[[ -z "$dir" ]] && dir='.'
|
|
[[ "$dir" != "/" ]] && dir="${dir/%\//}"
|
|
matches=$(eval "$1 $(printf %q "$dir")" | FZF_DEFAULT_OPTS="--height ${FZF_TMUX_HEIGHT:-40%} --reverse --scheme=path --bind=ctrl-z:ignore ${FZF_DEFAULT_OPTS-} ${FZF_COMPLETION_OPTS-} $2" __fzf_comprun "$4" -q "$leftover" | while read -r item; do
|
|
printf "%q " "${item%$3}$3"
|
|
done)
|
|
matches=${matches% }
|
|
[[ -z "$3" ]] && [[ "${__fzf_nospace_commands-}" = *" ${COMP_WORDS[0]} "* ]] && matches="$matches "
|
|
if [[ -n "$matches" ]]; then
|
|
COMPREPLY=( "$matches" )
|
|
else
|
|
COMPREPLY=( "$cur" )
|
|
fi
|
|
printf '\e[5n'
|
|
return 0
|
|
fi
|
|
dir=$(command dirname "$dir")
|
|
[[ "$dir" =~ /$ ]] || dir="$dir"/
|
|
done
|
|
else
|
|
shift
|
|
shift
|
|
shift
|
|
_fzf_handle_dynamic_completion "$cmd" "$@"
|
|
fi
|
|
}
|
|
|
|
_fzf_complete() {
|
|
# Split arguments around --
|
|
local args rest str_arg i sep
|
|
args=("$@")
|
|
sep=
|
|
for i in "${!args[@]}"; do
|
|
if [[ "${args[$i]}" = -- ]]; then
|
|
sep=$i
|
|
break
|
|
fi
|
|
done
|
|
if [[ -n "$sep" ]]; then
|
|
str_arg=
|
|
rest=("${args[@]:$((sep + 1)):${#args[@]}}")
|
|
args=("${args[@]:0:$sep}")
|
|
else
|
|
str_arg=$1
|
|
args=()
|
|
shift
|
|
rest=("$@")
|
|
fi
|
|
|
|
local cur selected trigger cmd post
|
|
post="$(caller 0 | command awk '{print $2}')_post"
|
|
type -t "$post" > /dev/null 2>&1 || post='command cat'
|
|
|
|
cmd="${COMP_WORDS[0]//[^A-Za-z0-9_=]/_}"
|
|
trigger=${FZF_COMPLETION_TRIGGER-'**'}
|
|
cur="${COMP_WORDS[COMP_CWORD]}"
|
|
if [[ "$cur" == *"$trigger" ]] && [[ $cur != *'$('* ]] && [[ $cur != *':='* ]] && [[ $cur != *'`'* ]]; then
|
|
cur=${cur:0:${#cur}-${#trigger}}
|
|
|
|
selected=$(FZF_DEFAULT_OPTS="--height ${FZF_TMUX_HEIGHT:-40%} --reverse --bind=ctrl-z:ignore ${FZF_DEFAULT_OPTS-} ${FZF_COMPLETION_OPTS-} $str_arg" __fzf_comprun "${rest[0]}" "${args[@]}" -q "$cur" | $post | command tr '\n' ' ')
|
|
selected=${selected% } # Strip trailing space not to repeat "-o nospace"
|
|
if [[ -n "$selected" ]]; then
|
|
COMPREPLY=("$selected")
|
|
else
|
|
COMPREPLY=("$cur")
|
|
fi
|
|
printf '\e[5n'
|
|
return 0
|
|
else
|
|
_fzf_handle_dynamic_completion "$cmd" "${rest[@]}"
|
|
fi
|
|
}
|
|
|
|
_fzf_path_completion() {
|
|
__fzf_generic_path_completion _fzf_compgen_path "-m" "" "$@"
|
|
}
|
|
|
|
# Deprecated. No file only completion.
|
|
_fzf_file_completion() {
|
|
_fzf_path_completion "$@"
|
|
}
|
|
|
|
_fzf_dir_completion() {
|
|
__fzf_generic_path_completion _fzf_compgen_dir "" "/" "$@"
|
|
}
|
|
|
|
_fzf_complete_kill() {
|
|
_fzf_proc_completion "$@"
|
|
}
|
|
|
|
_fzf_proc_completion() {
|
|
_fzf_complete -m --header-lines=1 --preview 'echo {}' --preview-window down:3:wrap --min-height 15 -- "$@" < <(
|
|
command ps -eo user,pid,ppid,start,time,command 2> /dev/null ||
|
|
command ps -eo user,pid,ppid,time,args # For BusyBox
|
|
)
|
|
}
|
|
|
|
_fzf_proc_completion_post() {
|
|
command awk '{print $2}'
|
|
}
|
|
|
|
# To use custom hostname lists, override __fzf_list_hosts.
|
|
# The function is expected to print hostnames, one per line as well as in the
|
|
# desired sorting and with any duplicates removed, to standard output.
|
|
if ! declare -F __fzf_list_hosts > /dev/null; then
|
|
__fzf_list_hosts() {
|
|
command cat <(command tail -n +1 ~/.ssh/config ~/.ssh/config.d/* /etc/ssh/ssh_config 2> /dev/null | command grep -i '^\s*host\(name\)\? ' | command awk '{for (i = 2; i <= NF; i++) print $1 " " $i}' | command grep -v '[*?%]') \
|
|
<(command grep -oE '^[[a-z0-9.,:-]+' ~/.ssh/known_hosts 2> /dev/null | command tr ',' '\n' | command tr -d '[' | command awk '{ print $1 " " $1 }') \
|
|
<(command grep -v '^\s*\(#\|$\)' /etc/hosts 2> /dev/null | command grep -Fv '0.0.0.0') |
|
|
command awk '{if (length($2) > 0) {print $2}}' | command sort -u
|
|
}
|
|
fi
|
|
|
|
_fzf_host_completion() {
|
|
_fzf_complete +m -- "$@" < <(__fzf_list_hosts)
|
|
}
|
|
|
|
# Values for $1 $2 $3 are described here
|
|
# https://www.gnu.org/software/bash/manual/html_node/Programmable-Completion.html
|
|
# > the first argument ($1) is the name of the command whose arguments are being completed,
|
|
# > the second argument ($2) is the word being completed,
|
|
# > and the third argument ($3) is the word preceding the word being completed on the current command line.
|
|
_fzf_complete_ssh() {
|
|
case $3 in
|
|
-i|-F|-E)
|
|
_fzf_path_completion "$@"
|
|
;;
|
|
*)
|
|
local user=
|
|
[[ "$2" =~ '@' ]] && user="${2%%@*}@"
|
|
_fzf_complete +m -- "$@" < <(__fzf_list_hosts | command awk -v user="$user" '{print user $0}')
|
|
;;
|
|
esac
|
|
}
|
|
|
|
_fzf_var_completion() {
|
|
_fzf_complete -m -- "$@" < <(
|
|
declare -xp | command sed -En 's|^declare [^ ]+ ([^=]+).*|\1|p'
|
|
)
|
|
}
|
|
|
|
_fzf_alias_completion() {
|
|
_fzf_complete -m -- "$@" < <(
|
|
alias | command sed -En 's|^alias ([^=]+).*|\1|p'
|
|
)
|
|
}
|
|
|
|
# fzf options
|
|
complete -o default -F _fzf_opts_completion fzf
|
|
# fzf-tmux is a thin fzf wrapper that has only a few more options than fzf
|
|
# itself. As a quick improvement we take fzf's completion. Adding the few extra
|
|
# fzf-tmux specific options (like `-w WIDTH`) are left as a future patch.
|
|
complete -o default -F _fzf_opts_completion fzf-tmux
|
|
|
|
d_cmds="${FZF_COMPLETION_DIR_COMMANDS:-cd pushd rmdir}"
|
|
a_cmds="
|
|
awk bat cat diff diff3
|
|
emacs emacsclient ex file ftp g++ gcc gvim head hg hx java
|
|
javac ld less more mvim nvim patch perl python ruby
|
|
sed sftp sort source tail tee uniq vi view vim wc xdg-open
|
|
basename bunzip2 bzip2 chmod chown curl cp dirname du
|
|
find git grep gunzip gzip hg jar
|
|
ln ls mv open rm rsync scp
|
|
svn tar unzip zip"
|
|
|
|
# Preserve existing completion
|
|
__fzf_orig_completion < <(complete -p $d_cmds $a_cmds 2> /dev/null)
|
|
|
|
if type _completion_loader > /dev/null 2>&1; then
|
|
_fzf_completion_loader=1
|
|
fi
|
|
|
|
__fzf_defc() {
|
|
local cmd func opts orig_var orig def
|
|
cmd="$1"
|
|
func="$2"
|
|
opts="$3"
|
|
orig_var="_fzf_orig_completion_${cmd//[^A-Za-z0-9_]/_}"
|
|
orig="${!orig_var-}"
|
|
if [[ -n "$orig" ]]; then
|
|
printf -v def "$orig" "$func"
|
|
eval "$def"
|
|
else
|
|
complete -F "$func" $opts "$cmd"
|
|
fi
|
|
}
|
|
|
|
# Anything
|
|
for cmd in $a_cmds; do
|
|
__fzf_defc "$cmd" _fzf_path_completion "-o default -o bashdefault"
|
|
done
|
|
|
|
# Directory
|
|
for cmd in $d_cmds; do
|
|
__fzf_defc "$cmd" _fzf_dir_completion "-o nospace -o dirnames"
|
|
done
|
|
|
|
# ssh
|
|
__fzf_defc ssh _fzf_complete_ssh "-o default -o bashdefault"
|
|
|
|
unset cmd d_cmds a_cmds
|
|
|
|
_fzf_setup_completion() {
|
|
local kind fn cmd
|
|
kind=$1
|
|
fn=_fzf_${1}_completion
|
|
if [[ $# -lt 2 ]] || ! type -t "$fn" > /dev/null; then
|
|
echo "usage: ${FUNCNAME[0]} path|dir|var|alias|host|proc COMMANDS..."
|
|
return 1
|
|
fi
|
|
shift
|
|
__fzf_orig_completion < <(complete -p "$@" 2> /dev/null)
|
|
for cmd in "$@"; do
|
|
case "$kind" in
|
|
dir) __fzf_defc "$cmd" "$fn" "-o nospace -o dirnames" ;;
|
|
var) __fzf_defc "$cmd" "$fn" "-o default -o nospace -v" ;;
|
|
alias) __fzf_defc "$cmd" "$fn" "-a" ;;
|
|
*) __fzf_defc "$cmd" "$fn" "-o default -o bashdefault" ;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
# Environment variables / Aliases / Hosts / Process
|
|
_fzf_setup_completion 'var' export unset printenv
|
|
_fzf_setup_completion 'alias' unalias
|
|
_fzf_setup_completion 'host' telnet
|
|
_fzf_setup_completion 'proc' kill
|