#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

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

}

_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=(

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

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

_hg_diff_opts=(

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

_hg_dryrun_opts=(

_hg_style_opts=(

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

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

_hg_commit_opts=(

_hg_remote_opts=(

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

  '*:unknown or missing files:_hg_addremove'

}

_hg_cmd_annotate() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

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

  '*:destination:_files'

}

_hg_cmd_bundle() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

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

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

}

_hg_cmd_clone() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

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

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

}

_hg_cmd_copy() {

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

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

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

}

_hg_cmd_export() {

  _arguments -s -w : $_hg_global_opts $_hg_diff_opts \

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

  '*:revision:_hg_tags'

}

_hg_cmd_grep() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

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

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

}

_hg_cmd_heads() {

  _arguments -s -w : $_hg_global_opts $_hg_style_opts \

}

_hg_cmd_help() {

  _arguments -s -w : $_hg_global_opts \

  '*:mercurial command:_hg_commands'

}

_hg_cmd_import() {

  _arguments -s -w : $_hg_global_opts \

  '*:patch:_files'

}

_hg_cmd_incoming() {

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

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

  ':source:_hg_remote'

}

_hg_cmd_init() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

  ':dir:_files -/'

}

_hg_cmd_locate() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

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

}

_hg_cmd_log() {

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

  '(-f --follow)--follow-first[only follow the first parent of merge changesets]' \

  '*'{-r,--rev}'[show the specified revision or range]:revision:_hg_tags' \

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

}

_hg_cmd_manifest() {

  _arguments -s -w : $_hg_global_opts \

  ':revision:_hg_tags'

}

_hg_cmd_outgoing() {

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

  ':destination:_hg_remote'

}

_hg_cmd_parents() {

  _arguments -s -w : $_hg_global_opts $_hg_style_opts \

  ':revision:_hg_tags'

}

_hg_cmd_paths() {

  _arguments -s -w : $_hg_global_opts \

  ':path:_hg_paths'

}

_hg_cmd_pull() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

  ':source:_hg_remote'

}

_hg_cmd_push() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

  ':destination:_hg_remote'

}

_hg_cmd_remove() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

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

}

_hg_cmd_rename() {

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

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

}

_hg_cmd_revert() {

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

  '--no-backup[do not save backup copies of files]' \

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

}

_hg_cmd_serve() {

  _arguments -s -w : $_hg_global_opts \

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

  '--stdio[for remote clients]' \

}

_hg_cmd_status() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

  '--rev[show difference from revision]:revision:_hg_tags' \

  '*:files:_files'

}

_hg_cmd_tag() {

  _arguments -s -w : $_hg_global_opts \