From 553f5eb417e843677339c6e2d20ff2778958ea17 Mon Sep 17 00:00:00 2001
From: Justin La Sotten <justin.lasotten@gmail.com>
Date: Thu, 5 Aug 2021 10:07:18 -0400
Subject: [PATCH] Adding support for AWS Profiles

---
 bedbugs.zsh-theme | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/bedbugs.zsh-theme b/bedbugs.zsh-theme
index 17f54cc..b457ace 100644
--- a/bedbugs.zsh-theme
+++ b/bedbugs.zsh-theme
@@ -81,6 +81,12 @@ prompt_virtualenv() {
   fi
 }
 
+prompt_aws_profile() {
+  if [[ -v AWS_PROFILE && ! -v AWS_PROFILE_DISABLE_PROMPT ]]; then
+    prompt_segment magenta white "AWS: $AWS_PROFILE"
+  fi
+}
+
 # Jobs: current count of running jobs
 prompt_jobs() {
   local jobs=$(jobs -l | wc -l)
@@ -102,6 +108,7 @@ build_prompt() {
   # Start 2nd line
   print
 
+  prompt_aws_profile
   prompt_virtualenv
   prompt_git
   prompt_sigil