#compdef set

local prev="$words[CURRENT-1]" expl

if [[ "$prev" = [-+]o ]]; then
  _description expl 'zsh option'
  compgen "$expl[@]" -o
elif [[ "$prev" = -A ]]; then
  _description expl array
  compgen "$expl[@]" -A
fi