blob: ea451e599e1616a4a7fe88a9cd316b3dd84cda7c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
# **** 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
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:
|
