XorA
/
bash-it
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Clean alias-completion plugin

pull/1872/head
John Fred Fadrigalan 2021-04-20 15:09:18 +08:00
parent 6ef505c707
commit 3a4476aa6a
No known key found for this signature in database
GPG Key ID: 4C23960F72FC911F
2 changed files with 60 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clean_files.txt
View File

@ -50,6 +50,7 @@ themes/purity
# plugins # plugins
# #
plugins/available/alias-completion.plugin.bash
plugins/available/basher.plugin.bash plugins/available/basher.plugin.bash
plugins/available/cmd-returned-notify.plugin.bash plugins/available/cmd-returned-notify.plugin.bash
plugins/available/docker-machine.plugin.bash plugins/available/docker-machine.plugin.bash
@ -104,6 +105,7 @@ aliases/available/vim.aliases.bash
aliases/available/git.aliases.bash aliases/available/git.aliases.bash
# tests # tests
test/alias-completion.plugin.bats
test/test_helper.bash test/test_helper.bash
# vendor init files # vendor init files

109
plugins/available/alias-completion.plugin.bash
View File

@ -1,3 +1,4 @@
# shellcheck shell=bash
# Load after the other completions to understand what needs to be completed # Load after the other completions to understand what needs to be completed
# BASH_IT_LOAD_PRIORITY: 365 # BASH_IT_LOAD_PRIORITY: 365
@ -17,54 +18,58 @@ about-plugin 'Automatic completion of aliases'
# Automatically add completion for all aliases to commands having completion functions # Automatically add completion for all aliases to commands having completion functions
function alias_completion { function alias_completion {
local namespace="alias_completion" local namespace="alias_completion"
# parse function based completion definitions, where capture group 2 => function and 3 => trigger # parse function based completion definitions, where capture group 2 => function and 3 => trigger
local compl_regex='complete( +[^ ]+)* -F ([^ ]+) ("[^"]+"|[^ ]+)' local compl_regex='complete( +[^ ]+)* -F ([^ ]+) ("[^"]+"|[^ ]+)'
# parse alias definitions, where capture group 1 => trigger, 2 => command, 3 => command arguments # parse alias definitions, where capture group 1 => trigger, 2 => command, 3 => command arguments
local alias_regex="alias( -- | )([^=]+)='(\"[^\"]+\"|[^ ]+)(( +[^ ]+)*)'" local alias_regex="alias( -- | )([^=]+)='(\"[^\"]+\"|[^ ]+)(( +[^ ]+)*)'"
# create array of function completion triggers, keeping multi-word triggers together # create array of function completion triggers, keeping multi-word triggers together
eval "local completions=($(complete -p | sed -Ene "/$compl_regex/s//'\3'/p"))" eval "local completions=($(complete -p | sed -Ene "/$compl_regex/s//'\3'/p"))"
(( ${#completions[@]} == 0 )) && return 0 ((${#completions[@]} == 0)) && return 0
# create temporary file for wrapper functions and completions # create temporary file for wrapper functions and completions
local tmp_file; tmp_file="$(mktemp -t "${namespace}-${RANDOM}XXXXXX")" || return 1 local tmp_file
tmp_file="$(mktemp -t "${namespace}-${RANDOM}XXXXXX")" || return 1
local completion_loader; completion_loader="$(complete -p -D 2>/dev/null | sed -Ene 's/.* -F ([^ ]*).*/\1/p')" local completion_loader
completion_loader="$(complete -p -D 2> /dev/null | sed -Ene 's/.* -F ([^ ]*).*/\1/p')"
# read in "<alias> '<aliased command>' '<command args>'" lines from defined aliases # read in "<alias> '<aliased command>' '<command args>'" lines from defined aliases
local line; while read line; do local line
eval "local alias_tokens; alias_tokens=($line)" 2>/dev/null || continue # some alias arg patterns cause an eval parse error while read -r line; do
local alias_name="${alias_tokens[0]}" alias_cmd="${alias_tokens[1]}" alias_args="${alias_tokens[2]# }" eval "local alias_tokens; alias_tokens=($line)" 2> /dev/null || continue # some alias arg patterns cause an eval parse error
local alias_name="${alias_tokens[0]}" alias_cmd="${alias_tokens[1]}" alias_args="${alias_tokens[2]# }"
# skip aliases to pipes, boolean control structures and other command lists # skip aliases to pipes, boolean control structures and other command lists
# (leveraging that eval errs out if $alias_args contains unquoted shell metacharacters) # (leveraging that eval errs out if $alias_args contains unquoted shell metacharacters)
eval "local alias_arg_words; alias_arg_words=($alias_args)" 2>/dev/null || continue eval "local alias_arg_words; alias_arg_words=($alias_args)" 2> /dev/null || continue
# avoid expanding wildcards # avoid expanding wildcards
read -a alias_arg_words <<< "$alias_args" read -ra alias_arg_words <<< "$alias_args"
# skip alias if there is no completion function triggered by the aliased command # skip alias if there is no completion function triggered by the aliased command
if [[ ! " ${completions[*]} " =~ " $alias_cmd " ]]; then if [[ ! " ${completions[*]} " =~ $alias_cmd ]]; then
if [[ -n "$completion_loader" ]]; then if [[ -n "$completion_loader" ]]; then
# force loading of completions for the aliased command # force loading of completions for the aliased command
eval "$completion_loader $alias_cmd" eval "$completion_loader $alias_cmd"
# 124 means completion loader was successful # 124 means completion loader was successful
[[ $? -eq 124 ]] || continue [[ $? -eq 124 ]] || continue
completions+=($alias_cmd) completions+=("$alias_cmd")
else else
continue continue
fi fi
fi fi
local new_completion="$(complete -p "$alias_cmd" 2>/dev/null)" local new_completion="$(complete -p "$alias_cmd" 2> /dev/null)"
# create a wrapper inserting the alias arguments if any # create a wrapper inserting the alias arguments if any
if [[ -n $alias_args ]]; then if [[ -n $alias_args ]]; then
local compl_func="${new_completion/#* -F /}"; compl_func="${compl_func%% *}" local compl_func="${new_completion/#* -F /}"
# avoid recursive call loops by ignoring our own functions compl_func="${compl_func%% *}"
if [[ "${compl_func#_$namespace::}" == $compl_func ]]; then # avoid recursive call loops by ignoring our own functions
local compl_wrapper="_${namespace}::${alias_name}" if [[ "${compl_func#_$namespace::}" == "$compl_func" ]]; then
echo "function $compl_wrapper { local compl_wrapper="_${namespace}::${alias_name}"
echo "function $compl_wrapper {
local compl_word=\$2 local compl_word=\$2
local prec_word=\$3 local prec_word=\$3
# check if prec_word is the alias itself. if so, replace it # check if prec_word is the alias itself. if so, replace it
@ -81,15 +86,17 @@ function alias_completion {
(( COMP_POINT += \${#COMP_LINE} )) (( COMP_POINT += \${#COMP_LINE} ))
$compl_func \"$alias_cmd\" \"\$compl_word\" \"\$prec_word\" $compl_func \"$alias_cmd\" \"\$compl_word\" \"\$prec_word\"
}" >> "$tmp_file" }" >> "$tmp_file"
new_completion="${new_completion/ -F $compl_func / -F $compl_wrapper }" new_completion="${new_completion/ -F $compl_func / -F $compl_wrapper }"
fi fi
fi fi
# replace completion trigger by alias # replace completion trigger by alias
if [[ -n $new_completion ]]; then if [[ -n $new_completion ]]; then
new_completion="${new_completion% *} $alias_name" new_completion="${new_completion% *} $alias_name"
echo "$new_completion" >> "$tmp_file" echo "$new_completion" >> "$tmp_file"
fi fi
done < <(alias -p | sed -Ene "s/$alias_regex/\2 '\3' '\4'/p") done < <(alias -p | sed -Ene "s/$alias_regex/\2 '\3' '\4'/p")
source "$tmp_file" && command rm -f "$tmp_file" source "$tmp_file" && command rm -f "$tmp_file"
}; alias_completion }
alias_completion