bugfixes + readme update

README.md

@@ -10,6 +10,8 @@
 
 This is the VSCode Extension for [SAML.to](https://saml.to). It allows for AWS role assumption within VSCode.
 
+![Assume an AWS Role in VSCode](resources/marketplace/assume-role.gif)
+
 ## Features
 
 - Assume AWS Roles in VSCode

TODO.md

@@ -3,3 +3,4 @@
 - Write to env?
 - Test with AWS Extension
 - Onboarding experience for brand new user
+- Set Provider/Org in assumeRole, save these to cache

package.json

@@ -134,8 +134,9 @@
   "dependencies": {
     "@aws-sdk/client-sts": "^3.458.0",
     "@octokit/rest": "^20.0.2",
+    "axios": "^1.6.2",
     "humanize-duration": "^3.31.0",
     "qrcode-svg": "^1.1.0",
     "which": "^4.0.0"
   }
-}
+}

src/aws.ts

@@ -5,6 +5,7 @@ import { AssumeRoleWithSAMLCommand, STSClient } from "@aws-sdk/client-sts";
 import { exec } from "./exec";
 import humanizeDuration from "humanize-duration";
 import { Configuration, ProfileName } from "./config";
+import { isAxiosError } from "axios";
 
 export const assumeAwsRole = (
   configuration: Configuration,
@@ -41,7 +42,7 @@ export const assumeAwsRole = (
             return {
               label: r.role.split("/").pop() || r.role,
               description: r.org,
-              detail: r.role,
+              detail: `${r.role}`,
               roleArn: r.role,
             };
           }),
@@ -157,12 +158,22 @@ export const assumeAwsRole = (
         }
       }
     } catch (e) {
+      // Clear State
+      configuration.assumeAws.lastRoleArn = null;
+
       if (!(e instanceof Error)) {
         throw e;
       }
-      vscode.window.showWarningMessage(
-        `[SAML.to] Unable to assume AWS role: ${e.message}`
-      );
+
+      if (isAxiosError(e)) {
+        vscode.window.showWarningMessage(
+          `[SAML.to] Unable to assume AWS role: ${e.response?.data?.message}`
+        );
+      } else {
+        vscode.window.showWarningMessage(
+          `[SAML.to] Unable to assume AWS role: ${e.message}`
+        );
+      }
     }
   };
 };

src/config.ts

@@ -113,6 +113,16 @@ class AssumeAwsConfiguration {
   }
 
   set lastRoleArn(roleArn: string | null) {
+    if (roleArn === null) {
+      this.context.workspaceState
+        .update("assumeAws.lastRoleArn", null)
+        .then(() => {});
+      this.context.globalState
+        .update("assumeAws.lastRoleArn", null)
+        .then(() => {});
+      return;
+    }
+
     let memento: vscode.Memento | undefined = undefined;
 
     if (this.rememberRole === "Global") {