Merge pull request #227 from danielshahaf/completion-set-context-foreach-dispatch-v1

[code/vcsh.git] / _vcsh

diff --git a/_vcsh b/_vcsh
index 6a21605f740afff657ced9a960374fd41c3bbe0d..6880f728eaf266b0573bd0da9acf556f48428658 100644 (file)
--- a/_vcsh
+++ b/_vcsh
@@ -1,12 +1,151 @@
 #compdef vcsh
 
-_arguments \
-       ':subcommand:((
-               help\:"display help"
-               list\:"list all repos"
-               use\:"use/enter repo"
-               run\:"run command on repo"
-               init\:"init & clone from repo"
-               clone\:"clone from repo"
-               exit\:"exit repo"
-       ))'
+function __vcsh_repositories () {
+       local expl
+       local -a repos
+       repos=( ${(f)"$(command vcsh list)"} )
+       _describe -t repos 'repositories' repos
+}
+
+function __vcsh_not_implemented_yet () {
+       _message "Subcommand completion '${1#*-}': not implemented yet"
+}
+
+function _vcsh-clone () {
+       __vcsh_not_implemented_yet "$0" #TODO
+}
+
+function _vcsh-delete () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+}
+
+function _vcsh-enter () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+}
+
+function _vcsh-foreach () {
+       _dispatch vcsh-foreach git
+}
+
+function _vcsh-help () {
+       _nothing
+}
+
+function _vcsh-init () {
+       _nothing
+}
+
+function _vcsh-list () {
+       _nothing
+}
+
+function _vcsh-list-tracked () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+}
+
+function _vcsh-list-untracked () {
+       _nothing
+}
+
+function _vcsh-pull () {
+       _nothing
+}
+
+function _vcsh-push () {
+       _nothing
+}
+
+function _vcsh-rename () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+       (( CURRENT == 3 )) && _message "new repository name"
+       (( CURRENT > 3 )) && _nothing
+}
+
+function _vcsh-run () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+       if (( CURRENT >= 3 )); then
+               words=( "${(@)words[3,-1]}" )
+               (( CURRENT -= 2 ))
+               _complete
+       fi
+}
+
+function _vcsh-status () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+}
+
+function _vcsh-upgrade () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+}
+
+function _vcsh-version () {
+       _nothing
+}
+
+function _vcsh-which () {
+       _files
+}
+
+function _vcsh-write-gitignore () {
+       (( CURRENT == 2 )) && __vcsh_repositories
+}
+
+function _vcsh () {
+       local curcontext="${curcontext}"
+       local state vcshcommand
+       local -a args subcommands
+
+       local VCSH_REPO_D
+        : ${VCSH_REPO_D:="${XDG_CONFIG_HOME:-"$HOME/.config"}/vcsh/repo.d"}
+
+       subcommands=(
+               "clone:clone an existing repository"
+               "commit:commit in all repositories"
+               "delete:delete an existing repository"
+               "enter:enter repository; spawn new <\$SHELL>"
+               "foreach:execute for all repos"
+               "help:display help"
+               "init:initialize an empty repository"
+               "list:list all local vcsh repositories"
+               "list-tracked:list all files tracked by vcsh"
+               "list-untracked:list all files not tracked by vcsh"
+               "pull:pull from all vcsh remotes"
+               "push:push to vcsh remotes"
+               "rename:rename a repository"
+               "run:run command with <\$GIT_DIR> and <\$GIT_WORK_TREE> set"
+               "status:show statuses of all/one vcsh repositories"
+               "upgrade:upgrade repository to currently recommended settings"
+               "version:print version information"
+               "which:find <substring> in name of any tracked file"
+               "write-gitignore:write .gitignore.d/<repo> via git ls-files"
+       )
+
+       args=(
+               '-c[source <file> prior to other configuration files]:config files:_path_files'
+               '-d[enable debug mode]'
+               '-v[enable verbose mode]'
+               '*:: :->subcommand_or_options_or_repo'
+       )
+
+       _arguments -C ${args} && return
+
+       if [[ ${state} == "subcommand_or_options_or_repo" ]]; then
+               if (( CURRENT == 1 )); then
+                       _describe -t subcommands 'vcsh sub-commands' subcommands
+                       __vcsh_repositories
+               else
+                       vcshcommand="${words[1]}"
+                       if ! (( ${+functions[_vcsh-$vcshcommand]} )); then
+                               # There is no handler function, so this is probably the name
+                               # of a repository. Act accordingly.
+                               # FIXME: this may want to use '_dispatch vcsh git'
+                               GIT_DIR=$VCSH_REPO_D/$words[1].git _dispatch git git
+                       else
+                               curcontext="${curcontext%:*:*}:vcsh-${vcshcommand}:"
+                               _call_function ret _vcsh-${vcshcommand}
+                       fi
+               fi
+       fi
+}
+
+_vcsh "$@"