local repos newFiles addedFiles commitMessage

typeset -A _hg_cmd_globals

_hg() {

  local cmd

  integer i=2

  _hg_cmd_globals=()

  while (( i < $#words ))

  do

    case "$words[$i]" in

      -R|--repository|--cwd|--config)

        # pass along arguments to hg completer

        _hg_cmd_globals+="$words[$i]"

        _hg_cmd_globals+="$words[$i+1]"

        (( i += 2 ))

        continue

      ;;

      -R*)

        _hg_cmd_globals+="$words[$i]"

       (( i++ ))

       continue

      ;;

      -*)

        # skip option

        (( i++ ))

        continue

      ;;

    esac

    if [[ -z "$cmd" ]]

    then

      cmd="$words[$i]"

      words[$i]=()

      (( CURRENT-- ))

    fi

    (( i++ ))

  done

  if [[ -z "$cmd" ]]

  then

    _arguments -s -w : $_hg_global_opts \

    ':mercurial command:_hg_commands'

    return

  fi

  # resolve abbreviations and aliases

  if ! (( $+functions[_hg_cmd_${cmd}] ))

  then

    local cmdexp

    (( $#_hg_cmd_list )) || _hg_get_commands

    cmdexp=$_hg_cmd_list[(r)${cmd}*]

    if [[ $cmdexp == $_hg_cmd_list[(R)${cmd}*] ]]

    then

      # might be nice to rewrite the command line with the expansion

      cmd="$cmdexp"

    fi

    if [[ -n $_hg_alias_list[$cmd] ]]

    then

      cmd=$_hg_alias_list[$cmd]

    fi

  fi

  if (( $+functions[_hg_cmd_${cmd}] ))

  then

    curcontext="${curcontext%:*:*}:hg-${cmd}:"

    _hg_cmd_${cmd}

    return

  fi

}

_hg_get_commands() {

  typeset -ga _hg_cmd_list

  typeset -gA _hg_alias_list

  local hline cmd cmdalias

  _call_program help hg --verbose help | while read -A hline

  do

    cmd="$hline[1]"

    case $cmd in

      *:)

        cmd=${cmd%:}

        _hg_cmd_list+=($cmd)

      ;;

      *,)

        cmd=${cmd%,}

        _hg_cmd_list+=($cmd)

        integer i=2

        while (( i <= $#hline ))

        do

          cmdalias=${hline[$i]%(:|,)}

          _hg_cmd_list+=($cmdalias)

          _hg_alias_list+=($cmdalias $cmd)

          (( i++ ))

        done

      ;;

    esac

  done

}

_hg_commands() {

  (( $#_hg_cmd_list )) || _hg_get_commands

  _describe -t hg-commands 'Mercurial command' _hg_cmd_list

}

_hg_tags() {

  typeset -a tags

  local tag rev

  _hg_cmd tags 2> /dev/null | while read tag rev

  do

    tags+=($tag)

  done

  (( $#tags )) && _describe -t hg-tags 'tags' tags

}

_hg_status() {

  status_files=(${(ps:\0:)"$(_hg_cmd status -0n$1 .)"})

}

_hg_unknown() {

  typeset -a status_files

  _hg_status u

  (( $#status_files )) && _describe -t hg-unknown-files 'unknown files' status_files

}

_hg_missing() {

  typeset -a status_files

  _hg_status d

  (( $#status_files )) && _describe -t hg-missing-files 'missing files' status_files

}

_hg_addremove() {

  _alternative "unknown files:unknown files:_hg_unknown" \

  "missing files:missing files:_hg_missing"

}

_hg_paths() {

  typeset -a paths pnames

  _hg_cmd paths 2> /dev/null | while read -A pnames

  do

    paths+=($pnames[1])

  done

  (( $#paths )) && _describe -t hg-paths 'repository aliases' paths

}

_hg_remote() {

  _alternative 'repository aliases:repository aliases:_hg_paths' \

  'directory:directory:_files -/'

}

_hg_qseries() {

  typeset -a patches

  patches=($(_hg_cmd qseries))

  (( $#patches )) && _describe -t hg-patches 'patches' patches

}

_hg_qapplied() {

  typeset -a patches

  patches=($(_hg_cmd qapplied))

  (( $#patches )) && _describe -t hg-applied-patches 'applied patches' patches

}

_hg_qunapplied() {

  typeset -a patches

  patches=($(_hg_cmd qunapplied))

  (( $#patches )) && _describe -t hg-unapplied-patches 'unapplied patches' patches

# Common options

_hg_global_opts=(

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

    '(-R)--repository[repository root directory]:repository:_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]'

)

_hg_pat_opts=(

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

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

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

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

_hg_diff_opts=(

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

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

  '(--git)-g[use git extended diff format]'

  '(-g)--git[use git extended diff format]'

  "--nodates[don't include dates in diff headers]")

_hg_dryrun_opts=(

  '(--dry-run)-n[do not perform actions, just print output]'

  '(-n)--dry-run[do not perform actions, just print output]')

_hg_style_opts=(

  '--style[display using template map file]:'

  '--template[display with template]:')

_hg_commit_opts=(

  '(-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')

_hg_remote_opts=(

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

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

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

_hg_cmd() {

  _call_program hg hg "$_hg_cmd_globals[@]" "$@"

}

_hg_cmd_add() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts $_hg_dryrun_opts \

  '*:unknown files:_hg_unknown'

}

_hg_cmd_addremove() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts $_hg_dryrun_opts \

  '(--similarity)-s[guess renamed files by similarity (0<=s<=100)]:' \

  '(-s)--similarity[guess renamed files by similarity (0<=s<=100)]:' \

  '*:unknown or missing files:_hg_addremove'

}

_hg_cmd_annotate() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

  '(--rev)-r+[annotate the specified revision]:revision:_hg_tags' \

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

  '(--follow)-f[follow file copies and renames]' \

  '(-f)--follow[follow file copies and renames]' \

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

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

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

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

  '(--date)-d[list the date]' \

  '(-d)--date[list the date]' \

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

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

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

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

  '*:files:_files -W $(_hg_cmd root)'

}

_hg_cmd_archive() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

  '--no-decode[do not pass files through decoders]' \

  '(--prefix)-p+[directory prefix for files in archive]:' \

  '(-p)--prefix[directory prefix for files in archive]:' \

  '(--rev)-r+[revision to distribute]:revision:_hg_tags' \

  '(-r)--rev[revision to distribute]:revision:_hg_tags' \

  '(--type)-t+[type of distribution to create]:archive type:(files tar tbz2 tgz uzip zip)' \

  '(-t)--type[type of distribution to create]:archive type:(files tar tbz2 tgz uzip zip)' \

  '*:destination:_files'

}

_hg_cmd_bundle() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

  '(--force)-f[run even when remote repository is unrelated]' \

  '(-f)--force[run even when remote repository is unrelated]' \

  '(2)*--base[a base changeset to specify instead of a destination]:revision:_hg_tags' \

  ':output file:_files' \

  ':destination repository:_files -/'

}

_hg_cmd_cat() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

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

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

  '(--rev)-r+[revision]:revision:_hg_tags' \

  '(-r)--rev[revision]:revision:_hg_tags' \

  '*:file:_files -W $(_hg_cmd root)'

}

_hg_cmd_clone() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

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

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

  '(--rev)-r+[a changeset you would like to have after cloning]:' \

  '(-r)--rev[a changeset you would like to have after cloning]:' \

  '--uncompressed[use uncompressed transfer (fast over LAN)]' \

  ':source repository:_hg_remote' \

  ':destination:_files -/'  

}

_hg_cmd_commit() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

  '(--addremove)-A[mark new/missing files as added/removed before committing]'

  '(-A)--addremove[mark new/missing files as added/removed before committing]'

  '(--message)-m+[use <text> as commit message]:text:' \

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

  '(--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:_files -W $(_hg_cmd root)'

}

_hg_cmd_copy() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts $_hg_dryrun_opts \

  '(--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]' \

  '*:file:_files -W $(_hg_cmd root)'

}

_hg_cmd_diff() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts $_hg_diff_opts \

  '*'{-r,--rev}'+[revision]:revision:_hg_tags ' \

  '(--show-function)-p[show which function each change is in]' \

  '(-p)--show-function[show which function each change is in]' \

  '(--ignore-all-space)-w[ignore white space when comparing lines]' \

  '(-w)--ignore-all-space[ignore white space when comparing lines]' \

  '(--ignore-space-change)-b[ignore changes in the amount of white space]' \

  '(-b)--ignore-space-change[ignore changes in the amount of white space]' \

  '(--ignore-blank-lines)-B[ignore changes whose lines are all blank]' \

  '(-B)--ignore-blank-lines[ignore changes whose lines are all blank]' \

  '*:file:_files -W $(_hg_cmd root)'

_hg_cmd_export() {

  _arguments -s -w : $_hg_global_opts $_hg_diff_opts \

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

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

  '--switch-parent[diff against the second parent]' \

  '*:revision:_hg_tags'

}

_hg_cmd_grep() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

  '(-print0)-0[end filenames with NUL]' \

  '(-0)--print0[end filenames with NUL]' \

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

  '(--follow)-f[follow changeset or file history]' \

  '(-f)--follow[follow changeset or file history]' \

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

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

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

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

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

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

  '*-r+[search in given revision range]:revision:_hg_tags' \

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

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

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

  '*:search pattern:_files -W $(_hg_cmd root)'

}

_hg_cmd_heads() {

  _arguments -s -w : $_hg_global_opts $_hg_style_opts \

  '(--rev)-r+[show only heads which are descendants of rev]:revision:_hg_tags' \

  '(-r)--rev[show only heads which are descendants of rev]:revision:_hg_tags'

}

_hg_cmd_help() {

  _arguments -s -w : $_hg_global_opts \

  '*:mercurial command:_hg_commands'

}

_hg_cmd_import() {

  _arguments -s -w : $_hg_global_opts \

  '(--strip)-p+[directory strip option for patch (default: 1)]:count:' \

  '(-p)--strip[directory strip option for patch (default: 1)]:count:' \

  '(--message)-m+[use <text> as commit message]:text:' \

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

  '(--force)-f[skip check for outstanding uncommitted changes]' \

  '(-f)--force[skip check for outstanding uncommitted changes]' \

  '*:patch:_files'

}

_hg_cmd_incoming() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts $_hg_style_opts \

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

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

  '(--force)-f[run even when the remote repository is unrelated]' \

  '(-f)--force[run even when the remote repository is unrelated]' \

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

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

  '(--rev)-r+[a specific revision up to which you would like to pull]' \

  '(-r)--rev[a specific revision up to which you would like to pull]' \

  '(--newest-first)-n[show newest record first]' \

  '(-n)--newest-first[show newest record first]' \

  '--bundle[file to store the bundles into]:bundle file:_files' \