# **** License **** # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License version 2 as # published by the Free Software Foundation. # # This program is distributed in the hope that it will be useful, but # WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # General Public License for more details. # # This code was originally developed by Vyatta, Inc. # Portions created by Vyatta are Copyright (C) 2006, 2007 Vyatta, Inc. # All Rights Reserved. # # Author: Tom Grennan # Date: 2007 # Description: setup bash completion for Vyatta operational commands # # **** End License **** _vyatta_op_init () { # empty and default line compeletion complete -E -F _vyatta_op_expand complete -D -F _vyatta_op_default_expand # create the top level aliases for the unambiguous portions of the commands # this is the only place we need an entire enumerated list of the subcommands for cmd in $( ls /opt/vyatta/share/vyatta-op/templates/ ); do for pos in $(seq 1 ${#cmd}); do case ${cmd:0:$pos} in for|do|done|if|fi|case|while|tr ) continue ;; *) ;; esac complete -F _vyatta_op_expand ${cmd:0:$pos} eval alias ${cmd:0:$pos}=\'_vyatta_op_run ${cmd:0:$pos}\' done done shopt -s histverify } _vyatta_op_get_node_def_field () { local file=$1 field=$2 sed -n '/^'"$field"':/,$ { # strip field name and hold rest of line s/[a-z]*: *// h :b # at EOF, print hold buffer and quit $ { x; p; q } # input next line n # if start of another field def, print hold buf and quit /^[a-z]*:/ { x; p; q } # add to hold buf and branch to input next line H bb }' $file } _vyatta_op_conv_node_path () { # is the node ok, ambiguous, or invalid local node_path local node local -a ARR node_path=$1 node=$2 ARR=( $(compgen -d $node_path/$node) ) if [[ "${#ARR[@]}" == "1" ]]; then echo ${ARR[0]##*/} elif [[ "${#ARR[@]}" == "0" ]]; then if [[ -d "${node_path}/node.tag" ]]; then echo "$node tag" else echo "$node invalid" fi elif [[ -d "$node_path/$node" ]]; then echo $node elif [[ "$VYATTA_USER_LEVEL_DIR" != "/opt/vyatta/etc/shell/level/admin" ]];then # special handling for unprivledged completions. # Since top level commands are different for unprivledged users # we need a handler to expand them properly. local -a filtered_cmds=() local -a allowed=( $(cat $VYATTA_USER_LEVEL_DIR/allowed-op.in) ) get_prefix_filtered_list $node allowed filtered_cmds if [[ "${#filtered_cmds[@]}" == "1" ]];then echo ${filtered_cmds[0]} else echo "${node} ambiguous" fi else echo "$node ambiguous" fi } _vyatta_op_conv_run_cmd () { # Substitue bash positional variables # for the same value in the expanded array local restore_shopts=$( shopt -p extglob nullglob | tr \\n \; ) shopt -s extglob shopt -u nullglob local run_cmd="$1" local line outline local -i inquote=0; local outcmd=''; local OIFS=$IFS local re="([^']*')(.*)" toggle_inquote() { if [[ $inquote == 0 ]]; then inquote=1 else inquote=0 fi } process_subline() { if [[ $inquote == 1 ]]; then outline+="$1" else outline+=$(sed -e 's/\$\([0-9]\)/\$\{args\[\1\]\}/g' <<<"$1") fi } run_cmd="${run_cmd/\"\$\@\"/${args[*]}}" run_cmd="${run_cmd/\$\*/${args[*]}}" run_cmd="${run_cmd//\\/\\\\}" IFS=$'\n' for line in ${run_cmd[@]}; do outline='' while [[ -n "$line" ]]; do if [[ "$line" =~ $re ]]; then process_subline "${BASH_REMATCH[1]}" toggle_inquote else process_subline "$line" fi line="${BASH_REMATCH[2]}" done outcmd+="$outline\n" done IFS=$OIFS eval "$restore_shopts" echo -ne "$outcmd" } _vyatta_op_run () { local -i estat local tpath=$vyatta_op_templates local restore_shopts=$( shopt -p extglob nullglob | tr \\n \; ) shopt -s extglob nullglob _vyatta_op_last_comp=${_vyatta_op_last_comp_init} false; estat=$? stty echo 2> /dev/null # turn echo on, this is a workaround for bug 7570 # not a fix we need to look at why the readline library # is getting confused on paged help text. i=1 declare -a args # array of expanded arguments for arg in "$@"; do local orig_arg=$arg if [[ $arg == "*" ]]; then arg="*" #leave user defined wildcards alone else arg=( $(_vyatta_op_conv_node_path $tpath $arg) ) # expand the arguments fi # output proper error message based on the above expansion if [[ "${arg[1]}" == "ambiguous" ]]; then echo -ne "\n Ambiguous command: ${args[@]} [$arg]\n" >&2 local -a cmds=( $(compgen -d $tpath/$arg) ) _vyatta_op_node_path=$tpath local comps=$(_vyatta_op_help $arg ${cmds[@]##*/}) echo -e "$comps\n" | sed -e 's/^P/ P/' eval $restore_shopts return 1 elif [[ "${arg[1]}" == "invalid" ]]; then echo -ne "\n Invalid command: ${args[@]} [$arg]\n\n" >&2 eval $restore_shopts return 1 fi if [ -f "$tpath/$arg/node.def" ] ; then tpath+=/$arg elif [ -f $tpath/node.tag/node.def ] ; then tpath+=/node.tag else echo -ne "\n Invalid command: ${args[@]} [$arg]\n\n" >&2 eval $restore_shopts return 1 fi if [[ "$arg" == "node.tag" ]]; then args[$i]=$orig_arg else args[$i]=$arg fi let "i+=1" done local run_cmd=$(_vyatta_op_get_node_def_field $tpath/node.def run) run_cmd=$(_vyatta_op_conv_run_cmd "$run_cmd") # convert the positional parameters local ret=0 local cmd_regex="^(LESSOPEN=|less|pager|tail|/opt/vyatta/bin/vyatta-tshark-interface-port.pl).*" if [ -n "$run_cmd" ]; then eval $restore_shopts if [[ -t 1 && "${args[1]}" == "show" && ! $run_cmd =~ $cmd_regex ]] ; then eval "($run_cmd) | ${VYATTA_PAGER:-cat}" else eval "$run_cmd" fi else echo -ne "\n Incomplete command: ${args[@]}\n\n" >&2 eval $restore_shopts ret=1 fi return $ret } ### Local Variables: ### mode: shell-script ### End: