Iterm2 Badge config: \(session.username)@\(session.hostname)\n\(user.kube_context)
Addition to .bashrc (or whatever runs on new shell):
function iterm2_print_user_vars() {
iterm2_set_user_var kube_context $(kubectl config current-context 2>/dev/null)
}