contrib/tg-completion.bash

   1 #
   2 # bash completion support for TopGit.
   3 #
   4 # Copyright (C) 2008 Jonas Fonseca <fonseca@diku.dk>
   5 # Copyright (C) 2006,2007 Shawn O. Pearce <spearce@spearce.org>
   6 # Based git's git-completion.sh: http://repo.or.cz/w/git/fastimport.git
   7 # Conceptually based on gitcompletion (http://gitweb.hawaga.org.uk/).
   8 # Distributed under the GNU General Public License, version 2.0.
   9 #
  10 # To use these routines:
  11 #
  12 #    1) Copy this file to somewhere (e.g. ~/.tg-completion.sh).
  13 #    2) Source it from your ~/.bashrc.
  14 #
  15 # Note: Make sure the tg script is in your PATH before you source this
  16 # script, so it can properly setup cached values.
  17
  18 case "$COMP_WORDBREAKS" in
  19 *:*) : great ;;
  20 *)   COMP_WORDBREAKS="$COMP_WORDBREAKS:"
  21 esac
  22
  23 ### {{{ Utilities
  24
  25 __tgdir ()
  26 {
  27         if [ -z "$1" ]; then
  28                 if [ -n "$__tg_dir" ]; then
  29                         echo "$__tg_dir"
  30                 elif [ -d .git ]; then
  31                         echo .git
  32                 else
  33                         git rev-parse --git-dir 2>/dev/null
  34                 fi
  35         elif [ -d "$1/.git" ]; then
  36                 echo "$1/.git"
  37         else
  38                 echo "$1"
  39         fi
  40 }
  41
  42 __tgcomp_1 ()
  43 {
  44         local c IFS=' '$'\t'$'\n'
  45         for c in $1; do
  46                 case "$c$2" in
  47                 --*=*) printf %s$'\n' "$c$2" ;;
  48                 *.)    printf %s$'\n' "$c$2" ;;
  49                 *)     printf %s$'\n' "$c$2 " ;;
  50                 esac
  51         done
  52 }
  53
  54 __tgcomp ()
  55 {
  56         local cur="${COMP_WORDS[COMP_CWORD]}"
  57         if [ $# -gt 2 ]; then
  58                 cur="$3"
  59         fi
  60         case "$cur" in
  61         --*=)
  62                 COMPREPLY=()
  63                 ;;
  64         *)
  65                 local IFS=$'\n'
  66                 COMPREPLY=($(compgen -P "$2" \
  67                         -W "$(__tgcomp_1 "$1" "$4")" \
  68                         -- "$cur"))
  69                 ;;
  70         esac
  71 }
  72
  73 __tg_heads ()
  74 {
  75         local cmd i is_hash=y dir="$(__tgdir "$1")"
  76         if [ -d "$dir" ]; then
  77                 git --git-dir="$dir" for-each-ref --format='%(refname:short)' \
  78                         refs/heads
  79                 return
  80         fi
  81         for i in $(git ls-remote "$1" 2>/dev/null); do
  82                 case "$is_hash,$i" in
  83                 y,*) is_hash=n ;;
  84                 n,*^{}) is_hash=y ;;
  85                 n,refs/heads/*) is_hash=y; echo "${i#refs/heads/}" ;;
  86                 n,*) is_hash=y; echo "$i" ;;
  87                 esac
  88         done
  89 }
  90
  91 __tg_refs ()
  92 {
  93         local cmd i is_hash=y dir="$(__tgdir "$1")"
  94         if [ -d "$dir" ]; then
  95                 if [ -e "$dir/HEAD" ]; then echo HEAD; fi
  96                 git --git-dir="$dir" for-each-ref --format='%(refname:short)' \
  97                         refs/tags refs/heads refs/remotes
  98                 return
  99         fi
 100         for i in $(git ls-remote "$dir" 2>/dev/null); do
 101                 case "$is_hash,$i" in
 102                 y,*) is_hash=n ;;
 103                 n,*^{}) is_hash=y ;;
 104                 n,refs/tags/*) is_hash=y; echo "${i#refs/tags/}" ;;
 105                 n,refs/heads/*) is_hash=y; echo "${i#refs/heads/}" ;;
 106                 n,refs/remotes/*) is_hash=y; echo "${i#refs/remotes/}" ;;
 107                 n,*) is_hash=y; echo "$i" ;;
 108                 esac
 109         done
 110 }
 111
 112 __tg_refs2 ()
 113 {
 114         local i
 115         for i in $(__tg_refs "$1"); do
 116                 echo "$i:$i"
 117         done
 118 }
 119
 120 __tg_refs_remotes ()
 121 {
 122         local cmd i is_hash=y
 123         for i in $(git ls-remote "$1" 2>/dev/null); do
 124                 case "$is_hash,$i" in
 125                 n,refs/heads/*)
 126                         is_hash=y
 127                         echo "$i:refs/remotes/$1/${i#refs/heads/}"
 128                         ;;
 129                 y,*) is_hash=n ;;
 130                 n,*^{}) is_hash=y ;;
 131                 n,refs/tags/*) is_hash=y;;
 132                 n,*) is_hash=y; ;;
 133                 esac
 134         done
 135 }
 136
 137 __tg_remotes ()
 138 {
 139         local i ngoff IFS=$'\n' d="$(__tgdir)"
 140         shopt -q nullglob || ngoff=1
 141         shopt -s nullglob
 142         for i in "$d/remotes"/*; do
 143                 echo ${i#$d/remotes/}
 144         done
 145         [ "$ngoff" ] && shopt -u nullglob
 146         for i in $(git --git-dir="$d" config --list); do
 147                 case "$i" in
 148                 remote.*.url=*)
 149                         i="${i#remote.}"
 150                         echo "${i/.url=*/}"
 151                         ;;
 152                 esac
 153         done
 154 }
 155
 156 __tg_find_subcommand ()
 157 {
 158         local word subcommand c=1
 159
 160         while [ $c -lt $COMP_CWORD ]; do
 161                 word="${COMP_WORDS[c]}"
 162                 for subcommand in $1; do
 163                         if [ "$subcommand" = "$word" ]; then
 164                                 echo "$subcommand"
 165                                 return
 166                         fi
 167                 done
 168                 c=$((++c))
 169         done
 170 }
 171
 172 __tg_complete_revlist ()
 173 {
 174         local pfx cur="${COMP_WORDS[COMP_CWORD]}"
 175         case "$cur" in
 176         *...*)
 177                 pfx="${cur%...*}..."
 178                 cur="${cur#*...}"
 179                 __tgcomp "$(__tg_refs)" "$pfx" "$cur"
 180                 ;;
 181         *..*)
 182                 pfx="${cur%..*}.."
 183                 cur="${cur#*..}"
 184                 __tgcomp "$(__tg_refs)" "$pfx" "$cur"
 185                 ;;
 186         *)
 187                 __tgcomp "$(__tg_refs)"
 188                 ;;
 189         esac
 190 }
 191
 192 __tg_topics ()
 193 {
 194         tg summary -t
 195 }
 196
 197 __tg_commands ()
 198 {
 199         if [ -n "$__tg_all_commandlist" ]; then
 200                 echo "$__tg_all_commandlist"
 201                 return
 202         fi
 203         local i IFS=" "$'\n'
 204         for i in $(tg help | sed -n 's/^Usage:.*(\(.*\)).*/\1/p' | tr '|' ' ')
 205         do
 206                 case $i in
 207                 *--*)             : helper pattern;;
 208                 *) echo $i;;
 209                 esac
 210         done
 211 }
 212 __tg_all_commandlist=
 213 __tg_all_commandlist="$(__tg_commands 2>/dev/null)"
 214
 215 __tg_complete_arg ()
 216 {
 217         if [ $COMP_CWORD -gt 2 ] && [ "${COMP_WORDS[$COMP_CWORD - 1]}" = "$1" ]; then
 218                 return 0
 219         fi
 220         return 1
 221 }
 222
 223 ### }}}
 224 ### {{{ Commands
 225
 226 _tg_create ()
 227 {
 228         local cmd="$1"
 229         local cur="${COMP_WORDS[COMP_CWORD]}"
 230
 231         # Name must be the first arg after the create command
 232         if [ $((cmd + 1)) = $COMP_CWORD ]; then
 233                 __tgcomp "$(__tg_topics)"
 234                 return
 235         fi
 236
 237         __tg_complete_arg "-r" && {
 238                 __tgcomp "$(__tg_remotes)"
 239                 return
 240         }
 241
 242         case "$cur" in
 243         -*)
 244                 __tgcomp "
 245                         -r
 246                 "
 247                 ;;
 248         *)
 249                 __tgcomp "$(__tg_refs)"
 250         esac
 251 }
 252
 253 _tg_delete ()
 254 {
 255         local cur="${COMP_WORDS[COMP_CWORD]}"
 256
 257         case "$cur" in
 258         -*)
 259                 __tgcomp "
 260                         -f
 261                 "
 262                 ;;
 263         *)
 264                 __tgcomp "$(__tg_topics)"
 265         esac
 266 }
 267
 268 _tg_depend ()
 269 {
 270         local subcommands="add"
 271         local subcommand="$(__git_find_subcommand "$subcommands")"
 272         if [ -z "$subcommand" ]; then
 273                 __tgcomp "$subcommands"
 274                 return
 275         fi
 276
 277         case "$subcommand" in
 278         add)
 279                 __tgcomp "$(__tg_refs)"
 280         esac
 281 }
 282
 283 _tg_export ()
 284 {
 285         local cur="${COMP_WORDS[COMP_CWORD]}"
 286
 287         __tg_complete_arg "--collapse" && {
 288                 __tgcomp "$(__tg_heads)"
 289                 return
 290         }
 291
 292         __tg_complete_arg "--quilt" && {
 293                 return
 294         }
 295
 296         case "$cur" in
 297         *)
 298                 __tgcomp "
 299                         --collapse
 300                         --quilt
 301                 "
 302         esac
 303 }
 304
 305 _tg_help ()
 306 {
 307         local cur="${COMP_WORDS[COMP_CWORD]}"
 308         case "$cur" in
 309         -*)
 310                 COMPREPLY=()
 311                 return
 312                 ;;
 313         esac
 314         __tgcomp "$(__tg_commands)"
 315 }
 316
 317 _tg_import ()
 318 {
 319         local cur="${COMP_WORDS[COMP_CWORD]}"
 320
 321         __tg_complete_arg "-p" && {
 322                 COMPREPLY=()
 323                 return
 324         }
 325
 326         case "$cur" in
 327         -*)
 328                 __tgcomp "
 329                         -p
 330                 "
 331                 ;;
 332         *)
 333                 __tg_complete_revlist
 334         esac
 335 }
 336
 337 _tg_info ()
 338 {
 339         local cur="${COMP_WORDS[COMP_CWORD]}"
 340
 341         case "$cur" in
 342         *)
 343                 __tgcomp "$(__tg_topics)"
 344         esac
 345 }
 346
 347 _tg_mail ()
 348 {
 349         local cur="${COMP_WORDS[COMP_CWORD]}"
 350
 351         case "$cur" in
 352         *)
 353                 __tgcomp "$(__tg_topics)"
 354         esac
 355 }
 356
 357 _tg_patch ()
 358 {
 359         local cur="${COMP_WORDS[COMP_CWORD]}"
 360
 361         case "$cur" in
 362         *)
 363                 __tgcomp "$(__tg_topics)"
 364         esac
 365 }
 366
 367 _tg_remote ()
 368 {
 369         local cur="${COMP_WORDS[COMP_CWORD]}"
 370
 371         case "$cur" in
 372         *)
 373                 __tgcomp "$(__tg_remotes)"
 374         esac
 375 }
 376
 377 _tg_summary ()
 378 {
 379         COMPREPLY=()
 380 }
 381
 382 _tg_update ()
 383 {
 384         COMPREPLY=()
 385 }
 386
 387 ### }}}
 388 ### {{{ tg completion
 389
 390 _tg ()
 391 {
 392         local i c=1 command __tg_dir
 393
 394         while [ $c -lt $COMP_CWORD ]; do
 395                 i="${COMP_WORDS[c]}"
 396                 case "$i" in
 397                 -r)
 398                         c=$((++c))
 399                         if [ $c -lt $COMP_CWORD ]; then
 400                                 __tgcomp "$(__tg_remotes)"
 401                                 return
 402                         fi
 403                         ;;
 404                 -h|--help) command="help"; break ;;
 405                 *) command="$i"; break ;;
 406                 esac
 407                 c=$((++c))
 408         done
 409
 410         if [ -z "$command" ]; then
 411                 case "${COMP_WORDS[COMP_CWORD]}" in
 412                 -*)     __tgcomp "
 413                                 -r
 414                                 -h
 415                                 --help
 416                         "
 417                         ;;
 418                 *)     __tgcomp "$(__tg_commands)" ;;
 419                 esac
 420                 return
 421         fi
 422
 423         case "$command" in
 424         create)      _tg_create "$c" ;;
 425         delete)      _tg_delete ;;
 426         depend)      _tg_depend ;;
 427         export)      _tg_export ;;
 428         help)        _tg_help ;;
 429         import)      _tg_import ;;
 430         info)        _tg_info ;;
 431         mail)        _tg_mail ;;
 432         patch)       _tg_patch ;;
 433         remote)      _tg_remote ;;
 434         summary)     _tg_summary ;;
 435         update)      _tg_update ;;
 436         *)           COMPREPLY=() ;;
 437         esac
 438 }
 439
 440 ### }}}
 441
 442 complete -o default -o nospace -F _tg tg
 443
 444 # The following are necessary only for Cygwin, and only are needed
 445 # when the user has tab-completed the executable name and consequently
 446 # included the '.exe' suffix.
 447 #
 448 if [ Cygwin = "$(uname -o 2>/dev/null)" ]; then
 449 complete -o default -o nospace -F _tg tg.exe
 450 fi