@aws/lsp-codewhisperer 0.0.6 → 0.0.10

package/out/language-server/chat/telemetry/chatTelemetryController.js

@@ -0,0 +1,256 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.convertToTelemetryUserIntent = exports.ChatTelemetryController = exports.CONVERSATION_ID_METRIC_KEY = void 0;
+const types_1 = require("../../telemetry/types");
+const clientTelemetry_1 = require("./clientTelemetry");
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const codeDiffTracker_1 = require("../../telemetry/codeDiffTracker");
+exports.CONVERSATION_ID_METRIC_KEY = 'cwsprChatConversationId';
+class ChatTelemetryController {
+    #activeTabId;
+    #tabTelemetryInfoByTabId;
+    #currentTriggerByTabId = {};
+    #credentialsProvider;
+    #telemetry;
+    #codeDiffTracker;
+    constructor(features) {
+        this.#tabTelemetryInfoByTabId = {};
+        this.#currentTriggerByTabId = {};
+        this.#telemetry = features.telemetry;
+        this.#credentialsProvider = features.credentialsProvider;
+        this.#telemetry.onClientTelemetry(params => this.#handleClientTelemetry(params));
+        this.#codeDiffTracker = new codeDiffTracker_1.CodeDiffTracker(features.workspace, features.logging, (entry, percentage) => this.emitModifyCodeMetric(entry, percentage));
+    }
+    get activeTabId() {
+        return this.#activeTabId;
+    }
+    set activeTabId(activeTabId) {
+        this.#activeTabId = activeTabId;
+    }
+    getCurrentTrigger(tabId) {
+        return this.#currentTriggerByTabId[tabId];
+    }
+    getConversationId(tabId) {
+        return tabId && this.#tabTelemetryInfoByTabId[tabId]?.conversationId;
+    }
+    removeConversation(tabId) {
+        delete this.#tabTelemetryInfoByTabId[tabId];
+    }
+    setConversationId(tabId, conversationId) {
+        this.updateTriggerInfo(tabId, { conversationId });
+    }
+    updateTriggerInfo(tabId, info) {
+        this.#tabTelemetryInfoByTabId[tabId] = { ...this.#tabTelemetryInfoByTabId[tabId], ...info };
+    }
+    getLastMessageTrigger(tabId) {
+        return this.#tabTelemetryInfoByTabId[tabId]?.lastMessageTrigger;
+    }
+    emitModifyCodeMetric(entry, percentage) {
+        this.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.ModifyCode,
+            data: {
+                cwsprChatMessageId: entry.messageId,
+                cwsprChatModificationPercentage: percentage ? percentage : 0,
+            },
+        });
+    }
+    emitChatMetric(metric) {
+        this.#telemetry.emitMetric({
+            ...metric,
+            data: {
+                ...metric.data,
+                credentialStartUrl: this.#credentialsProvider.getConnectionMetadata()?.sso?.startUrl,
+            },
+        });
+    }
+    emitConversationMetric(metric, tabId = this.activeTabId) {
+        const conversationId = this.getConversationId(tabId);
+        if (conversationId) {
+            this.#telemetry.emitMetric({
+                ...metric,
+                data: {
+                    ...metric.data,
+                    credentialStartUrl: this.#credentialsProvider.getConnectionMetadata()?.sso?.startUrl,
+                    [exports.CONVERSATION_ID_METRIC_KEY]: conversationId,
+                },
+            });
+        }
+    }
+    emitAddMessageMetric(tabId, metric) {
+        this.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.AddMessage,
+            data: {
+                cwsprChatHasCodeSnippet: metric.cwsprChatHasCodeSnippet,
+                cwsprChatTriggerInteraction: metric.cwsprChatTriggerInteraction,
+                cwsprChatMessageId: metric.cwsprChatMessageId,
+                cwsprChatUserIntent: metric.cwsprChatUserIntent,
+                cwsprChatProgrammingLanguage: metric.cwsprChatProgrammingLanguage,
+                cwsprChatResponseCodeSnippetCount: metric.cwsprChatResponseCodeSnippetCount,
+                cwsprChatResponseCode: metric.cwsprChatResponseCode,
+                cwsprChatSourceLinkCount: metric.cwsprChatSourceLinkCount,
+                cwsprChatReferencesCount: metric.cwsprChatReferencesCount,
+                cwsprChatFollowUpCount: metric.cwsprChatFollowUpCount,
+                cwsprTimeToFirstChunk: metric.cwsprTimeToFirstChunk,
+                cwsprChatFullResponseLatency: metric.cwsprChatFullResponseLatency,
+                cwsprChatTimeBetweenChunks: metric.cwsprChatTimeBetweenChunks,
+                cwsprChatRequestLength: metric.cwsprChatRequestLength,
+                cwsprChatResponseLength: metric.cwsprChatResponseLength,
+                cwsprChatConversationType: metric.cwsprChatConversationType,
+                cwsprChatActiveEditorTotalCharacters: metric.cwsprChatActiveEditorTotalCharacters,
+                cwsprChatActiveEditorImportCount: metric.cwsprChatActiveEditorImportCount,
+                // not possible: cwsprChatResponseType: metric.cwsprChatResponseType,
+            },
+        }, tabId);
+    }
+    emitStartConversationMetric(tabId, metric) {
+        this.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.StartConversation,
+            data: {
+                cwsprChatHasCodeSnippet: metric.cwsprChatHasCodeSnippet,
+                cwsprChatTriggerInteraction: metric.cwsprChatTriggerInteraction,
+                cwsprChatUserIntent: metric.cwsprChatUserIntent,
+                cwsprChatProgrammingLanguage: metric.cwsprChatProgrammingLanguage,
+                cwsprChatConversationType: metric.cwsprChatConversationType,
+            },
+        }, tabId);
+    }
+    emitInteractWithMessageMetric(tabId, metric) {
+        this.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.InteractWithMessage,
+            data: metric,
+        }, tabId);
+    }
+    emitMessageResponseError(tabId, metric) {
+        this.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.MessageResponseError,
+            data: {
+                cwsprChatHasCodeSnippet: metric.cwsprChatHasCodeSnippet,
+                cwsprChatTriggerInteraction: metric.cwsprChatTriggerInteraction,
+                cwsprChatUserIntent: metric.cwsprChatUserIntent,
+                cwsprChatProgrammingLanguage: metric.cwsprChatProgrammingLanguage,
+                cwsprChatActiveEditorTotalCharacters: metric.cwsprChatActiveEditorTotalCharacters,
+                cwsprChatActiveEditorImportCount: metric.cwsprChatActiveEditorImportCount,
+                cwsprChatRepsonseCode: metric.cwsprChatRepsonseCode,
+                cwsprChatRequestLength: metric.cwsprChatRequestLength,
+                cwsprChatConversationType: metric.cwsprChatConversationType,
+            },
+        }, tabId);
+    }
+    #enqueueCodeDiffEntry(params) {
+        const documentUri = params.textDocument?.uri;
+        const cursorRangeOrPosition = params.cursorState?.[0];
+        if (params.code && documentUri && cursorRangeOrPosition) {
+            const startPosition = 'position' in cursorRangeOrPosition ? cursorRangeOrPosition.position : cursorRangeOrPosition.range.start;
+            const endPosition = 'position' in cursorRangeOrPosition ? cursorRangeOrPosition.position : cursorRangeOrPosition.range.end;
+            this.#codeDiffTracker.enqueue({
+                messageId: params.messageId,
+                fileUrl: documentUri,
+                time: Date.now(),
+                originalString: params.code,
+                startPosition,
+                endPosition,
+            });
+        }
+    }
+    #handleClientTelemetry(params) {
+        if ((0, clientTelemetry_1.isClientTelemetryEvent)(params)) {
+            switch (params.name) {
+                case clientTelemetry_1.ChatUIEventName.AddMessage:
+                    // we are trusting that the notification comes just right before the request
+                    this.#currentTriggerByTabId[params.tabId] = params.triggerType;
+                    break;
+                case clientTelemetry_1.ChatUIEventName.TabAdd:
+                    this.#tabTelemetryInfoByTabId[params.tabId] = {
+                        ...this.#tabTelemetryInfoByTabId[params.tabId],
+                        startTrigger: {
+                            triggerType: params.triggerType,
+                        },
+                    };
+                    break;
+                case clientTelemetry_1.ChatUIEventName.EnterFocusChat:
+                    this.emitChatMetric({
+                        name: types_1.ChatTelemetryEventName.EnterFocusChat,
+                        data: {},
+                    });
+                    break;
+                case clientTelemetry_1.ChatUIEventName.ExitFocusChat:
+                    this.emitChatMetric({
+                        name: types_1.ChatTelemetryEventName.ExitFocusChat,
+                        data: {},
+                    });
+                    break;
+                case clientTelemetry_1.ChatUIEventName.Vote:
+                    this.emitConversationMetric({
+                        name: types_1.ChatTelemetryEventName.InteractWithMessage,
+                        data: {
+                            cwsprChatMessageId: params.messageId,
+                            cwsprChatInteractionType: params.vote === clientTelemetry_1.RelevancyVoteType.UP
+                                ? types_1.ChatInteractionType.Upvote
+                                : types_1.ChatInteractionType.Downvote,
+                        },
+                    });
+                    break;
+                case clientTelemetry_1.ChatUIEventName.InsertToCursorPosition:
+                case clientTelemetry_1.ChatUIEventName.CopyToClipboard:
+                    if (params.name === clientTelemetry_1.ChatUIEventName.InsertToCursorPosition) {
+                        this.#enqueueCodeDiffEntry(params);
+                    }
+                    this.emitConversationMetric({
+                        name: types_1.ChatTelemetryEventName.InteractWithMessage,
+                        data: {
+                            cwsprChatMessageId: params.messageId,
+                            cwsprChatInteractionType: params.name === clientTelemetry_1.ChatUIEventName.InsertToCursorPosition
+                                ? types_1.ChatInteractionType.InsertAtCursor
+                                : types_1.ChatInteractionType.CopySnippet,
+                            cwsprChatAcceptedCharactersLength: params.code?.length ?? 0,
+                            cwsprChatHasReference: Boolean(params.referenceTrackerInformation?.length),
+                            cwsprChatCodeBlockIndex: params.codeBlockIndex,
+                            cwsprChatTotalCodeBlocks: params.totalCodeBlocks,
+                        },
+                    });
+                    break;
+                case clientTelemetry_1.ChatUIEventName.LinkClick:
+                case clientTelemetry_1.ChatUIEventName.InfoLinkClick:
+                    this.emitInteractWithMessageMetric(params.tabId, {
+                        cwsprChatMessageId: params.messageId,
+                        cwsprChatInteractionType: types_1.ChatInteractionType.ClickBodyLink,
+                        cwsprChatInteractionTarget: params.link,
+                    });
+                    break;
+                case clientTelemetry_1.ChatUIEventName.SourceLinkClick:
+                    this.emitInteractWithMessageMetric(params.tabId, {
+                        cwsprChatMessageId: params.messageId,
+                        cwsprChatInteractionType: types_1.ChatInteractionType.ClickLink,
+                        cwsprChatInteractionTarget: params.link,
+                    });
+                    break;
+            }
+        }
+    }
+    dispose() {
+        this.#codeDiffTracker.shutdown();
+    }
+}
+exports.ChatTelemetryController = ChatTelemetryController;
+function convertToTelemetryUserIntent(userIntent) {
+    switch (userIntent) {
+        case codewhisperer_streaming_1.UserIntent.EXPLAIN_CODE_SELECTION:
+            return 'explainCodeSelection';
+        case codewhisperer_streaming_1.UserIntent.SUGGEST_ALTERNATE_IMPLEMENTATION:
+            return 'suggestAlternateImplementation';
+        case codewhisperer_streaming_1.UserIntent.APPLY_COMMON_BEST_PRACTICES:
+            return 'applyCommonBestPractices';
+        case codewhisperer_streaming_1.UserIntent.IMPROVE_CODE:
+            return 'improveCode';
+        case codewhisperer_streaming_1.UserIntent.CITE_SOURCES:
+            return 'citeSources';
+        case codewhisperer_streaming_1.UserIntent.EXPLAIN_LINE_BY_LINE:
+            return 'explainLineByLine';
+        case codewhisperer_streaming_1.UserIntent.SHOW_EXAMPLES:
+            return 'showExamples';
+        default:
+            return '';
+    }
+}
+exports.convertToTelemetryUserIntent = convertToTelemetryUserIntent;
+//# sourceMappingURL=chatTelemetryController.js.map

package/out/language-server/chat/telemetry/chatTelemetryController.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatTelemetryController.js","sourceRoot":"","sources":["../../../../src/language-server/chat/telemetry/chatTelemetryController.ts"],"names":[],"mappings":";;;AAEA,iDAM8B;AAE9B,uDAK0B;AAC1B,2EAA0D;AAE1D,qEAA0F;AAE7E,QAAA,0BAA0B,GAAG,yBAAyB,CAAA;AAgCnE,MAAa,uBAAuB;IAChC,YAAY,CAAS;IACrB,wBAAwB,CAA8C;IACtE,sBAAsB,GAAqC,EAAE,CAAA;IAC7D,oBAAoB,CAAiC;IACrD,UAAU,CAAuB;IACjC,gBAAgB,CAA8C;IAE9D,YAAY,QAAkB;QAC1B,IAAI,CAAC,wBAAwB,GAAG,EAAE,CAAA;QAClC,IAAI,CAAC,sBAAsB,GAAG,EAAE,CAAA;QAChC,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAA;QACpC,IAAI,CAAC,oBAAoB,GAAG,QAAQ,CAAC,mBAAmB,CAAA;QACxD,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC,CAAA;QAChF,IAAI,CAAC,gBAAgB,GAAG,IAAI,iCAAe,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CACpG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAC/C,CAAA;IACL,CAAC;IAED,IAAW,WAAW;QAClB,OAAO,IAAI,CAAC,YAAY,CAAA;IAC5B,CAAC;IAED,IAAW,WAAW,CAAC,WAA+B;QAClD,IAAI,CAAC,YAAY,GAAG,WAAW,CAAA;IACnC,CAAC;IAEM,iBAAiB,CAAC,KAAa;QAClC,OAAO,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAA;IAC7C,CAAC;IAEM,iBAAiB,CAAC,KAAc;QACnC,OAAO,KAAK,IAAI,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,cAAc,CAAA;IACxE,CAAC;IAEM,kBAAkB,CAAC,KAAa;QACnC,OAAO,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAA;IAC/C,CAAC;IAEM,iBAAiB,CAAC,KAAa,EAAE,cAAsB;QAC1D,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,CAAA;IACrD,CAAC;IAEM,iBAAiB,CAAC,KAAa,EAAE,IAAsC;QAC1E,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,GAAG,IAAI,EAAE,CAAA;IAC/F,CAAC;IAEM,qBAAqB,CAAC,KAAa;QACtC,OAAO,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,kBAAkB,CAAA;IACnE,CAAC;IAEM,oBAAoB,CAAC,KAAkC,EAAE,UAAkB;QAC9E,IAAI,CAAC,sBAAsB,CAAC;YACxB,IAAI,EAAE,8BAAsB,CAAC,UAAU;YACvC,IAAI,EAAE;gBACF,kBAAkB,EAAE,KAAK,CAAC,SAAS;gBACnC,+BAA+B,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;aAC/D;SACJ,CAAC,CAAA;IACN,CAAC;IAEM,cAAc,CACjB,MAA4D;QAE5D,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACvB,GAAG,MAAM;YACT,IAAI,EAAE;gBACF,GAAG,MAAM,CAAC,IAAI;gBACd,kBAAkB,EAAE,IAAI,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,EAAE,GAAG,EAAE,QAAQ;aACvF;SACJ,CAAC,CAAA;IACN,CAAC;IAEM,sBAAsB,CAIzB,MAEC,EACD,KAAK,GAAG,IAAI,CAAC,WAAW;QAExB,MAAM,cAAc,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAA;QACpD,IAAI,cAAc,EAAE;YAChB,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;gBACvB,GAAG,MAAM;gBACT,IAAI,EAAE;oBACF,GAAG,MAAM,CAAC,IAAI;oBACd,kBAAkB,EAAE,IAAI,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,EAAE,GAAG,EAAE,QAAQ;oBACpF,CAAC,kCAA0B,CAAC,EAAE,cAAc;iBAC/C;aACJ,CAAC,CAAA;SACL;IACL,CAAC;IAEM,oBAAoB,CAAC,KAAa,EAAE,MAA0C;QACjF,IAAI,CAAC,sBAAsB,CACvB;YACI,IAAI,EAAE,8BAAsB,CAAC,UAAU;YACvC,IAAI,EAAE;gBACF,uBAAuB,EAAE,MAAM,CAAC,uBAAuB;gBACvD,2BAA2B,EAAE,MAAM,CAAC,2BAA2B;gBAC/D,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;gBAC7C,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;gBAC/C,4BAA4B,EAAE,MAAM,CAAC,4BAA4B;gBACjE,iCAAiC,EAAE,MAAM,CAAC,iCAAiC;gBAC3E,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;gBACnD,wBAAwB,EAAE,MAAM,CAAC,wBAAwB;gBACzD,wBAAwB,EAAE,MAAM,CAAC,wBAAwB;gBACzD,sBAAsB,EAAE,MAAM,CAAC,sBAAsB;gBACrD,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;gBACnD,4BAA4B,EAAE,MAAM,CAAC,4BAA4B;gBACjE,0BAA0B,EAAE,MAAM,CAAC,0BAA0B;gBAC7D,sBAAsB,EAAE,MAAM,CAAC,sBAAsB;gBACrD,uBAAuB,EAAE,MAAM,CAAC,uBAAuB;gBACvD,yBAAyB,EAAE,MAAM,CAAC,yBAAyB;gBAC3D,oCAAoC,EAAE,MAAM,CAAC,oCAAoC;gBACjF,gCAAgC,EAAE,MAAM,CAAC,gCAAgC;gBACzE,qEAAqE;aACxE;SACJ,EACD,KAAK,CACR,CAAA;IACL,CAAC;IAEM,2BAA2B,CAAC,KAAa,EAAE,MAA0C;QACxF,IAAI,CAAC,sBAAsB,CACvB;YACI,IAAI,EAAE,8BAAsB,CAAC,iBAAiB;YAC9C,IAAI,EAAE;gBACF,uBAAuB,EAAE,MAAM,CAAC,uBAAuB;gBACvD,2BAA2B,EAAE,MAAM,CAAC,2BAA2B;gBAC/D,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;gBAC/C,4BAA4B,EAAE,MAAM,CAAC,4BAA4B;gBACjE,yBAAyB,EAAE,MAAM,CAAC,yBAAyB;aAC9D;SACJ,EACD,KAAK,CACR,CAAA;IACL,CAAC;IAEM,6BAA6B,CAChC,KAAa,EACb,MAAiE;QAEjE,IAAI,CAAC,sBAAsB,CACvB;YACI,IAAI,EAAE,8BAAsB,CAAC,mBAAmB;YAChD,IAAI,EAAE,MAAM;SACf,EACD,KAAK,CACR,CAAA;IACL,CAAC;IAEM,wBAAwB,CAAC,KAAa,EAAE,MAA0C;QACrF,IAAI,CAAC,sBAAsB,CACvB;YACI,IAAI,EAAE,8BAAsB,CAAC,oBAAoB;YACjD,IAAI,EAAE;gBACF,uBAAuB,EAAE,MAAM,CAAC,uBAAuB;gBACvD,2BAA2B,EAAE,MAAM,CAAC,2BAA2B;gBAC/D,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;gBAC/C,4BAA4B,EAAE,MAAM,CAAC,4BAA4B;gBACjE,oCAAoC,EAAE,MAAM,CAAC,oCAAoC;gBACjF,gCAAgC,EAAE,MAAM,CAAC,gCAAgC;gBACzE,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;gBACnD,sBAAsB,EAAE,MAAM,CAAC,sBAAsB;gBACrD,yBAAyB,EAAE,MAAM,CAAC,yBAAyB;aAC9D;SACJ,EACD,KAAK,CACR,CAAA;IACL,CAAC;IAED,qBAAqB,CAAC,MAAoC;QACtD,MAAM,WAAW,GAAG,MAAM,CAAC,YAAY,EAAE,GAAG,CAAA;QAC5C,MAAM,qBAAqB,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAA;QAErD,IAAI,MAAM,CAAC,IAAI,IAAI,WAAW,IAAI,qBAAqB,EAAE;YACrD,MAAM,aAAa,GACf,UAAU,IAAI,qBAAqB,CAAC,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,qBAAqB,CAAC,KAAK,CAAC,KAAK,CAAA;YAC5G,MAAM,WAAW,GACb,UAAU,IAAI,qBAAqB,CAAC,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,qBAAqB,CAAC,KAAK,CAAC,GAAG,CAAA;YAE1G,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC;gBAC1B,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE;gBAChB,cAAc,EAAE,MAAM,CAAC,IAAI;gBAC3B,aAAa;gBACb,WAAW;aACd,CAAC,CAAA;SACL;IACL,CAAC;IAED,sBAAsB,CAAC,MAAe;QAClC,IAAI,IAAA,wCAAsB,EAAC,MAAM,CAAC,EAAE;YAChC,QAAQ,MAAM,CAAC,IAAI,EAAE;gBACjB,KAAK,iCAAe,CAAC,UAAU;oBAC3B,4EAA4E;oBAC5E,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,WAAW,CAAA;oBAC9D,MAAK;gBACT,KAAK,iCAAe,CAAC,MAAM;oBACvB,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG;wBAC1C,GAAG,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,KAAK,CAAC;wBAC9C,YAAY,EAAE;4BACV,WAAW,EAAE,MAAM,CAAC,WAAW;yBAClC;qBACJ,CAAA;oBACD,MAAK;gBACT,KAAK,iCAAe,CAAC,cAAc;oBAC/B,IAAI,CAAC,cAAc,CAAC;wBAChB,IAAI,EAAE,8BAAsB,CAAC,cAAc;wBAC3C,IAAI,EAAE,EAAE;qBACX,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,iCAAe,CAAC,aAAa;oBAC9B,IAAI,CAAC,cAAc,CAAC;wBAChB,IAAI,EAAE,8BAAsB,CAAC,aAAa;wBAC1C,IAAI,EAAE,EAAE;qBACX,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,iCAAe,CAAC,IAAI;oBACrB,IAAI,CAAC,sBAAsB,CAAC;wBACxB,IAAI,EAAE,8BAAsB,CAAC,mBAAmB;wBAChD,IAAI,EAAE;4BACF,kBAAkB,EAAE,MAAM,CAAC,SAAS;4BACpC,wBAAwB,EACpB,MAAM,CAAC,IAAI,KAAK,mCAAiB,CAAC,EAAE;gCAChC,CAAC,CAAC,2BAAmB,CAAC,MAAM;gCAC5B,CAAC,CAAC,2BAAmB,CAAC,QAAQ;yBACzC;qBACJ,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,iCAAe,CAAC,sBAAsB,CAAC;gBAC5C,KAAK,iCAAe,CAAC,eAAe;oBAChC,IAAI,MAAM,CAAC,IAAI,KAAK,iCAAe,CAAC,sBAAsB,EAAE;wBACxD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAA;qBACrC;oBAED,IAAI,CAAC,sBAAsB,CAAC;wBACxB,IAAI,EAAE,8BAAsB,CAAC,mBAAmB;wBAChD,IAAI,EAAE;4BACF,kBAAkB,EAAE,MAAM,CAAC,SAAS;4BACpC,wBAAwB,EACpB,MAAM,CAAC,IAAI,KAAK,iCAAe,CAAC,sBAAsB;gCAClD,CAAC,CAAC,2BAAmB,CAAC,cAAc;gCACpC,CAAC,CAAC,2BAAmB,CAAC,WAAW;4BACzC,iCAAiC,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC;4BAC3D,qBAAqB,EAAE,OAAO,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC;4BAC1E,uBAAuB,EAAE,MAAM,CAAC,cAAc;4BAC9C,wBAAwB,EAAE,MAAM,CAAC,eAAe;yBACnD;qBACJ,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,iCAAe,CAAC,SAAS,CAAC;gBAC/B,KAAK,iCAAe,CAAC,aAAa;oBAC9B,IAAI,CAAC,6BAA6B,CAAC,MAAM,CAAC,KAAK,EAAE;wBAC7C,kBAAkB,EAAE,MAAM,CAAC,SAAS;wBACpC,wBAAwB,EAAE,2BAAmB,CAAC,aAAa;wBAC3D,0BAA0B,EAAE,MAAM,CAAC,IAAI;qBAC1C,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,iCAAe,CAAC,eAAe;oBAChC,IAAI,CAAC,6BAA6B,CAAC,MAAM,CAAC,KAAK,EAAE;wBAC7C,kBAAkB,EAAE,MAAM,CAAC,SAAS;wBACpC,wBAAwB,EAAE,2BAAmB,CAAC,SAAS;wBACvD,0BAA0B,EAAE,MAAM,CAAC,IAAI;qBAC1C,CAAC,CAAA;oBACF,MAAK;aACZ;SACJ;IACL,CAAC;IAEM,OAAO;QACV,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAA;IACpC,CAAC;CACJ;AArRD,0DAqRC;AAED,SAAgB,4BAA4B,CAAC,UAAuB;IAChE,QAAQ,UAAU,EAAE;QAChB,KAAK,oCAAU,CAAC,sBAAsB;YAClC,OAAO,sBAAsB,CAAA;QACjC,KAAK,oCAAU,CAAC,gCAAgC;YAC5C,OAAO,gCAAgC,CAAA;QAC3C,KAAK,oCAAU,CAAC,2BAA2B;YACvC,OAAO,0BAA0B,CAAA;QACrC,KAAK,oCAAU,CAAC,YAAY;YACxB,OAAO,aAAa,CAAA;QACxB,KAAK,oCAAU,CAAC,YAAY;YACxB,OAAO,aAAa,CAAA;QACxB,KAAK,oCAAU,CAAC,oBAAoB;YAChC,OAAO,mBAAmB,CAAA;QAC9B,KAAK,oCAAU,CAAC,aAAa;YACzB,OAAO,cAAc,CAAA;QACzB;YACI,OAAO,EAAE,CAAA;KAChB;AACL,CAAC;AAnBD,oEAmBC"}

package/out/language-server/chat/telemetry/chatTelemetryController.test.d.ts

@@ -0,0 +1 @@
+export {};

package/out/language-server/chat/telemetry/chatTelemetryController.test.js

@@ -0,0 +1,84 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const testing_1 = require("@aws/language-server-runtimes/testing");
+const ts_sinon_1 = require("ts-sinon");
+const types_1 = require("../../telemetry/types");
+const chatTelemetryController_1 = require("./chatTelemetryController");
+const assert = require("assert");
+const clientTelemetry_1 = require("./clientTelemetry");
+describe('TelemetryController', () => {
+    const mockTabId = 'mockTabId';
+    const mockConversationId = 'mockConversationId';
+    let testFeatures;
+    let telemetryController;
+    beforeEach(() => {
+        testFeatures = new testing_1.TestFeatures();
+        telemetryController = new chatTelemetryController_1.ChatTelemetryController(testFeatures);
+    });
+    it('able to set and get activeTabId', () => {
+        assert.strictEqual(telemetryController.activeTabId, undefined);
+        telemetryController.activeTabId = mockTabId;
+        assert.strictEqual(telemetryController.activeTabId, mockTabId);
+    });
+    it('able to set and get conversationId for a tab id', () => {
+        assert.strictEqual(telemetryController.getConversationId(mockTabId), undefined);
+        telemetryController.setConversationId(mockTabId, mockConversationId);
+        assert.strictEqual(telemetryController.getConversationId(mockTabId), mockConversationId);
+    });
+    it('able to remove conversation id by tab id from the map', () => {
+        telemetryController.setConversationId(mockTabId, mockConversationId);
+        assert.strictEqual(telemetryController.getConversationId(mockTabId), mockConversationId);
+        telemetryController.removeConversation(mockTabId);
+        assert.strictEqual(telemetryController.getConversationId(mockTabId), undefined);
+    });
+    it('handles enter focus client telemetry', () => {
+        const telemetryHandler = testFeatures.telemetry.onClientTelemetry.firstCall.firstArg;
+        telemetryHandler({
+            name: clientTelemetry_1.ChatUIEventName.EnterFocusChat,
+        });
+        ts_sinon_1.default.assert.calledOnceWithExactly(testFeatures.telemetry.emitMetric, {
+            name: types_1.ChatTelemetryEventName.EnterFocusChat,
+            data: { credentialStartUrl: undefined },
+        });
+    });
+    it('handles exit focus client telemetry', () => {
+        const telemetryHandler = testFeatures.telemetry.onClientTelemetry.firstCall.firstArg;
+        telemetryHandler({
+            name: clientTelemetry_1.ChatUIEventName.ExitFocusChat,
+        });
+        ts_sinon_1.default.assert.calledOnceWithExactly(testFeatures.telemetry.emitMetric, {
+            name: types_1.ChatTelemetryEventName.ExitFocusChat,
+            data: { credentialStartUrl: undefined },
+        });
+    });
+    it('does not handle unknown client telemetry', () => {
+        const telemetryHandler = testFeatures.telemetry.onClientTelemetry.firstCall.firstArg;
+        telemetryHandler({
+            params: { name: 'Unknown event' },
+        });
+        ts_sinon_1.default.assert.notCalled(testFeatures.telemetry.emitMetric);
+    });
+    it('does not emit metrics if conversation id is not present', () => {
+        telemetryController.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.EnterFocusConversation,
+            data: { credentialStartUrl: undefined },
+        });
+        ts_sinon_1.default.assert.notCalled(testFeatures.telemetry.emitMetric);
+    });
+    it('emits metrics only if conversation id for active tab is present', () => {
+        telemetryController.setConversationId(mockTabId, mockConversationId);
+        telemetryController.activeTabId = mockTabId;
+        telemetryController.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.EnterFocusConversation,
+            data: {},
+        });
+        ts_sinon_1.default.assert.calledOnceWithExactly(testFeatures.telemetry.emitMetric, {
+            name: types_1.ChatTelemetryEventName.EnterFocusConversation,
+            data: {
+                [chatTelemetryController_1.CONVERSATION_ID_METRIC_KEY]: mockConversationId,
+                credentialStartUrl: undefined,
+            },
+        });
+    });
+});
+//# sourceMappingURL=chatTelemetryController.test.js.map

package/out/language-server/chat/telemetry/chatTelemetryController.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatTelemetryController.test.js","sourceRoot":"","sources":["../../../../src/language-server/chat/telemetry/chatTelemetryController.test.ts"],"names":[],"mappings":";;AAAA,mEAAoE;AACpE,uCAA4B;AAC5B,iDAA8D;AAC9D,uEAA+F;AAC/F,iCAAiC;AACjC,uDAAmD;AAEnD,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACjC,MAAM,SAAS,GAAG,WAAW,CAAA;IAC7B,MAAM,kBAAkB,GAAG,oBAAoB,CAAA;IAE/C,IAAI,YAA0B,CAAA;IAC9B,IAAI,mBAA4C,CAAA;IAEhD,UAAU,CAAC,GAAG,EAAE;QACZ,YAAY,GAAG,IAAI,sBAAY,EAAE,CAAA;QACjC,mBAAmB,GAAG,IAAI,iDAAuB,CAAC,YAAY,CAAC,CAAA;IACnE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACvC,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAA;QAE9D,mBAAmB,CAAC,WAAW,GAAG,SAAS,CAAA;QAE3C,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAA;IAClE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACvD,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAA;QAE/E,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAA;QAEpE,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,kBAAkB,CAAC,CAAA;IAC5F,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC7D,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAA;QAEpE,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,kBAAkB,CAAC,CAAA;QAExF,mBAAmB,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAA;QAEjD,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAA;IACnF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC5C,MAAM,gBAAgB,GAAG,YAAY,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,CAAA;QAEpF,gBAAgB,CAAC;YACb,IAAI,EAAE,iCAAe,CAAC,cAAc;SACvC,CAAC,CAAA;QAEF,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,EAAE;YAClE,IAAI,EAAE,8BAAsB,CAAC,cAAc;YAC3C,IAAI,EAAE,EAAE,kBAAkB,EAAE,SAAS,EAAE;SAC1C,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC3C,MAAM,gBAAgB,GAAG,YAAY,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,CAAA;QAEpF,gBAAgB,CAAC;YACb,IAAI,EAAE,iCAAe,CAAC,aAAa;SACtC,CAAC,CAAA;QAEF,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,EAAE;YAClE,IAAI,EAAE,8BAAsB,CAAC,aAAa;YAC1C,IAAI,EAAE,EAAE,kBAAkB,EAAE,SAAS,EAAE;SAC1C,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAChD,MAAM,gBAAgB,GAAG,YAAY,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,CAAA;QAEpF,gBAAgB,CAAC;YACb,MAAM,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE;SACpC,CAAC,CAAA;QAEF,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;IAC7D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QAC/D,mBAAmB,CAAC,sBAAsB,CAAC;YACvC,IAAI,EAAE,8BAAsB,CAAC,sBAAsB;YACnD,IAAI,EAAE,EAAE,kBAAkB,EAAE,SAAS,EAAE;SAC1C,CAAC,CAAA;QAEF,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;IAC7D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACvE,mBAAmB,CAAC,iBAAiB,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAA;QACpE,mBAAmB,CAAC,WAAW,GAAG,SAAS,CAAA;QAE3C,mBAAmB,CAAC,sBAAsB,CAAC;YACvC,IAAI,EAAE,8BAAsB,CAAC,sBAAsB;YACnD,IAAI,EAAE,EAAE;SACX,CAAC,CAAA;QAEF,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,EAAE;YAClE,IAAI,EAAE,8BAAsB,CAAC,sBAAsB;YACnD,IAAI,EAAE;gBACF,CAAC,oDAA0B,CAAC,EAAE,kBAAkB;gBAChD,kBAAkB,EAAE,SAAS;aAChC;SACJ,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/chat/telemetry/clientTelemetry.d.ts

@@ -0,0 +1,60 @@
+import type * as ServerInterface from '@aws/language-server-runtimes/server-interface';
+import type * as ChatClientType from '@aws/chat-client-ui-types';
+export declare enum ChatUIEventName {
+    EnterFocusChat = "enterFocus",
+    ExitFocusChat = "exitFocus",
+    AddMessage = "addMessage",
+    SendToPrompt = "sendToPrompt",
+    TabAdd = "tabAdd",
+    CopyToClipboard = "copyToClipboard",
+    InsertToCursorPosition = "insertToCursorPosition",
+    Vote = "vote",
+    LinkClick = "linkClick",
+    InfoLinkClick = "infoLinkClick",
+    SourceLinkClick = "sourceLinkClick"
+}
+export declare enum RelevancyVoteType {
+    UP = "upvote",
+    DOWN = "downvote"
+}
+export type VoteTelemetryParams = BaseClientTelemetryParams<ChatUIEventName.Vote> & {
+    vote: RelevancyVoteType;
+    messageId: string;
+};
+export type CopyCodeToClipboardParams = BaseClientTelemetryParams<ChatUIEventName.CopyToClipboard> & {
+    messageId: string;
+    code?: string;
+    type?: ServerInterface.CodeSelectionType;
+    referenceTrackerInformation?: ServerInterface.ReferenceTrackerInformation[];
+    codeBlockIndex?: number;
+    totalCodeBlocks?: number;
+};
+export interface BaseClientTelemetryParams<TName extends ChatUIEventName> {
+    name: TName;
+    tabId: string;
+    eventId?: string;
+}
+export type TabAddParams = BaseClientTelemetryParams<ChatUIEventName.TabAdd> & {
+    triggerType: ChatClientType.TriggerType;
+};
+export type AddMessageParams = {
+    triggerType: ChatClientType.TriggerType;
+    tabId: string;
+    name: ChatUIEventName.AddMessage;
+};
+export type SendToPromptParams = ChatClientType.SendToPromptParams & BaseClientTelemetryParams<ChatUIEventName.SendToPrompt> & {
+    messageId: string;
+};
+export type InfoLinkClickParams = ServerInterface.InfoLinkClickParams & BaseClientTelemetryParams<ChatUIEventName.InfoLinkClick> & {
+    messageId: string;
+};
+export type LinkClickParams = ServerInterface.LinkClickParams & BaseClientTelemetryParams<ChatUIEventName.LinkClick> & {
+    messageId: string;
+};
+export type SourceLinkClickParams = ServerInterface.SourceLinkClickParams & BaseClientTelemetryParams<ChatUIEventName.SourceLinkClick>;
+export type InsertToCursorPositionParams = ServerInterface.InsertToCursorPositionParams & BaseClientTelemetryParams<ChatUIEventName.InsertToCursorPosition> & {
+    cursorState?: ServerInterface.CursorState[];
+    textDocument?: ServerInterface.TextDocumentIdentifier;
+};
+export type ClientTelemetryEvent = BaseClientTelemetryParams<ChatUIEventName.EnterFocusChat> | BaseClientTelemetryParams<ChatUIEventName.ExitFocusChat> | TabAddParams | AddMessageParams | SendToPromptParams | VoteTelemetryParams | CopyCodeToClipboardParams | InfoLinkClickParams | LinkClickParams | SourceLinkClickParams | InsertToCursorPositionParams;
+export declare function isClientTelemetryEvent(event: unknown): event is ClientTelemetryEvent;

package/out/language-server/chat/telemetry/clientTelemetry.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isClientTelemetryEvent = exports.RelevancyVoteType = exports.ChatUIEventName = void 0;
+const utils_1 = require("../../utils");
+var ChatUIEventName;
+(function (ChatUIEventName) {
+    ChatUIEventName["EnterFocusChat"] = "enterFocus";
+    ChatUIEventName["ExitFocusChat"] = "exitFocus";
+    ChatUIEventName["AddMessage"] = "addMessage";
+    ChatUIEventName["SendToPrompt"] = "sendToPrompt";
+    ChatUIEventName["TabAdd"] = "tabAdd";
+    ChatUIEventName["CopyToClipboard"] = "copyToClipboard";
+    ChatUIEventName["InsertToCursorPosition"] = "insertToCursorPosition";
+    ChatUIEventName["Vote"] = "vote";
+    ChatUIEventName["LinkClick"] = "linkClick";
+    ChatUIEventName["InfoLinkClick"] = "infoLinkClick";
+    ChatUIEventName["SourceLinkClick"] = "sourceLinkClick";
+})(ChatUIEventName = exports.ChatUIEventName || (exports.ChatUIEventName = {}));
+/* Chat client only telemetry - we should import these in the future */
+var RelevancyVoteType;
+(function (RelevancyVoteType) {
+    RelevancyVoteType["UP"] = "upvote";
+    RelevancyVoteType["DOWN"] = "downvote";
+})(RelevancyVoteType = exports.RelevancyVoteType || (exports.RelevancyVoteType = {}));
+const chatUIEventNameSet = new Set(Object.values(ChatUIEventName));
+function isClientTelemetryEvent(event) {
+    return (0, utils_1.isObject)(event) && typeof event.name === 'string' && chatUIEventNameSet.has(event.name);
+}
+exports.isClientTelemetryEvent = isClientTelemetryEvent;
+//# sourceMappingURL=clientTelemetry.js.map

package/out/language-server/chat/telemetry/clientTelemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"clientTelemetry.js","sourceRoot":"","sources":["../../../../src/language-server/chat/telemetry/clientTelemetry.ts"],"names":[],"mappings":";;;AAEA,uCAAsC;AAEtC,IAAY,eAYX;AAZD,WAAY,eAAe;IACvB,gDAA6B,CAAA;IAC7B,8CAA2B,CAAA;IAC3B,4CAAyB,CAAA;IACzB,gDAA6B,CAAA;IAC7B,oCAAiB,CAAA;IACjB,sDAAmC,CAAA;IACnC,oEAAiD,CAAA;IACjD,gCAAa,CAAA;IACb,0CAAuB,CAAA;IACvB,kDAA+B,CAAA;IAC/B,sDAAmC,CAAA;AACvC,CAAC,EAZW,eAAe,GAAf,uBAAe,KAAf,uBAAe,QAY1B;AAED,uEAAuE;AACvE,IAAY,iBAGX;AAHD,WAAY,iBAAiB;IACzB,kCAAa,CAAA;IACb,sCAAiB,CAAA;AACrB,CAAC,EAHW,iBAAiB,GAAjB,yBAAiB,KAAjB,yBAAiB,QAG5B;AAuED,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAS,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAA;AAE1E,SAAgB,sBAAsB,CAAC,KAAc;IACjD,OAAO,IAAA,gBAAQ,EAAC,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;AAClG,CAAC;AAFD,wDAEC"}

package/out/language-server/chat/utils.d.ts

@@ -1,4 +1,6 @@
-import { EditorState, GenerateAssistantResponseCommandInput } from '@amzn/codewhisperer-streaming';
-import { ChatParams } from '@aws/language-server-runtimes/server-interface';
-import { Result } from '../types';
-export declare function convertChatParamsToRequestInput(params: ChatParams, editorState?: EditorState): Result<GenerateAssistantResponseCommandInput, string>;
+import { ChatResult } from '@aws/language-server-runtimes/server-interface';
+type AuthFollowUpType = 'full-auth' | 're-auth' | 'missing_scopes' | 'use-supported-auth';
+export declare function getAuthFollowUpType(err: unknown): AuthFollowUpType | undefined;
+export declare function createAuthFollowUpResult(authType: AuthFollowUpType): ChatResult;
+export declare function getDefaultChatResponse(prompt?: string): ChatResult | undefined;
+export {};

package/out/language-server/chat/utils.js

@@ -1,34 +1,60 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.convertChatParamsToRequestInput = void 0;
-const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
-function convertChatParamsToRequestInput(params, editorState) {
-    const { prompt } = params;
-    if (prompt.prompt || prompt.escapedPrompt) {
-        // TODO: implement userInputMessageContext state when that is available, and diagnostic trigger type
-        return {
-            success: true,
-            data: {
-                conversationState: {
-                    chatTriggerType: codewhisperer_streaming_1.ChatTriggerType.MANUAL,
-                    currentMessage: {
-                        userInputMessage: {
-                            content: prompt.escapedPrompt ?? prompt.prompt,
-                            userInputMessageContext: editorState
-                                ? {
-                                    editorState,
-                                }
-                                : undefined,
-                        },
-                    },
-                },
-            },
-        };
+exports.getDefaultChatResponse = exports.createAuthFollowUpResult = exports.getAuthFollowUpType = void 0;
+const constants_1 = require("../constants");
+const constants_2 = require("./constants");
+const uuid_1 = require("uuid");
+const AUTH_ERROR_DEFINITION_LIST = [
+    {
+        match: (err) => err.message.startsWith(constants_1.MISSING_BEARER_TOKEN_ERROR),
+        authFollowType: 'full-auth',
+    },
+    {
+        match: (err) => err.message.startsWith(constants_1.INVALID_TOKEN),
+        authFollowType: 're-auth',
+    },
+    {
+        match: (err) => err.message.startsWith(constants_1.GENERIC_UNAUTHORIZED_ERROR),
+        authFollowType: 'full-auth',
+    },
+];
+function getAuthFollowUpType(err) {
+    return err instanceof Error
+        ? AUTH_ERROR_DEFINITION_LIST.find(definition => definition.match(err))?.authFollowType
+        : undefined;
+}
+exports.getAuthFollowUpType = getAuthFollowUpType;
+function createAuthFollowUpResult(authType) {
+    let pillText;
+    switch (authType) {
+        case 'full-auth':
+            pillText = 'Authenticate';
+            break;
+        case 'use-supported-auth':
+        case 'missing_scopes':
+            pillText = 'Enable Amazon Q';
+            break;
+        case 're-auth':
+            pillText = 'Re-authenticate';
+            break;
     }
     return {
-        success: false,
-        error: 'Invalid request input',
+        body: '',
+        followUp: {
+            text: '',
+            options: [{ pillText, type: authType }],
+        },
     };
 }
-exports.convertChatParamsToRequestInput = convertChatParamsToRequestInput;
+exports.createAuthFollowUpResult = createAuthFollowUpResult;
+function getDefaultChatResponse(prompt) {
+    if (prompt === constants_2.DEFAULT_HELP_FOLLOW_UP_PROMPT) {
+        return {
+            messageId: (0, uuid_1.v4)(),
+            body: constants_2.HELP_MESSAGE,
+        };
+    }
+    return undefined;
+}
+exports.getDefaultChatResponse = getDefaultChatResponse;
 //# sourceMappingURL=utils.js.map

package/out/language-server/chat/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/language-server/chat/utils.ts"],"names":[],"mappings":";;;AAAA,2EAAmH;AAInH,SAAgB,+BAA+B,CAC3C,MAAkB,EAClB,WAAyB;IAEzB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAA;IAEzB,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QACxC,oGAAoG;QACpG,OAAO;YACH,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,iBAAiB,EAAE;oBACf,eAAe,EAAE,yCAAe,CAAC,MAAM;oBACvC,cAAc,EAAE;wBACZ,gBAAgB,EAAE;4BACd,OAAO,EAAE,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,MAAM;4BAC9C,uBAAuB,EAAE,WAAW;gCAChC,CAAC,CAAC;oCACI,WAAW;iCACd;gCACH,CAAC,CAAC,SAAS;yBAClB;qBACJ;iBACJ;aACJ;SACJ,CAAA;IACL,CAAC;IAED,OAAO;QACH,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,uBAAuB;KACjC,CAAA;AACL,CAAC;AAhCD,0EAgCC"}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/language-server/chat/utils.ts"],"names":[],"mappings":";;;AACA,4CAAoG;AACpG,2CAAyE;AACzE,+BAAiC;AAMjC,MAAM,0BAA0B,GAA0B;IACtD;QACI,KAAK,EAAE,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,sCAA0B,CAAC;QACzE,cAAc,EAAE,WAAW;KAC9B;IACD;QACI,KAAK,EAAE,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,yBAAa,CAAC;QAC5D,cAAc,EAAE,SAAS;KAC5B;IACD;QACI,KAAK,EAAE,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,sCAA0B,CAAC;QACzE,cAAc,EAAE,WAAW;KAC9B;CACJ,CAAA;AAED,SAAgB,mBAAmB,CAAC,GAAY;IAC5C,OAAO,GAAG,YAAY,KAAK;QACvB,CAAC,CAAC,0BAA0B,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc;QACtF,CAAC,CAAC,SAAS,CAAA;AACnB,CAAC;AAJD,kDAIC;AAED,SAAgB,wBAAwB,CAAC,QAA0B;IAC/D,IAAI,QAAQ,CAAA;IACZ,QAAQ,QAAQ,EAAE;QACd,KAAK,WAAW;YACZ,QAAQ,GAAG,cAAc,CAAA;YACzB,MAAK;QACT,KAAK,oBAAoB,CAAC;QAC1B,KAAK,gBAAgB;YACjB,QAAQ,GAAG,iBAAiB,CAAA;YAC5B,MAAK;QACT,KAAK,SAAS;YACV,QAAQ,GAAG,iBAAiB,CAAA;YAC5B,MAAK;KACZ;IAED,OAAO;QACH,IAAI,EAAE,EAAE;QACR,QAAQ,EAAE;YACN,IAAI,EAAE,EAAE;YACR,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;SAC1C;KACJ,CAAA;AACL,CAAC;AAtBD,4DAsBC;AAED,SAAgB,sBAAsB,CAAC,MAAe;IAClD,IAAI,MAAM,KAAK,yCAA6B,EAAE;QAC1C,OAAO;YACH,SAAS,EAAE,IAAA,SAAI,GAAE;YACjB,IAAI,EAAE,wBAAY;SACrB,CAAA;KACJ;IAED,OAAO,SAAS,CAAA;AACpB,CAAC;AATD,wDASC"}

package/out/language-server/codeWhispererSecurityScanServer.js

@@ -35,21 +35,26 @@ const SecurityScanServerToken = (service) => ({ credentialsProvider, workspace,
         };
         try {
             if (!credentialsProvider.hasCredentials('bearer')) {
-                throw new Error('credentialsProvider does not have bearer token credentials');
+                throw new Error('Credentials provider does not have bearer token credentials');
             }
+            logging.log(`Parameters provided: ${JSON.stringify(params)}`);
             if (!params.arguments || params.arguments.length === 0) {
-                throw new Error(`Incorrect params provided. Params: ${params}`);
+                throw new Error(`Error: Invalid data.`);
             }
             const [arg] = params.arguments;
+            logging.log(`Arguments provided: ${JSON.stringify(arg)}`);
             const { ActiveFilePath: activeFilePath, ProjectPath: projectPath } = (0, utils_1.parseJson)(arg);
-            if (!activeFilePath || !projectPath) {
-                throw new Error(`Error: file path or project path not provided. Params: ${params}`);
+            if (!activeFilePath) {
+                throw new Error('Error: File to scan is missing.');
+            }
+            if (!projectPath) {
+                throw new Error('Error: Project is missing.');
             }
             const activeFilePathUri = (0, url_1.pathToFileURL)(activeFilePath).href;
             const document = await workspace.getTextDocument(activeFilePathUri);
             securityScanTelemetryEntry.codewhispererLanguage = (0, languageDetection_1.getSupportedLanguageId)(document);
             if (!document) {
-                throw new Error('Text document for given activeFilePath is undefined.');
+                throw new Error('Error: Text document for given file is missing.');
             }
             /**
              * Step 1: Generate context truncations
@@ -59,7 +64,7 @@ const SecurityScanServerToken = (service) => ({ credentialsProvider, workspace,
                 throw new Error(`"${document.languageId}" is not supported for security scan.`);
             }
             if (dependencyGraph.exceedsSizeLimit((await workspace.fs.getFileSize(activeFilePath)).size)) {
-                throw new Error(`Selected file larger than ${dependencyGraph.getReadableSizeLimit()}. Try a different file.`);
+                throw new Error(`Error: Selected file larger than ${dependencyGraph.getReadableSizeLimit()}. Try a different file.`);
             }
             const contextTruncationStartTime = perf_hooks_1.performance.now();
             const truncation = await dependencyGraph.generateTruncation(activeFilePath);
@@ -100,7 +105,7 @@ const SecurityScanServerToken = (service) => ({ credentialsProvider, workspace,
              */
             jobStatus = await scanHandler.pollScanJobStatus(scanJob.jobId);
             if (jobStatus === 'Failed') {
-                throw new Error('security scan job failed.');
+                throw new Error('Error: Security scan job failed.');
             }
             scanHandler.throwIfCancelled(token);
             /**