typeset -A kubectx_mapping

function kubectx_prompt_info() {
  if [ $commands[kubectl] ]; then
    local current_ctx=`kubectl config current-context`
    # use value in associative array if it exists, otherwise fall back to the context name
    echo "${kubectx_mapping[$current_ctx]:-$current_ctx}"
  fi
}