#compdef declare export integer float local readonly typeset

local expl state line func i use curcontext="$curcontext"
local -A allargs
local -a args

allargs=(
  A '(-E -F -L -R -T -U -Z -a -f -i -m)-A[specify that arguments refer to associative arrays]'
  E '(-A -F -L -R -T -U -Z -a -f -i -m)-E[floating point, use engineering notation on output]'
  F '(-A -E -L -R -T -U -Z -a -f -i -m)-F[floating point, use fixed point decimal on output]'
  L '(-A -E -F -f -i)-L+[left justify and remove leading blanks from value]:width'
  R '(-A -E -F -f -i)-R+[right justify and fill with leading blanks]:width'
  T '(-A -E -F -a -f -g -h -i -l -m -t)-T[tie scalar to array]'
  U '(-A -E -F -i)-U[keep array values unique and suppress alias expansion for functions]'
  Uf '(-E -F -i)-U[suppress alias expansion for functions]'
  Up '(-E -F -i)-U[keep array values unique]'
  Z '(-A -E -F -f -i)-Z+[right justify and fill with leading zeros]:width'
  a '(-A -E -F -T -f -i)-a[specify that arguments refer to arrays]'
  f '(-A -E -F -L -R -T -Z -a -g -h -i -l -r -x)-f[specify that arguments refer to functions]'
  g '(-T -f)-+g[do not restrict parameter to local scope]'
  h '(-T -f)-+h[hide specialness of parameter]'
  H '(-T -f)-+H[hide value of parameter in listings]'
  i '(-A -E -F -T -f)-+i[represent internally as an integer]'
  l '(-T -f)-l[convert the value to lowercase]'
  m '(-A -E -F -T -i)-m[treat arguments as patterns]'
  p '-p[output parameters in form of calls to typeset]'
  r '(-f)-+r[mark parameters as readonly]'
  t '(-T)-+t[tag parameters and turn on execution tracing for functions]'
  tf '(-T)-+t[turn on execution tracing for functions]'
  tp '(-T)-+t[tag parameters]'
  u '-u[convert the value to uppercase or mark function for autoloading]'
  uf '-u[mark function for autoloadling]'
  up '-u[convert the value to uppercase]'
  x '(-f)-+x[export parameter]'
)

use="AEFHLRTUZafghilmprtux"

case ${service} in
  float) use="EFHghlprtux";;
  functions)
    use="Umtu"
    func=f
  ;;
  integer)
    use="Hghilprtux"
    allargs[i]='-i[specify arithmetic base for output]' \
  ;;
  readonly) use="${use/r/}" ;;
  local) use="${use/f/}" ;&
  export) use="${${use/g/}/x/}" ;;
esac

[[ -z "${words[(r)-*f*]}" ]] || func=f
[[ -z "${words[(r)-*[aA]*]}" ]] || func=p
   
for ((i=1;i<=$#use;++i)); do
  args+=( ${allargs[${use[$i]}${${(s::)use[$i]}[(r)[Uut]]:+$func}]} )
done

_arguments -C -s -A "-*" -S "${args[@]}" '*:vars:->vars_eq'

if [[ "$state" = vars_eq ]]; then
  if [[ $func = f ]]; then
    _functions
  elif [[ "$PREFIX" = *\=* ]]; then
    compstate[parameter]="${PREFIX%%\=*}"
    compset -P 1 '*='
    _value
  elif (( $+opt_args[-a] || $+opt_args[-A] )); then
    _parameters -q
  else
    _parameters -q -S '='
  fi
fi
