#compdef -subscript- local expl ind osuf=']' flags sep if [[ "$1" = -q ]]; then osuf='] ' compquote osuf shift fi compset -P '\(([^\(\)]|\(*\))##\)' # remove subscript flags if [[ "$PREFIX" = :* ]]; then _wanted characters expl 'character class' \ compadd -p: -S ':]' alnum alpha ascii blank cntrl digit graph \ lower print punct space upper xdigit elif compset -P '\('; then local match compset -S '\)*' if [[ $PREFIX = (#b)*([bns])(?|)(*) ]]; then local f=$match[1] d=$match[2] e=$match[2] v=$match[3] [[ $f = s && ${(Pt)${compstate[parameter]}} != scalar* ]] && return 1 if [[ -z $d ]]; then _message -e delimiters 'delimiter' return else case $d in (\() e=\);; (\[) e=\];; (\{) e=\};; esac if [[ $v != *$e* ]]; then case $f in (s) _message 'separator string';; (b|n) [[ $v = <-># ]] && _message 'number' || return 1;; esac [[ -n $v && $SUFFIX$ISUFFIX != *$e* ]] && _message 'delimiter' return 0 fi fi fi case ${(Pt)${compstate[parameter]}} in assoc*) flags=( '(R k K i I)r[any one value matched by subscript as pattern]' '(r k K i I)R[all values matched by subscript as pattern]' '(r R K i I)k[any one value where subscript matched by key as pattern]' '(r R k i I)K[all values where subscript matched by key as pattern]' '(r R k K I)i[any one key matched by subscript as pattern]' '(r R k K i)I[all keys matched by subscript as pattern]' );; (|scalar*)) flags=( 'w[make subscripting work on words of scalar]' 's[specify word separator]' 'p[recognise escape sequences in subsequent s flag]' );& array*) flags=($flags 'n[Nth lowest/highest index with i/I/r/R flag]' 'b[begin with specified element]' '(r R k K i)I[highest index of value matched by subscript]' '(r R k K I)i[lowest index of value matched by subscript]' '(r k K i I)R[value matched by subscript at highest index]' '(R k K i I)r[value matched by subscript at lowest index]' );; esac _values -s '' 'subscript flags' $flags elif [[ ${(Pt)${compstate[parameter]}} = assoc* ]]; then local suf [[ "$RBUFFER" != (|\\)\]* ]] && suf="$osuf" _wanted association-keys expl 'association key' \ compadd -S "$suf" -k "$compstate[parameter]" elif [[ ${(Pt)${compstate[parameter]}} = array* ]]; then local list i j ret=1 disp _tags indexes parameters while _tags; do if _requested indexes; then ind=( {1..${#${(P)${compstate[parameter]}}}} ) if zstyle -T ":completion:${curcontext}:indexes" verbose; then list=() for i in "$ind[@]"; do if [[ "$i" = ${PREFIX}*${SUFFIX} ]]; then list=( "$list[@]" "${i}:$(print -D ${(P)${compstate[parameter]}[$i]})" ) else list=( "$list[@]" '' ) fi done zstyle -s ":completion:${curcontext}:indexes" list-separator sep || sep=-- zformat -a list " $sep " "$list[@]" disp=( -d list) else disp=() fi if [[ "$RBUFFER" = (|\\)\]* ]]; then _all_labels -V indexes expl 'array index' \ compadd -S '' "$disp[@]" -a ind && ret=0 else _all_labels -V indexes expl 'array index' \ compadd -S "$osuf" "$disp[@]" -a ind && ret=0 fi fi _requested parameters && _parameters && ret=0 (( ret )) || return 0 done return 1 else _dispatch -math- -math- fi