#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-6 Steve Borho

# Copyright (C) 2006-8 Brendan Cully <brendan@kublai.com>

#

# 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.

emulate -LR zsh

setopt extendedglob

local curcontext="$curcontext" state line

typeset -A _hg_cmd_globals

_hg() {

  local cmd _hg_root

  integer i=2

  _hg_cmd_globals=()

  while (( i < $#words ))

  do

    case "$words[$i]" in

      -R|--repository)

        eval _hg_root="$words[$i+1]"

        _hg_cmd_globals+=("$words[$i]" "$_hg_root")

        (( i += 2 ))

        continue

      ;;

      -R*)

        _hg_cmd_globals+="$words[$i]"

        eval _hg_root="${words[$i]#-R}"

       (( i++ ))

       continue

      ;;

      --cwd|--config)

        # pass along arguments to hg completer

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

        (( i += 2 ))

        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

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

  zstyle -s ":completion:$curcontext:" cache-policy update_policy

  if [[ -z "$update_policy" ]]

  then

    zstyle ":completion:$curcontext:" cache-policy _hg_cache_policy

  fi

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

  then

    _hg_cmd_${cmd}

  else

    # complete unknown commands normally

    _arguments -s -w : $_hg_global_opts \

      '*:files:_hg_files'

  fi

}

_hg_cache_policy() {

  typeset -a old

  # cache for a minute

  old=( "$1"(mm+10) )

  (( $#old )) && return 0

  return 1

}

_hg_get_commands() {

  typeset -ga _hg_cmd_list

  typeset -gA _hg_alias_list

  local hline cmd cmdalias

  _call_program hg hg debugcomplete -v 2>/dev/null | while read -A hline

  do

    cmd=$hline[1]

    _hg_cmd_list+=($cmd)

    for cmdalias in $hline[2,-1]

    do

      _hg_cmd_list+=($cmdalias)

      _hg_alias_list+=($cmdalias $cmd)

    done

  done

}

_hg_commands() {

  (( $#_hg_cmd_list )) || _hg_get_commands

  _describe -t commands 'mercurial command' _hg_cmd_list

}

_hg_revrange() {

  compset -P 1 '*:'

  _hg_tags "$@"

}

_hg_tags() {

  typeset -a tags

  local tag rev

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

  do

    tags+=(${tag/ #    [0-9]#:*})

  done

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

}

_hg_files() {

  if [[ -n "$_hg_root" ]]

  then

    [[ -d "$_hg_root/.hg" ]] || return

    case "$_hg_root" in

      /*)

        _files -W $_hg_root

      ;;

      *)

        _files -W $PWD/$_hg_root

      ;;

    esac

  else

    _files

  fi

}

_hg_status() {

  [[ -d $PREFIX ]] || PREFIX=$PREFIX:h

  status_files=(${(ps:\0:)"$(_hg_cmd status -0n$1 ./$PREFIX 2>/dev/null)"})

}

_hg_unknown() {

  typeset -a status_files

  _hg_status u

  _wanted files expl 'unknown files' _multi_parts / status_files

}

_hg_missing() {

  typeset -a status_files

  _hg_status d

  _wanted files expl 'missing files' _multi_parts / status_files

}

_hg_modified() {

  typeset -a status_files

  _hg_status m

  _wanted files expl 'modified files' _multi_parts / status_files

}

_hg_resolve() {

  local rstate rpah

  [[ -d $PREFIX ]] || PREFIX=$PREFIX:h

  _hg_cmd resolve -l ./$PREFIX 2> /dev/null | while read rstate rpath

  do

    [[ $rstate == 'R' ]] && resolved_files+=($rpath)

    [[ $rstate == 'U' ]] && unresolved_files+=($rpath)

  done

}

_hg_resolved() {

  typeset -a resolved_files unresolved_files

  _hg_resolve

  _wanted files expl 'resolved files' _multi_parts / resolved_files

}

_hg_unresolved() {

  typeset -a resolved_files unresolved_files

  _hg_resolve

  _wanted files expl 'unresolved files' _multi_parts / unresolved_files

}

_hg_config() {

    typeset -a items

    items=(${${(%f)"$(_call_program hg hg showconfig)"}%%\=*})

    (( $#items )) && _describe -t config 'config item' items

}

_hg_addremove() {

  _alternative 'files:unknown files:_hg_unknown' \

    'files:missing files:_hg_missing'

}

_hg_ssh_urls() {

  if [[ -prefix */ ]]

  then

    if zstyle -T ":completion:${curcontext}:files" remote-access

    then

      local host=${PREFIX%%/*}

      typeset -a remdirs

      compset -p $(( $#host + 1 ))

      local rempath=${(M)PREFIX##*/}

      local cacheid="hg:${host}-${rempath//\//_}"

      cacheid=${cacheid%[-_]}

      compset -P '*/'

      if _cache_invalid "$cacheid" || ! _retrieve_cache "$cacheid"

      then

        remdirs=(${${(M)${(f)"$(_call_program files ssh -a -x $host ls -1FL "${(q)rempath}" 2> /dev/null)"}##*/}%/})

        _store_cache "$cacheid" remdirs

      fi

      _describe -t directories 'remote directory' remdirs -S/

    else

      _message 'remote directory'

    fi

  else

    if compset -P '*@'

    then

      _hosts -S/

    else

      _alternative 'hosts:remote host name:_hosts -S/' \

        'users:user:_users -S@'

    fi

  fi

}

_hg_urls() {

  if compset -P bundle://

  then

    _files

  elif compset -P ssh://

  then

    _hg_ssh_urls

  elif [[ -prefix *: ]]

  then

    _urls

  else

    local expl

    compset -S '[^:]*'

    _wanted url-schemas expl 'URL schema' compadd -S '' - \

      http:// https:// ssh:// bundle://

  fi

}

_hg_paths() {

  typeset -a paths pnames

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

  do

    paths+=($pnames[1])

  done

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

}

_hg_remote() {

  _alternative 'path-aliases:repository alias:_hg_paths' \

    'directories:directory:_files -/' \

    'urls:URL:_hg_urls'

}

_hg_clone_dest() {

  _alternative 'directories:directory:_files -/' \

    'urls:URL:_hg_urls'

}

# Common options

_hg_global_opts=(

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

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

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

    '(--verbose -v)'{-v,--verbose}'[enable additional output]'

    '*--config[set/override config option]:defined config items:_hg_config'

    '(--quiet -q)'{-q,--quiet}'[suppress output]'

    '(--help -h)'{-h,--help}'[display help and exit]'

    '--debug[debug mode]'

    '--debugger[start debugger]'

    '--encoding[set the charset encoding (default: UTF8)]'

    '--encodingmode[set the charset encoding mode (default: strict)]'

    '--lsprof[print improved command execution profile]'

    '--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}'[include names matching the given patterns]:dir:_files -W $(_hg_cmd root) -/'

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

_hg_diff_opts=(

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

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

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

_hg_dryrun_opts=(

  '(--dry-run -n)'{-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)'{-e,--edit}'[edit commit message]'

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

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

_hg_remote_opts=(

  '(--ssh -e)'{-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)'{-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)'{-r+,--rev}'[annotate the specified revision]:revision:_hg_tags' \

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

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

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

  '(--date -d)'{-d,--date}'[list the date]' \

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

  '(--changeset -c)'{-c,--changeset}'[list the changeset]' \

  '*:files:_hg_files'

}

_hg_cmd_archive() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

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

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

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

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

  '*:destination:_files'

}

_hg_cmd_backout() {

  _arguments -s -w : $_hg_global_opts $_hg_pat_opts \

    '--merge[merge with old dirstate parent after backout]' \

    '(--date -d)'{-d+,--date}'[record datecode as commit date]:date code:' \

    '--parent[parent to choose when backing out merge]' \

    '(--user -u)'{-u+,--user}'[record user as commiter]:user:' \

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

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

    '(--logfile -l)'{-l+,--logfile}'[read commit message from <file>]:log file:_files -g \*.txt'

}

_hg_cmd_bundle() {

  _arguments -s -w : $_hg_global_opts $_hg_remote_opts \

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