ylib/zsh-autosuggestions
1
0
Fork 0
mirror of https://github.com/zsh-users/zsh-autosuggestions.git synced 2024-12-29 07:34:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Modify completion code to better fit our needs

Browse Source 
Only need the first completion result
This commit is contained in:
Eric Freese 2018-05-18 15:05:45 -06:00
parent f63afd5969
commit 4cca26ec84
2 changed files with 85 additions and 164 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

124
src/strategies/completion.zsh
View File

@ -9,114 +9,74 @@
_zsh_autosuggest_capture_completion() { _zsh_autosuggest_capture_completion() {
zpty $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME zsh -f -i zpty $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME zsh -f -i
# line buffer for pty output
local line local line
setopt rcquotes setopt rcquotes
() { () {
zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME source $1 # Initialize the pty env, blocking until null byte is seen
repeat 4; do zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME "source $1"
zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line '*'$'\0'
[[ $line == ok* ]] && return
done
echo 'error initializing.' >&2
exit 2
} =( <<< ' } =( <<< '
# no prompt! exec 2>/dev/null # Silence any error messages
PROMPT=
# load completion system
autoload compinit autoload compinit
compinit -d ~/.zcompdump_autosuggestions compinit -d ~/.zcompdump_autosuggestions
# never run a command
bindkey ''^M'' undefined # Exit as soon as completion is finished
bindkey ''^J'' undefined comppostfuncs=( exit )
bindkey ''^I'' complete-word
# send a line with null-byte at the end before and after completions are output # Never group stuff!
null-line () {
echo -E - $''\0''
}
compprefuncs=( null-line )
comppostfuncs=( null-line exit )
# never group stuff!
zstyle '':completion:*'' list-grouped false zstyle '':completion:*'' list-grouped false
# don''t insert tab when attempting completion on empty line
zstyle '':completion:*'' insert-tab false
# no list separator, this saves some stripping later on # no list separator, this saves some stripping later on
zstyle '':completion:*'' list-separator '''' zstyle '':completion:*'' list-separator ''''
# we use zparseopts # we use zparseopts
zmodload zsh/zutil zmodload zsh/zutil
# override compadd (this our hook) # override compadd (this our hook)
compadd () { compadd () {
# check if any of -O, -A or -D are given # Just delegate and leave if any of -O, -A or -D are given
if [[ ${@[1,(i)(-|--)]} == *-(O|A|D)\ * ]]; then if [[ ${@[1,(i)(-|--)]} == *-(O|A|D)\ * ]]; then
# if that is the case, just delegate and leave
builtin compadd "$@" builtin compadd "$@"
return $? return $?
fi fi
# ok, this concerns us!
# echo -E - got this: "$@"
# be careful with namespacing here, we don''t want to mess with stuff that
# should be passed to compadd!
typeset -a __hits __dscr __tmp
# do we have a description parameter?
# note we don''t use zparseopts here because of combined option parameters
# with arguments like -default- confuse it.
if (( $@[(I)-d] )); then # kind of a hack, $+@[(r)-d] doesn''t work because of line noise overload
# next param after -d
__tmp=${@[$[${@[(i)-d]}+1]]}
# description can be given as an array parameter name, or inline () array
if [[ $__tmp == \(* ]]; then
eval "__dscr=$__tmp"
else
__dscr=( "${(@P)__tmp}" )
fi
fi
# capture completions by injecting -A parameter into the compadd call.
# this takes care of matching for us.
builtin compadd -A __hits -D __dscr "$@"
# JESUS CHRIST IT TOOK ME FOREVER TO FIGURE OUT THIS OPTION WAS SET AND WAS MESSING WITH MY SHIT HERE
setopt localoptions norcexpandparam extendedglob setopt localoptions norcexpandparam extendedglob
# extract prefixes and suffixes from compadd call. we can''t do zsh''s cool
typeset -a __hits
# Capture completions by injecting -A parameter into the compadd call.
# This takes care of matching for us.
builtin compadd -A __hits "$@"
# Exit if no completion results
[[ -n $__hits ]] || return
# Extract prefixes and suffixes from compadd call. we can''t do zsh''s cool
# -r remove-func magic, but it''s better than nothing. # -r remove-func magic, but it''s better than nothing.
typeset -A apre hpre hsuf asuf typeset -A apre hpre hsuf asuf
zparseopts -E P:=apre p:=hpre S:=asuf s:=hsuf zparseopts -E P:=apre p:=hpre S:=asuf s:=hsuf
# append / to directories? we are only emulating -f in a half-assed way
# here, but it''s better than nothing.
integer dirsuf=0
# don''t be fooled by -default- >.>
if [[ -z $hsuf && "${${@//-default-/}% -# *}" == *-[[:alnum:]]#f* ]]; then
dirsuf=1
fi
# just drop
[[ -n $__hits ]] || return
# this is the point where we have all matches in $__hits and all
# descriptions in $__dscr!
# display all matches
local dsuf dscr
for i in {1..$#__hits}; do
# add a dir suffix?
(( dirsuf )) && [[ -d $__hits[$i] ]] && dsuf=/ || dsuf=
# description to be displayed afterwards
(( $#__dscr >= $i )) && dscr=" -- ${${__dscr[$i]}##$__hits[$i] #}" || dscr=
echo -E - $IPREFIX$apre$hpre$__hits[$i]$dsuf$hsuf$asuf
done
}
# signal success!
echo ok')
# Print the first match
echo -nE - $''\0''$IPREFIX$apre$hpre$__hits[1]$dsuf$hsuf$asuf$''\0''
}
# Signal setup completion by sending null byte
echo $''\0''
')
# Send the string and a tab to trigger completion
zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME "$*"$'\t' zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME "$*"$'\t'
integer tog=0 # Read up to the start of the first result
# read from the pty, and parse linewise zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line '*'$'\0'
while zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME; do :; done | while IFS= read -r line; do
if [[ $line == *$'\0\r' ]]; then
(( tog++ )) && return 0 || continue
fi
# display between toggles
(( tog )) && echo -E - $line
done
return 2 # Read the first result
zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line '*'$'\0'
# Print it, removing the trailing null byte
echo -E - ${line%$'\0'}
} }
_zsh_autosuggest_strategy_completion() { _zsh_autosuggest_strategy_completion() {

125
zsh-autosuggestions.zsh
View File

@ -506,114 +506,75 @@ zle -N autosuggest-toggle _zsh_autosuggest_widget_toggle
_zsh_autosuggest_capture_completion() { _zsh_autosuggest_capture_completion() {
zpty $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME zsh -f -i zpty $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME zsh -f -i
# line buffer for pty output
local line local line
setopt rcquotes setopt rcquotes
() { () {
zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME source $1 # Setup, blocking until null byte to signal completion
repeat 4; do zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME "source $1"
zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line '*'$'\0'
[[ $line == ok* ]] && return
done
echo 'error initializing.' >&2
exit 2
} =( <<< ' } =( <<< '
# no prompt! exec 2>/dev/null # Silence any error messages
PROMPT=
# load completion system
autoload compinit autoload compinit
compinit -d ~/.zcompdump_autosuggestions compinit -d ~/.zcompdump_autosuggestions
# never run a command
bindkey ''^M'' undefined # Exit as soon as completion is finished
bindkey ''^J'' undefined comppostfuncs=( exit )
bindkey ''^I'' complete-word
# send a line with null-byte at the end before and after completions are output # Never group stuff!
null-line () {
echo -E - $''\0''
}
compprefuncs=( null-line )
comppostfuncs=( null-line exit )
# never group stuff!
zstyle '':completion:*'' list-grouped false zstyle '':completion:*'' list-grouped false
# don''t insert tab when attempting completion on empty line
zstyle '':completion:*'' insert-tab false
# no list separator, this saves some stripping later on # no list separator, this saves some stripping later on
zstyle '':completion:*'' list-separator '''' zstyle '':completion:*'' list-separator ''''
# we use zparseopts # we use zparseopts
zmodload zsh/zutil zmodload zsh/zutil
# override compadd (this our hook) # override compadd (this our hook)
compadd () { compadd () {
# check if any of -O, -A or -D are given # Just delegate and leave if any of -O, -A or -D are given
if [[ ${@[1,(i)(-|--)]} == *-(O|A|D)\ * ]]; then if [[ ${@[1,(i)(-|--)]} == *-(O|A|D)\ * ]]; then
# if that is the case, just delegate and leave
builtin compadd "$@" builtin compadd "$@"
return $? return $?
fi fi
# ok, this concerns us!
# echo -E - got this: "$@"
# be careful with namespacing here, we don''t want to mess with stuff that
# should be passed to compadd!
typeset -a __hits __dscr __tmp
# do we have a description parameter?
# note we don''t use zparseopts here because of combined option parameters
# with arguments like -default- confuse it.
if (( $@[(I)-d] )); then # kind of a hack, $+@[(r)-d] doesn''t work because of line noise overload
# next param after -d
__tmp=${@[$[${@[(i)-d]}+1]]}
# description can be given as an array parameter name, or inline () array
if [[ $__tmp == \(* ]]; then
eval "__dscr=$__tmp"
else
__dscr=( "${(@P)__tmp}" )
fi
fi
# capture completions by injecting -A parameter into the compadd call.
# this takes care of matching for us.
builtin compadd -A __hits -D __dscr "$@"
# JESUS CHRIST IT TOOK ME FOREVER TO FIGURE OUT THIS OPTION WAS SET AND WAS MESSING WITH MY SHIT HERE
setopt localoptions norcexpandparam extendedglob setopt localoptions norcexpandparam extendedglob
# extract prefixes and suffixes from compadd call. we can''t do zsh''s cool
typeset -a __hits
# Capture completions by injecting -A parameter into the compadd call.
# This takes care of matching for us.
builtin compadd -A __hits "$@"
# Exit if no completion results
[[ -n $__hits ]] || return
# Extract prefixes and suffixes from compadd call. we can''t do zsh''s cool
# -r remove-func magic, but it''s better than nothing. # -r remove-func magic, but it''s better than nothing.
typeset -A apre hpre hsuf asuf typeset -A apre hpre hsuf asuf
zparseopts -E P:=apre p:=hpre S:=asuf s:=hsuf zparseopts -E P:=apre p:=hpre S:=asuf s:=hsuf
# append / to directories? we are only emulating -f in a half-assed way
# here, but it''s better than nothing.
integer dirsuf=0
# don''t be fooled by -default- >.>
if [[ -z $hsuf && "${${@//-default-/}% -# *}" == *-[[:alnum:]]#f* ]]; then
dirsuf=1
fi
# just drop
[[ -n $__hits ]] || return
# this is the point where we have all matches in $__hits and all
# descriptions in $__dscr!
# display all matches
local dsuf dscr
for i in {1..$#__hits}; do
# add a dir suffix?
(( dirsuf )) && [[ -d $__hits[$i] ]] && dsuf=/ || dsuf=
# description to be displayed afterwards
(( $#__dscr >= $i )) && dscr=" -- ${${__dscr[$i]}##$__hits[$i] #}" || dscr=
echo -E - $IPREFIX$apre$hpre$__hits[$i]$dsuf$hsuf$asuf
done
}
# signal success!
echo ok')
# Print the first match
echo -nE - $''\0''$IPREFIX$apre$hpre$__hits[1]$dsuf$hsuf$asuf$''\0''
}
# Signal setup completion by sending null byte
echo $''\0''
')
# Send the string and a tab to trigger completion
zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME "$*"$'\t' zpty -w $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME "$*"$'\t'
integer tog=0 # Read up to the start of the first result
# read from the pty, and parse linewise zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line '*'$'\0'
while zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME; do :; done | while IFS= read -r line; do
if [[ $line == *$'\0\r' ]]; then
(( tog++ )) && return 0 || continue
fi
# display between toggles
(( tog )) && echo -E - $line
done
return 2 # Read the first result
zpty -r $ZSH_AUTOSUGGEST_COMPLETIONS_PTY_NAME line '*'$'\0'
# Print it, removing the trailing null byte
echo -E - ${line%$'\0'}
} }
_zsh_autosuggest_strategy_completion() { _zsh_autosuggest_strategy_completion() {