# **** 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=( "$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 run_cmd
  run_cmd="$@"
  run_cmd="${run_cmd/\"\$\@\"/${args[*]}}"
  run_cmd="${run_cmd/\$\*/${args[*]}}"
  run_cmd=$(echo -e "$run_cmd" | sed -e 's/\$\([0-9]\)/\$\{args\[\1\]\}/g')
  echo -ne "$run_cmd"
}

_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=$?

    i=1
    declare -a args  # array of expanded arguments
    for arg in "$@"; do
        arg=( $(_vyatta_op_conv_node_path $tpath $arg) )  # expand the arguments
        # output proper error message based on the above expansion
        if [[ "${arg[1]}" == "ambiguous" ]]; then
          echo -ne "\n  Ambiguous command: ${args[@]} [$arg]\n"
          local -a cmds=( "$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"
          eval $restore_shopts
          return 1
        fi

        args[$i]=$arg
        if [ -f "$tpath/$arg/node.def" ] ; then
            tpath+=/$arg
        elif [ -f $tpath/node.tag/node.def ] ; then
            tpath+=/node.tag
        else
            echo "Invalid command" >&2
            eval $restore_shopts
            return 1
        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
      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
      ret=1
    fi
    eval $restore_shopts
    return $ret
}

###  Local Variables:
###  mode: shell-script
###  End: