From 102561a17943a10b5280fd6cbb786a749ead32b7 Mon Sep 17 00:00:00 2001
From: Cameron Reed <cameron@cam123.dev>
Date: Tue, 6 Aug 2024 08:22:26 -0600
Subject: [PATCH] Update zshrc

---
 common/users/cameron/home-manager/zsh.nix | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/common/users/cameron/home-manager/zsh.nix b/common/users/cameron/home-manager/zsh.nix
index 48c4422..ba6ff0a 100644
--- a/common/users/cameron/home-manager/zsh.nix
+++ b/common/users/cameron/home-manager/zsh.nix
@@ -90,8 +90,15 @@ function set_prompt() {
 	for i in $(seq 1 "''${#prefixes[@]}"); do
 		prefix+="%F{''${colors[$i]:-255}}''${prefixes[$i]}%f "
 	done
+
+    local user="${USER_OVERRIDE:-%n}"
+    local host="@${HOST_OVERRIDE:-%m}"
+
+    if [ -z "${SSH_CONNECTION}" ]; then
+        host=""
+    fi
 	
-	PROMPT="[''${prefix}%n@%m %1~]%(#.#.$) "
+	PROMPT="[''${prefix}${user}${host} %1~]%(#.#.$) "
 }
 
 typeset -a precmd_functions