Add license_setup_ended event

Author: anisaoshafi

src/plugins/setup.ts

@@ -1,3 +1,5 @@
+import { promiseHooks } from "node:v8";
+
 import { commands, ProgressLocation, window } from "vscode";
 
 import { createPlugin } from "../plugins.ts";
@@ -157,6 +159,7 @@ export default createPlugin(
 							// then there will be no license info to be reported by `localstack license info`.
 							// Also, an expired license could be cached.
 							// Activating the license pre-emptively to know its state during the setup process.
+							const licenseCheckStartedAt = new Date().toISOString();
 							const licenseIsValid = await minDelay(
 								activateLicense(outputChannel).then(() =>
 									checkIsLicenseValid(outputChannel),
@@ -173,15 +176,28 @@ export default createPlugin(
 								await activateLicenseUntilValid(
 									outputChannel,
 									cancellationToken,
+									telemetry,
+									origin_trigger,
+									licenseCheckStartedAt,
 								);
 							}
 
 							if (cancellationToken.isCancellationRequested) {
+								telemetry.track({
+									name: "license_setup_ended",
+									payload: {
+										namespace: "onboarding",
+										step_order: 3,
+										origin: origin_trigger,
+										auth_token: await readAuthToken(),
+										started_at: licenseCheckStartedAt,
+										ended_at: new Date().toISOString(),
+										status: "COMPLETED",
+									},
+								});
 								return;
 							}
 
-							//TODO add telemetry
-
 							/////////////////////////////////////////////////////////////////////
 							progress.report({
 								message: "Configuring AWS profiles...",
@@ -216,7 +232,7 @@ export default createPlugin(
 								name: "setup_ended",
 								payload: {
 									namespace: "onboarding",
-									steps: [1, 2, 3],
+									steps: [1, 2, 3, 4],
 									status: "COMPLETED",
 									auth_token: await readAuthToken(),
 								},

src/utils/configure-aws.ts

@@ -313,7 +313,7 @@ export async function configureAwsProfiles(options: {
 				payload: {
 					namespace: "onboarding",
 					origin: trigger,
-					step_order: 3,
+					step_order: 4,
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
@@ -347,7 +347,7 @@ export async function configureAwsProfiles(options: {
 			payload: {
 				namespace: "onboarding",
 				origin: trigger,
-				step_order: 3,
+				step_order: 4,
 				started_at: startedAt,
 				ended_at: new Date().toISOString(),
 				status: "SKIPPED",
@@ -381,7 +381,7 @@ export async function configureAwsProfiles(options: {
 				payload: {
 					namespace: "onboarding",
 					origin: trigger,
-					step_order: 3,
+					step_order: 4,
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
@@ -404,7 +404,7 @@ export async function configureAwsProfiles(options: {
 				payload: {
 					namespace: "onboarding",
 					origin: trigger,
-					step_order: 3,
+					step_order: 4,
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",
@@ -427,7 +427,7 @@ export async function configureAwsProfiles(options: {
 				payload: {
 					namespace: "onboarding",
 					origin: trigger,
-					step_order: 3,
+					step_order: 4,
 					started_at: startedAt,
 					ended_at: new Date().toISOString(),
 					status: "COMPLETED",

src/utils/license.ts

@@ -1,6 +1,8 @@
 import type { CancellationToken, LogOutputChannel } from "vscode";
 
+import { readAuthToken } from "./authenticate.ts";
 import { execLocalStack } from "./cli.ts";
+import type { Telemetry } from "./telemetry.ts";
 
 const LICENSE_VALIDITY_MARKER = "license validity: valid";
 
@@ -30,13 +32,28 @@ export async function activateLicense(outputChannel: LogOutputChannel) {
 export async function activateLicenseUntilValid(
 	outputChannel: LogOutputChannel,
 	cancellationToken: CancellationToken,
+	telemetry: Telemetry,
+	origin: "manual_trigger" | "extension_startup",
+	startedAt: string,
 ): Promise<void> {
 	while (true) {
 		if (cancellationToken.isCancellationRequested) {
 			break;
 		}
 		const licenseIsValid = await checkIsLicenseValid(outputChannel);
 		if (licenseIsValid) {
+			telemetry.track({
+				name: "license_setup_ended",
+				payload: {
+					namespace: "onboarding",
+					step_order: 3,
+					origin: origin,
+					auth_token: await readAuthToken(),
+					started_at: startedAt,
+					ended_at: new Date().toISOString(),
+					status: "COMPLETED",
+				},
+			});
 			break;
 		}
 		await activateLicense(outputChannel);

src/utils/telemetry.ts

@@ -46,11 +46,24 @@ type Events =
 			};
 	  }
 	| {
-			name: "aws_profile_configured";
+			name: "license_setup_ended";
 			payload: {
 				namespace: "onboarding";
 				origin: "manual_trigger" | "extension_startup";
 				step_order: 3;
+				auth_token?: string;
+				started_at: string;
+				ended_at: string;
+				status: "COMPLETED" | "FAILED" | "SKIPPED" | "CANCELLED";
+				errors?: string[];
+			};
+	  }
+	| {
+			name: "aws_profile_configured";
+			payload: {
+				namespace: "onboarding";
+				origin: "manual_trigger" | "extension_startup";
+				step_order: 4;
 				started_at: string;
 				ended_at: string;
 				status: "COMPLETED" | "FAILED" | "SKIPPED" | "CANCELLED";