#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 newFiles addedFiles includeExclude commitMessage

local _hg_state

_hg_state () {

  case "$state" in

    (tags)

      compadd $(hg tags 2> /dev/null |

        sed -e 's/[0-9]*:[a-f0-9]*$//; s/ *$//')

    ;;

    (qapplied)

      compadd $(hg qapplied)

    ;;

    (qunapplied)

      compadd $(hg qunapplied)

    ;;

  esac

}

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

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

# A lot of commands have these arguments

includeExclude=(

        '*-I-[include names matching the given patterns]:dir:_files -W $(hg root) -/'

        '*--include-[include names matching the given patterns]:dir:_files -W $(hg root) -/'

        '*-X-[exclude names matching the given patterns]:dir:_files -W $(hg root) -/'

        '*--exclude-[exclude names matching the given patterns]:dir:_files -W $(hg root) -/')

commitMessage=(

        '(-m --message -l --logfile --edit)-e[edit commit message]'

        '(-m --message -l --logfile -e)--edit[edit commit message]'

        '(-e --edit -l --logfile --message)-m[use <text> as commit message]:message:'

        '(-e --edit -l --logfile -m)--message[use <text> as commit message]:message:'

        '(-e --edit -m --message --logfile)-l[read the commit message from <file>]:log file:_files'

        '(-e --edit -m --message -l)--logfile[read the commit message from <file>]:log file:_files')

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

    _arguments -C -A "-*" \

    '(--repository)-R[repository root directory]:root:_files -/' \

    '(-R)--repository[repository root directory]:root:_files -/' \

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

    '(--noninteractive)-y[do not prompt, assume yes for any required answers]' \

    '(-y)--noninteractive[do not prompt, assume yes for any required answers]' \

    '(--verbose)-v[enable additional output]' \

    '(-v)--verbose[enable additional output]' \

    '(--quiet)-q[suppress output]' \

    '(-q)--quiet[suppress output]' \

    '(--help)-h[display help and exit]' \

    '(-h)--help[display help and exit]' \

    '--debug[debug mode]' \

    '--debugger[start debugger]' \

    '--traceback[print traceback on exception]' \

    '--time[time how long the command takes]' \

    '--profile[profile]' \

    '--version[output version information and exit]' \

    '*::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

    (add)

        newFiles=(${(ps:\0:)"$(hg status -0un .)"})

        _arguments $includeExclude \

        '*:file:->unknown'

        _wanted files expl 'unknown files' compadd -a newFiles

    ;;

    (addremove)

        _arguments $includeExclude \

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

    ;;

    (forget)

        addedFiles=(${(ps:\0:)"$(hg status -0an .)"})

        _arguments $includeExclude  \

        '*:file:->added'

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

    ;;

    (remove|rm)

        _arguments $includeExclude \

        '*:file:_files'

    ;;

    (copy|cp)

        _arguments $includeExclude \

        '(--after)-A[record a copy that has already occurred]' \

        '(-A)--after[record a copy that has already occurred]' \

        '(--force)-f[forcibly copy over an existing managed file]' \

        '(-f)--force[forcibly copy over an existing managed file]' \

        '(--parents)-p[append source path to dest]' \

        '(-p)--parents[append source path to dest]' \

        '*:files:_files'

    ;;

    (rename|mv)

        if (( CURRENT == 2 )); then

            _arguments $includeExclude \

            '(--after)-A[record a rename that has already occurred]' \

            '(-A)--after[record a rename that has already occurred]' \

            '(--force)-f[replace destination if it exists]' \

            '(-F)--force[replace destination if it exists]' \

            '(--parents)-p[append source path to dest]' \

            '(-p)--parents[append source path to dest]' \

            '*:files:_files'

        else

            _arguments '*:destination:_files'

        fi

    ;;

    (diff)

        _arguments $includeExclude \

        '*-r[revision]:revision:->tags' \

        '*--rev[revision]:revision:->tags' \

        '(--text)-a[treat all files as text]' \

        '(-a)--text[treat all files as text]' \

        '*:file:_files'

    ;;

    (status|st)

        _arguments $includeExclude \

        '(--no-status)-n[hide status prefix]' \

        '(-n)--no-status[hide status prefix]' \

        '(--print0)-0[end filenames with NUL, for use with xargs]' \

        '(-0)--print0[end filenames with NUL, for use with xargs]' \

        '(--modified)-m[show only modified files]' \

        '(-m)--modified[show only modified files]' \

        '(--added)-a[show only added files]' \

        '(-a)--added[show only added files]' \

        '(--removed)-r[show only removed files]' \

        '(-r)--removed[show only removed files]' \

        '(--unknown)-u[show only unknown files]' \

        '(-u)--unknown[show only unknown files]' \

        '*:search pattern, then files:_files'

    ;;

    (revert)

        addedFiles=(${(ps:\0:)"$(hg status -0amrn .)"})

        _arguments \

        '(--rev)-r[revision to revert to]:revision:->tags' \

        '(-r)--rev[revision to revert to]:revision:->tags' \

        '(--nonrecursive)-n[do not recurse into subdirectories]' \

        '(-n)--nonrecursive[do not recurse into subdirectories]' \

        '*:file:->modified'

        _wanted files expl 'mofified files' compadd -a addedFiles

    ;;

    (commit|ci)

        addedFiles=(${(ps:\0:)"$(hg status -0amrn .)"})

        _arguments $includeExclude \

        '(--addremove)-A[run addremove during commit]' \

        '(-A)--addremove[run addremove during commit]' \

        '(--message)-m[use <txt> as commit message]:string:' \

        '(-m)--message[use <txt> as commit message]:string:' \

        '(--logfile)-l[read commit message from <file>]:.log file:_file -g \*.txt' \

        '(-l)--logfile[read commit message from <file>]:.log file:_file -g \*.txt' \

        '(--date)-d[record datecode as commit date]:date code:' \

        '(-d)--date[record datecode as commit date]:date code:' \

        '(--user)-u[record user as commiter]:user:' \

        '(-u)--user[record user as commiter]:user:' \

        '*:file:->modified'

        _wanted files expl 'mofified files' compadd -a addedFiles

    ;;

    (cat)

        _arguments $includeExclude \

        '(--output)-o[print output to file with formatted name]:filespec:' \

        '(-o)--output[print output to file with formatted name]:filespec:' \

        '(--rev)-r[revision]:revision:->tags' \

        '(-r)--rev[revision]:revision:->tags' \

        '*:file:_files'

    ;;

    (annotate)

        _arguments $includeExclude \

        '(--rev)-r[annotate the specified revision]:revision:->tags' \

        '(-r)--rev[annotate the specified revision]:revision:->tags' \

        '(--text)-a[treat all files as text]' \

        '(-a)--text[treat all files as text]' \

        '(--user)-u[list the author]' \

        '(-u)--user[list the author]' \

        '(--changeset)-c[list the changeset]' \

        '(-c)--changeset[list the changeset]' \

        '(--number)-n[list the revision number (default)]' \

        '(-n)--number[list the revision number (default)]' \

        '*:files:_files'

    ;;

    (grep)

        _arguments $includeExclude \

        '*-r[search in given revision range]:revision:->tags' \

        '*--rev[search in given revision range]:revision:->tags' \

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

        '(-print0)-0[end filenames with NUL, for use with xargs]' \

        '(-0)--print0[end filenames with NUL, for use with xargs]' \

        '(--ignore-case)-i[ignore case when matching]' \

        '(-i)--ignore-case[ignore case when matching]' \

        '(--files-with-matches)-l[print names of files and revs that match]' \

        '(-l)--files-with-matches[print names of files and revs that match]' \

        '(--line-number)-n[print matching line numbers]' \

        '(-n)--line-number[print matching line numbers]' \

        '(--user)-u[print user who committed change]' \

        '(-u)--user[print user who committed change]' \

        '*:search pattern:'

    ;;

    (locate)

        _arguments $includeExclude \

        '(--rev)-r[search repository as it stood at revision]:revision:->tags' \

        '(-r)--rev[search repository as it stood at revision]:revision:->tags' \

        '(--print0)-0[end filenames with NUL, for use with xargs]' \

        '(-0)--print0[end filenames with NUL, for use with xargs]' \

        '(--fullpath)-f[print complete paths]' \

        '(-f)--fullpath[print complete paths]' \

        '*:search pattern:'

    ;;

    (log|history)

        '*-r[show the specified revision or range]:revision:->tags' \

        '*--rev[show the specified revision or range]:revision:->tags' \

        '(--no-merges -M --only-merges)-m[show only merge revisions]' \

        '(--no-merges -M -m)--only-merges[show only merge revisions]' \

        '(--only-merges -m --no-merges)-M[do not show merge revisions]' \

        '(--only-merges -m -M)--no-merges[do not show merge revisions]' \

        '(--keyword)-k[search for a keyword]:keyword:' \

        '(-k)--keyword[search for a keyword]:keyword:' \

        '(--branch)-b[show branches]' \

        '(-b)--branch[show branches]' \

        '(--patch)-p[show patch]' \

        '(-p)--patch[show patch]' \

        '*:file:_files'

    ;;

    (update|checkout|co)

        _arguments \

        '(--branch)-b[checkout the head of a specific branch]' \

        '(-b)--branch[checkout the head of a specific branch]' \

        '(-C --clean --merge)-m[allow merging of branches]' \

        '(-C --clean -m)--merge[allow merging of branches]' \

        '(-m --merge --clean)-C[overwrite locally modified files]' \

        '(-m --merge -C)--clean[overwrite locally modified files]' \

        '*:revision or tag:->tags'

    ;;

    (tag)

        _arguments \

        '(--local)-l[make the tag local]' \

        '(-l)--local[make the tag local]' \

        '(--message)-m[message for tag commit log entry]:string:' \

        '(-m)--message[message for tag commit log entry]:string:' \

        '(--date)-d[record datecode as commit date]:date code:' \

        '(-d)--date[record datecode as commit date]:date code:' \

        '(--user)-u[record user as commiter]:user:' \

        '(-u)--user[record user as commiter]:user:' \

        '*:name, then revision:->tags'

    ;;

    (clone)

        if (( CURRENT == 2 )); then

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

            _arguments \

            '(--no-update)-U[do not update the new working directory]' \

            '(-U)--no-update[do not update the new working directory]' \

            '(--ssh)-e[specify ssh command to use]:string:' \

            '(-e)--ssh[specify ssh command to use]:string:' \

            '--pull[use pull protocol to copy metadata]' \

            '--remotecmd[specify hg command to run on the remote side]:remote hg:' \

            '*:local repo:_files -/'

            _wanted source expl 'source repository' compadd -a repos

        elif (( CURRENT == 3 )); then

            _arguments '*:dest repo:_files -/'

        fi

    ;;

    (rawcommit)

        _arguments \

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

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

        '(--date)-d[record datecode as commit date]:date code:' \

        '(-d)--date[record datecode as commit date]:date code:' \

        '(--user)-u[record user as commiter]:user:' \

        '(-u)--user[record user as commiter]:user:' \

        '(--message)-m[use <txt> as commit message]:string:' \

        '(-m)--message[use <txt> as commit message]:string:' \

        '(--logfile)-l[read commit message from <file>]:.log file:_file -g \*.txt' \

        '(-l)--logfile[read commit message from <file>]:.log file:_file -g \*.txt' \

        '(--files)-F[file list]:file list:_files' \