From 7a047647f6f43f37161a8a4dc68400bf64ddfd78 Mon Sep 17 00:00:00 2001
From: Yogeshwaran Rajendran <Yogeshwaran.Rajendran@silabs.com>
Date: Mon, 17 Jul 2023 13:15:19 +0530
Subject: [PATCH] Testing

---
 themes/agnoster.zsh-theme | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/themes/agnoster.zsh-theme b/themes/agnoster.zsh-theme
index 99d678930..10f75859d 100644
--- a/themes/agnoster.zsh-theme
+++ b/themes/agnoster.zsh-theme
@@ -256,9 +256,17 @@ prompt_aws() {
 
 prompt_shell() {
   [[ -z "$SHELL_PROFILE" || "$SHOW_AWS_PROMPT" = false ]] && return
+  local ENV_PRPT="$SHELL_PROFILE"
+  case "$TERA_PROFILE" in
+    *qa*) ENV_PRPT="${SHELL_PROFILE}-qa" ;;
+    *stage*) ENV_PRPT="${SHELL_PROFILE}-st" ;;
+  esac
   case "$SHELL_PROFILE" in
-    *prod|*production*) prompt_segment red black  "${SHELL_PROFILE:gs/%/%%}" ;;
-    *) prompt_segment green black "${SHELL_PROFILE:gs/%/%%}" ;;
+    *prod|*production*) 
+          prompt_segment red black  "${ENV_PRPT:gs/%/%%}" ;;
+    *dev*) 
+          prompt_segment green black "${ENV_PRPT:gs/%/%%}" ;;
+    # *) prompt_segment green black "" ;;
   esac
 }
 ## Main prompt