2 # Copyright (c) - 2012 Simon Marchi <simon.marchi@polymtl.ca>
4 # This program is free software; you can redistribute it and/or modify it under
5 # the terms of the GNU General Public License as published by as published by
6 # the Free Software Foundation; only version 2 of the License.
8 # This program is distributed in the hope that it will be useful, but WITHOUT
9 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 # FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
13 # You should have received a copy of the GNU General Public License along with
14 # this program; if not, write to the Free Software Foundation, Inc., 51
15 # Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 _lttng_complete_sessions() {
19 # TODO, maybe have a lttng list --simple or something like that
23 _lttng_cmd_addcontext() {
24 local add_context_opts
25 add_context_opts=$(lttng add-context --list-options)
29 _lttng_complete_sessions
42 COMPREPLY=( $(compgen -W "${add_context_opts}" -- $cur) )
50 create_opts=$(lttng create --list-options)
61 COMPREPLY=( $(compgen -W "${create_opts}" -- $cur) )
67 _lttng_cmd_destroy() {
69 destroy_opts=$(lttng destroy --list-options)
73 COMPREPLY=( $(compgen -W "${destroy_opts}" -- $cur) )
76 _lttng_complete_sessions
81 _lttng_cmd_enablechannel() {
82 local enable_channel_opts
83 enable_channel_opts=$(lttng enable-channel --list-options)
87 _lttng_complete_sessions
94 COMPREPLY=( $(compgen -W "${enable_channel_opts}" -- $cur) )
100 _lttng_cmd_enableevent() {
101 local enable_event_opts
102 enable_event_opts=$(lttng enable-event --list-options)
106 _lttng_complete_sessions
122 COMPREPLY=( $(compgen -W "${enable_event_opts}" -- $cur) )
128 _lttng_cmd_disablechannel() {
129 local disable_channel_opts
130 disable_channel_opts=$(lttng disable-channel --list-options)
134 _lttng_complete_sessions
141 COMPREPLY=( $(compgen -W "${disable_channel_opts}" -- $cur) )
147 _lttng_cmd_disableevent() {
148 local disable_event_opts
149 disable_event_opts=$(lttng disable-event --list-options)
153 _lttng_complete_sessions
163 COMPREPLY=( $(compgen -W "${disable_event_opts}" -- $cur) )
171 list_opts=$(lttng list --list-options)
181 COMPREPLY=( $(compgen -W "${list_opts}" -- $cur) )
187 _lttng_cmd_setsession() {
188 local set_session_opts
189 set_session_opts=$(lttng set-session --list-options)
193 COMPREPLY=( $(compgen -W "${set_session_opts}" -- $cur) )
201 start_opts=$(lttng start --list-options)
205 COMPREPLY=( $(compgen -W "${start_opts}" -- $cur) )
208 _lttng_complete_sessions
215 stop_opts=$(lttng stop --list-options)
219 COMPREPLY=( $(compgen -W "${stop_opts}" -- $cur) )
222 _lttng_complete_sessions
227 _lttng_cmd_version() {
229 version_opts=$(lttng version --list-options)
233 COMPREPLY=( $(compgen -W "${version_opts}" -- $cur) )
238 _lttng_cmd_calibrate() {
240 calibrate_opts=$(lttng calibrate --list-options)
244 COMPREPLY=( $(compgen -W "${calibrate_opts}" -- $cur) )
251 view_opts=$(lttng view --list-options)
255 COMPREPLY=( $(compgen -W "${view_opts}" -- $cur) )
262 opts=$(lttng --list-options)
264 COMPREPLY=( $(compgen -W "${opts}" -- $cur) )
268 COMPREPLY=( $(compgen -W "$commands" -- $cur) )
271 _lttng_before_command() {
272 # Check if the previous word should alter the behavior
275 COMPREPLY=( $(compgen -g -- $cur) )
286 # If the current word starts with a dash, complete with options
290 # Otherwise complete with commands
296 _lttng_after_command() {
299 cmd_name=_lttng_cmd_${command//-/}
301 type -t $cmd_name | grep -q 'function' && $cmd_name
304 _lttng_is_command() {
305 for command in $commands; do
306 if [ "$1" == "$command" ]; then
315 local cur prev commands command_found command_found_index
317 # Get the current and previous word
318 _get_comp_words_by_ref cur prev
320 # Get the valid LTTng commands
321 commands=$(lttng --list-commands)
323 # The text of the found command
326 # The index of the found command in COMP_WORDS
327 command_found_index=-1
329 for (( i = 1 ; i < ${#COMP_WORDS[@]} ; i++ )); do
330 _lttng_is_command ${COMP_WORDS[$i]}
331 if [ $? -eq 0 ]; then
332 command_found=${COMP_WORDS[$i]}
333 command_found_index=$i
339 # Check if the cursor is before or after the command keyword
340 if [ -n "$command_found" ] && [ "$COMP_CWORD" -gt "$command_found_index" ]; then
343 _lttng_before_command
347 complete -F _lttng lttng