zsh-autosuggestions/src/strategies/predefined.zsh
2018-03-07 17:11:12 +08:00

107 lines
3.2 KiB
Bash

# zsh-augosuggestion: predefined strategy
# 1. search in history, returns result if it exists
# 2. search in predefined files
#
# configuration:
#
# ZSH_AUTOSUGGEST_PREDEFINE_PATH - user defined files separated by semicolon
#
# "~/.local/share/zsh_autosuggest/predefined" will be generated at the first time
#
# zsh-autosuggestions/predefined.txt is generated from tldr pages and
# will ship with predefined.zsh .
#
_zsh_autosuggest_script_path="${0:A:h}"
_zsh_autosuggest_data_home="${XDG_DATA_HOME:-$HOME/.local/share}/zsh_autosuggest"
_zsh_autosuggest_predefined_generate() {
local pbase="$_zsh_autosuggest_data_home"
local pname="$pbase/predefined"
local ptemp="$pbase/predefined.$[RANDOM]"
local suggests=()
local pwd=$(pwd)
[ ! -d "$pbase" ] && mkdir -p "$pbase" 2> /dev/null
# skip when ~/.zsh_autosuggest exists
[ -f "$pname" ] && return
echo "autosuggestions is generating: $pname"
# copy builtin predefine database
local txt="${_zsh_autosuggest_script_path}/predefined.txt"
[ -f "$txt" ] && cat "$txt" > "$ptemp"
# enumerate commands in $PATH and add them to ~/.zsh_autosuggest
for p ("${(@s/:/)PATH}"); do
[ ! -d "$p" ] && continue
cd "$p"
local files=($(command ls))
for fn in ${files}; do
if [ -x "$fn" ] && [[ "${fn:l}" != *.dll ]]; then
if [ -f "$fn" ] && [[ "${fn:l}" != *.nls ]]; then
# trim cygwin .exe/.cmd/.bat postfix
if [[ "$fn" == *.exe ]]; then
fn=${fn/%.exe/}
elif [[ "$fn" == *.cmd ]]; then
fn=${fn/%.cmd/}
elif [[ "$fn" == *.bat ]]; then
fn=${fn/%.bat/}
fi
if [[ ${#fn} -gt 1 ]]; then
suggests+=$fn
fi
fi
fi
done
done
cd "${pwd}"
# TODO: generate command parameters from completion database
print -l $suggests >> "$ptemp"
# atomic change file name
mv -f "$ptemp" "$pname"
}
_zsh_autosuggest_strategy_predefined() {
emulate -L zsh
setopt EXTENDED_GLOB
local prefix="${1//(#m)[\\*?[\]<>()|^~#]/\\$MATCH}"
# search the history at first
local result="${history[(r)${prefix}*]}"
# search the predefine files if nothing found in history
if [[ -z "$result" ]]; then
if (( ! ${+_ZSH_AUTOSUGGEST_PREDEFINE} )); then
# _zsh_autosuggest_predefined_generate
typeset -g _ZSH_AUTOSUGGEST_PREDEFINE=()
local pbase="$_zsh_autosuggest_data_home"
local pname="$pbase/predefined"
local names="${ZSH_AUTOSUGGEST_PREDEFINE_PATH};$pname"
local array=()
for i ("${(s:;:)names}"); do
if [[ -n "$i" ]] && [[ -f "$i" ]]; then
local temp=(${(f)"$(<$i)"})
array+=($temp)
fi
done
_ZSH_AUTOSUGGEST_PREDEFINE+=($array)
fi
result="${_ZSH_AUTOSUGGEST_PREDEFINE[(r)${prefix}*]}"
fi
typeset -g suggestion="$result"
}
_zsh_autosuggest_predefined_generate
# vim: set ts=4 sw=4 tw=0 et :