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

Improved criteria used to popup autosuggestions

Browse Source
This commit is contained in:
Thiago de Arruda 2013-10-29 10:46:39 -03:00
parent 4ca0501c25
commit 2d67624963
3 changed files with 74 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
autosuggestions.zsh
View File

@ -27,14 +27,14 @@ menu-complete reverse-menu-complete menu-expand-or-complete menu-select
accept-and-menu-complete accept-and-menu-complete
) )
pause-autosuggestions() { autosuggest-pause-autosuggestions() {
[[ -n $ZLE_AUTOSUGGESTING_PAUSED ]] && return [[ -n $ZLE_AUTOSUGGESTING_PAUSED ]] && return
local widget local widget
# When autosuggestions are disabled, kill the unmaterialized part # When autosuggestions are disabled, kill the unmaterialized part
RBUFFER='' RBUFFER=''
unset ZLE_AUTOSUGGESTING unset ZLE_AUTOSUGGESTING
ZLE_AUTOSUGGESTING_PAUSED=1 ZLE_AUTOSUGGESTING_PAUSED=1
zle -A self-insert paused-autosuggest-self-insert zle -A self-insert autosuggest-paused-self-insert
zle -A .magic-space magic-space zle -A .magic-space magic-space
zle -A .backward-delete-char backward-delete-char zle -A .backward-delete-char backward-delete-char
zle -A .accept-line accept-line zle -A .accept-line accept-line
@ -44,7 +44,7 @@ pause-autosuggestions() {
for widget in $ZLE_AUTOSUGGEST_COMPLETION_WIDGETS; do for widget in $ZLE_AUTOSUGGEST_COMPLETION_WIDGETS; do
eval "zle -A autosuggest-${widget}-orig $widget" eval "zle -A autosuggest-${widget}-orig $widget"
done done
highlight-suggested-text autosuggest-highlight-suggested-text
} }
enable-autosuggestions() { enable-autosuggestions() {
@ -53,9 +53,9 @@ enable-autosuggestions() {
unset ZLE_AUTOSUGGESTING_PAUSED unset ZLE_AUTOSUGGESTING_PAUSED
ZLE_AUTOSUGGESTING=1 ZLE_AUTOSUGGESTING=1
# Replace prediction widgets by versions that will also highlight RBUFFER # Replace prediction widgets by versions that will also highlight RBUFFER
zle -N self-insert autosuggest-self-insert zle -N self-insert autosuggest-insert-or-space
zle -N magic-space autosuggest-self-insert zle -N magic-space autosuggest-insert-or-space
zle -N backward-delete-char autosuggest-delete zle -N backward-delete-char autosuggest-backward-delete-char
zle -N accept-line autosuggest-accept-line zle -N accept-line autosuggest-accept-line
# Hook into some default widgets that should pause autosuggestion # Hook into some default widgets that should pause autosuggestion
# automatically # automatically
@ -69,13 +69,13 @@ enable-autosuggestions() {
zle -A autosuggest-tab $widget" zle -A autosuggest-tab $widget"
done done
if [[ $BUFFER != '' ]]; then if [[ $BUFFER != '' ]]; then
show-suggestion autosuggest-pop-suggestion
fi fi
} }
disable-autosuggestions() { disable-autosuggestions() {
if [[ -z $ZLE_AUTOSUGGESTING_PAUSED ]]; then if [[ -z $ZLE_AUTOSUGGESTING_PAUSED ]]; then
pause-autosuggestions autosuggest-pause-autosuggestions
fi fi
unset ZLE_AUTOSUGGESTING_PAUSED unset ZLE_AUTOSUGGESTING_PAUSED
zle -A .self-insert self-insert zle -A .self-insert self-insert
@ -90,7 +90,7 @@ toggle-autosuggestions() {
fi fi
} }
highlight-suggested-text() { autosuggest-highlight-suggested-text() {
if [[ -n $ZLE_AUTOSUGGESTING ]]; then if [[ -n $ZLE_AUTOSUGGESTING ]]; then
local color='fg=8' local color='fg=8'
[[ -n $AUTOSUGGESTION_HIGHLIGHT_COLOR ]] &&\ [[ -n $AUTOSUGGESTION_HIGHLIGHT_COLOR ]] &&\
@ -101,36 +101,36 @@ highlight-suggested-text() {
fi fi
} }
autosuggest-self-insert() { autosuggest-insert-or-space() {
setopt localoptions noshwordsplit noksharrays if [[ $LBUFFER == *$'\012'* ]] || (( PENDING )); then
if [[ ${RBUFFER[1]} == ${KEYS[-1]} ]]; then # Editing a multiline buffer or pasting in a chunk of text, dont
# Same as what's typed, just move on # autosuggest
zle .$WIDGET "$@"
elif [[ ${RBUFFER[1]} == ${KEYS[-1]} ]]; then
# Same as what's typed, just move on
((++CURSOR)) ((++CURSOR))
highlight-suggested-text autosuggest-highlight-suggested-text
else else
LBUFFER="$LBUFFER$KEYS" LBUFFER="$LBUFFER$KEYS"
show-suggestion autosuggest-pop-suggestion
fi fi
} }
# Taken from predict-on autosuggest-backward-delete-char() {
autosuggest-delete() { if ! (( $CURSOR )); then
if (( $#LBUFFER > 1 )); then zle .kill-whole-line
setopt localoptions noshwordsplit noksharrays return
# When editing a multiline buffer, it's unlikely prediction is wanted; fi
# or if the last widget was e.g. a motion, then probably the intent is
# to actually edit the line, not change the search prefix. if [[ $LBUFFER == *$'\012'* || $LASTWIDGET != (self-insert|magic-space|backward-delete-char) ]]; then
if [[ $LASTWIDGET != (self-insert|magic-space|backward-delete-char) ]]; then # When editing a multiline buffer or if the last widget was e.g. a motion,
LBUFFER="$LBUFFER[1,-2]" # then probably the intent is to actually edit the line, not change the
else # search prefix.
((--CURSOR)) LBUFFER="$LBUFFER[1,-2]"
zle .history-beginning-search-forward || RBUFFER="" else
return 0 ((--CURSOR))
fi zle .history-beginning-search-forward || RBUFFER=''
else
zle .kill-whole-line
fi fi
highlight-suggested-text
} }
# When autosuggesting, ignore RBUFFER which corresponds to the 'unmaterialized' # When autosuggesting, ignore RBUFFER which corresponds to the 'unmaterialized'
@ -141,50 +141,56 @@ autosuggest-accept-line() {
zle .accept-line zle .accept-line
} }
paused-autosuggest-self-insert() { autosuggest-paused-self-insert() {
if [[ $RBUFFER == '' ]]; then if [[ $RBUFFER == '' ]]; then
# Resume autosuggestions when inserting at the end of the line # Resume autosuggestions when inserting at the end of the line
enable-autosuggestions enable-autosuggestions
autosuggest-self-insert zle autosuggest-modify
else else
zle .self-insert zle .self-insert
fi fi
} }
autosuggest-get-suggested-completion() { autosuggest-get-suggested-completion() {
if (( $CURSOR == 0 )) || [[ ${LBUFFER[-1]} == ' ' ]]; then
RBUFFER=''
return
fi
local words last_word local words last_word
local suggestion=$(autosuggest-first-completion $LBUFFER) local suggestion="$(autosuggest-first-completion ${LBUFFER})"
words=(${(z)LBUFFER}) words=(${(z)LBUFFER})
last_word=${words[-1]} last_word=${words[-1]}
suggestion=${suggestion:$#last_word} suggestion=${suggestion:$#last_word}
RBUFFER="$suggestion" RBUFFER="$suggestion"
} }
show-suggestion() { autosuggest-pop-suggestion() {
[[ -n $ZLE_DISABLE_AUTOSUGGEST || $LBUFFER == '' ]] && return [[ -n $ZLE_DISABLE_AUTOSUGGEST || $LBUFFER == '' ]] && return
zle .history-beginning-search-backward ||\ zle .history-beginning-search-backward ||\
autosuggest-get-suggested-completion autosuggest-get-suggested-completion
highlight-suggested-text autosuggest-highlight-suggested-text
} }
autosuggest-pause() { autosuggest-pause() {
pause-autosuggestions autosuggest-pause-autosuggestions
zle autosuggest-${WIDGET}-orig "$@" zle autosuggest-${WIDGET}-orig "$@"
} }
autosuggest-tab() { autosuggest-tab() {
RBUFFER='' RBUFFER=''
zle autosuggest-${WIDGET}-orig "$@" zle autosuggest-${WIDGET}-orig "$@"
autosuggest-highlight-suggested-text
} }
accept-suggested-small-word() { accept-suggested-small-word() {
zle .vi-forward-word zle .vi-forward-word
highlight-suggested-text autosuggest-highlight-suggested-text
} }
accept-suggested-word() { accept-suggested-word() {
zle .forward-word zle .forward-word
highlight-suggested-text autosuggest-highlight-suggested-text
} }
zle -N toggle-autosuggestions zle -N toggle-autosuggestions

12
completion-client.zsh
View File

@ -9,9 +9,13 @@ autosuggest-ensure-server() {
local pid_file="$server_dir/pid" local pid_file="$server_dir/pid"
local socket_path="$server_dir/socket" local socket_path="$server_dir/socket"
[[ -S $socket_path && -r $pid_file ]] && \ if [[ ! -S $socket_path || ! -r $pid_file ]] || ! kill -0 $(<$pid_file) &> /dev/null; then
kill -0 $(<$pid_file) &> /dev/null || \ if which setsid &> /dev/null; then
zsh $AUTOSUGGEST_SERVER_SCRIPT $server_dir $pid_file $socket_path &! setsid zsh $AUTOSUGGEST_SERVER_SCRIPT $server_dir $pid_file $socket_path &!
else
zsh $AUTOSUGGEST_SERVER_SCRIPT $server_dir $pid_file $socket_path &!
fi
fi
integer remaining_tries=10 integer remaining_tries=10
# wait until the process is listening # wait until the process is listening
@ -28,7 +32,7 @@ autosuggest-first-completion() {
local connection=$REPLY local connection=$REPLY
local completion local completion
print -u $connection - $1 print -u $connection - $1
while read -u $connection completion; do while IFS= read -r -u $connection completion; do
print - ${completion} print - ${completion}
done done
# close fd # close fd

25
completion-server.zsh
View File

@ -2,20 +2,25 @@
# Based on: # Based on:
# https://github.com/Valodim/zsh-capture-completion/blob/master/capture.zsh # https://github.com/Valodim/zsh-capture-completion/blob/master/capture.zsh
# close stdio if [[ -n $ZLE_AUTOSUGGEST_SERVER_LOG ]]; then
exec &> /dev/null exec &>> "$HOME/.autosuggest-server.log"
else
exec &> /dev/null
fi
exec < /dev/null exec < /dev/null
zmodload zsh/zpty zmodload zsh/zpty
zmodload zsh/net/socket zmodload zsh/net/socket
setopt noglob setopt noglob
print "autosuggestion server started, pid: $$"
# Start an interactive zsh connected to a zpty # Start an interactive zsh connected to a zpty
zpty z ZLE_DISABLE_AUTOSUGGEST=1 zsh -i zpty z ZLE_DISABLE_AUTOSUGGEST=1 zsh -i
print 'interactive shell started'
# Source the init script # Source the init script
zpty -w z "source '${0:a:h}/completion-server-init.zsh'" zpty -w z "source '${0:a:h}/completion-server-init.zsh'"
# read all completions and return the longest match # read everything until a line containing the byte 0 is found
read-to-null() { read-to-null() {
while zpty -r z chunk; do while zpty -r z chunk; do
[[ $chunk == *$'\0'* ]] && break [[ $chunk == *$'\0'* ]] && break
@ -26,6 +31,7 @@ read-to-null() {
# wait for ok from shell # wait for ok from shell
read-to-null &> /dev/null read-to-null &> /dev/null
print 'interactive shell ready'
# listen on a socket for completion requests # listen on a socket for completion requests
server_dir=$1 server_dir=$1
@ -34,7 +40,9 @@ socket_path=$3
cleanup() { cleanup() {
print 'removing socket and pid file...'
rm -f $socket_path $pid_file rm -f $socket_path $pid_file
print "autosuggestion server stopped, pid: $$"
} }
trap cleanup TERM INT HUP EXIT trap cleanup TERM INT HUP EXIT
@ -47,14 +55,18 @@ while ! zsocket -l $socket_path; do
else else
exit 1 exit 1
fi fi
print "will retry listening on '$socket_path'"
done done
print $$ > $pid_file
server=$REPLY server=$REPLY
print "server listening on '$socket_path'"
print $$ > $pid_file
while zsocket -a $server &> /dev/null; do while zsocket -a $server &> /dev/null; do
connection=$REPLY connection=$REPLY
print "connection accepted, fd: $connection"
# connection accepted, read the request and send response # connection accepted, read the request and send response
while read -u $connection prefix &> /dev/null; do while read -u $connection prefix &> /dev/null; do
# send the prefix to be completed followed by a TAB to force # send the prefix to be completed followed by a TAB to force
@ -63,7 +75,7 @@ while zsocket -a $server &> /dev/null; do
zpty -r z chunk &> /dev/null # read empty line before completions zpty -r z chunk &> /dev/null # read empty line before completions
local current='' local current=''
# read completions one by one, storing the longest match # read completions one by one, storing the longest match
read-to-null | while read line; do read-to-null | while IFS= read -r line; do
(( $#line > $#current )) && current=$line (( $#line > $#current )) && current=$line
done done
# send the longest completion back to the client, strip the last # send the longest completion back to the client, strip the last
@ -75,6 +87,7 @@ while zsocket -a $server &> /dev/null; do
fi fi
# close fd # close fd
exec {connection}>&- exec {connection}>&-
print "connection closed, fd: $connection"
# clear input buffer # clear input buffer
zpty -w z $'\n' zpty -w z $'\n'
done done