From 0c627eb31de1b816e98fa17a11ecf675c4e2b366 Mon Sep 17 00:00:00 2001
From: romkatv <roman.perepelitsa@gmail.com>
Date: Tue, 27 Aug 2019 10:57:54 +0200
Subject: [PATCH] add `proxy` prompt segment

---
 config/p10k-classic.zsh |  7 +++++++
 config/p10k-lean.zsh    |  7 +++++++
 internal/icons.zsh      |  5 +++++
 internal/p10k.zsh       | 10 ++++++++++
 4 files changed, 29 insertions(+)

diff --git a/config/p10k-classic.zsh b/config/p10k-classic.zsh
index aab768ef..6864ba47 100644
--- a/config/p10k-classic.zsh
+++ b/config/p10k-classic.zsh
@@ -65,6 +65,7 @@ fi
       # =========================[ Line #2 ]=========================
       newline
       # public_ip             # public IP address
+      # proxy                 # system-wide http/https/ftp proxy
       # battery               # internal battery
       # example               # example user-defined segment (see prompt_example function below)
   )
@@ -626,6 +627,12 @@ fi
   # Custom icon.
   # typeset -g POWERLEVEL9K_PUBLIC_IP_VISUAL_IDENTIFIER_EXPANSION='⭐'
 
+  #########################[ proxy: system-wide http/https/ftp proxy ]##########################
+  # Proxy color.
+  typeset -g POWERLEVEL9K_PROXY_FOREGROUND=68
+  # Custom icon.
+  # typeset -g POWERLEVEL9K_PROXY_VISUAL_IDENTIFIER_EXPANSION='⭐'
+
   ################################[ battery: internal battery ]#################################
   # Show battery in red when it's below this level and not connected to power supply.
   typeset -g POWERLEVEL9K_BATTERY_LOW_THRESHOLD=20
diff --git a/config/p10k-lean.zsh b/config/p10k-lean.zsh
index fdc41f30..62ed0bad 100644
--- a/config/p10k-lean.zsh
+++ b/config/p10k-lean.zsh
@@ -65,6 +65,7 @@ fi
       # =========================[ Line #2 ]=========================
       newline
       # public_ip             # public IP address
+      # proxy                 # system-wide http/https/ftp proxy
       # battery               # internal battery
       # example               # example user-defined segment (see prompt_example function below)
   )
@@ -609,6 +610,12 @@ fi
   # Custom icon.
   # typeset -g POWERLEVEL9K_PUBLIC_IP_VISUAL_IDENTIFIER_EXPANSION='⭐'
 
+  #########################[ proxy: system-wide http/https/ftp proxy ]##########################
+  # Proxy color.
+  typeset -g POWERLEVEL9K_PROXY_FOREGROUND=68
+  # Custom icon.
+  # typeset -g POWERLEVEL9K_PROXY_VISUAL_IDENTIFIER_EXPANSION='⭐'
+
   ################################[ battery: internal battery ]#################################
   # Show battery in red when it's below this level and not connected to power supply.
   typeset -g POWERLEVEL9K_BATTERY_LOW_THRESHOLD=20
diff --git a/internal/icons.zsh b/internal/icons.zsh
index 425e57c5..c74fe909 100755
--- a/internal/icons.zsh
+++ b/internal/icons.zsh
@@ -112,6 +112,7 @@ function _p9k_init_icons() {
         LARAVEL_ICON                   ''
         RANGER_ICON                    '\u2B50'               # ⭐
         TERRAFORM_ICON                 '\U1F6E0\u00A0'        # 🛠️
+        PROXY_ICON                     '\u21AF'               # ↯
       )
     ;;
     'awesome-fontconfig')
@@ -211,6 +212,7 @@ function _p9k_init_icons() {
         LARAVEL_ICON                   ''
         RANGER_ICON                    '\u2B50'               # ⭐
         TERRAFORM_ICON                 '\U1F6E0\u00A0'        # 🛠️
+        PROXY_ICON                     '\u21AF'               # ↯
       )
     ;;
     'awesome-mapped-fontconfig')
@@ -314,6 +316,7 @@ function _p9k_init_icons() {
         LARAVEL_ICON                   ''
         RANGER_ICON                    '\u2B50'                                       # ⭐
         TERRAFORM_ICON                 '\U1F6E0\u00A0'                                # 🛠️
+        PROXY_ICON                     '\u21AF'               # ↯
       )
     ;;
     'nerdfont-complete'|'nerdfont-fontconfig')
@@ -414,6 +417,7 @@ function _p9k_init_icons() {
         LARAVEL_ICON                   '\ue73f'               # 
         RANGER_ICON                    '\u2B50'               # ⭐
         TERRAFORM_ICON                 '\U1F6E0\u00A0'        # 🛠️
+        PROXY_ICON                     '\u21AF'               # ↯
       )
     ;;
     *)
@@ -513,6 +517,7 @@ function _p9k_init_icons() {
         LARAVEL_ICON                   ''
         RANGER_ICON                    '\u2B50'               # ⭐
         TERRAFORM_ICON                 '\U1F6E0\u00A0'        # 🛠️
+        PROXY_ICON                     '\u21AF'               # ↯
       )
     ;;
   esac
diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index cff575b0..5caaea15 100755
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -3052,6 +3052,16 @@ function prompt_terraform() {
   [[ $ws == default ]] || _p9k_prompt_segment $0 $_p9k_color1 blue TERRAFORM_ICON 0 '' $ws
 }
 
+function prompt_proxy() {
+  local -U p=(
+    $all_proxy $http_proxy $https_proxy $ftp_proxy
+    $ALL_PROXY $HTTP_PROXY $HTTPS_PROXY $FTP_PROXY)
+  p=(${(@)${(@)${(@)p#*://}##*@}%%/*})
+  (( $#p )) || return
+  (( $#p == 1 )) || p=("")
+  _p9k_prompt_segment $0 $_p9k_color1 blue PROXY_ICON 0 '' "$p[1]"
+}
+
 _p9k_preexec() {
   if (( $+_p9k_real_zle_rprompt_indent )); then
     if [[ -n $_p9k_real_zle_rprompt_indent ]]; then