#compdef kontext

__kubectl_context_list() {
  local -a ctx_list_disp ctx_list
  local expl
  IFS=$'\n' ctx_list=($(kubectl config get-contexts -o name))
  _wanted ctx expl 'kubernetes context' compadd "$@" "$ctx_list_disp[@]" - "${(@)ctx_list}"
}

__kontext() {
  _arguments \
    ':ctx arg:__kubectl_context_list' \
    - set1 '-h[help]' \
    - set2 '-n[namespace]'
}

compdef __kontext kontext