blob: bc01dd2ff09697ae72d1c2ba5f2c4dc09d38d736 (
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
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
|
# **** 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
run_cmd=$1
run_cmd="${run_cmd/\"\$\@\"/${args[*]}}"
run_cmd="${run_cmd/\$\*/${args[*]}}"
inquote=0;
outcmd='';
OIFS=$IFS
IFS=$'\n'
regex="'[^']+'"
for line in ${run_cmd[@]}; do
outline=''
if [[ "$line" =~ $regex ]]; then
IFS=$OIFS
for word in $line; do
if [[ $word =~ "'" ]]; then
if [[ $inquote == 0 ]]; then
inquote=1
else
inquote=0
outline+="$word "
continue
fi
fi
if [[ $inquote == 0 ]] &&
[[ "$word" =~ \$[0-9] ]]; then
word=$( sed -e 's/\$\([0-9]\)/\$\{args\[\1\]\}/g' <<<"$word" )
fi
outline+="$word "
done
IFS=$'\n'
line=$outline
else
if [[ "$line" =~ \$[0-9] ]]; then
line=$( sed -e 's/\$\([0-9]\)/\$\{args\[\1\]\}/g' <<<"$line" )
fi
fi
outcmd+="$line\n"
done
run_cmd=$outcmd
IFS=$OIFS
eval "$restore_shopts"
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=$?
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:
|
