From eaf13dcec8e06666740e74b74c5a8e7831e8cb0c Mon Sep 17 00:00:00 2001
From: "Morley, Jonathan" <jmorley@cvent.com>
Date: Mon, 26 Jul 2021 11:24:27 -0400
Subject: [PATCH] retry on error posting to AWS

---
 src/config/profile.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/config/profile.rs b/src/config/profile.rs
index cfba7389..6ea9a335 100644
--- a/src/config/profile.rs
+++ b/src/config/profile.rs
@@ -25,7 +25,13 @@ impl ProfileConfig {
         default_role: Option<String>,
     ) -> Result<(String, Self)> {
         let response = client.get_saml_response(link.link_url.clone()).await?;
-        let aws_response = response.post_to_aws().await?;
+        let aws_response = match response.post_to_aws().await {
+            Err(e) => {
+                warn!("Caught error trying to login to AWS: {}, trying again", e);
+                response.post_to_aws().await
+            }
+            ok => ok,
+        }?;
         let aws_response_text = aws_response.text().await?;
 
         let roles = response.clone().roles;