dotfiles: .zfun/zsh-autosuggestions/zsh-autosuggestions.zsh annotate

annotate .zfun/zsh-autosuggestions/zsh-autosuggestions.zsh @ 516:dbe413154ccd

env: wire up pyenv

author	Augie Fackler <raf@durin42.com>
date	Thu, 02 Dec 2021 22:08:45 -0500
parents	e1ce8897030d
children

rev	line source
467 e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	1 # Fish-like fast/unobtrusive autosuggestions for zsh.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	2 # https://github.com/zsh-users/zsh-autosuggestions
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	3 # v0.5.0
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	4 # Copyright (c) 2013 Thiago de Arruda
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	5 # Copyright (c) 2016-2018 Eric Freese
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	6 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	7 # Permission is hereby granted, free of charge, to any person
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	8 # obtaining a copy of this software and associated documentation
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	9 # files (the "Software"), to deal in the Software without
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	10 # restriction, including without limitation the rights to use,
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	11 # copy, modify, merge, publish, distribute, sublicense, and/or sell
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	12 # copies of the Software, and to permit persons to whom the
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	13 # Software is furnished to do so, subject to the following
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	14 # conditions:
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	15 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	16 # The above copyright notice and this permission notice shall be
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	17 # included in all copies or substantial portions of the Software.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	18 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	19 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	20 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	21 # OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	22 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	23 # HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	24 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	25 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	26 # OTHER DEALINGS IN THE SOFTWARE.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	27
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	28 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	29 # Setup #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	30 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	31
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	32 # Precmd hooks for initializing the library and starting pty's
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	33 autoload -Uz add-zsh-hook
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	34
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	35 # Asynchronous suggestions are generated in a pty
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	36 zmodload zsh/zpty
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	37
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	38 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	39 # Global Configuration Variables #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	40 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	41
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	42 # Color to use when highlighting suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	43 # Uses format of `region_highlight`
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	44 # More info: http://zsh.sourceforge.net/Doc/Release/Zsh-Line-Editor.html#Zle-Widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	45 : ${ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE='fg=8'}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	46
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	47 # Prefix to use when saving original versions of bound widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	48 : ${ZSH_AUTOSUGGEST_ORIGINAL_WIDGET_PREFIX=autosuggest-orig-}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	49
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	50 # Strategies to use to fetch a suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	51 # Will try each strategy in order until a suggestion is returned
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	52 (( ! ${+ZSH_AUTOSUGGEST_STRATEGY} )) && ZSH_AUTOSUGGEST_STRATEGY=(history)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	53
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	54 # Widgets that clear the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	55 (( ! ${+ZSH_AUTOSUGGEST_CLEAR_WIDGETS} )) && ZSH_AUTOSUGGEST_CLEAR_WIDGETS=(
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	56 history-search-forward
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	57 history-search-backward
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	58 history-beginning-search-forward
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	59 history-beginning-search-backward
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	60 history-substring-search-up
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	61 history-substring-search-down
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	62 up-line-or-beginning-search
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	63 down-line-or-beginning-search
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	64 up-line-or-history
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	65 down-line-or-history
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	66 accept-line
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	67 )
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	68
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	69 # Widgets that accept the entire suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	70 (( ! ${+ZSH_AUTOSUGGEST_ACCEPT_WIDGETS} )) && ZSH_AUTOSUGGEST_ACCEPT_WIDGETS=(
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	71 forward-char
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	72 end-of-line
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	73 vi-forward-char
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	74 vi-end-of-line
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	75 vi-add-eol
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	76 )
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	77
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	78 # Widgets that accept the entire suggestion and execute it
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	79 (( ! ${+ZSH_AUTOSUGGEST_EXECUTE_WIDGETS} )) && ZSH_AUTOSUGGEST_EXECUTE_WIDGETS=(
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	80 )
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	81
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	82 # Widgets that accept the suggestion as far as the cursor moves
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	83 (( ! ${+ZSH_AUTOSUGGEST_PARTIAL_ACCEPT_WIDGETS} )) && ZSH_AUTOSUGGEST_PARTIAL_ACCEPT_WIDGETS=(
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	84 forward-word
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	85 emacs-forward-word
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	86 vi-forward-word
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	87 vi-forward-word-end
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	88 vi-forward-blank-word
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	89 vi-forward-blank-word-end
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	90 vi-find-next-char
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	91 vi-find-next-char-skip
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	92 )
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	93
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	94 # Widgets that should be ignored (globbing supported but must be escaped)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	95 (( ! ${+ZSH_AUTOSUGGEST_IGNORE_WIDGETS} )) && ZSH_AUTOSUGGEST_IGNORE_WIDGETS=(
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	96 orig-\*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	97 beep
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	98 run-help
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	99 set-local-history
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	100 which-command
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	101 yank
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	102 yank-pop
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	103 )
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	104
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	105 # Max size of buffer to trigger autosuggestion. Leave null for no upper bound.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	106 : ${ZSH_AUTOSUGGEST_BUFFER_MAX_SIZE=}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	107
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	108 # Pty name for calculating autosuggestions asynchronously
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	109 : ${ZSH_AUTOSUGGEST_ASYNC_PTY_NAME=zsh_autosuggest_pty}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	110
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	111 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	112 # Utility Functions #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	113 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	114
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	115 _zsh_autosuggest_escape_command() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	116 setopt localoptions EXTENDED_GLOB
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	117
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	118 # Escape special chars in the string (requires EXTENDED_GLOB)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	119 echo -E "${1//(#m)[\"\'\\()\[\]\|*?~]/\\$MATCH}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	120 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	121
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	122 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	123 # Feature Detection #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	124 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	125
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	126 _zsh_autosuggest_feature_detect_zpty_returns_fd() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	127 typeset -g _ZSH_AUTOSUGGEST_ZPTY_RETURNS_FD
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	128 typeset -h REPLY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	129
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	130 zpty zsh_autosuggest_feature_detect '{ zshexit() { kill -KILL $$; sleep 1 } }'
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	131
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	132 if (( REPLY )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	133 _ZSH_AUTOSUGGEST_ZPTY_RETURNS_FD=1
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	134 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	135 _ZSH_AUTOSUGGEST_ZPTY_RETURNS_FD=0
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	136 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	137
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	138 zpty -d zsh_autosuggest_feature_detect
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	139 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	140
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	141 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	142 # Widget Helpers #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	143 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	144
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	145 _zsh_autosuggest_incr_bind_count() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	146 if ((${+_ZSH_AUTOSUGGEST_BIND_COUNTS[$1]})); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	147 ((_ZSH_AUTOSUGGEST_BIND_COUNTS[$1]++))
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	148 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	149 _ZSH_AUTOSUGGEST_BIND_COUNTS[$1]=1
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	150 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	151
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	152 typeset -gi bind_count=$_ZSH_AUTOSUGGEST_BIND_COUNTS[$1]
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	153 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	154
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	155 _zsh_autosuggest_get_bind_count() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	156 if ((${+_ZSH_AUTOSUGGEST_BIND_COUNTS[$1]})); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	157 typeset -gi bind_count=$_ZSH_AUTOSUGGEST_BIND_COUNTS[$1]
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	158 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	159 typeset -gi bind_count=0
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	160 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	161 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	162
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	163 # Bind a single widget to an autosuggest widget, saving a reference to the original widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	164 _zsh_autosuggest_bind_widget() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	165 typeset -gA _ZSH_AUTOSUGGEST_BIND_COUNTS
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	166
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	167 local widget=$1
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	168 local autosuggest_action=$2
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	169 local prefix=$ZSH_AUTOSUGGEST_ORIGINAL_WIDGET_PREFIX
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	170
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	171 local -i bind_count
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	172
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	173 # Save a reference to the original widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	174 case $widgets[$widget] in
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	175 # Already bound
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	176 user:_zsh_autosuggest_(bound\|orig)_*);;
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	177
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	178 # User-defined widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	179 user:*)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	180 _zsh_autosuggest_incr_bind_count $widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	181 zle -N $prefix${bind_count}-$widget ${widgets[$widget]#*:}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	182 ;;
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	183
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	184 # Built-in widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	185 builtin)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	186 _zsh_autosuggest_incr_bind_count $widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	187 eval "_zsh_autosuggest_orig_${(q)widget}() { zle .${(q)widget} }"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	188 zle -N $prefix${bind_count}-$widget _zsh_autosuggest_orig_$widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	189 ;;
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	190
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	191 # Completion widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	192 completion:*)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	193 _zsh_autosuggest_incr_bind_count $widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	194 eval "zle -C $prefix${bind_count}-${(q)widget} ${${(s.:.)widgets[$widget]}[2,3]}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	195 ;;
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	196 esac
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	197
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	198 _zsh_autosuggest_get_bind_count $widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	199
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	200 # Pass the original widget's name explicitly into the autosuggest
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	201 # function. Use this passed in widget name to call the original
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	202 # widget instead of relying on the $WIDGET variable being set
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	203 # correctly. $WIDGET cannot be trusted because other plugins call
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	204 # zle without the `-w` flag (e.g. `zle self-insert` instead of
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	205 # `zle self-insert -w`).
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	206 eval "_zsh_autosuggest_bound_${bind_count}_${(q)widget}() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	207 _zsh_autosuggest_widget_$autosuggest_action $prefix$bind_count-${(q)widget} \$@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	208 }"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	209
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	210 # Create the bound widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	211 zle -N -- $widget _zsh_autosuggest_bound_${bind_count}_$widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	212 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	213
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	214 # Map all configured widgets to the right autosuggest widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	215 _zsh_autosuggest_bind_widgets() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	216 emulate -L zsh
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	217
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	218 local widget
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	219 local ignore_widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	220
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	221 ignore_widgets=(
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	222 .\*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	223 _\*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	224 zle-\*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	225 autosuggest-\*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	226 $ZSH_AUTOSUGGEST_ORIGINAL_WIDGET_PREFIX\*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	227 $ZSH_AUTOSUGGEST_IGNORE_WIDGETS
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	228 )
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	229
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	230 # Find every widget we might want to bind and bind it appropriately
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	231 for widget in ${${(f)"$(builtin zle -la)"}:#${(j:\|:)~ignore_widgets}}; do
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	232 if [[ -n ${ZSH_AUTOSUGGEST_CLEAR_WIDGETS[(r)$widget]} ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	233 _zsh_autosuggest_bind_widget $widget clear
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	234 elif [[ -n ${ZSH_AUTOSUGGEST_ACCEPT_WIDGETS[(r)$widget]} ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	235 _zsh_autosuggest_bind_widget $widget accept
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	236 elif [[ -n ${ZSH_AUTOSUGGEST_EXECUTE_WIDGETS[(r)$widget]} ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	237 _zsh_autosuggest_bind_widget $widget execute
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	238 elif [[ -n ${ZSH_AUTOSUGGEST_PARTIAL_ACCEPT_WIDGETS[(r)$widget]} ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	239 _zsh_autosuggest_bind_widget $widget partial_accept
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	240 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	241 # Assume any unspecified widget might modify the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	242 _zsh_autosuggest_bind_widget $widget modify
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	243 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	244 done
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	245 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	246
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	247 # Given the name of an original widget and args, invoke it, if it exists
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	248 _zsh_autosuggest_invoke_original_widget() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	249 # Do nothing unless called with at least one arg
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	250 (( $# )) \|\| return 0
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	251
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	252 local original_widget_name="$1"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	253
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	254 shift
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	255
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	256 if (( ${+widgets[$original_widget_name]} )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	257 zle $original_widget_name -- $@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	258 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	259 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	260
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	261 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	262 # Highlighting #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	263 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	264
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	265 # If there was a highlight, remove it
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	266 _zsh_autosuggest_highlight_reset() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	267 typeset -g _ZSH_AUTOSUGGEST_LAST_HIGHLIGHT
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	268
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	269 if [[ -n "$_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	270 region_highlight=("${(@)region_highlight:#$_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT}")
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	271 unset _ZSH_AUTOSUGGEST_LAST_HIGHLIGHT
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	272 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	273 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	274
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	275 # If there's a suggestion, highlight it
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	276 _zsh_autosuggest_highlight_apply() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	277 typeset -g _ZSH_AUTOSUGGEST_LAST_HIGHLIGHT
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	278
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	279 if (( $#POSTDISPLAY )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	280 typeset -g _ZSH_AUTOSUGGEST_LAST_HIGHLIGHT="$#BUFFER $(($#BUFFER + $#POSTDISPLAY)) $ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	281 region_highlight+=("$_ZSH_AUTOSUGGEST_LAST_HIGHLIGHT")
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	282 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	283 unset _ZSH_AUTOSUGGEST_LAST_HIGHLIGHT
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	284 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	285 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	286
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	287 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	288 # Autosuggest Widget Implementations #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	289 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	290
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	291 # Disable suggestions
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	292 _zsh_autosuggest_disable() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	293 typeset -g _ZSH_AUTOSUGGEST_DISABLED
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	294 _zsh_autosuggest_clear
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	295 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	296
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	297 # Enable suggestions
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	298 _zsh_autosuggest_enable() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	299 unset _ZSH_AUTOSUGGEST_DISABLED
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	300
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	301 if (( $#BUFFER )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	302 _zsh_autosuggest_fetch
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	303 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	304 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	305
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	306 # Toggle suggestions (enable/disable)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	307 _zsh_autosuggest_toggle() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	308 if [[ -n "${_ZSH_AUTOSUGGEST_DISABLED+x}" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	309 _zsh_autosuggest_enable
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	310 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	311 _zsh_autosuggest_disable
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	312 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	313 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	314
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	315 # Clear the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	316 _zsh_autosuggest_clear() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	317 # Remove the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	318 unset POSTDISPLAY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	319
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	320 _zsh_autosuggest_invoke_original_widget $@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	321 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	322
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	323 # Modify the buffer and get a new suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	324 _zsh_autosuggest_modify() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	325 emulate -L zsh
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	326
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	327 local -i retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	328
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	329 # Only available in zsh >= 5.4
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	330 local -i KEYS_QUEUED_COUNT
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	331
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	332 # Save the contents of the buffer/postdisplay
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	333 local orig_buffer="$BUFFER"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	334 local orig_postdisplay="$POSTDISPLAY"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	335
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	336 # Clear suggestion while waiting for next one
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	337 unset POSTDISPLAY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	338
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	339 # Original widget may modify the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	340 _zsh_autosuggest_invoke_original_widget $@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	341 retval=$?
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	342
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	343 # Don't fetch a new suggestion if there's more input to be read immediately
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	344 if (( $PENDING > 0 )) \|\| (( $KEYS_QUEUED_COUNT > 0 )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	345 POSTDISPLAY="$orig_postdisplay"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	346 return $retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	347 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	348
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	349 # Optimize if manually typing in the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	350 if (( $#BUFFER > $#orig_buffer )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	351 local added=${BUFFER#$orig_buffer}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	352
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	353 # If the string added matches the beginning of the postdisplay
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	354 if [[ "$added" = "${orig_postdisplay:0:$#added}" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	355 POSTDISPLAY="${orig_postdisplay:$#added}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	356 return $retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	357 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	358 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	359
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	360 # Don't fetch a new suggestion if the buffer hasn't changed
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	361 if [[ "$BUFFER" = "$orig_buffer" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	362 POSTDISPLAY="$orig_postdisplay"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	363 return $retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	364 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	365
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	366 # Bail out if suggestions are disabled
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	367 if [[ -n "${_ZSH_AUTOSUGGEST_DISABLED+x}" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	368 return $?
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	369 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	370
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	371 # Get a new suggestion if the buffer is not empty after modification
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	372 if (( $#BUFFER > 0 )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	373 if [[ -z "$ZSH_AUTOSUGGEST_BUFFER_MAX_SIZE" ]] \|\| (( $#BUFFER <= $ZSH_AUTOSUGGEST_BUFFER_MAX_SIZE )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	374 _zsh_autosuggest_fetch
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	375 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	376 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	377
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	378 return $retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	379 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	380
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	381 # Fetch a new suggestion based on what's currently in the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	382 _zsh_autosuggest_fetch() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	383 if zpty -t "$ZSH_AUTOSUGGEST_ASYNC_PTY_NAME" &>/dev/null; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	384 _zsh_autosuggest_async_request "$BUFFER"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	385 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	386 local suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	387 _zsh_autosuggest_fetch_suggestion "$BUFFER"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	388 _zsh_autosuggest_suggest "$suggestion"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	389 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	390 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	391
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	392 # Offer a suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	393 _zsh_autosuggest_suggest() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	394 emulate -L zsh
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	395
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	396 local suggestion="$1"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	397
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	398 if [[ -n "$suggestion" ]] && (( $#BUFFER )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	399 POSTDISPLAY="${suggestion#$BUFFER}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	400 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	401 unset POSTDISPLAY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	402 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	403 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	404
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	405 # Accept the entire suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	406 _zsh_autosuggest_accept() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	407 local -i max_cursor_pos=$#BUFFER
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	408
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	409 # When vicmd keymap is active, the cursor can't move all the way
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	410 # to the end of the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	411 if [[ "$KEYMAP" = "vicmd" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	412 max_cursor_pos=$((max_cursor_pos - 1))
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	413 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	414
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	415 # Only accept if the cursor is at the end of the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	416 if [[ $CURSOR = $max_cursor_pos ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	417 # Add the suggestion to the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	418 BUFFER="$BUFFER$POSTDISPLAY"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	419
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	420 # Remove the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	421 unset POSTDISPLAY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	422
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	423 # Move the cursor to the end of the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	424 CURSOR=${#BUFFER}
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	425 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	426
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	427 _zsh_autosuggest_invoke_original_widget $@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	428 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	429
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	430 # Accept the entire suggestion and execute it
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	431 _zsh_autosuggest_execute() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	432 # Add the suggestion to the buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	433 BUFFER="$BUFFER$POSTDISPLAY"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	434
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	435 # Remove the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	436 unset POSTDISPLAY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	437
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	438 # Call the original `accept-line` to handle syntax highlighting or
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	439 # other potential custom behavior
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	440 _zsh_autosuggest_invoke_original_widget "accept-line"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	441 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	442
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	443 # Partially accept the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	444 _zsh_autosuggest_partial_accept() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	445 local -i retval cursor_loc
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	446
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	447 # Save the contents of the buffer so we can restore later if needed
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	448 local original_buffer="$BUFFER"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	449
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	450 # Temporarily accept the suggestion.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	451 BUFFER="$BUFFER$POSTDISPLAY"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	452
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	453 # Original widget moves the cursor
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	454 _zsh_autosuggest_invoke_original_widget $@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	455 retval=$?
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	456
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	457 # Normalize cursor location across vi/emacs modes
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	458 cursor_loc=$CURSOR
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	459 if [[ "$KEYMAP" = "vicmd" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	460 cursor_loc=$((cursor_loc + 1))
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	461 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	462
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	463 # If we've moved past the end of the original buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	464 if (( $cursor_loc > $#original_buffer )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	465 # Set POSTDISPLAY to text right of the cursor
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	466 POSTDISPLAY="${BUFFER[$(($cursor_loc + 1)),$#BUFFER]}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	467
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	468 # Clip the buffer at the cursor
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	469 BUFFER="${BUFFER[1,$cursor_loc]}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	470 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	471 # Restore the original buffer
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	472 BUFFER="$original_buffer"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	473 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	474
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	475 return $retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	476 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	477
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	478 for action in clear modify fetch suggest accept partial_accept execute enable disable toggle; do
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	479 eval "_zsh_autosuggest_widget_$action() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	480 local -i retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	481
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	482 _zsh_autosuggest_highlight_reset
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	483
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	484 _zsh_autosuggest_$action \$@
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	485 retval=\$?
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	486
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	487 _zsh_autosuggest_highlight_apply
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	488
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	489 zle -R
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	490
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	491 return \$retval
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	492 }"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	493 done
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	494
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	495 zle -N autosuggest-fetch _zsh_autosuggest_widget_fetch
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	496 zle -N autosuggest-suggest _zsh_autosuggest_widget_suggest
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	497 zle -N autosuggest-accept _zsh_autosuggest_widget_accept
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	498 zle -N autosuggest-clear _zsh_autosuggest_widget_clear
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	499 zle -N autosuggest-execute _zsh_autosuggest_widget_execute
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	500 zle -N autosuggest-enable _zsh_autosuggest_widget_enable
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	501 zle -N autosuggest-disable _zsh_autosuggest_widget_disable
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	502 zle -N autosuggest-toggle _zsh_autosuggest_widget_toggle
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	503
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	504 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	505 # History Suggestion Strategy #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	506 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	507 # Suggests the most recent history item that matches the given
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	508 # prefix.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	509 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	510
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	511 _zsh_autosuggest_strategy_history() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	512 # Reset options to defaults and enable LOCAL_OPTIONS
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	513 emulate -L zsh
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	514
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	515 # Enable globbing flags so that we can use (#m)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	516 setopt EXTENDED_GLOB
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	517
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	518 # Escape backslashes and all of the glob operators so we can use
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	519 # this string as a pattern to search the $history associative array.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	520 # - (#m) globbing flag enables setting references for match data
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	521 # TODO: Use (b) flag when we can drop support for zsh older than v5.0.8
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	522 local prefix="${1//(#m)[\\*?[\]<>()\|^~#]/\\$MATCH}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	523
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	524 # Get the history items that match
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	525 # - (r) subscript flag makes the pattern match on values
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	526 typeset -g suggestion="${history[(r)${prefix}*]}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	527 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	528
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	529 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	530 # Match Previous Command Suggestion Strategy #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	531 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	532 # Suggests the most recent history item that matches the given
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	533 # prefix and whose preceding history item also matches the most
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	534 # recently executed command.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	535 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	536 # For example, suppose your history has the following entries:
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	537 # - pwd
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	538 # - ls foo
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	539 # - ls bar
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	540 # - pwd
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	541 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	542 # Given the history list above, when you type 'ls', the suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	543 # will be 'ls foo' rather than 'ls bar' because your most recently
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	544 # executed command (pwd) was previously followed by 'ls foo'.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	545 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	546 # Note that this strategy won't work as expected with ZSH options that don't
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	547 # preserve the history order such as `HIST_IGNORE_ALL_DUPS` or
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	548 # `HIST_EXPIRE_DUPS_FIRST`.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	549
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	550 _zsh_autosuggest_strategy_match_prev_cmd() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	551 # Reset options to defaults and enable LOCAL_OPTIONS
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	552 emulate -L zsh
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	553
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	554 # Enable globbing flags so that we can use (#m)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	555 setopt EXTENDED_GLOB
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	556
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	557 # TODO: Use (b) flag when we can drop support for zsh older than v5.0.8
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	558 local prefix="${1//(#m)[\\*?[\]<>()\|^~#]/\\$MATCH}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	559
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	560 # Get all history event numbers that correspond to history
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	561 # entries that match pattern $prefix*
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	562 local history_match_keys
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	563 history_match_keys=(${(k)history[(R)$prefix*]})
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	564
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	565 # By default we use the first history number (most recent history entry)
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	566 local histkey="${history_match_keys[1]}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	567
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	568 # Get the previously executed command
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	569 local prev_cmd="$(_zsh_autosuggest_escape_command "${history[$((HISTCMD-1))]}")"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	570
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	571 # Iterate up to the first 200 history event numbers that match $prefix
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	572 for key in "${(@)history_match_keys[1,200]}"; do
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	573 # Stop if we ran out of history
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	574 [[ $key -gt 1 ]] \|\| break
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	575
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	576 # See if the history entry preceding the suggestion matches the
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	577 # previous command, and use it if it does
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	578 if [[ "${history[$((key - 1))]}" == "$prev_cmd" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	579 histkey="$key"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	580 break
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	581 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	582 done
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	583
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	584 # Give back the matched history entry
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	585 typeset -g suggestion="$history[$histkey]"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	586 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	587
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	588 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	589 # Fetch Suggestion #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	590 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	591 # Loops through all specified strategies and returns a suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	592 # from the first strategy to provide one.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	593 #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	594
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	595 _zsh_autosuggest_fetch_suggestion() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	596 typeset -g suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	597 local -a strategies
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	598
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	599 # Ensure we are working with an array
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	600 strategies=(${=ZSH_AUTOSUGGEST_STRATEGY})
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	601
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	602 for strategy in $strategies; do
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	603 # Try to get a suggestion from this strategy
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	604 _zsh_autosuggest_strategy_$strategy "$1"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	605
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	606 # Break once we've found a suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	607 [[ -n "$suggestion" ]] && break
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	608 done
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	609 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	610
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	611 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	612 # Async #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	613 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	614
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	615 # Zpty process is spawned running this function
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	616 _zsh_autosuggest_async_server() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	617 emulate -R zsh
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	618
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	619 # There is a bug in zpty module (fixed in zsh/master) by which a
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	620 # zpty that exits will kill all zpty processes that were forked
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	621 # before it. Here we set up a zsh exit hook to SIGKILL the zpty
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	622 # process immediately, before it has a chance to kill any other
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	623 # zpty processes.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	624 zshexit() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	625 kill -KILL $$
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	626 sleep 1 # Block for long enough for the signal to come through
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	627 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	628
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	629 # Don't add any extra carriage returns
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	630 stty -onlcr
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	631
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	632 # Don't translate carriage returns to newlines
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	633 stty -icrnl
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	634
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	635 # Silence any error messages
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	636 exec 2>/dev/null
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	637
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	638 local last_pid
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	639
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	640 while IFS='' read -r -d $'\0' query; do
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	641 # Kill last bg process
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	642 kill -KILL $last_pid &>/dev/null
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	643
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	644 # Run suggestion search in the background
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	645 (
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	646 local suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	647 _zsh_autosuggest_fetch_suggestion "$query"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	648 echo -n -E "$suggestion"$'\0'
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	649 ) &
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	650
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	651 last_pid=$!
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	652 done
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	653 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	654
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	655 _zsh_autosuggest_async_request() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	656 # Write the query to the zpty process to fetch a suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	657 zpty -w -n $ZSH_AUTOSUGGEST_ASYNC_PTY_NAME "${1}"$'\0'
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	658 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	659
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	660 # Called when new data is ready to be read from the pty
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	661 # First arg will be fd ready for reading
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	662 # Second arg will be passed in case of error
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	663 _zsh_autosuggest_async_response() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	664 setopt LOCAL_OPTIONS EXTENDED_GLOB
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	665
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	666 local suggestion
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	667
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	668 zpty -rt $ZSH_AUTOSUGGEST_ASYNC_PTY_NAME suggestion '*'$'\0' 2>/dev/null
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	669 zle autosuggest-suggest -- "${suggestion%%$'\0'##}"
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	670 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	671
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	672 _zsh_autosuggest_async_pty_create() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	673 # With newer versions of zsh, REPLY stores the fd to read from
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	674 typeset -h REPLY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	675
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	676 # If we won't get a fd back from zpty, try to guess it
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	677 if (( ! $_ZSH_AUTOSUGGEST_ZPTY_RETURNS_FD )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	678 integer -l zptyfd
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	679 exec {zptyfd}>&1 # Open a new file descriptor (above 10).
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	680 exec {zptyfd}>&- # Close it so it's free to be used by zpty.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	681 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	682
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	683 # Fork a zpty process running the server function
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	684 zpty -b $ZSH_AUTOSUGGEST_ASYNC_PTY_NAME _zsh_autosuggest_async_server
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	685
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	686 # Store the fd so we can remove the handler later
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	687 if (( REPLY )); then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	688 _ZSH_AUTOSUGGEST_PTY_FD=$REPLY
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	689 else
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	690 _ZSH_AUTOSUGGEST_PTY_FD=$zptyfd
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	691 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	692
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	693 # Set up input handler from the zpty
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	694 zle -F $_ZSH_AUTOSUGGEST_PTY_FD _zsh_autosuggest_async_response
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	695 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	696
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	697 _zsh_autosuggest_async_pty_destroy() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	698 # Remove the input handler
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	699 zle -F $_ZSH_AUTOSUGGEST_PTY_FD &>/dev/null
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	700
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	701 # Destroy the zpty
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	702 zpty -d $ZSH_AUTOSUGGEST_ASYNC_PTY_NAME &>/dev/null
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	703 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	704
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	705 _zsh_autosuggest_async_pty_recreate() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	706 _zsh_autosuggest_async_pty_destroy
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	707 _zsh_autosuggest_async_pty_create
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	708 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	709
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	710 _zsh_autosuggest_async_start() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	711 typeset -g _ZSH_AUTOSUGGEST_PTY_FD
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	712
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	713 _zsh_autosuggest_feature_detect_zpty_returns_fd
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	714 _zsh_autosuggest_async_pty_recreate
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	715
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	716 # We recreate the pty to get a fresh list of history events
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	717 add-zsh-hook precmd _zsh_autosuggest_async_pty_recreate
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	718 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	719
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	720 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	721 # Start #
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	722 #--------------------------------------------------------------------#
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	723
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	724 # Start the autosuggestion widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	725 _zsh_autosuggest_start() {
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	726 add-zsh-hook -d precmd _zsh_autosuggest_start
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	727
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	728 _zsh_autosuggest_bind_widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	729
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	730 # Re-bind widgets on every precmd to ensure we wrap other wrappers.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	731 # Specifically, highlighting breaks if our widgets are wrapped by
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	732 # zsh-syntax-highlighting widgets. This also allows modifications
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	733 # to the widget list variables to take effect on the next precmd.
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	734 add-zsh-hook precmd _zsh_autosuggest_bind_widgets
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	735
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	736 if [[ -n "${ZSH_AUTOSUGGEST_USE_ASYNC+x}" ]]; then
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	737 _zsh_autosuggest_async_start
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	738 fi
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	739 }
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	740
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	741 # Start the autosuggestion widgets on the next precmd
e1ce8897030d zsh: import df6f6f9ff41 of zsh-autosuggestions Augie Fackler <raf@durin42.com> parents: diff changeset	742 add-zsh-hook precmd _zsh_autosuggest_start

Mercurial > dotfiles

annotate .zfun/zsh-autosuggestions/zsh-autosuggestions.zsh @ 516:dbe413154ccd