From ba39eb3421f7aa955a24516040006e3a22254528 Mon Sep 17 00:00:00 2001
From: romkatv <roman.perepelitsa@gmail.com>
Date: Mon, 29 Jul 2019 14:36:21 +0200
Subject: [PATCH] support POWERLEVEL9K_CONTEXT_${state}_TEMPLATE

---
 internal/p10k.zsh | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index d573e50b..ad051d90 100755
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -1153,20 +1153,19 @@ prompt_public_ip() {
 prompt_context() {
   if ! _p9k_cache_get $0 "${(%):-%#}"; then
     local -i enabled=1
-    local content='' state=''
-    if [[ $_POWERLEVEL9K_ALWAYS_SHOW_CONTEXT == 1 || -z $DEFAULT_USER || $_P9K_SSH == 1 ]]; then
-      content=$_POWERLEVEL9K_CONTEXT_TEMPLATE
-    else
+    local content
+    if [[ $_POWERLEVEL9K_ALWAYS_SHOW_CONTEXT == 0 && -n $DEFAULT_USER && $_P9K_SSH == 0 ]]; then
       local user="$(whoami)"
-      if [[ $user != $DEFAULT_USER ]]; then
-        content=$_POWERLEVEL9K_CONTEXT_TEMPLATE
-      elif (( _POWERLEVEL9K_ALWAYS_SHOW_USER )); then
-        content="${user//\%/%%}"
-      else
-        enabled=0
+      if [[ $user == $DEFAULT_USER ]]; then
+        if (( _POWERLEVEL9K_ALWAYS_SHOW_USER )); then
+          content="${user//\%/%%}"
+        else
+          enabled=0
+        fi
       fi
     fi
 
+    local state
     if (( enabled )); then
       state="DEFAULT"
       if [[ "${(%):-%#}" == '#' ]]; then
@@ -1180,6 +1179,16 @@ prompt_context() {
       elif [[ -n "$SUDO_COMMAND" ]]; then
         state="SUDO"
       fi
+
+      if [[ -z $content ]]; then
+        local var=POWERLEVEL9K_CONTEXT_${state}_TEMPLATE
+        if (( $+parameters[$var] )); then
+          content=${(P)var}
+          content=${(g::)content}
+        else
+          content=$_POWERLEVEL9K_CONTEXT_TEMPLATE
+        fi
+      fi
     fi
 
     _p9k_cache_set "$enabled" "$state" "$content"