From 9fdbd4a3e75d9dbec35be8c25175d6f0fdefa6b3 Mon Sep 17 00:00:00 2001
From: Melvyn de Kort <m.de.kort@portbase.com>
Date: Wed, 16 Oct 2019 12:03:22 +0200
Subject: [PATCH] Add support for awsume

---
 internal/p10k.zsh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index 7f731888..d1c5e9e4 100644
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -934,7 +934,7 @@ prompt_anaconda() {
 ################################################################
 # AWS Profile
 prompt_aws() {
-  local aws_profile="${AWS_VAULT:-${AWS_PROFILE:-$AWS_DEFAULT_PROFILE}}"
+  local aws_profile="${AWS_VAULT:-${AWSUME_PROFILE:-${AWS_PROFILE:-$AWS_DEFAULT_PROFILE}}}"
   if [[ "$aws_profile" != (default|) ]]; then
     _p9k_prompt_segment "$0" red white 'AWS_ICON' 0 '' "${aws_profile//\%/%%}"
   fi