2016-07-20 05:04:18 +02:00
|
|
|
|
|
|
|
#--------------------------------------------------------------------#
|
|
|
|
# Async #
|
|
|
|
#--------------------------------------------------------------------#
|
|
|
|
|
|
|
|
_zsh_autosuggest_async_fetch_suggestion() {
|
2017-01-25 03:58:06 +01:00
|
|
|
# Send the prefix to the pty to fetch a suggestion
|
2017-01-25 05:58:17 +01:00
|
|
|
zpty -w -n $ZSH_AUTOSUGGEST_PTY_NAME "${1}"$'\0'
|
2016-07-20 05:04:18 +02:00
|
|
|
}
|
|
|
|
|
2017-01-25 03:53:26 +01:00
|
|
|
# Pty is spawned running this function
|
2017-01-25 03:55:38 +01:00
|
|
|
_zsh_autosuggest_async_suggestion_server() {
|
2017-01-25 03:58:25 +01:00
|
|
|
emulate -R zsh
|
|
|
|
|
2017-01-25 06:27:09 +01:00
|
|
|
# Output only newlines (not carriage return + newline)
|
|
|
|
stty -onlcr
|
|
|
|
|
2017-01-25 06:48:30 +01:00
|
|
|
local strategy=$1
|
2017-01-25 08:00:53 +01:00
|
|
|
local last_pid
|
2017-01-25 06:48:30 +01:00
|
|
|
|
2017-01-25 05:58:17 +01:00
|
|
|
while IFS='' read -r -d $'\0' prefix; do
|
2016-07-20 05:04:18 +02:00
|
|
|
# Kill last bg process
|
2017-01-25 08:00:53 +01:00
|
|
|
kill -KILL $last_pid &>/dev/null
|
2016-07-20 05:04:18 +02:00
|
|
|
|
|
|
|
# Run suggestion search in the background
|
2017-01-25 07:59:38 +01:00
|
|
|
(echo -n -E "$($strategy "$prefix")"$'\0') &
|
2017-01-25 08:00:53 +01:00
|
|
|
|
|
|
|
last_pid=$!
|
2016-07-20 05:04:18 +02:00
|
|
|
done
|
|
|
|
}
|
|
|
|
|
2017-01-25 03:53:26 +01:00
|
|
|
# Called when new data is ready to be read from the pty
|
|
|
|
# First arg will be fd ready for reading
|
|
|
|
# Second arg will be passed in case of error
|
2016-07-20 05:04:18 +02:00
|
|
|
_zsh_autosuggest_async_suggestion_ready() {
|
2017-01-25 03:58:52 +01:00
|
|
|
local suggestion
|
|
|
|
|
|
|
|
zpty -rt $ZSH_AUTOSUGGEST_PTY_NAME suggestion '*'$'\0' 2>/dev/null
|
|
|
|
zle _autosuggest-show-suggestion "${suggestion%$'\0'}"
|
2016-07-20 05:04:18 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
# Recreate the pty to get a fresh list of history events
|
|
|
|
_zsh_autosuggest_async_recreate_pty() {
|
|
|
|
typeset -g _ZSH_AUTOSUGGEST_PTY_FD
|
|
|
|
|
|
|
|
# Kill the old pty
|
|
|
|
if [ -n "$_ZSH_AUTOSUGGEST_PTY_FD" ]; then
|
2017-01-25 03:53:26 +01:00
|
|
|
# Remove the input handler
|
2016-07-20 05:04:18 +02:00
|
|
|
zle -F $_ZSH_AUTOSUGGEST_PTY_FD
|
2017-01-25 03:53:26 +01:00
|
|
|
|
|
|
|
# Destroy the pty
|
2016-07-20 05:04:18 +02:00
|
|
|
zpty -d $ZSH_AUTOSUGGEST_PTY_NAME &>/dev/null
|
|
|
|
fi
|
|
|
|
|
2017-01-27 00:50:19 +01:00
|
|
|
# With newer versions of zsh, REPLY stores the fd to read from
|
2016-07-20 05:04:18 +02:00
|
|
|
typeset -h REPLY
|
2017-01-25 03:53:26 +01:00
|
|
|
|
2017-01-27 00:50:19 +01:00
|
|
|
# If we won't get a fd back from zpty, try to guess it
|
|
|
|
if [ $_ZSH_AUTOSUGGEST_ZPTY_RETURNS_FD -eq 0 ]; then
|
|
|
|
integer -l zptyfd
|
|
|
|
exec {zptyfd}>&1 # Open a new file descriptor (above 10).
|
|
|
|
exec {zptyfd}>&- # Close it so it's free to be used by zpty.
|
|
|
|
fi
|
|
|
|
|
2017-01-25 03:53:26 +01:00
|
|
|
# Start a new pty running the server function
|
2017-01-25 06:48:30 +01:00
|
|
|
zpty -b $ZSH_AUTOSUGGEST_PTY_NAME "_zsh_autosuggest_async_suggestion_server _zsh_autosuggest_strategy_$ZSH_AUTOSUGGEST_STRATEGY"
|
2017-01-25 03:53:26 +01:00
|
|
|
|
|
|
|
# Store the fd so we can destroy this pty later
|
2017-01-27 00:50:19 +01:00
|
|
|
if (( REPLY )); then
|
|
|
|
_ZSH_AUTOSUGGEST_PTY_FD=$REPLY
|
|
|
|
else
|
|
|
|
_ZSH_AUTOSUGGEST_PTY_FD=$zptyfd
|
|
|
|
fi
|
|
|
|
|
2017-01-25 03:53:26 +01:00
|
|
|
|
|
|
|
# Set up input handler from the pty
|
2016-07-20 05:04:18 +02:00
|
|
|
zle -F $_ZSH_AUTOSUGGEST_PTY_FD _zsh_autosuggest_async_suggestion_ready
|
|
|
|
}
|