btrfs-completion

   1 # original by Alfredo Esteban <aedelatorre at xxxxxxxxx>
   2 # (http://www.spinics.net/lists/linux-btrfs/msg15899.html)
   3 # edited by Joseph Wang <joequant at gmail.com>
   4 # (http://lists.alioth.debian.org/pipermail/bash-completion-devel/2013-June/004868.html)
   5 # edited by John C F <john.ch.fr at gmail.com> on 2015-02-02
   6
   7 _btrfs_devs()
   8 {
   9     local DEVS
  10     DEVS=''; while read dev; do DEVS+="$dev "; done < <(lsblk -pnro name)
  11         COMPREPLY+=( $( compgen -W "$DEVS" -- "$cur" ) )
  12 }
  13
  14 _btrfs_mnts()
  15 {
  16         local MNTS
  17         MNTS=''
  18         while read mnt; do MNTS+="$mnt "
  19         done < <(mount -t btrfs | awk '{print $3}')
  20         COMPREPLY+=( $( compgen -W "$MNTS" -- "$cur" ) )
  21 }
  22
  23 _btrfs()
  24 {
  25         local cur prev words cword
  26     _init_completion || return
  27
  28     COMPREPLY=()
  29
  30         local cmd=${words[1]}
  31
  32     commands='subvolume filesystem balance device scrub check rescue restore inspect-internal property send receive quota qgroup replace help version'
  33     commands_subvolume='create delete list snapshot find-new get-default set-default show sync'
  34     commands_filesystem='defragment sync resize show df du label usage'
  35     commands_balance='start pause cancel resume status'
  36     commands_device='scan add delete remove ready stats usage'
  37     commands_scrub='start cancel resume status'
  38     commands_rescue='chunk-recover super-recover zero-log'
  39     commands_inspect_internal='inode-resolve logical-resolve subvolid-resolve rootid min-dev-size dump-tree dump-super tree-stats'
  40     commands_property='get set list'
  41     commands_quota='enable disable rescan'
  42     commands_qgroup='assign remove create destroy show limit'
  43     commands_replace='start status cancel'
  44
  45         if [[ "$cur" == -* && $cword -le 3 && "$cmd" != "help" ]]; then
  46                 COMPREPLY=( $( compgen -W '--help' -- "$cur" ) )
  47                 return 0
  48         fi
  49
  50     if [[ $cword -eq 1 ]]; then
  51         COMPREPLY=( $( compgen -W "$commands" -- "$cur" ) )
  52         return 0
  53     elif [[ $cword -eq 2 ]]; then
  54         case $cmd in
  55             subvolume)
  56                 opts="$commands_subvolume"
  57                 ;;
  58             filesystem)
  59                 opts="$commands_filesystem"
  60                 ;;
  61             balance)
  62                 opts="$commands_balance"
  63                 ;;
  64             device)
  65                 opts="$commands_device"
  66                 ;;
  67             scrub)
  68                 opts="$commands_scrub"
  69                 ;;
  70             check)
  71                 _btrfs_devs
  72                                 return 0
  73                 ;;
  74             rescue)
  75                 opts="$commands_rescue"
  76                 ;;
  77             restore)
  78                 _btrfs_devs
  79                                 return 0
  80                 ;;
  81             inspect-internal)
  82                 opts="$commands_inspect_internal"
  83                 ;;
  84             property)
  85                 opts="$commands_property"
  86                 ;;
  87             send|receive)
  88                 _filedir -d
  89                 return 0
  90                 ;;
  91             quota)
  92                 opts="$commands_quota"
  93                 ;;
  94             qgroup)
  95                 opts="$commands_qgroup"
  96                 ;;
  97             replace)
  98                 opts="$commands_replace"
  99                 ;;
 100             help)
 101                 opts="--full"
 102                 ;;
 103             version)
 104                 return 0
 105                 ;;
 106         esac
 107         COMPREPLY=( $( compgen -W "$opts" -- "$cur" ) )
 108         return 0
 109     elif [[ $cword -eq 3 ]]; then
 110         case $cmd in
 111             filesystem)
 112                                 case $prev in
 113                                         defragment)
 114                                                 _filedir
 115                                                 return 0
 116                                                 ;;
 117                                         df|usage)
 118                                                 _btrfs_mnts
 119                                                 return 0
 120                                                 ;;
 121                                         label)
 122                                                 _btrfs_mnts
 123                                                 _btrfs_devs
 124                                                 return 0
 125                                                 ;;
 126                                 esac
 127                 ;;
 128             device|rescue)
 129                                 _btrfs_devs
 130                 return 0
 131                 ;;
 132                         inspect-internal)
 133                                 case $prev in
 134                                         min-dev-size)
 135                                                 _btrfs_mnts
 136                                                 return 0
 137                                                 ;;
 138                                         rootid)
 139                                                 _filedir
 140                                                 return 0
 141                                                 ;;
 142                                 esac
 143                                 ;;
 144                         receive)
 145                                 case $prev in
 146                                         -f)
 147                                                 _filedir
 148                                                 return 0
 149                                                 ;;
 150                                 esac
 151                                 ;;
 152             replace)
 153                 case $prev in
 154                     status|cancel)
 155                                                 _btrfs_mnts
 156                         return 0
 157                         ;;
 158                     start)
 159                                                 _btrfs_devs
 160                         return 0
 161                         ;;
 162                 esac
 163                 ;;
 164                         subvolume)
 165                                 case $prev in
 166                                         list)
 167                                                 _btrfs_mnts
 168                                                 return 0
 169                                                 ;;
 170                                 esac
 171                                 ;;
 172         esac
 173     fi
 174
 175     _filedir -d
 176     return 0
 177 }
 178
 179 complete -F _btrfs btrfs