From 1447b38896dcc97fe665dcb324966d2c151f07f7 Mon Sep 17 00:00:00 2001
From: Chen Asraf <chenasrafil@gmail.com>
Date: Wed, 25 Jan 2023 01:33:51 +0200
Subject: [PATCH] feat: add hostname to prompt

---
 lambda-mod.zsh-theme | 31 ++++++++++++++++---------------
 1 file changed, 16 insertions(+), 15 deletions(-)

diff --git a/lambda-mod.zsh-theme b/lambda-mod.zsh-theme
index 6be00e4..10903ad 100644
--- a/lambda-mod.zsh-theme
+++ b/lambda-mod.zsh-theme
@@ -7,29 +7,31 @@ if [[ "$USER" == "root" ]]; then USERCOLOR="red"; else USERCOLOR="yellow"; fi
 # return anything in this case. So wrap it in another function and check
 # for an empty string.
 function check_git_prompt_info() {
-    if type git &>/dev/null && git rev-parse --git-dir > /dev/null 2>&1; then
-        if [[ -z $(git_prompt_info 2> /dev/null) ]]; then
-            echo "%{$fg[blue]%}detached-head%{$reset_color%}) $(git_prompt_status)
+  if type git &>/dev/null && git rev-parse --git-dir >/dev/null 2>&1; then
+    if [[ -z $(git_prompt_info 2>/dev/null) ]]; then
+      echo "%{$fg[blue]%}detached-head%{$reset_color%}) $(git_prompt_status)
 %{$fg[yellow]%}→ "
-        else
-            echo "$(git_prompt_info 2> /dev/null) $(git_prompt_status)
-%{$fg_bold[cyan]%}→ "
-        fi
     else
-        echo "%{$fg_bold[cyan]%}→ "
+      echo "$(git_prompt_info 2>/dev/null) $(git_prompt_status)
+%{$fg_bold[cyan]%}→ "
     fi
+  else
+    echo "%{$fg_bold[cyan]%}→ "
+  fi
 }
 
 function get_right_prompt() {
-    if type git &>/dev/null && git rev-parse --git-dir > /dev/null 2>&1; then
-        echo -n "$(git_prompt_short_sha)%{$reset_color%}"
-    else
-        echo -n "%{$reset_color%}"
-    fi
+  if type git &>/dev/null && git rev-parse --git-dir >/dev/null 2>&1; then
+    echo -n "$(git_prompt_short_sha)%{$reset_color%}"
+  else
+    echo -n "%{$reset_color%}"
+  fi
 }
 
+HOST_SUF=$([[ "$LAMBDA_MOD_HOST_SUF" == "true" ]] && echo "@%m" || echo "")
+
 PROMPT=$'\n'$LAMBDA'\
- %{$fg_bold[$USERCOLOR]%}%n\
+ %{$fg_bold[$USERCOLOR]%}%n@'${HOST_SUF}'\
  %{$fg_no_bold[magenta]%}[%'${LAMBDA_MOD_N_DIR_LEVELS:-3}'~]\
  $(check_git_prompt_info)\
 %{$reset_color%}'
@@ -53,7 +55,6 @@ ZSH_THEME_GIT_PROMPT_UNTRACKED="%{$fg_bold[cyan]%}?"
 # Format for git_prompt_ahead()
 ZSH_THEME_GIT_PROMPT_AHEAD=" %{$fg_bold[white]%}^"
 
-
 # Format for git_prompt_long_sha() and git_prompt_short_sha()
 ZSH_THEME_GIT_PROMPT_SHA_BEFORE=" %{$fg_bold[white]%}[%{$fg_bold[blue]%}"
 ZSH_THEME_GIT_PROMPT_SHA_AFTER="%{$fg_bold[white]%}]"