Show AWS_PROFILE in shell prompt

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "awscredx"
-version = "0.3.0"
+version = "0.4.0"
 authors = ["Alexei Samokvalov <[email protected]>"]
 edition = "2018"
 

README.md

@@ -5,7 +5,8 @@ Assume AWS roles under the motto "don't make me think".
 `awscredx` has a goal to make the role assumption on the command line simple and intuitive.
 
 ## How to use it
-Install it by calling `awscredx init`.
+Download the [binary](https://github.com/sam701/awscredx/releases/latest).
+Add `awscredx` into your `PATH` and call `awscredx init`.
 It will print what it has done.
 
 TODO: add a screenshot

src/init/script.fish

@@ -2,12 +2,35 @@
 # Do not edit.
 
 set -x AWSCREDX_SCRIPT_VERSION "@version@"
+functions -c fish_prompt _original_fish_prompt
+
+
+if test -e $HOME/.config/fish/functions/fish_prompt.fish
+  cat $HOME/.config/fish/functions/fish_prompt.fish | grep AWS_PROFILE > /dev/null
+  if test $status -eq 0
+    set _original_prompt_has_aws_profile 1
+  end
+end
 
 function assume
   set -l output ("@bin@" assume $argv)
   set -l s $status
   if test $s -eq 0
     eval "$output"
+    if test -z "$_original_prompt_has_aws_profile"
+      function fish_prompt
+        set -l old_status $status
+
+        echo -n "["
+        set_color brmagenta
+        echo -n $AWS_PROFILE
+        set_color normal
+        echo -n "] "
+
+        echo -n "exit $old_status" | .
+        _original_fish_prompt
+      end
+    end
   else
     return $s
   end

src/init/script.sh

@@ -2,11 +2,13 @@
 # Do not edit.
 
 export AWSCREDX_SCRIPT_VERSION="@version@"
+_ORIGINAL_PS1="${PS1:-}"
 function assume {
   out=$("@bin@" assume "$@")
   s=$?
   if [[ $s == 0 ]]; then
     $out
+    PS1="[\e[1;35m$AWS_PROFILE\e[m] ${_ORIGINAL_PS1:-}"
   else
     return $s
   fi