add auth_token to other events after auth_token_configured

Author: anisaoshafi

src/plugins/setup.ts

@@ -5,6 +5,7 @@ import {
 	checkIsAuthenticated,
 	requestAuthentication,
 	saveAuthToken,
+	readAuthToken,
 } from "../utils/authenticate.ts";
 import { configureAwsProfiles } from "../utils/configure-aws.ts";
 import { runInstallProcess } from "../utils/install.ts";
@@ -100,6 +101,7 @@ export default createPlugin(
 										namespace: "onboarding",
 										steps: [1, 2, 3],
 										status: "CANCELLED",
+										auth_token: await readAuthToken(),
 									},
 								});
 								return;
@@ -236,6 +238,7 @@ export default createPlugin(
 									namespace: "onboarding",
 									steps: [1, 2, 3],
 									status: "COMPLETED",
+									auth_token: await readAuthToken(),
 								},
 							});
 						},

src/utils/authenticate.ts

@@ -131,7 +131,7 @@ export async function checkIsAuthenticated() {
 			return false;
 		}
 		return true;
-	} catch (error) {
+	} catch {
 		return false;
 	}
 }
@@ -143,3 +143,21 @@ function isAuthTokenPresent(authObject: unknown) {
 		AUTH_TOKEN_KEY in authObject
 	);
 }
+
+// Reads the auth token from the auth.json file for logging in the user
+export async function readAuthToken(): Promise<string> {
+	try {
+		const authJson = await fs.readFile(LOCALSTACK_AUTH_FILENAME, "utf-8");
+		const authObject = JSON.parse(authJson) as unknown;
+		if (!isAuthTokenPresent(authObject)) {
+			return "";
+		}
+		const authToken = authObject[AUTH_TOKEN_KEY];
+		if (typeof authToken !== "string") {
+			return "";
+		}
+		return authToken;
+	} catch {
+		return "";
+	}
+}

src/utils/configure-aws.ts

@@ -5,6 +5,7 @@ import * as path from "node:path";
 
 import { window } from "vscode";
 
+import { readAuthToken } from "./authenticate.ts";
 import { parseIni, serializeIni, updateIniSection } from "./ini-parser.ts";
 import type { IniFile, IniSection } from "./ini-parser.ts";
 import type { Telemetry } from "./telemetry.ts";
@@ -296,6 +297,8 @@ export async function configureAwsProfiles(options: {
 	const credentialsNeedsOverride =
 		checkIfCredentialsNeedsOverride(credentialsSection);
 
+	const authToken = await readAuthToken();
+
 	// means sections exist, but we need to check what's inside
 	if (credentialsSection && configSection) {
 		if (!configNeedsOverride && !credentialsNeedsOverride) {
@@ -314,6 +317,7 @@ export async function configureAwsProfiles(options: {
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
+					auth_token: authToken,
 				},
 			});
 			return;
@@ -347,6 +351,7 @@ export async function configureAwsProfiles(options: {
 				started_at: startedAt,
 				ended_at: new Date().toISOString(),
 				status: "SKIPPED",
+				auth_token: authToken,
 			},
 		});
 		return;
@@ -380,6 +385,7 @@ export async function configureAwsProfiles(options: {
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
+					auth_token: authToken,
 				},
 			});
 		} else if (configNeedsOverride) {
@@ -402,6 +408,7 @@ export async function configureAwsProfiles(options: {
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
+					auth_token: authToken,
 				},
 			});
 		} else if (credentialsNeedsOverride) {
@@ -424,6 +431,7 @@ export async function configureAwsProfiles(options: {
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
+					auth_token: authToken,
 				},
 			});
 		}

src/utils/manage.ts

@@ -4,6 +4,7 @@ import type { ExtensionContext, LogOutputChannel, MessageItem } from "vscode";
 import { commands, env, Uri, window } from "vscode";
 import { tr } from "zod/v4/locales";
 
+import { readAuthToken } from "./authenticate.ts";
 import { spawnLocalStack } from "./cli.ts";
 import { exec } from "./exec.ts";
 import { checkIsLicenseValid } from "./license.ts";
@@ -94,6 +95,7 @@ export async function startLocalStack(
 		command: "localstack.viewLogs",
 	});
 
+	const authToken = await readAuthToken();
 	try {
 		await spawnLocalStack(
 			[
@@ -132,6 +134,7 @@ export async function startLocalStack(
 				namespace: "emulator",
 				status: "COMPLETED",
 				emulator_session_id: emulatorSessionId,
+				auth_token: authToken,
 			},
 		});
 	} catch (error) {
@@ -155,6 +158,7 @@ export async function startLocalStack(
 				namespace: "emulator",
 				status: "FAILED",
 				errors: [String(error)],
+				auth_token: authToken,
 			},
 		});
 	}
@@ -166,6 +170,7 @@ export async function stopLocalStack(
 ) {
 	void showInformationMessage("Stopping LocalStack.");
 
+	const authToken = await readAuthToken();
 	try {
 		// get session id before killing container
 		const emulatorSessionId = await fetchLocalStackSessionId();
@@ -180,6 +185,7 @@ export async function stopLocalStack(
 				namespace: "emulator",
 				status: "COMPLETED",
 				emulator_session_id: emulatorSessionId,
+				auth_token: authToken,
 			},
 		});
 	} catch (error) {
@@ -194,6 +200,7 @@ export async function stopLocalStack(
 				namespace: "emulator",
 				status: "FAILED",
 				errors: [String(error)],
+				auth_token: authToken,
 			},
 		});
 	}

src/utils/telemetry.ts

@@ -62,6 +62,7 @@ type Events =
 				ended_at: string;
 				status: "COMPLETED" | "FAILED" | "SKIPPED" | "CANCELLED";
 				errors?: string[];
+				auth_token: string;
 			};
 	  }
 	| {
@@ -70,6 +71,7 @@ type Events =
 				namespace: "onboarding";
 				steps: number[];
 				status: "COMPLETED" | "FAILED" | "CANCELLED";
+				auth_token: string;
 			};
 	  }
 	| {
@@ -79,6 +81,7 @@ type Events =
 				status: "COMPLETED" | "FAILED";
 				emulator_session_id?: string;
 				errors?: string[];
+				auth_token: string;
 			};
 	  }
 	| {
@@ -88,6 +91,7 @@ type Events =
 				status: "COMPLETED" | "FAILED";
 				emulator_session_id?: string;
 				errors?: string[];
+				auth_token: string;
 			};
 	  };