#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

tags=($(hg tags 2> /dev/null | sed -e 's/[0-9]*:[a-f0-9]\{40\}$//; s/ *$//'))
subcmds=($(hg -v help | sed -e '1,/^list of commands:/d' \
      -e '/^global options:/,$d' -e '/^ [^ ]/!d; s/[,:]//g;'))

if [[ $service == "hg" ]]; then
    _arguments -C -A "-*" \
    '-R+[repository root directory]' \
    '-y[non-interactive]' \
    '-v[verbose]' \
    '-q[quiet]' \
    '--time[time how long the command takes]' \
    '--profile[profile]' \
    '-h-[display help and exit]' \
    '--version-[output version information and exit]' \
    '--cwd[change working directory]:new working directory:_files -/' \
    '*::command:->subcmd' && return 0

    if (( CURRENT == 1 )); then
        _wanted commands expl 'hg command' compadd -a subcmds
        return
    fi
    service="$words[1]"
    curcontext="${curcontext%:*}=$service:"
fi

case $service in
    (addremove)
        _arguments \
        '-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) )
        _arguments \
        '-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)
        _arguments \
        '-I[include path in search]:dir:_files -W $(hg root) -/' \
        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \
        '*:file:_files'
    ;;

    (cat)
        _arguments \
        '-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'
    ;;

    (checkout|update|up|co)
        _arguments \
        '-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
    ;;

    (commit|ci)
        _arguments \
        '-I[include path in search]:dir:_files -W $(hg root) -/' \
        '-X[exclude path in search]:dir:_files -W $(hg root) -/' \
        '-A[run addremove during commit]' \
        '-m[commit message]:string:' \
        '-d[date code]:string:' \
        '-u[user]:string:' \
        '*:file:_files'
    ;;

    (tag)
        _arguments \
        '-l[make the tag local]:' \
        '-m[commit message]:string:' \
        '-d[date code]:string:' \
        '-u[user]:string:' \
        '*:name and revision:'
    ;;

    (clone)
        _arguments \
        '-U[skip update after cloning]' \
        '-e[ssh command]:string:' \
        '--pull[use pull protocol to copy metadata]' \
        '--remotecmd[remote hg command]:command:->subcmd'
    ;;

    (export)
        _arguments \
        '-o[output to a file]:file:' \
        '-a-[tread all files as text]' \
        '*:revision:->revs'
        _wanted revs expl 'revision or tag' compadd -a tags
    ;;

    (heads)
        _arguments '-b[find branch info]'
    ;;

    (outgoing|out)
        _arguments '-p[show patch]'
    ;;

    (paths)
        _arguments '*:symbolic name:(default default-push)'
    ;;

    (init)
        _arguments '*:new repo directory:_files -/'
    ;;

    (unbundle)
        _arguments '*:changegroup file:_files'
    ;;

    (manifest)
        _arguments \
        '*:revision:->revs'
        _wanted revs expl 'revision or tag' compadd -a tags
    ;;

    (parents)
        _arguments \
        '*:revision:->revs'
        _wanted revs expl 'revision or tag' compadd -a tags
    ;;

    (serve)
        _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]'
    ;;

    (pull)
        repos=( $(hg paths | sed -e 's/^.*= //') )
        _arguments \
        '-u[update working directory]' \
        '-e[ssh command]:remote commands:' \
        '--remotecmd[remote hg command]:command:->subcmd' \
        '*:pull source:->repo'
        _wanted source expl 'source repository' compadd -a repos
    ;;

	(push)
        repos=( $(hg paths | sed -e 's/^.*= //') )
        _arguments \
        '-f[force push]' \
        '-e[ssh command]:remote commands:' \
        '--remotecmd[remote hg command]:command:->subcmd' \
        '*:pull source:->repo'
        _wanted source expl 'source repository' compadd -a repos
    ;;

    (id|identify)
    ;;

    (recover)
    ;;

    (rawcommit)
        _arguments \
        '-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'
    ;;

    (copy|cp)
        _arguments \
        '-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:'
    ;;

    (rename|mv)
        _arguments \
        '-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:'
    ;;

    (forget)
        addedFiles=( $(hg status -an) )
        _arguments \
        '-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
    ;;

    (import|patch)
        _arguments \
        '-p[path strip (default: 1)]:count:' \
        '-f[skip check for outstanding changes]' \
        '-b[base path]:file:_files -W $(hg root)' \
        '*:patch file:_files'
    ;;

    (incoming|in)
        _arguments \
        '-p[show patch]' \
        '*:mercurial repository:_files'
    ;;

    (diff)
        _arguments \
        '-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'
    ;;

    (log|history)
        _arguments \
        '-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'
    ;;

    (grep)
        _arguments \
        '-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'
    ;;

    (status)
        _arguments \
        '-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'
    ;;

    (locate)
        _arguments \
        '-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:'
    ;;

    (help)
        _wanted commands expl 'hg command' compadd -a subcmds
    ;;

    (root|undo|view|verify|version)
        # no arguments for these commands
    ;;

    (*)
        _message "unknown hg command completion: $service"
    ;;
esac