@aws/lsp-codewhisperer 0.0.6 → 0.0.9

package/out/language-server/chat/contexts/documentContext.js

@@ -10,21 +10,30 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _DocumentContextExtractor_characterLimits;
+var _DocumentContextExtractor_characterLimits, _DocumentContextExtractor_logger, _DocumentContextExtractor_documentSymbolExtractor;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DocumentContextExtractor = void 0;
 const languageDetection_1 = require("../../languageDetection");
+const documentFqnExtractor_1 = require("./documentFqnExtractor");
 const utils_1 = require("./utils");
 class DocumentContextExtractor {
-    constructor(characterLimits = DocumentContextExtractor.DEFAULT_CHARACTER_LIMIT) {
+    constructor(config) {
         _DocumentContextExtractor_characterLimits.set(this, void 0);
-        __classPrivateFieldSet(this, _DocumentContextExtractor_characterLimits, characterLimits, "f");
+        _DocumentContextExtractor_logger.set(this, void 0);
+        _DocumentContextExtractor_documentSymbolExtractor.set(this, void 0);
+        const { characterLimits, ...fqnConfig } = config ?? {};
+        __classPrivateFieldSet(this, _DocumentContextExtractor_logger, config?.logger, "f");
+        __classPrivateFieldSet(this, _DocumentContextExtractor_characterLimits, characterLimits ?? DocumentContextExtractor.DEFAULT_CHARACTER_LIMIT, "f");
+        __classPrivateFieldSet(this, _DocumentContextExtractor_documentSymbolExtractor, new documentFqnExtractor_1.DocumentFqnExtractor(fqnConfig), "f");
+    }
+    dispose() {
+        __classPrivateFieldGet(this, _DocumentContextExtractor_documentSymbolExtractor, "f").dispose();
     }
     /**
      * From the given the cursor state, we want to give Q context up to the characters limit
      * on both sides of the cursor.
      */
-    async extractEditorState(document, cursorState) {
+    async extractDocumentContext(document, cursorState) {
         const targetRange = 'position' in cursorState
             ? {
                 start: cursorState.position,
@@ -33,27 +42,27 @@ class DocumentContextExtractor {
             : cursorState.range;
         const codeBlockRange = (0, utils_1.getExtendedCodeBlockRange)(document, targetRange, __classPrivateFieldGet(this, _DocumentContextExtractor_characterLimits, "f"));
         const rangeWithinCodeBlock = (0, utils_1.getSelectionWithinExtendedRange)(targetRange, codeBlockRange);
-        return {
-            document: await this.extractDocumentContext(document, codeBlockRange),
-            cursorState: rangeWithinCodeBlock ? { range: rangeWithinCodeBlock } : undefined,
-        };
-    }
-    /**
-     * Extract document context from the given range inside a document
-     */
-    async extractDocumentContext(document, codeBlockRange) {
-        const text = document.getText(codeBlockRange);
         const languageId = (0, languageDetection_1.getLanguageId)(document);
-        const relativeFilePath = document.uri;
+        let documentSymbols = [];
+        try {
+            // best effort to extract symbols
+            documentSymbols = await __classPrivateFieldGet(this, _DocumentContextExtractor_documentSymbolExtractor, "f").extractDocumentSymbols(document, codeBlockRange, languageId);
+        }
+        catch (e) {
+            __classPrivateFieldGet(this, _DocumentContextExtractor_logger, "f")?.log(`Error extracting document symbols but continuing on. ${e instanceof Error ? e.message : 'Unknown error'}`);
+        }
         return {
-            text,
+            cursorState: rangeWithinCodeBlock ? { range: rangeWithinCodeBlock } : undefined,
+            text: document.getText(codeBlockRange),
             programmingLanguage: languageId ? { languageName: languageId } : undefined,
-            relativeFilePath,
-            documentSymbols: [],
+            relativeFilePath: document.uri,
+            documentSymbols,
+            hasCodeSnippet: Boolean(rangeWithinCodeBlock),
+            totalEditorCharacters: document.getText().length,
         };
     }
 }
 exports.DocumentContextExtractor = DocumentContextExtractor;
-_DocumentContextExtractor_characterLimits = new WeakMap();
+_DocumentContextExtractor_characterLimits = new WeakMap(), _DocumentContextExtractor_logger = new WeakMap(), _DocumentContextExtractor_documentSymbolExtractor = new WeakMap();
 DocumentContextExtractor.DEFAULT_CHARACTER_LIMIT = 9000;
 //# sourceMappingURL=documentContext.js.map

package/out/language-server/chat/contexts/documentContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"documentContext.js","sourceRoot":"","sources":["../../../../src/language-server/chat/contexts/documentContext.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAGA,+DAAuD;AACvD,mCAAoF;AAEpF,MAAa,wBAAwB;IAKjC,YAAY,kBAA0B,wBAAwB,CAAC,uBAAuB;QAFtF,4DAAwB;QAGpB,uBAAA,IAAI,6CAAoB,eAAe,MAAA,CAAA;IAC3C,CAAC;IACD;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,QAAsB,EAAE,WAAwB;QAC5E,MAAM,WAAW,GACb,UAAU,IAAI,WAAW;YACrB,CAAC,CAAC;gBACI,KAAK,EAAE,WAAW,CAAC,QAAQ;gBAC3B,GAAG,EAAE,WAAW,CAAC,QAAQ;aAC5B;YACH,CAAC,CAAC,WAAW,CAAC,KAAK,CAAA;QAE3B,MAAM,cAAc,GAAG,IAAA,iCAAyB,EAAC,QAAQ,EAAE,WAAW,EAAE,uBAAA,IAAI,iDAAiB,CAAC,CAAA;QAE9F,MAAM,oBAAoB,GAAG,IAAA,uCAA+B,EAAC,WAAW,EAAE,cAAc,CAAC,CAAA;QAEzF,OAAO;YACH,QAAQ,EAAE,MAAM,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,cAAc,CAAC;YACrE,WAAW,EAAE,oBAAoB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC,SAAS;SAClF,CAAA;IACL,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,sBAAsB,CAC/B,QAAsB,EACtB,cAAqB;QAErB,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;QAC7C,MAAM,UAAU,GAAG,IAAA,iCAAa,EAAC,QAAQ,CAAC,CAAA;QAC1C,MAAM,gBAAgB,GAAG,QAAQ,CAAC,GAAG,CAAA;QAErC,OAAO;YACH,IAAI;YACJ,mBAAmB,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS;YAC1E,gBAAgB;YAChB,eAAe,EAAE,EAAE;SACtB,CAAA;IACL,CAAC;;AAhDL,4DAiDC;;AAhD2B,gDAAuB,GAAG,IAAI,AAAP,CAAO"}
+{"version":3,"file":"documentContext.js","sourceRoot":"","sources":["../../../../src/language-server/chat/contexts/documentContext.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAGA,+DAAuD;AAEvD,iEAAyF;AACzF,mCAAoF;AAcpF,MAAa,wBAAwB;IAOjC,YAAY,MAAuC;QAJnD,4DAAwB;QACxB,mDAA6B;QAC7B,oEAA8C;QAG1C,MAAM,EAAE,eAAe,EAAE,GAAG,SAAS,EAAE,GAAG,MAAM,IAAI,EAAE,CAAA;QAEtD,uBAAA,IAAI,oCAAW,MAAM,EAAE,MAAM,MAAA,CAAA;QAC7B,uBAAA,IAAI,6CAAoB,eAAe,IAAI,wBAAwB,CAAC,uBAAuB,MAAA,CAAA;QAC3F,uBAAA,IAAI,qDAA4B,IAAI,2CAAoB,CAAC,SAAS,CAAC,MAAA,CAAA;IACvE,CAAC;IAEM,OAAO;QACV,uBAAA,IAAI,yDAAyB,CAAC,OAAO,EAAE,CAAA;IAC3C,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,sBAAsB,CAAC,QAAsB,EAAE,WAAwB;QAChF,MAAM,WAAW,GACb,UAAU,IAAI,WAAW;YACrB,CAAC,CAAC;gBACI,KAAK,EAAE,WAAW,CAAC,QAAQ;gBAC3B,GAAG,EAAE,WAAW,CAAC,QAAQ;aAC5B;YACH,CAAC,CAAC,WAAW,CAAC,KAAK,CAAA;QAE3B,MAAM,cAAc,GAAG,IAAA,iCAAyB,EAAC,QAAQ,EAAE,WAAW,EAAE,uBAAA,IAAI,iDAAiB,CAAC,CAAA;QAE9F,MAAM,oBAAoB,GAAG,IAAA,uCAA+B,EAAC,WAAW,EAAE,cAAc,CAAC,CAAA;QAEzF,MAAM,UAAU,GAAG,IAAA,iCAAa,EAAC,QAAQ,CAAC,CAAA;QAE1C,IAAI,eAAe,GAAqB,EAAE,CAAA;QAE1C,IAAI;YACA,iCAAiC;YACjC,eAAe,GAAG,MAAM,uBAAA,IAAI,yDAAyB,CAAC,sBAAsB,CACxE,QAAQ,EACR,cAAc,EACd,UAAU,CACb,CAAA;SACJ;QAAC,OAAO,CAAC,EAAE;YACR,uBAAA,IAAI,wCAAQ,EAAE,GAAG,CACb,wDAAwD,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAC7G,CAAA;SACJ;QAED,OAAO;YACH,WAAW,EAAE,oBAAoB,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC,SAAS;YAC/E,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC;YACtC,mBAAmB,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS;YAC1E,gBAAgB,EAAE,QAAQ,CAAC,GAAG;YAC9B,eAAe;YACf,cAAc,EAAE,OAAO,CAAC,oBAAoB,CAAC;YAC7C,qBAAqB,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,MAAM;SACnD,CAAA;IACL,CAAC;;AA9DL,4DA+DC;;AA9D2B,gDAAuB,GAAG,IAAI,CAAA"}

package/out/language-server/chat/contexts/documentContext.test.js

@@ -1,23 +1,31 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const assert = require("assert");
+const ts_sinon_1 = require("ts-sinon");
 const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
 const documentContext_1 = require("./documentContext");
+const documentFqnExtractor_1 = require("./documentFqnExtractor");
 describe('DocumentContext', () => {
     const mockTypescriptCodeBlock = `function test() {
     console.log('test')
 }`;
     const mockTSDocument = vscode_languageserver_textdocument_1.TextDocument.create('file://test.ts', 'typescript', 1, mockTypescriptCodeBlock);
+    beforeEach(() => {
+        ts_sinon_1.default.stub(documentFqnExtractor_1.DocumentFqnExtractor.prototype, 'extractDocumentSymbols').resolves([]);
+    });
+    afterEach(() => {
+        ts_sinon_1.default.restore();
+    });
     describe('documentContextExtractor.extractEditorState', () => {
         it('extracts editor state for range selection', async () => {
-            const documentContextExtractor = new documentContext_1.DocumentContextExtractor(19);
+            const documentContextExtractor = new documentContext_1.DocumentContextExtractor({ characterLimits: 19 });
             const expected = {
-                document: {
-                    programmingLanguage: { languageName: 'typescript' },
-                    relativeFilePath: 'file://test.ts',
-                    documentSymbols: [],
-                    text: "console.log('test')",
-                },
+                programmingLanguage: { languageName: 'typescript' },
+                relativeFilePath: 'file://test.ts',
+                documentSymbols: [],
+                text: "console.log('test')",
+                hasCodeSnippet: true,
+                totalEditorCharacters: mockTypescriptCodeBlock.length,
                 cursorState: {
                     range: {
                         start: {
@@ -31,7 +39,7 @@ describe('DocumentContext', () => {
                     },
                 },
             };
-            const result = await documentContextExtractor.extractEditorState(mockTSDocument, {
+            const result = await documentContextExtractor.extractDocumentContext(mockTSDocument, {
                 // highlighting "log"
                 range: {
                     start: {
@@ -47,14 +55,14 @@ describe('DocumentContext', () => {
             assert.deepStrictEqual(result, expected);
         });
         it('extracts editor state for collapsed position', async () => {
-            const documentContextExtractor = new documentContext_1.DocumentContextExtractor(19);
+            const documentContextExtractor = new documentContext_1.DocumentContextExtractor({ characterLimits: 19 });
             const expected = {
-                document: {
-                    programmingLanguage: { languageName: 'typescript' },
-                    relativeFilePath: 'file://test.ts',
-                    documentSymbols: [],
-                    text: "console.log('test')",
-                },
+                programmingLanguage: { languageName: 'typescript' },
+                relativeFilePath: 'file://test.ts',
+                documentSymbols: [],
+                text: "console.log('test')",
+                hasCodeSnippet: true,
+                totalEditorCharacters: mockTypescriptCodeBlock.length,
                 cursorState: {
                     range: {
                         start: {
@@ -68,7 +76,7 @@ describe('DocumentContext', () => {
                     },
                 },
             };
-            const result = await documentContextExtractor.extractEditorState(mockTSDocument, {
+            const result = await documentContextExtractor.extractDocumentContext(mockTSDocument, {
                 // highlighting "o" in "log"
                 range: {
                     start: {
@@ -84,17 +92,17 @@ describe('DocumentContext', () => {
             assert.deepStrictEqual(result, expected);
         });
         it('returns undefined cursorState if the end position was collapsed', async () => {
-            const documentContextExtractor = new documentContext_1.DocumentContextExtractor(0);
+            const documentContextExtractor = new documentContext_1.DocumentContextExtractor({ characterLimits: 0 });
             const expected = {
-                document: {
-                    programmingLanguage: { languageName: 'typescript' },
-                    relativeFilePath: 'file://test.ts',
-                    documentSymbols: [],
-                    text: '',
-                },
+                programmingLanguage: { languageName: 'typescript' },
+                relativeFilePath: 'file://test.ts',
+                documentSymbols: [],
+                text: '',
+                hasCodeSnippet: false,
+                totalEditorCharacters: mockTypescriptCodeBlock.length,
                 cursorState: undefined,
             };
-            const result = await documentContextExtractor.extractEditorState(mockTSDocument, {
+            const result = await documentContextExtractor.extractDocumentContext(mockTSDocument, {
                 range: {
                     start: {
                         line: 1,
@@ -109,39 +117,33 @@ describe('DocumentContext', () => {
             assert.deepStrictEqual(result, expected);
         });
     });
-    describe('extractDocumentContext', () => {
-        it('extract document context with the code block range correctly', async () => {
-            const documentContextExtractor = new documentContext_1.DocumentContextExtractor();
-            const expectedResult = {
-                programmingLanguage: { languageName: 'typescript' },
-                relativeFilePath: 'file://test.ts',
-                documentSymbols: [],
-                text: "console.log('test')",
-            };
-            const result = await documentContextExtractor.extractDocumentContext(mockTSDocument, {
-                start: { line: 1, character: 4 },
-                end: { line: 1, character: 23 },
-            });
-            assert.deepStrictEqual(result, expectedResult);
-        });
-        it('handles other languages correctly', async () => {
-            const documentContextExtractor = new documentContext_1.DocumentContextExtractor();
-            const mockGoCodeBLock = `func main() {
+    it('handles other languages correctly', async () => {
+        const documentContextExtractor = new documentContext_1.DocumentContextExtractor({ characterLimits: 19 });
+        const mockGoCodeBLock = `func main() {
     fmt.Println("test")
 }`;
-            const mockDocument = vscode_languageserver_textdocument_1.TextDocument.create('file://test.go', 'go', 1, mockGoCodeBLock);
-            const expectedResult = {
-                programmingLanguage: { languageName: 'go' },
-                relativeFilePath: 'file://test.go',
-                documentSymbols: [],
-                text: 'fmt.Println("test")',
-            };
-            const result = await documentContextExtractor.extractDocumentContext(mockDocument, {
+        const mockDocument = vscode_languageserver_textdocument_1.TextDocument.create('file://test.go', 'go', 1, mockGoCodeBLock);
+        const expectedResult = {
+            programmingLanguage: { languageName: 'go' },
+            relativeFilePath: 'file://test.go',
+            documentSymbols: [],
+            text: 'fmt.Println("test")',
+            totalEditorCharacters: mockGoCodeBLock.length,
+            hasCodeSnippet: true,
+            cursorState: {
+                range: {
+                    start: { line: 0, character: 0 },
+                    end: { line: 0, character: 19 },
+                },
+            },
+        };
+        const result = await documentContextExtractor.extractDocumentContext(mockDocument, {
+            range: {
                 start: { line: 1, character: 4 },
                 end: { line: 1, character: 23 },
-            });
-            assert.deepStrictEqual(result, expectedResult);
+            },
         });
+        assert.deepStrictEqual(result, expectedResult);
     });
 });
 //# sourceMappingURL=documentContext.test.js.map

package/out/language-server/chat/contexts/documentContext.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"documentContext.test.js","sourceRoot":"","sources":["../../../../src/language-server/chat/contexts/documentContext.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,2FAAiE;AACjE,uDAA4D;AAE5D,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,MAAM,uBAAuB,GAAG;;EAElC,CAAA;IACE,MAAM,cAAc,GAAG,iDAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,YAAY,EAAE,CAAC,EAAE,uBAAuB,CAAC,CAAA;IAEtG,QAAQ,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,EAAE,CAAC,CAAA;YACjE,MAAM,QAAQ,GAAgB;gBAC1B,QAAQ,EAAE;oBACN,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;oBACnD,gBAAgB,EAAE,gBAAgB;oBAClC,eAAe,EAAE,EAAE;oBACnB,IAAI,EAAE,qBAAqB;iBAC9B;gBACD,WAAW,EAAE;oBACT,KAAK,EAAE;wBACH,KAAK,EAAE;4BACH,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,CAAC;yBACf;wBACD,GAAG,EAAE;4BACD,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,EAAE;yBAChB;qBACJ;iBACJ;aACJ,CAAA;YAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,kBAAkB,CAAC,cAAc,EAAE;gBAC7E,qBAAqB;gBACrB,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;iBACJ;aACJ,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,EAAE,CAAC,CAAA;YACjE,MAAM,QAAQ,GAAgB;gBAC1B,QAAQ,EAAE;oBACN,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;oBACnD,gBAAgB,EAAE,gBAAgB;oBAClC,eAAe,EAAE,EAAE;oBACnB,IAAI,EAAE,qBAAqB;iBAC9B;gBACD,WAAW,EAAE;oBACT,KAAK,EAAE;wBACH,KAAK,EAAE;4BACH,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,CAAC;yBACf;wBACD,GAAG,EAAE;4BACD,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,EAAE;yBAChB;qBACJ;iBACJ;aACJ,CAAA;YAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,kBAAkB,CAAC,cAAc,EAAE;gBAC7E,4BAA4B;gBAC5B,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;iBACJ;aACJ,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,CAAC,CAAC,CAAA;YAEhE,MAAM,QAAQ,GAAgB;gBAC1B,QAAQ,EAAE;oBACN,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;oBACnD,gBAAgB,EAAE,gBAAgB;oBAClC,eAAe,EAAE,EAAE;oBACnB,IAAI,EAAE,EAAE;iBACX;gBACD,WAAW,EAAE,SAAS;aACzB,CAAA;YAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,kBAAkB,CAAC,cAAc,EAAE;gBAC7E,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;iBACJ;aACJ,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,EAAE,CAAA;YAE/D,MAAM,cAAc,GAA4B;gBAC5C,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;gBACnD,gBAAgB,EAAE,gBAAgB;gBAClC,eAAe,EAAE,EAAE;gBACnB,IAAI,EAAE,qBAAqB;aAC9B,CAAA;YACD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,cAAc,EAAE;gBACjF,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;gBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;aAClC,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,EAAE,CAAA;YAE/D,MAAM,eAAe,GAAG;;EAElC,CAAA;YACU,MAAM,YAAY,GAAG,iDAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,EAAE,eAAe,CAAC,CAAA;YAEpF,MAAM,cAAc,GAA4B;gBAC5C,mBAAmB,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;gBAC3C,gBAAgB,EAAE,gBAAgB;gBAClC,eAAe,EAAE,EAAE;gBACnB,IAAI,EAAE,qBAAqB;aAC9B,CAAA;YACD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,YAAY,EAAE;gBAC/E,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;gBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;aAClC,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}
+{"version":3,"file":"documentContext.test.js","sourceRoot":"","sources":["../../../../src/language-server/chat/contexts/documentContext.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,uCAA4B;AAC5B,2FAAiE;AACjE,uDAA6E;AAC7E,iEAA6D;AAE7D,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,MAAM,uBAAuB,GAAG;;EAElC,CAAA;IACE,MAAM,cAAc,GAAG,iDAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,YAAY,EAAE,CAAC,EAAE,uBAAuB,CAAC,CAAA;IAEtG,UAAU,CAAC,GAAG,EAAE;QACZ,kBAAK,CAAC,IAAI,CAAC,2CAAoB,CAAC,SAAS,EAAE,wBAAwB,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAA;IACrF,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,kBAAK,CAAC,OAAO,EAAE,CAAA;IACnB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAA;YACtF,MAAM,QAAQ,GAAoB;gBAC9B,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;gBACnD,gBAAgB,EAAE,gBAAgB;gBAClC,eAAe,EAAE,EAAE;gBACnB,IAAI,EAAE,qBAAqB;gBAC3B,cAAc,EAAE,IAAI;gBACpB,qBAAqB,EAAE,uBAAuB,CAAC,MAAM;gBACrD,WAAW,EAAE;oBACT,KAAK,EAAE;wBACH,KAAK,EAAE;4BACH,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,CAAC;yBACf;wBACD,GAAG,EAAE;4BACD,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,EAAE;yBAChB;qBACJ;iBACJ;aACJ,CAAA;YAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,cAAc,EAAE;gBACjF,qBAAqB;gBACrB,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;iBACJ;aACJ,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAA;YACtF,MAAM,QAAQ,GAAoB;gBAC9B,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;gBACnD,gBAAgB,EAAE,gBAAgB;gBAClC,eAAe,EAAE,EAAE;gBACnB,IAAI,EAAE,qBAAqB;gBAC3B,cAAc,EAAE,IAAI;gBACpB,qBAAqB,EAAE,uBAAuB,CAAC,MAAM;gBACrD,WAAW,EAAE;oBACT,KAAK,EAAE;wBACH,KAAK,EAAE;4BACH,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,CAAC;yBACf;wBACD,GAAG,EAAE;4BACD,IAAI,EAAE,CAAC;4BACP,SAAS,EAAE,EAAE;yBAChB;qBACJ;iBACJ;aACJ,CAAA;YAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,cAAc,EAAE;gBACjF,4BAA4B;gBAC5B,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;iBACJ;aACJ,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,CAAA;YAErF,MAAM,QAAQ,GAAoB;gBAC9B,mBAAmB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;gBACnD,gBAAgB,EAAE,gBAAgB;gBAClC,eAAe,EAAE,EAAE;gBACnB,IAAI,EAAE,EAAE;gBACR,cAAc,EAAE,KAAK;gBACrB,qBAAqB,EAAE,uBAAuB,CAAC,MAAM;gBACrD,WAAW,EAAE,SAAS;aACzB,CAAA;YAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,cAAc,EAAE;gBACjF,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,EAAE;qBAChB;iBACJ;aACJ,CAAC,CAAA;YAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,wBAAwB,GAAG,IAAI,0CAAwB,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAA;QAEtF,MAAM,eAAe,GAAG;;EAE9B,CAAA;QACM,MAAM,YAAY,GAAG,iDAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,EAAE,eAAe,CAAC,CAAA;QAEpF,MAAM,cAAc,GAAoB;YACpC,mBAAmB,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;YAC3C,gBAAgB,EAAE,gBAAgB;YAClC,eAAe,EAAE,EAAE;YACnB,IAAI,EAAE,qBAAqB;YAC3B,qBAAqB,EAAE,eAAe,CAAC,MAAM;YAC7C,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE;gBACT,KAAK,EAAE;oBACH,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;oBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;iBAClC;aACJ;SACJ,CAAA;QACD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,YAAY,EAAE;YAC/E,KAAK,EAAE;gBACH,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;gBAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;aAClC;SACJ,CAAC,CAAA;QAEF,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/chat/contexts/documentFqnExtractor.d.ts

@@ -0,0 +1,22 @@
+import { DocumentSymbol } from '@amzn/codewhisperer-streaming';
+import { Range, TextDocument } from 'vscode-languageserver-textdocument';
+import { Features } from '../../types';
+export interface DocumentFqnExtractorConfig {
+    nameMinLength?: number;
+    nameMaxLength?: number;
+    maxSymbols?: number;
+    timeout?: number;
+    logger?: Features['logging'];
+}
+export declare class DocumentFqnExtractor {
+    #private;
+    static readonly DEFAULT_CONFIG: {
+        nameMinLength: number;
+        nameMaxLength: number;
+        maxSymbols: number;
+    };
+    static readonly FQN_SUPPORTED_LANGUAGE_SET: Set<string>;
+    constructor(config?: DocumentFqnExtractorConfig);
+    dispose(): void;
+    extractDocumentSymbols(document: TextDocument, range: Range, languageId?: string): Promise<DocumentSymbol[]>;
+}

package/out/language-server/chat/contexts/documentFqnExtractor.js

@@ -0,0 +1,106 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _DocumentFqnExtractor_instances, _DocumentFqnExtractor_nameMinLength, _DocumentFqnExtractor_nameMaxLength, _DocumentFqnExtractor_maxSymbols, _DocumentFqnExtractor_workerPool, _DocumentFqnExtractor_extractSymbols, _DocumentFqnExtractor_extractNames, _DocumentFqnExtractor_findNamesInRange;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DocumentFqnExtractor = void 0;
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const lsp_fqn_1 = require("@aws/lsp-fqn");
+class DocumentFqnExtractor {
+    constructor(config) {
+        _DocumentFqnExtractor_instances.add(this);
+        _DocumentFqnExtractor_nameMinLength.set(this, void 0);
+        _DocumentFqnExtractor_nameMaxLength.set(this, void 0);
+        _DocumentFqnExtractor_maxSymbols.set(this, void 0);
+        _DocumentFqnExtractor_workerPool.set(this, void 0);
+        const { nameMaxLength, nameMinLength, maxSymbols, timeout, logger } = config ?? {};
+        __classPrivateFieldSet(this, _DocumentFqnExtractor_nameMinLength, nameMinLength ?? DocumentFqnExtractor.DEFAULT_CONFIG.nameMinLength, "f");
+        __classPrivateFieldSet(this, _DocumentFqnExtractor_nameMaxLength, nameMaxLength ?? DocumentFqnExtractor.DEFAULT_CONFIG.nameMaxLength, "f");
+        __classPrivateFieldSet(this, _DocumentFqnExtractor_maxSymbols, maxSymbols ?? DocumentFqnExtractor.DEFAULT_CONFIG.maxSymbols, "f");
+        __classPrivateFieldSet(this, _DocumentFqnExtractor_workerPool, new lsp_fqn_1.FqnWorkerPool({
+            timeout,
+            logger: logger
+                ? {
+                    // binding for log function (e.g. winston) that relies on "this"
+                    log: logger.log.bind(logger),
+                    error: logger.log.bind(logger),
+                }
+                : undefined,
+        }), "f");
+    }
+    dispose() {
+        __classPrivateFieldGet(this, _DocumentFqnExtractor_workerPool, "f").dispose();
+    }
+    async extractDocumentSymbols(document, range, languageId = document.languageId) {
+        return DocumentFqnExtractor.FQN_SUPPORTED_LANGUAGE_SET.has(languageId)
+            ? __classPrivateFieldGet(this, _DocumentFqnExtractor_instances, "m", _DocumentFqnExtractor_extractSymbols).call(this, document, range, languageId)
+            : [];
+    }
+}
+exports.DocumentFqnExtractor = DocumentFqnExtractor;
+_DocumentFqnExtractor_nameMinLength = new WeakMap(), _DocumentFqnExtractor_nameMaxLength = new WeakMap(), _DocumentFqnExtractor_maxSymbols = new WeakMap(), _DocumentFqnExtractor_workerPool = new WeakMap(), _DocumentFqnExtractor_instances = new WeakSet(), _DocumentFqnExtractor_extractSymbols = async function _DocumentFqnExtractor_extractSymbols(document, range, languageId) {
+    const names = await __classPrivateFieldGet(this, _DocumentFqnExtractor_instances, "m", _DocumentFqnExtractor_extractNames).call(this, document, range, languageId);
+    const documentSymbols = [];
+    for (const name of names) {
+        if (documentSymbols.length >= __classPrivateFieldGet(this, _DocumentFqnExtractor_maxSymbols, "f")) {
+            break;
+        }
+        const sourceSymbolString = name.source.join('.');
+        const symbolFqn = {
+            name: name.symbol.join('.') ?? '',
+            type: codewhisperer_streaming_1.SymbolType.USAGE,
+            source: sourceSymbolString ? sourceSymbolString : undefined,
+        };
+        if (symbolFqn.name.length >= __classPrivateFieldGet(this, _DocumentFqnExtractor_nameMinLength, "f") &&
+            symbolFqn.name.length < __classPrivateFieldGet(this, _DocumentFqnExtractor_nameMaxLength, "f") &&
+            (symbolFqn.source === undefined ||
+                (symbolFqn.source.length >= __classPrivateFieldGet(this, _DocumentFqnExtractor_nameMinLength, "f") && symbolFqn.source.length < __classPrivateFieldGet(this, _DocumentFqnExtractor_nameMaxLength, "f")))) {
+            documentSymbols.push(symbolFqn);
+        }
+    }
+    return documentSymbols;
+}, _DocumentFqnExtractor_extractNames = async function _DocumentFqnExtractor_extractNames(document, range, languageId) {
+    const result = await __classPrivateFieldGet(this, _DocumentFqnExtractor_instances, "m", _DocumentFqnExtractor_findNamesInRange).call(this, document.getText(), range, languageId);
+    if (!result.success || !result.data.fullyQualified) {
+        return [];
+    }
+    const dedupedUsedFullyQualifiedNames = Object.fromEntries(result.data.fullyQualified.usedSymbols.map((name) => [
+        JSON.stringify([name.source, name.symbol]),
+        { source: name.source, symbol: name.symbol },
+    ]));
+    return Object.values(dedupedUsedFullyQualifiedNames).sort((name, other) => name.source.length + name.symbol.length - (other.source.length + other.symbol.length));
+}, _DocumentFqnExtractor_findNamesInRange = function _DocumentFqnExtractor_findNamesInRange(fileText, selection, languageId) {
+    return __classPrivateFieldGet(this, _DocumentFqnExtractor_workerPool, "f").exec({
+        /**
+         * [\ue000-\uf8ff]: Private Use Area in Unicode
+         * \ud83c[\udf00-\udfff]: Presentation Symbols
+         * \ud83d[\udc00-\uddff]: Presentation Symbols
+         */
+        fileText: fileText.replace(/([\uE000-\uF8FF]|\uD83C[\uDF00-\uDFFF]|\uD83D[\uDC00-\uDDFF])/g, ''),
+        selection,
+        languageId,
+    });
+};
+DocumentFqnExtractor.DEFAULT_CONFIG = {
+    nameMinLength: 1,
+    nameMaxLength: 256,
+    maxSymbols: 25,
+};
+DocumentFqnExtractor.FQN_SUPPORTED_LANGUAGE_SET = new Set([
+    'python',
+    'java',
+    'javascript',
+    'typescript',
+    'javascriptreact',
+    'typescriptreact',
+]);
+//# sourceMappingURL=documentFqnExtractor.js.map

package/out/language-server/chat/contexts/documentFqnExtractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"documentFqnExtractor.js","sourceRoot":"","sources":["../../../../src/language-server/chat/contexts/documentFqnExtractor.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2EAA0E;AAC1E,0CAAwH;AAYxH,MAAa,oBAAoB;IAsB7B,YAAY,MAAmC;;QAN/C,sDAAsB;QACtB,sDAAsB;QACtB,mDAAmB;QAEnB,mDAA2B;QAGvB,MAAM,EAAE,aAAa,EAAE,aAAa,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,EAAE,CAAA;QAClF,uBAAA,IAAI,uCAAkB,aAAa,IAAI,oBAAoB,CAAC,cAAc,CAAC,aAAa,MAAA,CAAA;QACxF,uBAAA,IAAI,uCAAkB,aAAa,IAAI,oBAAoB,CAAC,cAAc,CAAC,aAAa,MAAA,CAAA;QACxF,uBAAA,IAAI,oCAAe,UAAU,IAAI,oBAAoB,CAAC,cAAc,CAAC,UAAU,MAAA,CAAA;QAC/E,uBAAA,IAAI,oCAAe,IAAI,uBAAa,CAAC;YACjC,OAAO;YACP,MAAM,EAAE,MAAM;gBACV,CAAC,CAAC;oBACI,gEAAgE;oBAChE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;oBAC5B,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;iBACjC;gBACH,CAAC,CAAC,SAAS;SAClB,CAAC,MAAA,CAAA;IACN,CAAC;IAEM,OAAO;QACV,uBAAA,IAAI,wCAAY,CAAC,OAAO,EAAE,CAAA;IAC9B,CAAC;IAEM,KAAK,CAAC,sBAAsB,CAC/B,QAAsB,EACtB,KAAY,EACZ,UAAU,GAAG,QAAQ,CAAC,UAAU;QAEhC,OAAO,oBAAoB,CAAC,0BAA0B,CAAC,GAAG,CAAC,UAAU,CAAC;YAClE,CAAC,CAAC,uBAAA,IAAI,6EAAgB,MAApB,IAAI,EAAiB,QAAQ,EAAE,KAAK,EAAE,UAAmC,CAAC;YAC5E,CAAC,CAAC,EAAE,CAAA;IACZ,CAAC;;AAnDL,oDAsHC;sSAjEG,KAAK,+CAAiB,QAAsB,EAAE,KAAY,EAAE,UAAiC;IACzF,MAAM,KAAK,GAAG,MAAM,uBAAA,IAAI,2EAAc,MAAlB,IAAI,EAAe,QAAQ,EAAE,KAAK,EAAE,UAAU,CAAC,CAAA;IAEnE,MAAM,eAAe,GAAqB,EAAE,CAAA;IAE5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACtB,IAAI,eAAe,CAAC,MAAM,IAAI,uBAAA,IAAI,wCAAY,EAAE;YAC5C,MAAK;SACR;QAED,MAAM,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAChD,MAAM,SAAS,GAAG;YACd,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE;YACjC,IAAI,EAAE,oCAAU,CAAC,KAAK;YACtB,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;SAC9D,CAAA;QAED,IACI,SAAS,CAAC,IAAI,CAAC,MAAM,IAAI,uBAAA,IAAI,2CAAe;YAC5C,SAAS,CAAC,IAAI,CAAC,MAAM,GAAG,uBAAA,IAAI,2CAAe;YAC3C,CAAC,SAAS,CAAC,MAAM,KAAK,SAAS;gBAC3B,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI,uBAAA,IAAI,2CAAe,IAAI,SAAS,CAAC,MAAM,CAAC,MAAM,GAAG,uBAAA,IAAI,2CAAe,CAAC,CAAC,EACxG;YACE,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;SAClC;KACJ;IAED,OAAO,eAAe,CAAA;AAC1B,CAAC,uCAED,KAAK,6CACD,QAAsB,EACtB,KAAY,EACZ,UAAiC;IAEjC,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,+EAAkB,MAAtB,IAAI,EAAmB,QAAQ,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,CAAA;IAElF,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE;QAChD,OAAO,EAAE,CAAA;KACZ;IAED,MAAM,8BAA8B,GAA0C,MAAM,CAAC,WAAW,CAC5F,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAwB,EAAE,EAAE,CAAC;QACrE,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE;KAC/C,CAAC,CACL,CAAA;IAED,OAAO,MAAM,CAAC,MAAM,CAAC,8BAA8B,CAAC,CAAC,IAAI,CACrD,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CACzG,CAAA;AACL,CAAC,2FAEiB,QAAgB,EAAE,SAAgB,EAAE,UAAiC;IACnF,OAAO,uBAAA,IAAI,wCAAY,CAAC,IAAI,CAAC;QACzB;;;;WAIG;QACH,QAAQ,EAAE,QAAQ,CAAC,OAAO,CAAC,gEAAgE,EAAE,EAAE,CAAC;QAChG,SAAS;QACT,UAAU;KACb,CAAC,CAAA;AACN,CAAC;AApHsB,mCAAc,GAAG;IACpC,aAAa,EAAE,CAAC;IAChB,aAAa,EAAE,GAAG;IAClB,UAAU,EAAE,EAAE;CACjB,CAAA;AAEsB,+CAA0B,GAAG,IAAI,GAAG,CAAC;IACxD,QAAQ;IACR,MAAM;IACN,YAAY;IACZ,YAAY;IACZ,iBAAiB;IACjB,iBAAiB;CACpB,CAAC,CAAA"}

package/out/language-server/chat/contexts/documentFqnExtractor.test.d.ts

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

package/out/language-server/chat/contexts/documentFqnExtractor.test.js

@@ -0,0 +1,94 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const lsp_fqn_1 = require("@aws/lsp-fqn");
+const assert = require("assert");
+const ts_sinon_1 = require("ts-sinon");
+const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
+const documentFqnExtractor_1 = require("./documentFqnExtractor");
+const mockData_1 = require("./mockData");
+describe('DocumentFQNExtractor', () => {
+    let extractorStub;
+    const typescriptDocument = vscode_languageserver_textdocument_1.TextDocument.create('file:///test.ts', 'typescript', 1, 'test');
+    const mockRange = { start: { line: 0, character: 0 }, end: { line: 0, character: 0 } };
+    let documentFqnExtractor;
+    beforeEach(() => {
+        extractorStub = ts_sinon_1.default.stub(lsp_fqn_1.FqnWorkerPool.prototype, 'exec');
+        documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor();
+    });
+    afterEach(() => {
+        extractorStub.restore();
+    });
+    it('returns symbols in the right shape', async () => {
+        const documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor();
+        extractorStub.returns(Promise.resolve({ success: true, data: mockData_1.mockExtractedSymbols }));
+        const documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange);
+        assert.deepStrictEqual(documentSymbols, mockData_1.expectedExtractedNames);
+        ts_sinon_1.default.assert.calledOnceWithExactly(extractorStub, {
+            fileText: typescriptDocument.getText(),
+            selection: mockRange,
+            languageId: typescriptDocument.languageId,
+        });
+    });
+    it('returns empty array if language id is not supported', async () => {
+        const documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor();
+        extractorStub.returns(Promise.resolve({ success: true, data: mockData_1.mockExtractedSymbols }));
+        const documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange, 'lolcode');
+        assert.deepStrictEqual(documentSymbols, []);
+    });
+    it('resolves to empty array if not successful', async () => {
+        extractorStub.resolves({ success: false, data: mockData_1.mockExtractedSymbols });
+        let documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange);
+        assert.deepStrictEqual(documentSymbols, []);
+        extractorStub.resolves({ success: false, data: undefined });
+        documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange);
+        assert.deepStrictEqual(documentSymbols, []);
+    });
+    it('uses language id if passed', async () => {
+        const documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor();
+        extractorStub.resolves({ success: true, data: mockData_1.mockExtractedSymbols });
+        const documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange, 'python');
+        assert.deepStrictEqual(documentSymbols, mockData_1.expectedExtractedNames);
+        ts_sinon_1.default.assert.calledOnceWithExactly(extractorStub, {
+            fileText: typescriptDocument.getText(),
+            selection: mockRange,
+            languageId: 'python',
+        });
+    });
+    it('dedups symbols', async () => {
+        const documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor();
+        extractorStub.resolves({
+            success: true,
+            data: {
+                fullyQualified: {
+                    ...mockData_1.mockExtractedSymbols.fullyQualified,
+                    usedSymbols: [
+                        ...mockData_1.mockExtractedSymbols.fullyQualified.usedSymbols.slice(0, 4),
+                        ...mockData_1.mockExtractedSymbols.fullyQualified.usedSymbols.slice(3, 6),
+                        ...mockData_1.mockExtractedSymbols.fullyQualified.usedSymbols.slice(5),
+                    ],
+                },
+            },
+        });
+        const documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange);
+        assert.deepStrictEqual(documentSymbols, mockData_1.expectedExtractedNames);
+    });
+    it('returns no more than the limit of symbols specify', async () => {
+        const documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor({ maxSymbols: 3 });
+        extractorStub.resolves({
+            success: true,
+            data: mockData_1.mockExtractedSymbols,
+        });
+        const documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange);
+        assert.deepStrictEqual(documentSymbols, mockData_1.expectedExtractedNames.slice(0, 3));
+    });
+    it('filters out symbols if either name or source does not conform to the length limit', async () => {
+        const documentFqnExtractor = new documentFqnExtractor_1.DocumentFqnExtractor({ nameMinLength: 5, nameMaxLength: 8 });
+        extractorStub.resolves({
+            success: true,
+            data: mockData_1.mockExtractedSymbols,
+        });
+        const documentSymbols = await documentFqnExtractor.extractDocumentSymbols(typescriptDocument, mockRange);
+        assert.deepStrictEqual(documentSymbols, [{ name: 'mkdir', type: 'USAGE', source: 'node:fs' }]);
+    });
+});
+//# sourceMappingURL=documentFqnExtractor.test.js.map

package/out/language-server/chat/contexts/documentFqnExtractor.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"documentFqnExtractor.test.js","sourceRoot":"","sources":["../../../../src/language-server/chat/contexts/documentFqnExtractor.test.ts"],"names":[],"mappings":";;AACA,0CAA4C;AAC5C,iCAAgC;AAChC,uCAA4B;AAC5B,2FAAiE;AACjE,iEAA6D;AAC7D,yCAAyE;AAEzE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IAClC,IAAI,aAA8B,CAAA;IAElC,MAAM,kBAAkB,GAAG,iDAAY,CAAC,MAAM,CAAC,iBAAiB,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,CAAC,CAAA;IAC1F,MAAM,SAAS,GAAU,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAAA;IAC7F,IAAI,oBAA0C,CAAA;IAE9C,UAAU,CAAC,GAAG,EAAE;QACZ,aAAa,GAAG,kBAAK,CAAC,IAAI,CAAC,uBAAa,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;QAC3D,oBAAoB,GAAG,IAAI,2CAAoB,EAAE,CAAA;IACrD,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,aAAa,CAAC,OAAO,EAAE,CAAA;IAC3B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,oBAAoB,GAAG,IAAI,2CAAoB,EAAE,CAAA;QAEvD,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,+BAAoB,EAAE,CAAC,CAAC,CAAA;QACrF,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAA;QAExG,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,iCAAsB,CAAC,CAAA;QAC/D,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,aAAa,EAAE;YAC9C,QAAQ,EAAE,kBAAkB,CAAC,OAAO,EAAE;YACtC,SAAS,EAAE,SAAS;YACpB,UAAU,EAAE,kBAAkB,CAAC,UAAU;SAC5C,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,oBAAoB,GAAG,IAAI,2CAAoB,EAAE,CAAA;QAEvD,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,+BAAoB,EAAE,CAAC,CAAC,CAAA;QACrF,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CACrE,kBAAkB,EAClB,SAAS,EACT,SAAS,CACZ,CAAA;QAED,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,EAAE,CAAC,CAAA;IAC/C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACvD,aAAa,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,+BAAoB,EAAE,CAAC,CAAA;QAEtE,IAAI,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAA;QAEtG,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,EAAE,CAAC,CAAA;QAE3C,aAAa,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAA;QAE3D,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAA;QAElG,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,EAAE,CAAC,CAAA;IAC/C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,oBAAoB,GAAG,IAAI,2CAAoB,EAAE,CAAA;QAEvD,aAAa,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,+BAAoB,EAAE,CAAC,CAAA;QACrE,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CACrE,kBAAkB,EAClB,SAAS,EACT,QAAQ,CACX,CAAA;QAED,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,iCAAsB,CAAC,CAAA;QAC/D,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,aAAa,EAAE;YAC9C,QAAQ,EAAE,kBAAkB,CAAC,OAAO,EAAE;YACtC,SAAS,EAAE,SAAS;YACpB,UAAU,EAAE,QAAQ;SACvB,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC5B,MAAM,oBAAoB,GAAG,IAAI,2CAAoB,EAAE,CAAA;QAEvD,aAAa,CAAC,QAAQ,CAAC;YACnB,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,cAAc,EAAE;oBACZ,GAAG,+BAAoB,CAAC,cAAc;oBACtC,WAAW,EAAE;wBACT,GAAG,+BAAoB,CAAC,cAAc,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;wBAC9D,GAAG,+BAAoB,CAAC,cAAc,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;wBAC9D,GAAG,+BAAoB,CAAC,cAAc,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;qBAC9D;iBACJ;aACJ;SACJ,CAAC,CAAA;QAEF,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAA;QAExG,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,iCAAsB,CAAC,CAAA;IACnE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,oBAAoB,GAAG,IAAI,2CAAoB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAA;QAExE,aAAa,CAAC,QAAQ,CAAC;YACnB,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,+BAAoB;SAC7B,CAAC,CAAA;QACF,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAA;QAExG,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,iCAAsB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IAC/E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,oBAAoB,GAAG,IAAI,2CAAoB,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAA;QAE7F,aAAa,CAAC,QAAQ,CAAC;YACnB,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,+BAAoB;SAC7B,CAAC,CAAA;QACF,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAA;QAExG,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAA;IAClG,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/chat/contexts/mockData.d.ts

@@ -0,0 +1,37 @@
+export declare const mockExtractedSymbols: {
+    fullyQualified: {
+        declaredSymbols: {
+            source: string[];
+            symbol: never[];
+            extent: {
+                start: {
+                    line: number;
+                    character: number;
+                };
+                end: {
+                    line: number;
+                    character: number;
+                };
+            };
+        }[];
+        usedSymbols: {
+            source: string[];
+            symbol: string[];
+            extent: {
+                start: {
+                    line: number;
+                    character: number;
+                };
+                end: {
+                    line: number;
+                    character: number;
+                };
+            };
+        }[];
+    };
+};
+export declare const expectedExtractedNames: {
+    name: string;
+    type: string;
+    source: string;
+}[];