about summary refs log tree commit diff
path: root/Completion/Unix/Command/_stgit
diff options
context:
space:
mode:
Diffstat (limited to 'Completion/Unix/Command/_stgit')
-rw-r--r--Completion/Unix/Command/_stgit29
1 files changed, 27 insertions, 2 deletions
diff --git a/Completion/Unix/Command/_stgit b/Completion/Unix/Command/_stgit
index 69b50004c..c8a3c8c80 100644
--- a/Completion/Unix/Command/_stgit
+++ b/Completion/Unix/Command/_stgit
@@ -50,8 +50,33 @@ subcmds=(
   'status:show the tree status'
 )
 
+local curcontext="$curcontext"
+local subcmd
+local ret=1
+
 if (( CURRENT == 2 )); then
-  _describe -t commands 'stgit command' subcmds
+  _describe -t commands 'stgit command' subcmds && ret=0
 else
-  _files
+  shift words
+  (( CURRENT-- ))
+  subcmd="$words[1]"
+  curcontext="${curcontext%:*}-${subcmd}:"
+
+  case $subcmd in
+    (push)
+      _wanted -V "unapplied patches" expl "patch" compadd $(stg unapplied) \
+		&& ret=0
+    ;;
+    (pop)
+      _wanted -V "applied patches" expl "patch" compadd $(stg applied) && ret=0
+    ;;
+    (edit|files|goto|rename|log|float|delete|sink|mail|sync|show|pick|hide)
+      _wanted -V "patches" expl "patch" compadd $(stg series --noprefix) && ret=0
+    ;;
+    (*)
+      _files
+    ;;
+  esac
 fi
+
+return ret