From 12ff176c5597f52d7417eb4a572e3a62075a527e Mon Sep 17 00:00:00 2001
From: Pablo Diaz <padilo@gmail.com>
Date: Tue, 30 Oct 2018 22:54:46 +0100
Subject: [PATCH] aws session token support

---
 plugins/available/aws.plugin.bash | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plugins/available/aws.plugin.bash b/plugins/available/aws.plugin.bash
index a8941e07..81d88f97 100644
--- a/plugins/available/aws.plugin.bash
+++ b/plugins/available/aws.plugin.bash
@@ -37,7 +37,7 @@ function __awskeys_help {
 function __awskeys_get {
     local ln=$(grep -n "\[ *$1 *\]" ~/.aws/credentials | cut -d ":" -f 1)
     if [[ -n "${ln}" ]]; then
-        tail -n +${ln} ~/.aws/credentials | egrep -m 2 "aws_access_key_id|aws_secret_access_key"
+        tail -n +${ln} ~/.aws/credentials | egrep -m 3 "aws_access_key_id|aws_secret_access_key|aws_session_token"
     fi
 }
 
@@ -79,7 +79,7 @@ function __awskeys_export {
 }
 
 function __awskeys_unset {
-    unset AWS_PROFILE AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY
+    unset AWS_PROFILE AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN
 }
 
 function __awskeys_comp {