@aws/lsp-codewhisperer 0.0.32 → 0.0.33

package/out/language-server/agenticChat/tools/chatDb/util.js

@@ -0,0 +1,255 @@
+"use strict";
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FileSystemAdapter = exports.historyPath = exports.TabCollection = void 0;
+exports.messageToStreamingMessage = messageToStreamingMessage;
+exports.messageToChatMessage = messageToChatMessage;
+exports.chatMessageToMessage = chatMessageToMessage;
+exports.updateOrCreateConversation = updateOrCreateConversation;
+exports.groupTabsByDate = groupTabsByDate;
+const path = require("path");
+// Ported from https://github.com/aws/aws-toolkit-vscode/blob/master/packages/core/src/shared/db/chatDb/util.ts
+exports.TabCollection = 'tabs';
+exports.historyPath = path.join('.aws', 'amazonq', 'history');
+const TabTypes = [
+    'cwc',
+    'featuredev',
+    'gumby',
+    'review',
+    'testgen',
+    'doc',
+    'agentWalkthrough',
+    'welcome',
+    'unknown',
+];
+/**
+ * Converts Message to codewhisperer-streaming ChatMessage
+ */
+function messageToStreamingMessage(msg) {
+    return msg.type === 'answer'
+        ? {
+            assistantResponseMessage: {
+                messageId: msg.messageId,
+                content: msg.body,
+                toolUses: msg.toolUses || [],
+            },
+        }
+        : {
+            userInputMessage: {
+                content: msg.body,
+                userIntent: msg.userIntent,
+                origin: msg.origin || 'IDE',
+                userInputMessageContext: msg.userInputMessageContext || {},
+            },
+        };
+}
+/**
+ * Converts Message to LSP Protocol ChatMessage
+ */
+function messageToChatMessage(msg) {
+    const chatMessages = [
+        {
+            body: msg.body,
+            type: msg.type,
+            codeReference: msg.codeReference,
+            relatedContent: msg.relatedContent && msg.relatedContent?.content.length > 0 ? msg.relatedContent : undefined,
+        },
+    ];
+    // Check if there are any toolUses with explanations that should be displayed as directive messages
+    if (msg.toolUses && msg.toolUses.length > 0) {
+        for (const toolUse of msg.toolUses) {
+            if (toolUse.input && typeof toolUse.input === 'object') {
+                const input = toolUse.input;
+                if (input.explanation) {
+                    chatMessages.push({
+                        body: input.explanation,
+                        type: 'directive',
+                    });
+                }
+            }
+        }
+    }
+    return chatMessages;
+}
+/**
+ * Converts codewhisperer-streaming ChatMessage to Message
+ */
+function chatMessageToMessage(chatMessage) {
+    if ('userInputMessage' in chatMessage) {
+        const userInputMessage = chatMessage.userInputMessage;
+        return {
+            body: userInputMessage.content || '',
+            type: 'prompt',
+            userIntent: userInputMessage.userIntent,
+            origin: userInputMessage.origin || 'IDE',
+            userInputMessageContext: userInputMessage.userInputMessageContext || {},
+        };
+    }
+    else if ('assistantResponseMessage' in chatMessage) {
+        const assistantResponseMessage = chatMessage.assistantResponseMessage;
+        return {
+            body: assistantResponseMessage.content || '',
+            type: 'answer',
+            messageId: assistantResponseMessage.messageId,
+            toolUses: assistantResponseMessage.toolUses || [],
+        };
+    }
+    else {
+        // Default fallback for unexpected message format
+        return {
+            body: '',
+            type: 'prompt',
+        };
+    }
+}
+/**
+ *
+ * This adapter implements the LokiPersistenceAdapter interface for file system operations using web-compatible shared fs utils.
+ * It provides methods for loading, saving, and deleting databases, as well as ensuring
+ * the existence of the directory.
+ *
+ * Error Handling:
+ * - All methods use try-catch blocks to to prevent breaking the application
+ * - In case of errors, the callback functions are used to communicate the error state
+ *   without throwing exceptions.
+ *
+ */
+class FileSystemAdapter {
+    #directory;
+    #workspace;
+    constructor(workspace, directory) {
+        this.#directory = directory;
+        this.#workspace = workspace;
+    }
+    async ensureDirectory() {
+        await this.#workspace.fs.mkdir(this.#directory, { recursive: true });
+    }
+    async loadDatabase(dbname, callback) {
+        try {
+            await this.ensureDirectory();
+            const filename = path.join(this.#directory, dbname);
+            try {
+                const data = await this.#workspace.fs.readFile(filename, { encoding: 'utf8' });
+                callback(data);
+            }
+            catch (err) {
+                // File doesn't exist
+                callback(undefined);
+            }
+        }
+        catch (err) {
+            callback(err);
+        }
+    }
+    async saveDatabase(dbname, dbstring, callback) {
+        try {
+            await this.ensureDirectory();
+            const filename = path.join(this.#directory, dbname);
+            await this.#workspace.fs.writeFile(filename, dbstring, { mode: 0o600 });
+            callback(undefined);
+        }
+        catch (err) {
+            callback(err);
+        }
+    }
+    async deleteDatabase(dbname, callback) {
+        const filename = path.join(this.#directory, dbname);
+        try {
+            await this.#workspace.fs.rm(filename);
+            callback(undefined);
+        }
+        catch (err) {
+            callback(err);
+        }
+    }
+}
+exports.FileSystemAdapter = FileSystemAdapter;
+function updateOrCreateConversation(conversations, conversationId, newMessage, clientType) {
+    const existingConversation = conversations.find(conv => conv.conversationId === conversationId);
+    if (existingConversation) {
+        return conversations.map(conv => conv.conversationId === conversationId ? { ...conv, messages: [...conv.messages, newMessage] } : conv);
+    }
+    else {
+        return [
+            ...conversations,
+            {
+                conversationId,
+                clientType,
+                messages: [newMessage],
+            },
+        ];
+    }
+}
+function groupTabsByDate(tabs) {
+    const now = new Date();
+    const today = new Date(now.setHours(0, 0, 0, 0));
+    const yesterday = new Date(today);
+    yesterday.setDate(yesterday.getDate() - 1);
+    const lastWeek = new Date(today);
+    lastWeek.setDate(lastWeek.getDate() - 7);
+    const lastMonth = new Date(today);
+    lastMonth.setMonth(lastMonth.getMonth() - 1);
+    // Sort tabs by updatedAt in descending order
+    const sortedTabs = [...tabs]
+        .map(tab => ({ ...tab, updatedAt: new Date(tab.updatedAt) }))
+        .sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+    // Helper function to convert Tab to DetailedListItem
+    const tabToDetailedListItem = (tab) => ({
+        icon: getTabTypeIcon(tab.tabType),
+        // Show open tabs as bold (in markdown)
+        description: tab.isOpen ? `**${tab.title}**` : tab.title,
+        id: tab.historyId,
+    });
+    const tabGroups = [
+        {
+            name: 'Today',
+            tabs: sortedTabs.filter(tab => tab.updatedAt >= today),
+        },
+        {
+            name: 'Yesterday',
+            tabs: sortedTabs.filter(tab => tab.updatedAt >= yesterday && tab.updatedAt < today),
+        },
+        {
+            name: 'Last Week',
+            tabs: sortedTabs.filter(tab => tab.updatedAt >= lastWeek && tab.updatedAt < yesterday),
+        },
+        {
+            name: 'Last Month',
+            tabs: sortedTabs.filter(tab => tab.updatedAt >= lastMonth && tab.updatedAt < lastWeek),
+        },
+        {
+            name: 'Older',
+            tabs: sortedTabs.filter(tab => tab.updatedAt < lastMonth),
+        },
+    ];
+    // Convert to DetailedListItemGroup[] and filter out empty groups
+    return tabGroups
+        .filter(group => group.tabs.length > 0)
+        .map(group => ({
+        groupName: group.name,
+        icon: 'calendar',
+        items: group.tabs.map(tabToDetailedListItem),
+    }));
+}
+function getTabTypeIcon(tabType) {
+    switch (tabType) {
+        case 'cwc':
+            return 'chat';
+        case 'doc':
+            return 'file';
+        case 'review':
+            return 'bug';
+        case 'gumby':
+            return 'transform';
+        case 'testgen':
+            return 'check-list';
+        case 'featuredev':
+            return 'code-block';
+        default:
+            return 'chat';
+    }
+}
+//# sourceMappingURL=util.js.map

package/out/language-server/agenticChat/tools/chatDb/util.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"util.js","sourceRoot":"","sources":["../../../../../src/language-server/agenticChat/tools/chatDb/util.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AA0EH,8DAiBC;AAKD,oDA0BC;AAKD,oDAyBC;AAkED,gEAsBC;AAED,0CAsDC;AAtSD,6BAA4B;AAmB5B,+GAA+G;AAElG,QAAA,aAAa,GAAG,MAAM,CAAA;AAEtB,QAAA,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAA;AAElE,MAAM,QAAQ,GAAG;IACb,KAAK;IACL,YAAY;IACZ,OAAO;IACP,QAAQ;IACR,SAAS;IACT,KAAK;IACL,kBAAkB;IAClB,SAAS;IACT,SAAS;CACH,CAAA;AAkCV;;GAEG;AACH,SAAgB,yBAAyB,CAAC,GAAY;IAClD,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ;QACxB,CAAC,CAAC;YACI,wBAAwB,EAAE;gBACtB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,OAAO,EAAE,GAAG,CAAC,IAAI;gBACjB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;aAC/B;SACJ;QACH,CAAC,CAAC;YACI,gBAAgB,EAAE;gBACd,OAAO,EAAE,GAAG,CAAC,IAAI;gBACjB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,KAAK;gBAC3B,uBAAuB,EAAE,GAAG,CAAC,uBAAuB,IAAI,EAAE;aAC7D;SACJ,CAAA;AACX,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,GAAY;IAC7C,MAAM,YAAY,GAAkB;QAChC;YACI,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,cAAc,EACV,GAAG,CAAC,cAAc,IAAI,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;SACpG;KACJ,CAAA;IAED,mGAAmG;IACnG,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1C,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAY,CAAA;gBAClC,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;oBACpB,YAAY,CAAC,IAAI,CAAC;wBACd,IAAI,EAAE,KAAK,CAAC,WAAW;wBACvB,IAAI,EAAE,WAAW;qBACpB,CAAC,CAAA;gBACN,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,YAAY,CAAA;AACvB,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,WAA6B;IAC9D,IAAI,kBAAkB,IAAI,WAAW,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,WAAW,CAAC,gBAAoC,CAAA;QACzE,OAAO;YACH,IAAI,EAAE,gBAAgB,CAAC,OAAO,IAAI,EAAE;YACpC,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,gBAAgB,CAAC,UAAU;YACvC,MAAM,EAAE,gBAAgB,CAAC,MAAM,IAAI,KAAK;YACxC,uBAAuB,EAAE,gBAAgB,CAAC,uBAAuB,IAAI,EAAE;SAC1E,CAAA;IACL,CAAC;SAAM,IAAI,0BAA0B,IAAI,WAAW,EAAE,CAAC;QACnD,MAAM,wBAAwB,GAAG,WAAW,CAAC,wBAAoD,CAAA;QACjG,OAAO;YACH,IAAI,EAAE,wBAAwB,CAAC,OAAO,IAAI,EAAE;YAC5C,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,wBAAwB,CAAC,SAAS;YAC7C,QAAQ,EAAE,wBAAwB,CAAC,QAAQ,IAAI,EAAE;SACpD,CAAA;IACL,CAAC;SAAM,CAAC;QACJ,iDAAiD;QACjD,OAAO;YACH,IAAI,EAAE,EAAE;YACR,IAAI,EAAE,QAAQ;SACjB,CAAA;IACL,CAAC;AACL,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAa,iBAAiB;IAC1B,UAAU,CAAA;IACV,UAAU,CAAA;IACV,YAAY,SAAoB,EAAE,SAAiB;QAC/C,IAAI,CAAC,UAAU,GAAG,SAAS,CAAA;QAC3B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAA;IAC/B,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IACxE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,QAAoD;QACnF,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,eAAe,EAAE,CAAA;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAA;YAEnD,IAAI,CAAC;gBACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAA;gBAC9E,QAAQ,CAAC,IAAI,CAAC,CAAA;YAClB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACX,qBAAqB;gBACrB,QAAQ,CAAC,SAAS,CAAC,CAAA;YACvB,CAAC;QACL,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAChB,QAAQ,CAAC,GAAG,CAAC,CAAA;QACjB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,QAAgB,EAAE,QAA0C;QAC3F,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,eAAe,EAAE,CAAA;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAA;YAEnD,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;YACvE,QAAQ,CAAC,SAAS,CAAC,CAAA;QACvB,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAChB,QAAQ,CAAC,GAAG,CAAC,CAAA;QACjB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAc,EAAE,QAA0C;QAC3E,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAA;QACnD,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAA;YACrC,QAAQ,CAAC,SAAS,CAAC,CAAA;QACvB,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAChB,QAAQ,CAAC,GAAG,CAAC,CAAA;QACjB,CAAC;IACL,CAAC;CACJ;AAlDD,8CAkDC;AAED,SAAgB,0BAA0B,CACtC,aAA6B,EAC7B,cAAsB,EACtB,UAAmB,EACnB,UAAkB;IAElB,MAAM,oBAAoB,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,cAAc,KAAK,cAAc,CAAC,CAAA;IAE/F,IAAI,oBAAoB,EAAE,CAAC;QACvB,OAAO,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAC5B,IAAI,CAAC,cAAc,KAAK,cAAc,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CACxG,CAAA;IACL,CAAC;SAAM,CAAC;QACJ,OAAO;YACH,GAAG,aAAa;YAChB;gBACI,cAAc;gBACd,UAAU;gBACV,QAAQ,EAAE,CAAC,UAAU,CAAC;aACzB;SACJ,CAAA;IACL,CAAC;AACL,CAAC;AAED,SAAgB,eAAe,CAAC,IAAW;IACvC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAA;IACtB,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IAChD,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAA;IACjC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;IAC1C,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAA;IAChC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;IACxC,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAA;IACjC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAA;IAE5C,6CAA6C;IAC7C,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,CAAC;SACvB,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;SAC5D,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAA;IAElE,qDAAqD;IACrD,MAAM,qBAAqB,GAAG,CAAC,GAAQ,EAAoB,EAAE,CAAC,CAAC;QAC3D,IAAI,EAAE,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC;QACjC,uCAAuC;QACvC,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK;QACxD,EAAE,EAAE,GAAG,CAAC,SAAS;KACpB,CAAC,CAAA;IAEF,MAAM,SAAS,GAAG;QACd;YACI,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,SAAS,IAAI,KAAK,CAAC;SACzD;QACD;YACI,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,SAAS,IAAI,SAAS,IAAI,GAAG,CAAC,SAAS,GAAG,KAAK,CAAC;SACtF;QACD;YACI,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,SAAS,IAAI,QAAQ,IAAI,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC;SACzF;QACD;YACI,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,SAAS,IAAI,SAAS,IAAI,GAAG,CAAC,SAAS,GAAG,QAAQ,CAAC;SACzF;QACD;YACI,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC;SAC5D;KACJ,CAAA;IAED,iEAAiE;IACjE,OAAO,SAAS;SACX,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;SACtC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACX,SAAS,EAAE,KAAK,CAAC,IAAI;QACrB,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,qBAAqB,CAAC;KAC/C,CAAC,CAAC,CAAA;AACX,CAAC;AAED,SAAS,cAAc,CAAC,OAAgB;IACpC,QAAQ,OAAO,EAAE,CAAC;QACd,KAAK,KAAK;YACN,OAAO,MAAM,CAAA;QACjB,KAAK,KAAK;YACN,OAAO,MAAM,CAAA;QACjB,KAAK,QAAQ;YACT,OAAO,KAAK,CAAA;QAChB,KAAK,OAAO;YACR,OAAO,WAAW,CAAA;QACtB,KAAK,SAAS;YACV,OAAO,YAAY,CAAA;QACvB,KAAK,YAAY;YACb,OAAO,YAAY,CAAA;QACvB;YACI,OAAO,MAAM,CAAA;IACrB,CAAC;AACL,CAAC"}

package/out/language-server/agenticChat/tools/codeSearch.d.ts

@@ -0,0 +1,43 @@
+import { InvokeOutput } from './toolShared';
+import { Features } from '@aws/language-server-runtimes/server-interface/server';
+import { RelevantTextDocument } from '@amzn/codewhisperer-streaming';
+import { LineInfo } from '../context/agenticChatTriggerContext';
+export interface CodeSearchParams {
+    query: string;
+}
+export type CodeSearchOutput = RelevantTextDocument & LineInfo;
+export declare class CodeSearch {
+    private readonly logging;
+    private readonly workspace;
+    private readonly lsp;
+    constructor(features: Pick<Features, 'logging' | 'workspace' | 'lsp'>);
+    validate(params: CodeSearchParams): Promise<void>;
+    queueDescription(params: CodeSearchParams, updates: WritableStream, requiresAcceptance: boolean): Promise<void>;
+    invoke(params: CodeSearchParams): Promise<InvokeOutput>;
+    private getOrSetSearchPath;
+    private executeCodeSearch;
+    /**
+     * Parses chunks from vector index search into CodeSearchOutput format
+     * Based on the queryRelevantDocuments method pattern
+     */
+    private parseChunksToCodeSearchOutput;
+    private createOutput;
+    getSpec(): {
+        readonly name: "codeSearch";
+        readonly description: "Find snippets of code from the codebase most relevant to the search query.\nThis is a semantic search tool, so the query should ask for something semantically matching what is needed.\nUnless there is a clear reason to use your own search query, please just reuse the user's exact query with their wording.\nTheir exact wording/phrasing can often be helpful for the semantic search query. Keeping the same exact question format can also be helpful.";
+        readonly inputSchema: {
+            readonly type: "object";
+            readonly properties: {
+                readonly query: {
+                    readonly type: "string";
+                    readonly description: "The search query to find relevant code.";
+                };
+                readonly explanation: {
+                    readonly type: "string";
+                    readonly description: "One sentence explanation as to why this tool is being used, and how it contributes to the goal";
+                };
+            };
+            readonly required: readonly ["query"];
+        };
+    };
+}

package/out/language-server/agenticChat/tools/codeSearch.js

@@ -0,0 +1,168 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeSearch = void 0;
+const toolShared_1 = require("./toolShared");
+const workspaceUtils_1 = require("@aws/lsp-core/out/util/workspaceUtils");
+const localProjectContextController_1 = require("../../../shared/localProjectContextController");
+const path = require("path");
+class CodeSearch {
+    logging;
+    workspace;
+    lsp;
+    constructor(features) {
+        this.logging = features.logging;
+        this.workspace = features.workspace;
+        this.lsp = features.lsp;
+    }
+    async validate(params) {
+        if (!params.query || params.query.trim().length === 0) {
+            throw new Error('Code search query cannot be empty.');
+        }
+        const searchPath = this.getOrSetSearchPath();
+        if (searchPath) {
+            await (0, toolShared_1.validatePath)(searchPath, this.workspace.fs.exists);
+        }
+    }
+    async queueDescription(params, updates, requiresAcceptance) {
+        const writer = updates.getWriter();
+        const closeWriter = async (w) => {
+            await w.close();
+            w.releaseLock();
+        };
+        if (!requiresAcceptance) {
+            await writer.write('');
+            await closeWriter(writer);
+            return;
+        }
+        const path = this.getOrSetSearchPath();
+        await writer.write(`Performing code search for "${params.query}" in ${path}`);
+        await closeWriter(writer);
+    }
+    async invoke(params) {
+        const path = this.getOrSetSearchPath();
+        try {
+            const results = await this.executeCodeSearch(params.query);
+            return this.createOutput(!results || results.length === 0 ? 'No code matches found for code search.' : results);
+        }
+        catch (error) {
+            this.logging.error(`Failed to perform code search for "${params.query}" in workspace "${path}": ${error.message || error}`);
+            throw new Error(`Failed to perform code search for "${params.query}" in workspace"${path}": ${error.message || error}`);
+        }
+    }
+    getOrSetSearchPath(path) {
+        let searchPath = '';
+        if (path && path.trim().length !== 0) {
+            searchPath = path;
+        }
+        else {
+            // Handle optional path parameter
+            // Use current workspace folder as default if path is not provided
+            const workspaceFolders = (0, workspaceUtils_1.getWorkspaceFolderPaths)(this.lsp);
+            if (workspaceFolders && workspaceFolders.length !== 0) {
+                this.logging.debug(`Using default workspace folder: ${workspaceFolders[0]}`);
+                searchPath = workspaceFolders[0];
+            }
+        }
+        return searchPath;
+    }
+    async executeCodeSearch(query) {
+        this.logging.info(`Executing code search for "${query}" in "${path}"`);
+        const localProjectContextController = await localProjectContextController_1.LocalProjectContextController.getInstance();
+        if (!localProjectContextController.isEnabled) {
+            this.logging.warn(`Error during code search: local project context controller is disabled`);
+            throw new Error(`Error during code search: Amazon Q Workspace Index disabled,
+                 please update the configuration to enable Amazon Q workspace Index`);
+        }
+        try {
+            // TODO: we need to handle the validation of workspace indexing status once localProjectContextController support
+            // check the indexing status.
+            // Use the localProjectContextController to query the vector index
+            const searchResults = await localProjectContextController.queryVectorIndex({ query: query });
+            const sanitizedSearchResults = this.parseChunksToCodeSearchOutput(searchResults);
+            this.logging.info(`Code searched succeed with num of results: "${sanitizedSearchResults.length}"`);
+            return sanitizedSearchResults;
+        }
+        catch (error) {
+            this.logging.error(`Error during code search: ${error.message || error}`);
+            throw error;
+        }
+    }
+    /**
+     * Parses chunks from vector index search into CodeSearchOutput format
+     * Based on the queryRelevantDocuments method pattern
+     */
+    parseChunksToCodeSearchOutput(chunks) {
+        const codeSearchResults = [];
+        if (!chunks) {
+            return codeSearchResults;
+        }
+        for (const chunk of chunks) {
+            // Extract content and context
+            const text = chunk.content || '';
+            const relativeFilePath = chunk.relativePath ?? path.basename(chunk.filePath);
+            // Extract line information
+            const startLine = chunk.startLine ?? -1;
+            const endLine = chunk.endLine ?? -1;
+            // Create the base search result
+            const baseSearchResult = {
+                text,
+                relativeFilePath,
+                startLine,
+                endLine,
+            };
+            // Add programming language information if available
+            if (chunk.programmingLanguage && chunk.programmingLanguage !== 'unknown') {
+                codeSearchResults.push({
+                    ...baseSearchResult,
+                    programmingLanguage: {
+                        languageName: chunk.programmingLanguage,
+                    },
+                });
+            }
+            else {
+                codeSearchResults.push(baseSearchResult);
+            }
+        }
+        return codeSearchResults;
+    }
+    createOutput(content) {
+        if (typeof content === 'string') {
+            return {
+                output: {
+                    kind: 'text',
+                    content: content,
+                },
+            };
+        }
+        else {
+            return {
+                output: {
+                    kind: 'json',
+                    content: content,
+                },
+            };
+        }
+    }
+    getSpec() {
+        return {
+            name: 'codeSearch',
+            description: "Find snippets of code from the codebase most relevant to the search query.\nThis is a semantic search tool, so the query should ask for something semantically matching what is needed.\nUnless there is a clear reason to use your own search query, please just reuse the user's exact query with their wording.\nTheir exact wording/phrasing can often be helpful for the semantic search query. Keeping the same exact question format can also be helpful.",
+            inputSchema: {
+                type: 'object',
+                properties: {
+                    query: {
+                        type: 'string',
+                        description: 'The search query to find relevant code.',
+                    },
+                    explanation: {
+                        type: 'string',
+                        description: 'One sentence explanation as to why this tool is being used, and how it contributes to the goal',
+                    },
+                },
+                required: ['query'],
+            },
+        };
+    }
+}
+exports.CodeSearch = CodeSearch;
+//# sourceMappingURL=codeSearch.js.map

package/out/language-server/agenticChat/tools/codeSearch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codeSearch.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/tools/codeSearch.ts"],"names":[],"mappings":";;;AAAA,6CAAoG;AAEpG,0EAA+E;AAC/E,iGAA6F;AAI7F,6BAA6B;AAQ7B,MAAa,UAAU;IACF,OAAO,CAAqB;IAC5B,SAAS,CAAuB;IAChC,GAAG,CAAiB;IACrC,YAAY,QAAyD;QACjE,IAAI,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAA;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAA;QACnC,IAAI,CAAC,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAA;IAC3B,CAAC;IAEM,KAAK,CAAC,QAAQ,CAAC,MAAwB;QAC1C,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAA;QACzD,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAE5C,IAAI,UAAU,EAAE,CAAC;YACb,MAAM,IAAA,yBAAY,EAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,CAAA;QAC5D,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,MAAwB,EAAE,OAAuB,EAAE,kBAA2B;QACxG,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAA;QAClC,MAAM,WAAW,GAAG,KAAK,EAAE,CAA8B,EAAE,EAAE;YACzD,MAAM,CAAC,CAAC,KAAK,EAAE,CAAA;YACf,CAAC,CAAC,WAAW,EAAE,CAAA;QACnB,CAAC,CAAA;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACtB,MAAM,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACtB,MAAM,WAAW,CAAC,MAAM,CAAC,CAAA;YACzB,OAAM;QACV,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;QACtC,MAAM,MAAM,CAAC,KAAK,CAAC,+BAA+B,MAAM,CAAC,KAAK,QAAQ,IAAI,EAAE,CAAC,CAAA;QAC7E,MAAM,WAAW,CAAC,MAAM,CAAC,CAAA;IAC7B,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,MAAwB;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAEtC,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YAC1D,OAAO,IAAI,CAAC,YAAY,CACpB,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,wCAAwC,CAAC,CAAC,CAAC,OAAO,CACxF,CAAA;QACL,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,CAAC,KAAK,CACd,sCAAsC,MAAM,CAAC,KAAK,mBAAmB,IAAI,MAAM,KAAK,CAAC,OAAO,IAAI,KAAK,EAAE,CAC1G,CAAA;YACD,MAAM,IAAI,KAAK,CACX,sCAAsC,MAAM,CAAC,KAAK,kBAAkB,IAAI,MAAM,KAAK,CAAC,OAAO,IAAI,KAAK,EAAE,CACzG,CAAA;QACL,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,IAAa;QACpC,IAAI,UAAU,GAAG,EAAE,CAAA;QACnB,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,UAAU,GAAG,IAAI,CAAA;QACrB,CAAC;aAAM,CAAC;YACJ,iCAAiC;YACjC,kEAAkE;YAClE,MAAM,gBAAgB,GAAG,IAAA,wCAAuB,EAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YAC1D,IAAI,gBAAgB,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mCAAmC,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;gBAC5E,UAAU,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAA;YACpC,CAAC;QACL,CAAC;QACD,OAAO,UAAU,CAAA;IACrB,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,KAAa;QACzC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,8BAA8B,KAAK,SAAS,IAAI,GAAG,CAAC,CAAA;QACtE,MAAM,6BAA6B,GAAG,MAAM,6DAA6B,CAAC,WAAW,EAAE,CAAA;QAEvF,IAAI,CAAC,6BAA6B,CAAC,SAAS,EAAE,CAAC;YAC3C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,wEAAwE,CAAC,CAAA;YAC3F,MAAM,IAAI,KAAK,CAAC;oFACwD,CAAC,CAAA;QAC7E,CAAC;QACD,IAAI,CAAC;YACD,iHAAiH;YACjH,6BAA6B;YAC7B,kEAAkE;YAClE,MAAM,aAAa,GAAG,MAAM,6BAA6B,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;YAC5F,MAAM,sBAAsB,GAAG,IAAI,CAAC,6BAA6B,CAAC,aAAa,CAAC,CAAA;YAChF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,+CAA+C,sBAAsB,CAAC,MAAM,GAAG,CAAC,CAAA;YAClG,OAAO,sBAAsB,CAAA;QACjC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,6BAA6B,KAAK,CAAC,OAAO,IAAI,KAAK,EAAE,CAAC,CAAA;YACzE,MAAM,KAAK,CAAA;QACf,CAAC;IACL,CAAC;IAED;;;OAGG;IACK,6BAA6B,CAAC,MAAe;QACjD,MAAM,iBAAiB,GAAuB,EAAE,CAAA;QAChD,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,OAAO,iBAAiB,CAAA;QAC5B,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,8BAA8B;YAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAA;YAChC,MAAM,gBAAgB,GAAG,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;YAE5E,2BAA2B;YAC3B,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC,CAAA;YACvC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC,CAAA;YAEnC,gCAAgC;YAChC,MAAM,gBAAgB,GAAG;gBACrB,IAAI;gBACJ,gBAAgB;gBAChB,SAAS;gBACT,OAAO;aACV,CAAA;YAED,oDAAoD;YACpD,IAAI,KAAK,CAAC,mBAAmB,IAAI,KAAK,CAAC,mBAAmB,KAAK,SAAS,EAAE,CAAC;gBACvE,iBAAiB,CAAC,IAAI,CAAC;oBACnB,GAAG,gBAAgB;oBACnB,mBAAmB,EAAE;wBACjB,YAAY,EAAE,KAAK,CAAC,mBAAmB;qBAC1C;iBACJ,CAAC,CAAA;YACN,CAAC;iBAAM,CAAC;gBACJ,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;YAC5C,CAAC;QACL,CAAC;QAED,OAAO,iBAAiB,CAAA;IAC5B,CAAC;IAEO,YAAY,CAAC,OAAuB;QACxC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO;gBACH,MAAM,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;iBACnB;aACJ,CAAA;QACL,CAAC;aAAM,CAAC;YACJ,OAAO;gBACH,MAAM,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;iBACnB;aACJ,CAAA;QACL,CAAC;IACL,CAAC;IAEM,OAAO;QACV,OAAO;YACH,IAAI,EAAE,YAAY;YAClB,WAAW,EACP,kcAAkc;YACtc,WAAW,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACR,KAAK,EAAE;wBACH,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,yCAAyC;qBACzD;oBACD,WAAW,EAAE;wBACT,IAAI,EAAE,QAAQ;wBACd,WAAW,EACP,gGAAgG;qBACvG;iBACJ;gBACD,QAAQ,EAAE,CAAC,OAAO,CAAC;aACtB;SACK,CAAA;IACd,CAAC;CACJ;AAlLD,gCAkLC"}

package/out/language-server/agenticChat/tools/executeBash.d.ts

@@ -0,0 +1,70 @@
+import { CommandValidation, ExplanatoryParams, InvokeOutput } from './toolShared';
+import { CancellationToken } from 'vscode-languageserver';
+import { Features } from '../../types';
+export declare enum CommandCategory {
+    ReadOnly = 0,
+    Mutate = 1,
+    Destructive = 2
+}
+export declare const splitOperators: Set<string>;
+export declare const splitOperatorsArray: string[];
+export declare const commandCategories: Map<string, CommandCategory>;
+export declare const maxToolResponseSize: number;
+export declare const lineCount: number;
+export declare const destructiveCommandWarningMessage = "WARNING: Potentially destructive command detected:\n\n";
+export declare const mutateCommandWarningMessage = "Mutation command:\n\n";
+export declare const outOfWorkspaceWarningmessage = "Execution out of workspace scope:\n\n";
+/**
+ * Parameters for executing a command on the system shell.
+ * Works cross-platform: uses cmd.exe on Windows and bash on Unix-like systems.
+ */
+export interface ExecuteBashParams extends ExplanatoryParams {
+    command: string;
+    cwd?: string;
+}
+/**
+ * Output from executing a command on the system shell.
+ * Format is consistent across platforms (Windows, macOS, Linux).
+ */
+export interface ExecuteBashOutput {
+    exitStatus: string;
+    stdout: string;
+    stderr: string;
+}
+export declare class ExecuteBash {
+    private childProcess?;
+    private readonly logging;
+    private readonly lsp;
+    constructor(features: Pick<Features, 'logging' | 'lsp'> & Partial<Features>);
+    validate(command: string): Promise<void>;
+    private static handleTimestampedChunk;
+    requiresAcceptance(params: ExecuteBashParams, approvedPaths?: Set<string>): Promise<CommandValidation>;
+    private looksLikePath;
+    invoke(params: ExecuteBashParams, cancellationToken?: CancellationToken, updates?: WritableStream): Promise<InvokeOutput>;
+    private static handleChunk;
+    private static truncateSafelyWithSuffix;
+    private static whichCommand;
+    queueDescription(command: string, updates: WritableStream): Promise<void>;
+    getSpec(): {
+        readonly name: "executeBash";
+        readonly description: "Execute the specified command on the system shell (bash on Unix/Linux/macOS, cmd.exe on Windows).";
+        readonly inputSchema: {
+            readonly type: "object";
+            readonly properties: {
+                readonly explanation: {
+                    readonly type: "string";
+                    readonly description: "One sentence explanation as to why this tool is being used, and how it contributes to the goal.";
+                };
+                readonly command: {
+                    readonly type: "string";
+                    readonly description: "Command to execute on the system shell. On Windows, this will run in cmd.exe; on Unix-like systems, this will run in bash.";
+                };
+                readonly cwd: {
+                    readonly type: "string";
+                    readonly description: "Parameter to set the current working directory for the command execution.";
+                };
+            };
+            readonly required: readonly ["command", "cwd"];
+        };
+    };
+}