#compdef hg

# Zsh completion script for mercurial.  Rename this file to _hg and copy

# it into your zsh function path (/usr/share/zsh/site-functions for

# instance)

#

# Copyright (C) 2005 Steve Borho

#

# This is free software; you can redistribute it and/or modify it under

# the terms of the GNU General Public License as published by the Free

# Software Foundation; either version 2 of the License, or (at your

# option) any later version.

#

local curcontext="$curcontext" state line

typeset -A opt_args

local subcmds repos tags newFiles addedFiles

subcmds=($(hg -v help | sed -e '1,/^list of commands:/d' \

      -e '/^global options:/,$d' -e '/^ [^ ]/!d; s/[,:]//g;'))

if [[ $service == "hg" ]]; then

    '--cwd[change working directory]:new working directory:_files -/' \

fi

case $service in

    (addremove)

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '*:directories:_files -/'  # assume they want to add/remove a dir

    ;;

    (add)

        newFiles=( $(hg status -un) )

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/'

        _wanted files expl 'unknown files' compadd -a newFiles

    ;;

    (remove|rm)

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '*:file:_files'

    ;;

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-o[output to file]:file:' \

        '-r[revision]:revision:($tags)' \

        '*:file:_files'

        '-b[checkout the head of a specific branch]:tag:' \

        '-m[allow merging of conflicts]' \

        '-C[overwrite locally modified files]' \

        '*:revision:->revs'

        _wanted revs expl 'revision or tag' compadd -a tags

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '*:file:_files'

        '*:name and revision:'

        '-o[output to a file]:file:' \

        '-a-[tread all files as text]' \

        '*:revision:->revs'

        _wanted revs expl 'revision or tag' compadd -a tags

    ;;

        _arguments '*:changegroup file:_files'

    ;;

        '*:revision:->revs'

        _wanted revs expl 'revision or tag' compadd -a tags

    ;;

        '*:revision:->revs'

        _wanted revs expl 'revision or tag' compadd -a tags

    ;;

        _arguments \

        '-A[access log file]:log file:_files' \

        '-E[error log file]:log file:_files' \

        '-p[listen port]:listen port:' \

        '-a[interface address]:interface address:' \

        '-n[repository name]:repository name:' \

        '--stdio[for remote clients]' \

        '-t[template directory]:template dir:_files -/' \

        '--style[template style]:style' \

        '-6[use IPv6 in addition to IPv4]'

    ;;

        repos=( $(hg paths | sed -e 's/^.*= //') )

        '-u[update working directory]' \

        '-e[ssh command]:remote commands:' \

        '*:pull source:->repo'

    ;;

        repos=( $(hg paths | sed -e 's/^.*= //') )

        '-f[force push]' \

        '-e[ssh command]:remote commands:' \

        '*:pull source:->repo'

    ;;

        '-p[parent]:revision:($tags)' \

        '-d[date]:date:' \

        '-u[user]:user:' \

        '-F[file list]:file list:_files' \

        '-m[commit message]:string:' \

        '-l[commit message file]:message file:_files -g *.txt' \

        '*:files to commit:_files'

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-A-[record a copy after it has happened]' \

        '-f[replace destination if it exists]' \

        '-p[append source path to dest]' \

        '*:destination:'

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-A-[record a copy after it has happened]' \

        '-f[replace destination if it exists]' \

        '-p[append source path to dest]' \

        '*:destination:'

        addedFiles=( $(hg status -an) )

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/'

        _wanted files expl 'newly added files' compadd -a addedFiles

        '-p[path strip (default: 1)]:count:' \

        '-f[skip check for outstanding changes]' \

        '-b[base path]:file:_files -W $(hg root)' \

        '*:patch file:_files'

        '-p[show patch]' \

        '*:mercurial repository:_files'

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-r[revision]:revision:($tags)' \

        '-a-[tread all files as text]' \

        '*:file:_files'

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-r[revision]:revision:($tags)' \

        '-b[show branches]' \

        '-p[show patch]' \

        '*:file:_files'

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-0[end fields with NUL]' \

        '--all[print all revisions with matches]' \

        '-i[ignore case]' \

        '-l[print names of files and revs with matches]' \

        '-n[print line numbers]' \

        '-r[search in revision rev]:revision:($tags)' \

        '-u[print user who made change]' \

        '*:search pattern, then files:_files'

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '-0[end filenames with NUL]' \

        '-m[show only modified files]' \

        '-a[show only added files]' \

        '-r[show only removed files]' \

        '-u[show only unknown files]' \

        '-n[hide status prefix]' \

        '*:search pattern, then files:_files'

        '-r[search in revision rev]:revision:($tags)' \

        '-0[end fields with NUL]' \

        '-f[print complete paths]' \

        '-I[include path in search]:dir:_files -W $(hg root) -/' \

        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \

        '*:search pattern:'

    (root|undo|view|verify|version)

        # no arguments for these commands

    ;;

esac