extras/lttng-bash_completion

   1 #
   2 # Copyright (c) - 2012 Simon Marchi <simon.marchi@polymtl.ca>
   3 #
   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.
   7 #
   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
  11 # more details.
  12 #
  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.
  16 #
  17
  18 _lttng_complete_sessions() {
  19         # TODO
  20         # This code does nothing for now. When there is a mecanism to get the
  21         # existing sessions, use it to fill the sessions variable.
  22         local sessions
  23         sessions=""
  24         COMPREPLY=( $(compgen -W "${sessions}" -- $cur) )
  25         return
  26 }
  27
  28 _lttng_cmd_addcontext() {
  29         local add_context_opts
  30         add_context_opts=$(lttng add-context --list-options)
  31
  32         case $prev in
  33         --session|-s)
  34                 _lttng_complete_sessions
  35                 return
  36                 ;;
  37         --channel|-c)
  38                 return
  39                 ;;
  40         --type|-t)
  41                 return
  42                 ;;
  43         esac
  44
  45         case $cur in
  46         -*)
  47                 COMPREPLY=( $(compgen -W "${add_context_opts}" -- $cur) )
  48                 return
  49                 ;;
  50         esac
  51 }
  52
  53 _lttng_cmd_create() {
  54         local create_opts
  55         create_opts=$(lttng create --list-options)
  56
  57         case $prev in
  58         --output|-o)
  59                 _filedir -d
  60                 return
  61                 ;;
  62         esac
  63
  64         case $cur in
  65         -*)
  66                 COMPREPLY=( $(compgen -W "${create_opts}" -- $cur) )
  67                 return
  68                 ;;
  69         esac
  70 }
  71
  72 _lttng_cmd_destroy() {
  73         local destroy_opts
  74         destroy_opts=$(lttng destroy --list-options)
  75
  76         case $cur in
  77         -*)
  78                 COMPREPLY=( $(compgen -W "${destroy_opts}" -- $cur) )
  79                 ;;
  80         *)
  81                 _lttng_complete_sessions
  82                 ;;
  83         esac
  84 }
  85
  86 _lttng_cmd_enablechannel() {
  87         local enable_channel_opts
  88         enable_channel_opts=$(lttng enable-channel --list-options)
  89
  90         case $prev in
  91         --session|-s)
  92                 _lttng_complete_sessions
  93                 return
  94                 ;;
  95         esac
  96
  97         case $cur in
  98         -*)
  99                 COMPREPLY=( $(compgen -W "${enable_channel_opts}" -- $cur) )
 100                 return
 101                 ;;
 102         esac
 103 }
 104
 105 _lttng_cmd_enableevent() {
 106         local enable_event_opts
 107         enable_event_opts=$(lttng enable-event --list-options)
 108
 109         case $prev in
 110         --session|-s)
 111                 _lttng_complete_sessions
 112                 return
 113                 ;;
 114         --channel|-c)
 115                 return
 116                 ;;
 117         --probe)
 118                 return
 119                 ;;
 120         --function)
 121                 return
 122                 ;;
 123         esac
 124
 125         case $cur in
 126         -*)
 127                 COMPREPLY=( $(compgen -W "${enable_event_opts}" -- $cur) )
 128                 return
 129                 ;;
 130         esac
 131 }
 132
 133 _lttng_cmd_disablechannel() {
 134         local disable_channel_opts
 135         disable_channel_opts=$(lttng disable-channel --list-options)
 136
 137         case $prev in
 138         --session|-s)
 139                 _lttng_complete_sessions
 140                 return
 141                 ;;
 142         esac
 143
 144         case $cur in
 145         -*)
 146                 COMPREPLY=( $(compgen -W "${disable_channel_opts}" -- $cur) )
 147                 return
 148                 ;;
 149         esac
 150 }
 151
 152 _lttng_cmd_disableevent() {
 153         local disable_event_opts
 154         disable_event_opts=$(lttng disable-event --list-options)
 155
 156         case $prev in
 157         --session|-s)
 158                 _lttng_complete_sessions
 159                 return
 160                 ;;
 161         --channel|-c)
 162                 return
 163                 ;;
 164         esac
 165
 166         case $cur in
 167         -*)
 168                 COMPREPLY=( $(compgen -W "${disable_event_opts}" -- $cur) )
 169                 return
 170                 ;;
 171         esac
 172 }
 173
 174 _lttng_cmd_list() {
 175         local list_opts
 176         list_opts=$(lttng list --list-options)
 177
 178         case $prev in
 179         --channel|-c)
 180                 return
 181                 ;;
 182         esac
 183
 184         case $cur in
 185         -*)
 186                 COMPREPLY=( $(compgen -W "${list_opts}" -- $cur) )
 187                 return
 188                 ;;
 189         *)
 190                 _lttng_complete_sessions
 191                 return
 192         esac
 193 }
 194
 195 _lttng_cmd_setsession() {
 196         local set_session_opts
 197         set_session_opts=$(lttng set-session --list-options)
 198
 199         case $cur in
 200         -*)
 201                 COMPREPLY=( $(compgen -W "${set_session_opts}" -- $cur) )
 202                 return
 203                 ;;
 204         *)
 205                 _lttng_complete_sessions
 206                 return
 207                 ;;
 208         esac
 209 }
 210
 211 _lttng_cmd_start() {
 212         local start_opts
 213         start_opts=$(lttng start --list-options)
 214
 215         case $cur in
 216         -*)
 217                 COMPREPLY=( $(compgen -W "${start_opts}" -- $cur) )
 218                 ;;
 219         *)
 220                 _lttng_complete_sessions
 221                 ;;
 222         esac
 223 }
 224
 225 _lttng_cmd_stop() {
 226         local stop_opts
 227         stop_opts=$(lttng stop --list-options)
 228
 229         case $cur in
 230         -*)
 231                 COMPREPLY=( $(compgen -W "${stop_opts}" -- $cur) )
 232                 ;;
 233         *)
 234                 _lttng_complete_sessions
 235                 ;;
 236         esac
 237 }
 238
 239 _lttng_cmd_version() {
 240         local version_opts
 241         version_opts=$(lttng version --list-options)
 242
 243         case $cur in
 244         -*)
 245                 COMPREPLY=( $(compgen -W "${version_opts}" -- $cur) )
 246                 ;;
 247         esac
 248 }
 249
 250 _lttng_cmd_calibrate() {
 251         local calibrate_opts
 252         calibrate_opts=$(lttng calibrate --list-options)
 253
 254         case $cur in
 255         -*)
 256                 COMPREPLY=( $(compgen -W "${calibrate_opts}" -- $cur) )
 257                 ;;
 258         esac
 259 }
 260
 261 _lttng_cmd_view() {
 262         local view_opts
 263                 view_opts=$(lttng view --list-options)
 264
 265                 case $cur in
 266                 -*)
 267                 COMPREPLY=( $(compgen -W "${view_opts}" -- $cur) )
 268                 ;;
 269         esac
 270 }
 271
 272 _lttng_opts() {
 273         local opts
 274         opts=$(lttng --list-options)
 275
 276         COMPREPLY=( $(compgen -W "${opts}" -- $cur) )
 277 }
 278
 279 _lttng_commands() {
 280         COMPREPLY=( $(compgen -W "$commands" -- $cur) )
 281 }
 282
 283 _lttng_before_command() {
 284         # Check if the previous word should alter the behavior
 285         case $prev in
 286         --group|-g)
 287                 COMPREPLY=( $(compgen -g -- $cur) )
 288                 return
 289                 ;;
 290         --sessiond-path)
 291                 _filedir
 292                 return
 293                 ;;
 294         esac
 295
 296         case $cur in
 297         -*)
 298                 # If the current word starts with a dash, complete with options
 299                 _lttng_opts
 300                 ;;
 301         *)
 302                 # Otherwise complete with commands
 303                 _lttng_commands
 304                 ;;
 305         esac
 306 }
 307
 308 _lttng_after_command() {
 309         local cmd_name
 310
 311         cmd_name=_lttng_cmd_${command//-/}
 312
 313         type -t $cmd_name | grep -q 'function' && $cmd_name
 314 }
 315
 316 _lttng_is_command() {
 317         for command in $commands; do
 318                 if [ "$1" == "$command" ]; then
 319                         return 0
 320                 fi
 321         done
 322
 323         return 1
 324 }
 325
 326 _lttng() {
 327         local cur prev commands command_found command_found_index
 328
 329         # Get the current and previous word
 330         _get_comp_words_by_ref cur prev
 331
 332         # Get the valid LTTng commands
 333         commands=$(lttng --list-commands)
 334
 335         # The text of the found command
 336         command_found=""
 337
 338         # The index of the found command in COMP_WORDS
 339         command_found_index=-1
 340
 341         for (( i = 1 ; i < ${#COMP_WORDS[@]} ; i++ )); do
 342                 _lttng_is_command ${COMP_WORDS[$i]}
 343                 if [ $? -eq 0 ]; then
 344                         command_found=${COMP_WORDS[$i]}
 345                         command_found_index=$i
 346                         break
 347                 fi
 348
 349         done
 350
 351         # Check if the cursor is before or after the command keyword
 352         if [ -n "$command_found" ] && [ "$COMP_CWORD" -gt "$command_found_index" ]; then
 353                 _lttng_after_command
 354         else
 355                 _lttng_before_command
 356         fi
 357 }
 358
 359 complete -F _lttng lttng