From 5284763155cfe49f1a1561166bee08938f0bb5d7 Mon Sep 17 00:00:00 2001 From: hkobew Date: Fri, 24 Jan 2025 13:18:48 -0500 Subject: [PATCH] telemtry: implement telemetry for amazonQLSP --- packages/amazonq/src/lsp/activation.ts | 7 +++++-- packages/core/src/amazonq/index.ts | 1 + 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/packages/amazonq/src/lsp/activation.ts b/packages/amazonq/src/lsp/activation.ts index bdf4a59ab1a..2361da768f3 100644 --- a/packages/amazonq/src/lsp/activation.ts +++ b/packages/amazonq/src/lsp/activation.ts @@ -7,11 +7,14 @@ import vscode from 'vscode' import { startLanguageServer } from './client' import { AmazonQLSPResolver } from './lspInstaller' import { ToolkitError } from 'aws-core-vscode/shared' +import { lspSetupStage } from 'aws-core-vscode/amazonq' export async function activate(ctx: vscode.ExtensionContext): Promise { try { - const installResult = await new AmazonQLSPResolver().resolve() - await startLanguageServer(ctx, installResult.resourcePaths) + await lspSetupStage('all', async () => { + const installResult = await new AmazonQLSPResolver().resolve() + await lspSetupStage('launch', async () => await startLanguageServer(ctx, installResult.resourcePaths)) + }) } catch (err) { const e = err as ToolkitError void vscode.window.showInformationMessage(`Unable to launch amazonq language server: ${e.message}`) diff --git a/packages/core/src/amazonq/index.ts b/packages/core/src/amazonq/index.ts index c5abbf7658e..0c1b1803995 100644 --- a/packages/core/src/amazonq/index.ts +++ b/packages/core/src/amazonq/index.ts @@ -17,6 +17,7 @@ export { } from './onboardingPage/walkthrough' export { LspController } from './lsp/lspController' export { LspClient } from './lsp/lspClient' +export { lspSetupStage } from './lsp/util' export { api } from './extApi' export { AmazonQChatViewProvider } from './webview/webView' export { init as cwChatAppInit } from '../codewhispererChat/app'