|
#compdef hgtk
# Zsh completion script for hgtk.
# copy it into your zsh function path (/usr/share/zsh/site-functions)
#
# Copyright (C) 2009-2010 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.
emulate -LR zsh
setopt extendedglob
local curcontext="$curcontext" state line
typeset -A _hg_cmd_globals
_hgtk() {
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 : $_hgtk_global_opts \
':hgtk command:_hgtk_commands'
return
fi
# resolve abbreviations and aliases
if ! (( $+functions[_hgtk_cmd_${cmd}] ))
then
local cmdexp
(( $#_hgtk_cmd_list )) || _hgtk_get_commands
cmdexp=$_hgtk_cmd_list[(r)${cmd}*]
if [[ $cmdexp == $_hgtk_cmd_list[(R)${cmd}*] ]]
then
# might be nice to rewrite the command line with the expansion
cmd="$cmdexp"
fi
if [[ -n $_hgtk_alias_list[$cmd] ]]
then
cmd=$_hgtk_alias_list[$cmd]
fi
fi
curcontext="${curcontext%:*:*}:hgtk-${cmd}:"
zstyle -s ":completion:$curcontext:" cache-policy update_policy
if [[ -z "$update_policy" ]]
then
zstyle ":completion:$curcontext:" cache-policy _hgtk_cache_policy
fi
if (( $+functions[_hgtk_cmd_${cmd}] ))
then
_hgtk_cmd_${cmd}
else
# complete unknown commands normally
_arguments -s -w : $_hgtk_global_opts \
'*:files:_hg_files'
fi
}
_hgtk_cache_policy() {
typeset -a old
# cache for a minute
old=( "$1"(mm+10) )
(( $#old )) && return 0
return 1
}
_hgtk_get_commands() {
typeset -ga _hgtk_cmd_list
typeset -gA _hgtk_alias_list
local hline cmd cmdalias
_call_program hgtk hgtk debugcomplete -v 2>/dev/null | while read -A hline
do
cmd=$hline[1]
_hgtk_cmd_list+=($cmd)
for cmdalias in $hline[2,-1]
do
_hgtk_cmd_list+=($cmdalias)
_hgtk_alias_list+=($cmdalias $cmd)
done
done
}
_hgtk_commands() {
(( $#_hgtk_cmd_list )) || _hgtk_get_commands
_describe -t commands 'hgtk command' _hgtk_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
_hgtk_global_opts=(
'(--repository -R)'{-R+,--repository}'[repository root directory]:repository:_files -/'
'(--verbose -v)'{-v,--verbose}'[enable additional output]'
'(--quiet -q)'{-q,--quiet}'[suppress output]'
'(--help -h)'{-h,--help}'[display help and exit]'
'--debugger[start debugger]'
)
_hg_cmd() {
_call_program hg hg "$_hg_cmd_globals[@]" "$@"
}
_hgtk_cmd_add() {
_arguments -s -w : $_hgtk_global_opts \
'*:unknown files:_hg_unknown'
}
_hgtk_cmd_annotate() {
_arguments -s -w : $_hgtk_global_opts \
'*:files:_hg_files'
}
_hgtk_cmd_clone() {
_arguments -s -w : $_hgtk_global_opts \
':source repository:_hg_remote' \
':destination:_hg_clone_dest'
}
_hgtk_cmd_commit() {
_arguments -s -w : $_hgtk_global_opts \
'(--date -d)'{-d+,--date}'[record datecode as commit date]:date code:' \
'(--user -u)'{-u+,--user}'[record user as commiter]:user:' \
'*:file:_hg_files'
}
_hgtk_cmd_vdiff() {
typeset -A opt_args
_arguments -s -w : $_hgtk_global_opts \
'*'{-r,--rev}'+[revision]:revision:_hg_revrange' \
'*:file:->diff_files'
if [[ $state == 'diff_files' ]]
then
if [[ -n $opt_args[-r] ]]
then
_hg_files
else
_hg_modified
fi
fi
}
_hgtk_cmd_grep() {
_arguments -s -w : $_hgtk_global_opts \
'*:files:_hg_files'
}
_hgtk_cmd_help() {
_arguments -s -w : $_hgtk_global_opts \
'*:hgtk command:_hgtk_commands'
}
_hgtk_cmd_init() {
_arguments -s -w : $_hgtk_global_opts \
':dir:_files -/'
}
_hgtk_cmd_log() {
_arguments -s -w : $_hgtk_global_opts \
'(--limit -l)'{-l+,--limit}'[limit number of changes displayed]:' \
'*'{-r,--rev}'[show the specified revision or range]:revision:_hg_revrange' \
'*:files:_hg_files'
}
_hgtk_cmd_pull() {
_arguments -s -w : $_hgtk_global_opts
}
_hgtk_cmd_push() {
_arguments -s -w : $_hgtk_global_opts
}
_hgtk_cmd_rename() {
_arguments -s -w : $_hgtk_global_opts \
'*:file:_hg_files'
}
_hgtk_cmd_serve() {
_arguments -s -w : $_hgtk_global_opts
}
_hgtk_cmd_status() {
_arguments -s -w : $_hgtk_global_opts \
'--rev[show difference from revision]:revision:_hg_tags' \
'*:files:_files'
}
_hgtk_cmd_update() {
_arguments -s -w : $_hgtk_global_opts \
'(--rev -r)'{-r+,--rev}'[revision]:revision:_hg_tags' \
':revision:_hg_tags'
}
_hgtk "$@"
|
Loading...