@aws/lsp-codewhisperer 0.0.14 → 0.0.16

package/out/language-server/utilities/supplementalContextUtil/crossFileContextUtil.test.js

@@ -0,0 +1,157 @@
+"use strict";
+// Port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/24840fda8559a3e3ace3517ad9844db76680dc50/packages/core/src/test/shared/filesystemUtilities.test.ts
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const path = require("path");
+const sinon = require("sinon");
+const crossFile = require("./crossFileContextUtil");
+const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
+const testing_1 = require("@aws/language-server-runtimes/testing");
+const testUtils_1 = require("../../testUtils");
+const crossFileContextUtil_1 = require("./crossFileContextUtil");
+const constants_1 = require("../../models/constants");
+describe('crossFileContextUtil', function () {
+    const fakeCancellationToken = {
+        isCancellationRequested: false,
+        onCancellationRequested: sinon.spy(),
+    };
+    let features;
+    beforeEach(() => {
+        features = new testing_1.TestFeatures();
+    });
+    afterEach(() => {
+        features.dispose();
+    });
+    function openDocument(uri, languageId, version = 1, content = '') {
+        const document = vscode_languageserver_textdocument_1.TextDocument.create(uri, languageId, version, content);
+        features.openDocument(document);
+        return document;
+    }
+    describe('fetchSupplementalContextForSrc', function () {
+        describe('should fetch 3 chunks and each chunk should contains 10 lines', async function () {
+            async function assertCorrectCodeChunk() {
+                openDocument('file:///CrossFile.java', 'java', 1, testUtils_1.SAMPLE_FILE_OF_60_LINES_IN_JAVA);
+                const currentDocument = openDocument('file:///TargetFile.java', 'java');
+                const actual = await crossFile.fetchSupplementalContextForSrc(currentDocument, { line: 0, character: 0 }, features.workspace, fakeCancellationToken);
+                assert.ok(actual);
+                assert.ok(actual.supplementalContextItems.length === 3);
+                assert.strictEqual(actual.supplementalContextItems[0].content.split('\n').length, 10);
+                assert.strictEqual(actual.supplementalContextItems[1].content.split('\n').length, 10);
+                assert.strictEqual(actual.supplementalContextItems[2].content.split('\n').length, 10);
+            }
+            it('control group', async function () {
+                await assertCorrectCodeChunk();
+            });
+        });
+    });
+    describe('non supported language should return undefined', function () {
+        it('c++', async function () {
+            const currentDocument = openDocument('file:///TestFile.cpp', 'cpp', 1, 'content');
+            const actual = await crossFile.fetchSupplementalContextForSrc(currentDocument, { line: 0, character: 0 }, features.workspace, fakeCancellationToken);
+            assert.strictEqual(actual, undefined);
+        });
+        it('ruby', async function () {
+            const currentDocument = openDocument('file:///testfile.rb', 'ruby', 1, 'content');
+            const actual = await crossFile.fetchSupplementalContextForSrc(currentDocument, { line: 0, character: 0 }, features.workspace, fakeCancellationToken);
+            assert.strictEqual(actual, undefined);
+        });
+    });
+    describe('getCrossFileCandidate', function () {
+        it('should return opened files, exclude test files and sorted ascendingly by file distance', async function () {
+            const targetFile = path.join('file://', 'src', 'service', 'microService', 'CodeWhispererFileContextProvider.java');
+            const fileWithDistance3 = path.join('file://', 'src', 'service', 'CodewhispererRecommendationService.java');
+            const fileWithDistance5 = path.join('file://', 'src', 'util', 'CodeWhispererConstants.java');
+            const fileWithDistance6 = path.join('file://', 'src', 'ui', 'popup', 'CodeWhispererPopupManager.java');
+            const fileWithDistance7 = path.join('file://', 'src', 'ui', 'popup', 'components', 'CodeWhispererPopup.java');
+            const fileWithDistance8 = path.join('file://', 'src', 'ui', 'popup', 'components', 'actions', 'AcceptRecommendationAction.java');
+            const testFile1 = path.join('file://', 'test', 'service', 'CodeWhispererFileContextProviderTest.java');
+            const testFile2 = path.join('file://', 'test', 'ui', 'CodeWhispererPopupManagerTest.java');
+            const expectedFilePaths = [
+                fileWithDistance3,
+                fileWithDistance5,
+                fileWithDistance6,
+                fileWithDistance7,
+                fileWithDistance8,
+            ];
+            const shuffledFilePaths = (0, testUtils_1.shuffleList)(expectedFilePaths);
+            for (const filePath of shuffledFilePaths) {
+                openDocument(filePath, 'java');
+            }
+            openDocument(testFile1, 'java');
+            openDocument(testFile2, 'java');
+            const targetDocument = openDocument(targetFile, 'java');
+            const actual = await crossFile.getCrossFileCandidates(targetDocument, features.workspace);
+            assert.ok(actual.length === 5);
+            actual.forEach((actualFile, index) => {
+                const expectedFile = expectedFilePaths[index];
+                assert.strictEqual(expectedFile, actualFile.uri);
+            });
+        });
+    });
+    // describe('partial support - control group', function () {
+    // Not implemented, user group filters are not used in VSCode and tests are no-op
+    // https://github.com/aws/aws-toolkit-vscode/blob/a2daf60dca5e5699b8b25f6dc84708295f042c38/packages/amazonq/test/unit/codewhisperer/util/crossFileContextUtil.test.ts#L144
+    // })
+    // describe('partial support - crossfile group', function () {
+    // Not implemented, user group filters are not used in VSCode and tests are no-op
+    // https://github.com/aws/aws-toolkit-vscode/blob/a2daf60dca5e5699b8b25f6dc84708295f042c38/packages/amazonq/test/unit/codewhisperer/util/crossFileContextUtil.test.ts#L144
+    // })
+    describe('full support', function () {
+        const fileExtLists = ['.java', '.js', '.ts', '.py', '.tsx', '.jsx'];
+        function convertToExtensionMap(input) {
+            const result = {};
+            for (const [language, extensions] of Object.entries(input)) {
+                for (const extension of extensions) {
+                    if (!result[extension]) {
+                        result[extension] = [];
+                    }
+                    result[extension].push(language);
+                }
+            }
+            return result;
+        }
+        const extToLanguageIdMap = convertToExtensionMap(crossFileContextUtil_1.supportedLanguageToDialects);
+        fileExtLists.forEach(fileExt => {
+            extToLanguageIdMap[fileExt].forEach(languageId => {
+                it('should be non empty', async function () {
+                    const document = openDocument(`file:///file-1${fileExt}`, languageId, 1, 'content-1');
+                    openDocument(`file:///file-2${fileExt}`, languageId, 1, 'content-2');
+                    openDocument(`file:///file-3${fileExt}`, languageId, 1, 'content-3');
+                    openDocument(`file:///file-4${fileExt}`, languageId, 1, 'content-4');
+                    const actual = await crossFile.fetchSupplementalContextForSrc(document, { line: 0, character: 0 }, features.workspace, fakeCancellationToken);
+                    assert.ok(actual && actual.supplementalContextItems.length !== 0);
+                });
+            });
+        });
+    });
+    describe('splitFileToChunks', function () {
+        it('should split file to a chunk of 2 lines', function () {
+            const document = vscode_languageserver_textdocument_1.TextDocument.create('file:///file.py', 'python', 1, 'line_1\nline_2\nline_3\nline_4\nline_5\nline_6\nline_7');
+            const chunks = crossFile.splitFileToChunks(document, 2);
+            assert.strictEqual(chunks.length, 4);
+            assert.strictEqual(chunks[0].content, 'line_1\nline_2');
+            assert.strictEqual(chunks[1].content, 'line_3\nline_4');
+            assert.strictEqual(chunks[2].content, 'line_5\nline_6');
+            assert.strictEqual(chunks[3].content, 'line_7');
+        });
+        it('should split file to a chunk of 5 lines', function () {
+            const document = vscode_languageserver_textdocument_1.TextDocument.create('file:///file.py', 'python', 1, 'line_1\nline_2\nline_3\nline_4\nline_5\nline_6\nline_7');
+            const chunks = crossFile.splitFileToChunks(document, 5);
+            assert.strictEqual(chunks.length, 2);
+            assert.strictEqual(chunks[0].content, 'line_1\nline_2\nline_3\nline_4\nline_5');
+            assert.strictEqual(chunks[1].content, 'line_6\nline_7');
+        });
+        it('codewhisperer crossfile config should use 10 lines', function () {
+            const document = vscode_languageserver_textdocument_1.TextDocument.create('file:///testfile.java', 'java', 1, testUtils_1.SAMPLE_FILE_OF_60_LINES_IN_JAVA);
+            const chunks = crossFile.splitFileToChunks(document, constants_1.crossFileContextConfig.numberOfLinesEachChunk);
+            assert.strictEqual(chunks.length, 6);
+        });
+        it('should split file with Windows line endings', function () {
+            const document = vscode_languageserver_textdocument_1.TextDocument.create('file:///testfile.java', 'java', 1, testUtils_1.SAMPLE_FILE_OF_60_LINES_IN_JAVA.replaceAll('\n', '\r\n'));
+            const chunks = crossFile.splitFileToChunks(document, constants_1.crossFileContextConfig.numberOfLinesEachChunk);
+            assert.strictEqual(chunks.length, 6);
+        });
+    });
+});
+//# sourceMappingURL=crossFileContextUtil.test.js.map

package/out/language-server/utilities/supplementalContextUtil/crossFileContextUtil.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"crossFileContextUtil.test.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/crossFileContextUtil.test.ts"],"names":[],"mappings":";AAAA,mDAAmD;AACnD,oJAAoJ;;AAEpJ,iCAAgC;AAChC,6BAA4B;AAC5B,+BAA8B;AAC9B,oDAAmD;AACnD,2FAAiE;AACjE,mEAAoE;AAEpE,+CAA8E;AAC9E,iEAAoE;AACpE,sDAA+D;AAE/D,QAAQ,CAAC,sBAAsB,EAAE;IAC7B,MAAM,qBAAqB,GAAsB;QAC7C,uBAAuB,EAAE,KAAK;QAC9B,uBAAuB,EAAE,KAAK,CAAC,GAAG,EAAE;KACvC,CAAA;IACD,IAAI,QAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACZ,QAAQ,GAAG,IAAI,sBAAY,EAAE,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,QAAQ,CAAC,OAAO,EAAE,CAAA;IACtB,CAAC,CAAC,CAAA;IAEF,SAAS,YAAY,CAAC,GAAW,EAAE,UAAkB,EAAE,OAAO,GAAG,CAAC,EAAE,UAAkB,EAAE;QACpF,MAAM,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;QACvE,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAA;QAE/B,OAAO,QAAQ,CAAA;IACnB,CAAC;IAED,QAAQ,CAAC,gCAAgC,EAAE;QACvC,QAAQ,CAAC,+DAA+D,EAAE,KAAK;YAC3E,KAAK,UAAU,sBAAsB;gBACjC,YAAY,CAAC,wBAAwB,EAAE,MAAM,EAAE,CAAC,EAAE,2CAA+B,CAAC,CAAA;gBAClF,MAAM,eAAe,GAAG,YAAY,CAAC,yBAAyB,EAAE,MAAM,CAAC,CAAA;gBAEvE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,8BAA8B,CACzD,eAAe,EACf,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EACzB,QAAQ,CAAC,SAAS,EAClB,qBAAqB,CACxB,CAAA;gBACD,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAA;gBACjB,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,wBAAwB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAA;gBAEvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;gBACrF,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;gBACrF,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;YACzF,CAAC;YAED,EAAE,CAAC,eAAe,EAAE,KAAK;gBACrB,MAAM,sBAAsB,EAAE,CAAA;YAClC,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,gDAAgD,EAAE;QACvD,EAAE,CAAC,KAAK,EAAE,KAAK;YACX,MAAM,eAAe,GAAG,YAAY,CAAC,sBAAsB,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,CAAC,CAAA;YACjF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,8BAA8B,CACzD,eAAe,EACf,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EACzB,QAAQ,CAAC,SAAS,EAClB,qBAAqB,CACxB,CAAA;YACD,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,MAAM,EAAE,KAAK;YACZ,MAAM,eAAe,GAAG,YAAY,CAAC,qBAAqB,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,CAAC,CAAA;YACjF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,8BAA8B,CACzD,eAAe,EACf,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EACzB,QAAQ,CAAC,SAAS,EAClB,qBAAqB,CACxB,CAAA;YACD,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,uBAAuB,EAAE;QAC9B,EAAE,CAAC,wFAAwF,EAAE,KAAK;YAC9F,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CACxB,SAAS,EACT,KAAK,EACL,SAAS,EACT,cAAc,EACd,uCAAuC,CAC1C,CAAA;YACD,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,yCAAyC,CAAC,CAAA;YAC3G,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,6BAA6B,CAAC,CAAA;YAC5F,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,gCAAgC,CAAC,CAAA;YACtG,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAC/B,SAAS,EACT,KAAK,EACL,IAAI,EACJ,OAAO,EACP,YAAY,EACZ,yBAAyB,CAC5B,CAAA;YACD,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAC/B,SAAS,EACT,KAAK,EACL,IAAI,EACJ,OAAO,EACP,YAAY,EACZ,SAAS,EACT,iCAAiC,CACpC,CAAA;YACD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,2CAA2C,CAAC,CAAA;YACtG,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,oCAAoC,CAAC,CAAA;YAE1F,MAAM,iBAAiB,GAAG;gBACtB,iBAAiB;gBACjB,iBAAiB;gBACjB,iBAAiB;gBACjB,iBAAiB;gBACjB,iBAAiB;aACpB,CAAA;YAED,MAAM,iBAAiB,GAAG,IAAA,uBAAW,EAAC,iBAAiB,CAAC,CAAA;YAExD,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;gBACvC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YAClC,CAAC;YAED,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;YAC/B,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;YAC/B,MAAM,cAAc,GAAG,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAA;YAEvD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,sBAAsB,CAAC,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAA;YAEzF,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAA;YAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;gBACjC,MAAM,YAAY,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAA;gBAC7C,MAAM,CAAC,WAAW,CAAC,YAAY,EAAE,UAAU,CAAC,GAAG,CAAC,CAAA;YACpD,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,4DAA4D;IAC5D,iFAAiF;IACjF,0KAA0K;IAC1K,KAAK;IAEL,8DAA8D;IAC9D,iFAAiF;IACjF,0KAA0K;IAC1K,KAAK;IAEL,QAAQ,CAAC,cAAc,EAAE;QACrB,MAAM,YAAY,GAAG,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAEnE,SAAS,qBAAqB,CAAC,KAA4C;YACvE,MAAM,MAAM,GAA6B,EAAE,CAAA;YAE3C,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzD,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;oBACjC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;wBACrB,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,CAAA;oBAC1B,CAAC;oBACD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;gBACpC,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAA;QACjB,CAAC;QACD,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,kDAA2B,CAAC,CAAA;QAE7E,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC3B,kBAAkB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;gBAC7C,EAAE,CAAC,qBAAqB,EAAE,KAAK;oBAC3B,MAAM,QAAQ,GAAG,YAAY,CAAC,iBAAiB,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,CAAC,CAAA;oBACrF,YAAY,CAAC,iBAAiB,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,CAAC,CAAA;oBACpE,YAAY,CAAC,iBAAiB,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,CAAC,CAAA;oBACpE,YAAY,CAAC,iBAAiB,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,CAAC,CAAA;oBAEpE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,8BAA8B,CACzD,QAAQ,EACR,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EACzB,QAAQ,CAAC,SAAS,EAClB,qBAAqB,CACxB,CAAA;oBAED,MAAM,CAAC,EAAE,CAAC,MAAM,IAAI,MAAM,CAAC,wBAAwB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAA;gBACrE,CAAC,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,mBAAmB,EAAE;QAC1B,EAAE,CAAC,yCAAyC,EAAE;YAC1C,MAAM,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAChC,iBAAiB,EACjB,QAAQ,EACR,CAAC,EACD,wDAAwD,CAC3D,CAAA;YAED,MAAM,MAAM,GAAG,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;YAEvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;YACpC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAA;YACvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAA;YACvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAA;YACvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yCAAyC,EAAE;YAC1C,MAAM,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAChC,iBAAiB,EACjB,QAAQ,EACR,CAAC,EACD,wDAAwD,CAC3D,CAAA;YAED,MAAM,MAAM,GAAG,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;YAEvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;YACpC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,wCAAwC,CAAC,CAAA;YAC/E,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE;YACrD,MAAM,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAAC,uBAAuB,EAAE,MAAM,EAAE,CAAC,EAAE,2CAA+B,CAAC,CAAA;YAEzG,MAAM,MAAM,GAAG,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,kCAAsB,CAAC,sBAAsB,CAAC,CAAA;YACnG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6CAA6C,EAAE;YAC9C,MAAM,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAChC,uBAAuB,EACvB,MAAM,EACN,CAAC,EACD,2CAA+B,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAC3D,CAAA;YAED,MAAM,MAAM,GAAG,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,kCAAsB,CAAC,sBAAsB,CAAC,CAAA;YACnG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/utilities/supplementalContextUtil/rankBm25.d.ts

@@ -0,0 +1,32 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export interface BM25Document {
+    content: string;
+    /** The score that the document recieves. */
+    score: number;
+    index: number;
+}
+export declare abstract class BM25 {
+    protected readonly corpus: string[];
+    protected readonly tokenizer: (str: string) => string[];
+    protected readonly k1: number;
+    protected readonly b: number;
+    protected readonly epsilon: number;
+    protected readonly corpusSize: number;
+    protected readonly avgdl: number;
+    protected readonly idf: Map<string, number>;
+    protected readonly docLen: number[];
+    protected readonly docFreqs: Map<string, number>[];
+    protected readonly nd: Map<string, number>;
+    constructor(corpus: string[], tokenizer: ((str: string) => string[]) | undefined, k1: number, b: number, epsilon: number);
+    abstract calIdf(nd: Map<string, number>): void;
+    abstract score(query: string): BM25Document[];
+    topN(query: string, n: number): BM25Document[];
+}
+export declare class BM25Okapi extends BM25 {
+    constructor(corpus: string[], tokenizer?: (str: string) => string[]);
+    calIdf(nd: Map<string, number>): void;
+    score(query: string): BM25Document[];
+}

package/out/language-server/utilities/supplementalContextUtil/rankBm25.js

@@ -0,0 +1,113 @@
+"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.BM25Okapi = exports.BM25 = void 0;
+class BM25 {
+    corpus;
+    tokenizer;
+    k1;
+    b;
+    epsilon;
+    corpusSize;
+    avgdl;
+    idf = new Map();
+    docLen = [];
+    docFreqs = [];
+    nd = new Map();
+    constructor(corpus, tokenizer = defaultTokenizer, k1, b, epsilon) {
+        this.corpus = corpus;
+        this.tokenizer = tokenizer;
+        this.k1 = k1;
+        this.b = b;
+        this.epsilon = epsilon;
+        this.corpusSize = corpus.length;
+        let numDoc = 0;
+        corpus
+            .map(document => {
+            return tokenizer(document);
+        })
+            .forEach(document => {
+            this.docLen.push(document.length);
+            numDoc += document.length;
+            const frequencies = new Map();
+            document.forEach(word => {
+                frequencies.set(word, (frequencies.get(word) || 0) + 1);
+            });
+            this.docFreqs.push(frequencies);
+            frequencies.forEach((freq, word) => {
+                this.nd.set(word, (this.nd.get(word) || 0) + 1);
+            });
+        });
+        this.avgdl = numDoc / this.corpusSize;
+        this.calIdf(this.nd);
+    }
+    topN(query, n) {
+        const notSorted = this.score(query);
+        const sorted = notSorted.sort((a, b) => b.score - a.score);
+        return sorted.slice(0, Math.min(n, sorted.length));
+    }
+}
+exports.BM25 = BM25;
+class BM25Okapi extends BM25 {
+    constructor(corpus, tokenizer = defaultTokenizer) {
+        super(corpus, tokenizer, 1.5, 0.75, 0.25);
+    }
+    calIdf(nd) {
+        let idfSum = 0;
+        const negativeIdfs = [];
+        for (const [word, freq] of nd) {
+            const idf = Math.log(this.corpusSize - freq + 0.5) - Math.log(freq + 0.5);
+            this.idf.set(word, idf);
+            idfSum += idf;
+            if (idf < 0) {
+                negativeIdfs.push(word);
+            }
+        }
+        const averageIdf = idfSum / this.idf.size;
+        const eps = this.epsilon * averageIdf;
+        for (const word of negativeIdfs) {
+            this.idf.set(word, eps);
+        }
+    }
+    score(query) {
+        const queryWords = defaultTokenizer(query);
+        return this.docFreqs.map((docFreq, index) => {
+            let score = 0;
+            queryWords.forEach((queryWord, _) => {
+                const queryWordFreqForDocument = docFreq.get(queryWord) || 0;
+                const numerator = (this.idf.get(queryWord) || 0.0) * queryWordFreqForDocument * (this.k1 + 1);
+                const denominator = queryWordFreqForDocument + this.k1 * (1 - this.b + (this.b * this.docLen[index]) / this.avgdl);
+                score += numerator / denominator;
+            });
+            return {
+                content: this.corpus[index],
+                score: score,
+                index: index,
+            };
+        });
+    }
+}
+exports.BM25Okapi = BM25Okapi;
+// TODO: This is a very simple tokenizer, we want to replace this by more sophisticated one.
+function defaultTokenizer(content) {
+    const regex = /\w+/g;
+    const words = content.split(' ');
+    const result = [];
+    for (const word of words) {
+        const wordList = findAll(word, regex);
+        result.push(...wordList);
+    }
+    return result;
+}
+function findAll(str, re) {
+    let match;
+    const matches = [];
+    while ((match = re.exec(str)) !== null) {
+        matches.push(match[0]);
+    }
+    return matches;
+}
+//# sourceMappingURL=rankBm25.js.map

package/out/language-server/utilities/supplementalContextUtil/rankBm25.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rankBm25.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/rankBm25.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAaH,MAAsB,IAAI;IASC;IACA;IACA;IACA;IACA;IAZJ,UAAU,CAAQ;IAClB,KAAK,CAAQ;IACb,GAAG,GAAwB,IAAI,GAAG,EAAE,CAAA;IACpC,MAAM,GAAa,EAAE,CAAA;IACrB,QAAQ,GAA0B,EAAE,CAAA;IACpC,EAAE,GAAwB,IAAI,GAAG,EAAE,CAAA;IAEtD,YACuB,MAAgB,EAChB,YAAuC,gBAAgB,EACvD,EAAU,EACV,CAAS,EACT,OAAe;QAJf,WAAM,GAAN,MAAM,CAAU;QAChB,cAAS,GAAT,SAAS,CAA8C;QACvD,OAAE,GAAF,EAAE,CAAQ;QACV,MAAC,GAAD,CAAC,CAAQ;QACT,YAAO,GAAP,OAAO,CAAQ;QAElC,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,MAAM,CAAA;QAE/B,IAAI,MAAM,GAAG,CAAC,CAAA;QACd,MAAM;aACD,GAAG,CAAC,QAAQ,CAAC,EAAE;YACZ,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAA;QAC9B,CAAC,CAAC;aACD,OAAO,CAAC,QAAQ,CAAC,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;YACjC,MAAM,IAAI,QAAQ,CAAC,MAAM,CAAA;YAEzB,MAAM,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAA;YAC7C,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;gBACpB,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAA;YAC3D,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;YAE/B,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;gBAC/B,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAA;YACnD,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEN,IAAI,CAAC,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC,UAAU,CAAA;QAErC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACxB,CAAC;IAMD,IAAI,CAAC,KAAa,EAAE,CAAS;QACzB,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QACnC,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAA;QAC1D,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;IACtD,CAAC;CACJ;AAnDD,oBAmDC;AAED,MAAa,SAAU,SAAQ,IAAI;IAC/B,YAAY,MAAgB,EAAE,YAAuC,gBAAgB;QACjF,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,CAAA;IAC7C,CAAC;IAED,MAAM,CAAC,EAAuB;QAC1B,IAAI,MAAM,GAAG,CAAC,CAAA;QAEd,MAAM,YAAY,GAAa,EAAE,CAAA;QACjC,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,GAAG,IAAI,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,CAAA;YACzE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;YACvB,MAAM,IAAI,GAAG,CAAA;YAEb,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;gBACV,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC3B,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAA;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,GAAG,UAAU,CAAA;QACrC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YAC9B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;QAC3B,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAa;QACf,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YACxC,IAAI,KAAK,GAAG,CAAC,CAAA;YACb,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE;gBAChC,MAAM,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;gBAC5D,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,GAAG,wBAAwB,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;gBAC7F,MAAM,WAAW,GACb,wBAAwB,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAA;gBAElG,KAAK,IAAI,SAAS,GAAG,WAAW,CAAA;YACpC,CAAC,CAAC,CAAA;YAEF,OAAO;gBACH,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC3B,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,KAAK;aACf,CAAA;QACL,CAAC,CAAC,CAAA;IACN,CAAC;CACJ;AA9CD,8BA8CC;AAED,4FAA4F;AAC5F,SAAS,gBAAgB,CAAC,OAAe;IACrC,MAAM,KAAK,GAAG,MAAM,CAAA;IACpB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAChC,MAAM,MAAM,GAAG,EAAE,CAAA;IACjB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QACrC,MAAM,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAA;IAC5B,CAAC;IAED,OAAO,MAAM,CAAA;AACjB,CAAC;AAED,SAAS,OAAO,CAAC,GAAW,EAAE,EAAU;IACpC,IAAI,KAA6B,CAAA;IACjC,MAAM,OAAO,GAAa,EAAE,CAAA;IAE5B,OAAO,CAAC,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACrC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;IAC1B,CAAC;IAED,OAAO,OAAO,CAAA;AAClB,CAAC"}

package/out/language-server/utilities/supplementalContextUtil/supplementalContextUtil.d.ts

@@ -0,0 +1,5 @@
+import { CodeWhispererSupplementalContext } from '../../models/model';
+import { CancellationToken, Logging, Position, TextDocument, Workspace } from '@aws/language-server-runtimes/server-interface';
+export declare class CancellationError extends Error {
+}
+export declare function fetchSupplementalContext(document: TextDocument, position: Position, workspace: Workspace, logging: Logging, cancellationToken: CancellationToken): Promise<CodeWhispererSupplementalContext | undefined>;

package/out/language-server/utilities/supplementalContextUtil/supplementalContextUtil.js

@@ -0,0 +1,57 @@
+"use strict";
+// Port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/9d8ddbd85f4533e539a58e76f7c46883d8e50a79/packages/core/src/codewhisperer/util/supplementalContext/supplementalContextUtil.ts
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CancellationError = void 0;
+exports.fetchSupplementalContext = fetchSupplementalContext;
+const crossFileContextUtil_1 = require("./crossFileContextUtil");
+const codeParsingUtil_1 = require("./codeParsingUtil");
+class CancellationError extends Error {
+}
+exports.CancellationError = CancellationError;
+async function fetchSupplementalContext(document, position, workspace, logging, cancellationToken) {
+    const timesBeforeFetching = performance.now();
+    const isUtg = (0, codeParsingUtil_1.isTestFile)(document.uri, {
+        languageId: document.languageId,
+        fileContent: document.getText(),
+    });
+    try {
+        let supplementalContextValue;
+        if (isUtg) {
+            return;
+        }
+        else {
+            supplementalContextValue = await (0, crossFileContextUtil_1.fetchSupplementalContextForSrc)(document, position, workspace, cancellationToken);
+        }
+        if (supplementalContextValue) {
+            return {
+                isUtg: isUtg,
+                isProcessTimeout: false,
+                supplementalContextItems: supplementalContextValue.supplementalContextItems,
+                contentsLength: supplementalContextValue.supplementalContextItems.reduce((acc, curr) => acc + curr.content.length, 0),
+                latency: performance.now() - timesBeforeFetching,
+                strategy: supplementalContextValue.strategy,
+            };
+        }
+        else {
+            return undefined;
+        }
+    }
+    catch (err) {
+        if (err instanceof CancellationError) {
+            return {
+                isUtg: isUtg,
+                isProcessTimeout: true,
+                supplementalContextItems: [],
+                contentsLength: 0,
+                latency: performance.now() - timesBeforeFetching,
+                strategy: 'Empty',
+            };
+        }
+        else {
+            logging.log(`Fail to fetch supplemental context for target file ${document.uri}: ${err}`);
+            return undefined;
+        }
+    }
+}
+//# sourceMappingURL=supplementalContextUtil.js.map

package/out/language-server/utilities/supplementalContextUtil/supplementalContextUtil.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"supplementalContextUtil.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/supplementalContextUtil.ts"],"names":[],"mappings":";AAAA,mDAAmD;AACnD,8KAA8K;;;AAe9K,4DA4DC;AAzED,iEAAuE;AACvE,uDAA8C;AAU9C,MAAa,iBAAkB,SAAQ,KAAK;CAAG;AAA/C,8CAA+C;AAExC,KAAK,UAAU,wBAAwB,CAC1C,QAAsB,EACtB,QAAkB,EAClB,SAAoB,EACpB,OAAgB,EAChB,iBAAoC;IAEpC,MAAM,mBAAmB,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;IAE7C,MAAM,KAAK,GAAG,IAAA,4BAAU,EAAC,QAAQ,CAAC,GAAG,EAAE;QACnC,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,WAAW,EAAE,QAAQ,CAAC,OAAO,EAAE;KAClC,CAAC,CAAA;IAEF,IAAI,CAAC;QACD,IAAI,wBAEW,CAAA;QAEf,IAAI,KAAK,EAAE,CAAC;YACR,OAAM;QACV,CAAC;aAAM,CAAC;YACJ,wBAAwB,GAAG,MAAM,IAAA,qDAA8B,EAC3D,QAAQ,EACR,QAAQ,EACR,SAAS,EACT,iBAAiB,CACpB,CAAA;QACL,CAAC;QAED,IAAI,wBAAwB,EAAE,CAAC;YAC3B,OAAO;gBACH,KAAK,EAAE,KAAK;gBACZ,gBAAgB,EAAE,KAAK;gBACvB,wBAAwB,EAAE,wBAAwB,CAAC,wBAAwB;gBAC3E,cAAc,EAAE,wBAAwB,CAAC,wBAAwB,CAAC,MAAM,CACpE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EACxC,CAAC,CACJ;gBACD,OAAO,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,mBAAmB;gBAChD,QAAQ,EAAE,wBAAwB,CAAC,QAAQ;aAC9C,CAAA;QACL,CAAC;aAAM,CAAC;YACJ,OAAO,SAAS,CAAA;QACpB,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACX,IAAI,GAAG,YAAY,iBAAiB,EAAE,CAAC;YACnC,OAAO;gBACH,KAAK,EAAE,KAAK;gBACZ,gBAAgB,EAAE,IAAI;gBACtB,wBAAwB,EAAE,EAAE;gBAC5B,cAAc,EAAE,CAAC;gBACjB,OAAO,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,mBAAmB;gBAChD,QAAQ,EAAE,OAAO;aACpB,CAAA;QACL,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,GAAG,CAAC,sDAAsD,QAAQ,CAAC,GAAG,KAAK,GAAG,EAAE,CAAC,CAAA;YACzF,OAAO,SAAS,CAAA;QACpB,CAAC;IACL,CAAC;AACL,CAAC"}

package/out/language-server/utilities/supplementalContextUtil/supplementalContextUtil.test.js

@@ -0,0 +1,88 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const sinon = require("sinon");
+const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const supplementalContextUtil_1 = require("./supplementalContextUtil");
+const crossFileContextUtil = require("./crossFileContextUtil");
+const codeParsingUtil = require("./codeParsingUtil");
+describe('fetchSupplementalContext', function () {
+    let workspace;
+    let logging;
+    let cancellationToken;
+    let document;
+    let position;
+    let crossFileContextStub;
+    let isTestFileStub;
+    let performanceStub;
+    beforeEach(() => {
+        document = vscode_languageserver_textdocument_1.TextDocument.create('file:///somefile.js', 'javascript', 1, 'console.log("Hello, World!");');
+        position = server_interface_1.Position.create(0, 0);
+        workspace = {};
+        logging = {
+            log: sinon.stub(),
+        };
+        cancellationToken = {
+            isCancellationRequested: false,
+            onCancellationRequested: sinon.stub(),
+        };
+        crossFileContextStub = sinon.stub(crossFileContextUtil, 'fetchSupplementalContextForSrc');
+        isTestFileStub = sinon.stub(codeParsingUtil, 'isTestFile');
+        performanceStub = sinon.stub({ now: () => 0 });
+        sinon.stub(global, 'performance').value(performanceStub);
+    });
+    afterEach(() => {
+        sinon.restore();
+    });
+    it('should return supplemental context for non-test files', async function () {
+        performanceStub.now.onFirstCall().returns(0);
+        performanceStub.now.onSecondCall().returns(100); // 100ms elapsed time
+        isTestFileStub.returns(false);
+        crossFileContextStub.returns({
+            supplementalContextItems: [{ content: 'test content', filePath: 'somefile.js' }],
+            strategy: 'OpenTabs_BM25',
+        });
+        const expectedContext = {
+            isUtg: false,
+            isProcessTimeout: false,
+            latency: 100,
+            contentsLength: 12,
+            supplementalContextItems: [{ content: 'test content', filePath: 'somefile.js' }],
+            strategy: 'OpenTabs_BM25',
+        };
+        const result = await (0, supplementalContextUtil_1.fetchSupplementalContext)(document, position, workspace, logging, cancellationToken);
+        assert.deepStrictEqual(result, expectedContext);
+    });
+    it('should return undefined for test files', async function () {
+        isTestFileStub.returns(true);
+        const result = await (0, supplementalContextUtil_1.fetchSupplementalContext)(document, position, workspace, logging, cancellationToken);
+        assert.strictEqual(result, undefined);
+    });
+    it('should return empty context when CancellationError is received', async function () {
+        isTestFileStub.returns(false);
+        crossFileContextStub.throws(new supplementalContextUtil_1.CancellationError());
+        performanceStub.now.onFirstCall().returns(0);
+        performanceStub.now.onSecondCall().returns(100); // 100ms elapsed time
+        const expectedContext = {
+            contentsLength: 0,
+            isProcessTimeout: true,
+            isUtg: false,
+            latency: 100,
+            strategy: 'Empty',
+            supplementalContextItems: [],
+        };
+        const result = await (0, supplementalContextUtil_1.fetchSupplementalContext)(document, position, workspace, logging, cancellationToken);
+        assert.deepStrictEqual(result, expectedContext);
+    });
+    it('should handle errors and return undefined', async function () {
+        isTestFileStub.returns(false);
+        crossFileContextStub.throws(new Error('Some error'));
+        const result = await (0, supplementalContextUtil_1.fetchSupplementalContext)(document, position, workspace, logging, cancellationToken);
+        assert.strictEqual(result, undefined);
+        sinon.assert.calledWithMatch(
+        // @ts-ignore
+        logging.log, 'Fail to fetch supplemental context for target file file:///somefile.js');
+    });
+});
+//# sourceMappingURL=supplementalContextUtil.test.js.map

package/out/language-server/utilities/supplementalContextUtil/supplementalContextUtil.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"supplementalContextUtil.test.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/supplementalContextUtil.test.ts"],"names":[],"mappings":";;AAAA,iCAAgC;AAChC,+BAA8B;AAC9B,2FAAiE;AACjE,qFAAgH;AAChH,uEAAuF;AACvF,+DAA8D;AAC9D,qDAAoD;AAGpD,QAAQ,CAAC,0BAA0B,EAAE;IACjC,IAAI,SAAoB,CAAA;IACxB,IAAI,OAAgB,CAAA;IACpB,IAAI,iBAAoC,CAAA;IACxC,IAAI,QAAsB,CAAA;IAC1B,IAAI,QAAkB,CAAA;IACtB,IAAI,oBAAqC,CAAA;IACzC,IAAI,cAA+B,CAAA;IACnC,IAAI,eAAkE,CAAA;IAEtE,UAAU,CAAC,GAAG,EAAE;QACZ,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAAC,qBAAqB,EAAE,YAAY,EAAE,CAAC,EAAE,+BAA+B,CAAC,CAAA;QACvG,QAAQ,GAAG,2BAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;QAChC,SAAS,GAAG,EAAe,CAAA;QAC3B,OAAO,GAAG;YACN,GAAG,EAAE,KAAK,CAAC,IAAI,EAAE;SACT,CAAA;QACZ,iBAAiB,GAAG;YAChB,uBAAuB,EAAE,KAAK;YAC9B,uBAAuB,EAAE,KAAK,CAAC,IAAI,EAAE;SACxC,CAAA;QACD,oBAAoB,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,EAAE,gCAAgC,CAAC,CAAA;QACzF,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,CAAC,CAAA;QAE1D,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAA;QAC9C,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAA;IAC5D,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,KAAK,CAAC,OAAO,EAAE,CAAA;IACnB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,KAAK;QAC7D,eAAe,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;QAC5C,eAAe,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA,CAAC,qBAAqB;QACrE,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC7B,oBAAoB,CAAC,OAAO,CAAC;YACzB,wBAAwB,EAAE,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;YAChF,QAAQ,EAAE,eAAe;SAC5B,CAAC,CAAA;QAEF,MAAM,eAAe,GAAqC;YACtD,KAAK,EAAE,KAAK;YACZ,gBAAgB,EAAE,KAAK;YACvB,OAAO,EAAE,GAAG;YACZ,cAAc,EAAE,EAAE;YAClB,wBAAwB,EAAE,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;YAChF,QAAQ,EAAE,eAAe;SAC5B,CAAA;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,kDAAwB,EAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAA;QAExG,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,eAAe,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wCAAwC,EAAE,KAAK;QAC9C,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAE5B,MAAM,MAAM,GAAG,MAAM,IAAA,kDAAwB,EAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAA;QAExG,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gEAAgE,EAAE,KAAK;QACtE,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC7B,oBAAoB,CAAC,MAAM,CAAC,IAAI,2CAAiB,EAAE,CAAC,CAAA;QACpD,eAAe,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;QAC5C,eAAe,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA,CAAC,qBAAqB;QACrE,MAAM,eAAe,GAAG;YACpB,cAAc,EAAE,CAAC;YACjB,gBAAgB,EAAE,IAAI;YACtB,KAAK,EAAE,KAAK;YACZ,OAAO,EAAE,GAAG;YACZ,QAAQ,EAAE,OAAO;YACjB,wBAAwB,EAAE,EAAE;SAC/B,CAAA;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,kDAAwB,EAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAA;QAExG,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,eAAe,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK;QACjD,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC7B,oBAAoB,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAA;QAEpD,MAAM,MAAM,GAAG,MAAM,IAAA,kDAAwB,EAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAA;QAExG,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;QACrC,KAAK,CAAC,MAAM,CAAC,eAAe;QACxB,aAAa;QACb,OAAO,CAAC,GAAG,EACX,wEAAwE,CAC3E,CAAA;IACL,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/utilities/textUtils.d.ts

@@ -0,0 +1 @@
+export declare function undefinedIfEmpty(str: string | undefined): string | undefined;

package/out/language-server/utilities/textUtils.js

@@ -0,0 +1,12 @@
+"use strict";
+// Partial port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/0c4289b1a0b5d294cc352f7d2e7e586937ac0318/packages/core/src/shared/utilities/textUtilities.ts#L391
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.undefinedIfEmpty = undefinedIfEmpty;
+function undefinedIfEmpty(str) {
+    if (str && str.trim().length > 0) {
+        return str;
+    }
+    return undefined;
+}
+//# sourceMappingURL=textUtils.js.map

package/out/language-server/utilities/textUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"textUtils.js","sourceRoot":"","sources":["../../../src/language-server/utilities/textUtils.ts"],"names":[],"mappings":";AAAA,2DAA2D;AAC3D,mJAAmJ;;AAEnJ,4CAMC;AAND,SAAgB,gBAAgB,CAAC,GAAuB;IACpD,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO,GAAG,CAAA;IACd,CAAC;IAED,OAAO,SAAS,CAAA;AACpB,CAAC"}

package/out/language-server/utilities/textUtils.test.js

@@ -0,0 +1,21 @@
+"use strict";
+// Partial port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/0c4289b1a0b5d294cc352f7d2e7e586937ac0318/packages/core/src/test/shared/utilities/textUtilities.test.ts
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const textUtils_1 = require("./textUtils");
+describe('undefinedIfEmpty', function () {
+    const cases = [
+        { input: undefined, output: undefined, case: 'return undefined if input is undefined' },
+        { input: '', output: undefined, case: 'return undefined if input is empty string' },
+        { input: '   ', output: undefined, case: 'return undefined if input is blank' },
+        { input: 'foo', output: 'foo', case: 'return str if input is not empty' },
+        { input: ' foo ', output: ' foo ', case: 'return original str without trim' },
+    ];
+    cases.forEach(testCases => {
+        it(testCases.case, function () {
+            assert.strictEqual((0, textUtils_1.undefinedIfEmpty)(testCases.input), testCases.output);
+        });
+    });
+});
+//# sourceMappingURL=textUtils.test.js.map

package/out/language-server/utilities/textUtils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"textUtils.test.js","sourceRoot":"","sources":["../../../src/language-server/utilities/textUtils.test.ts"],"names":[],"mappings":";AAAA,2DAA2D;AAC3D,wJAAwJ;;AAExJ,iCAAgC;AAChC,2CAA8C;AAE9C,QAAQ,CAAC,kBAAkB,EAAE;IACzB,MAAM,KAAK,GAA8E;QACrF,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,wCAAwC,EAAE;QACvF,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,2CAA2C,EAAE;QACnF,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,oCAAoC,EAAE;QAC/E,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,kCAAkC,EAAE;QACzE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,kCAAkC,EAAE;KAChF,CAAA;IAED,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;QACtB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE;YACf,MAAM,CAAC,WAAW,CAAC,IAAA,4BAAgB,EAAC,SAAS,CAAC,KAAK,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAA;QAC3E,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/utils.js

@@ -1,6 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getUserAgent = exports.flattenMetric = exports.getBearerTokenFromProvider = exports.getErrorMessage = exports.parseJson = exports.getCompletionType = exports.isNullish = exports.isObject = exports.isAwsError = void 0;
+exports.getUserAgent = exports.flattenMetric = void 0;
+exports.isAwsError = isAwsError;
+exports.isObject = isObject;
+exports.isNullish = isNullish;
+exports.getCompletionType = getCompletionType;
+exports.parseJson = parseJson;
+exports.getErrorMessage = getErrorMessage;
+exports.getBearerTokenFromProvider = getBearerTokenFromProvider;
 const constants_1 = require("./constants");
 function isAwsError(error) {
     if (error === undefined) {
@@ -8,7 +15,6 @@ function isAwsError(error) {
     }
     return error instanceof Error && hasCode(error) && hasTime(error);
 }
-exports.isAwsError = isAwsError;
 function hasCode(error) {
     return typeof error.code === 'string';
 }
@@ -18,16 +24,13 @@ function hasTime(error) {
 function isObject(value) {
     return Boolean(value) && typeof value === 'object';
 }
-exports.isObject = isObject;
 function isNullish(value) {
     return value === null || value === undefined;
 }
-exports.isNullish = isNullish;
 function getCompletionType(suggestion) {
     const nonBlankLines = suggestion.content.split('\n').filter(line => line.trim() !== '').length;
     return nonBlankLines > 1 ? 'Block' : 'Line';
 }
-exports.getCompletionType = getCompletionType;
 function parseJson(jsonString) {
     try {
         return JSON.parse(jsonString);
@@ -36,14 +39,12 @@ function parseJson(jsonString) {
         throw new Error(`error while parsing string: ${jsonString}`);
     }
 }
-exports.parseJson = parseJson;
 function getErrorMessage(error) {
     if (error instanceof Error) {
         return error.message;
     }
     return String(error);
 }
-exports.getErrorMessage = getErrorMessage;
 function getBearerTokenFromProvider(credentialsProvider) {
     if (!credentialsProvider.hasCredentials('bearer')) {
         throw new Error(constants_1.MISSING_BEARER_TOKEN_ERROR);
@@ -54,7 +55,6 @@ function getBearerTokenFromProvider(credentialsProvider) {
     }
     return credentials.token;
 }
-exports.getBearerTokenFromProvider = getBearerTokenFromProvider;
 const flattenMetric = (obj, prefix = '') => {
     const flattened = {};
     Object.keys(obj).forEach(key => {

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

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/language-server/utils.ts"],"names":[],"mappings":";;;AAQA,2CAAwD;AAGxD,SAAgB,UAAU,CAAC,KAAc;IACrC,IAAI,KAAK,KAAK,SAAS,EAAE;QACrB,OAAO,KAAK,CAAA;KACf;IAED,OAAO,KAAK,YAAY,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,CAAA;AACrE,CAAC;AAND,gCAMC;AAED,SAAS,OAAO,CAAI,KAAQ;IACxB,OAAO,OAAQ,KAA4B,CAAC,IAAI,KAAK,QAAQ,CAAA;AACjE,CAAC;AAED,SAAS,OAAO,CAAC,KAAY;IACzB,OAAQ,KAA4B,CAAC,IAAI,YAAY,IAAI,CAAA;AAC7D,CAAC;AAED,SAAgB,QAAQ,CAAC,KAAc;IACnC,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAA;AACtD,CAAC;AAFD,4BAEC;AAED,SAAgB,SAAS,CAAC,KAAc;IACpC,OAAO,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,CAAA;AAChD,CAAC;AAFD,8BAEC;AAED,SAAgB,iBAAiB,CAAC,UAAsB;IACpD,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,CAAA;IAE9F,OAAO,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAA;AAC/C,CAAC;AAJD,8CAIC;AAED,SAAgB,SAAS,CAAC,UAAkB;IACxC,IAAI;QACA,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAA;KAChC;IAAC,MAAM;QACJ,MAAM,IAAI,KAAK,CAAC,+BAA+B,UAAU,EAAE,CAAC,CAAA;KAC/D;AACL,CAAC;AAND,8BAMC;AAED,SAAgB,eAAe,CAAC,KAAU;IACtC,IAAI,KAAK,YAAY,KAAK,EAAE;QACxB,OAAO,KAAK,CAAC,OAAO,CAAA;KACvB;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAA;AACxB,CAAC;AALD,0CAKC;AAED,SAAgB,0BAA0B,CAAC,mBAAwC;IAC/E,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE;QAC/C,MAAM,IAAI,KAAK,CAAC,sCAA0B,CAAC,CAAA;KAC9C;IAED,MAAM,WAAW,GAAG,mBAAmB,CAAC,cAAc,CAAC,QAAQ,CAAsB,CAAA;IAErF,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE;QACpB,MAAM,IAAI,KAAK,CAAC,sCAA0B,CAAC,CAAA;KAC9C;IAED,OAAO,WAAW,CAAC,KAAK,CAAA;AAC5B,CAAC;AAZD,gEAYC;AAEM,MAAM,aAAa,GAAG,CAAC,GAAQ,EAAE,MAAM,GAAG,EAAE,EAAE,EAAE;IACnD,MAAM,SAAS,GAAQ,EAAE,CAAA;IAEzB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QAC3B,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAA;QAEtB,IAAI,MAAM,KAAK,EAAE,EAAE;YACf,GAAG,GAAG,GAAG,GAAG,GAAG,CAAA;SAClB;QAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE;YAC7C,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,IAAA,qBAAa,EAAC,KAAK,EAAE,MAAM,GAAG,GAAG,CAAC,CAAC,CAAA;SAC/D;aAAM;YACH,SAAS,CAAC,MAAM,GAAG,GAAG,CAAC,GAAG,KAAK,CAAA;SAClC;IACL,CAAC,CAAC,CAAA;IAEF,OAAO,SAAS,CAAA;AACpB,CAAC,CAAA;AAlBY,QAAA,aAAa,iBAkBzB;AAED,MAAM,iBAAiB,GAAG,sBAAsB,CAAA;AACzC,MAAM,YAAY,GAAG,CAAC,gBAAkC,EAAE,UAAuB,EAAU,EAAE;IAChG,MAAM,MAAM,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;IAEnD,MAAM,KAAK,GAAa,EAAE,CAAA;IAE1B,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IAE7B,sCAAsC;IACtC,IAAI,UAAU,EAAE,IAAI,EAAE;QAClB,UAAU,CAAC,OAAO;YACd,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;YAChE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAA;KAC5C;IAED,iCAAiC;IACjC,iEAAiE;IACjE,oEAAoE;IACpE,IAAI,gBAAgB,EAAE,qBAAqB,EAAE,GAAG,EAAE;QAC9C,MAAM,EAAE,UAAU,EAAE,GAAG,gBAAgB,EAAE,qBAAqB,EAAE,GAAG,CAAA;QACnE,MAAM,EAAE,SAAS,EAAE,GAAG,UAAU,IAAI,EAAE,CAAA;QAEtC,IAAI,SAAS,EAAE;YACX,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC,CAAA;SACzG;QAED,IAAI,UAAU,EAAE;YACZ,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC,CAAA;SAC5G;QAED,IAAI,UAAU,EAAE,QAAQ,EAAE;YACtB,KAAK,CAAC,IAAI,CAAC,YAAY,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;SACjD;KACJ;SAAM;QACH,qFAAqF;QACrF,MAAM,EAAE,UAAU,EAAE,GAAG,gBAAgB,IAAI,EAAE,CAAA;QAC7C,IAAI,UAAU,EAAE;YACZ,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC,CAAA;SAC5G;KACJ;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAC1B,CAAC,CAAA;AAzCY,QAAA,YAAY,gBAyCxB"}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/language-server/utils.ts"],"names":[],"mappings":";;;AAWA,gCAMC;AAUD,4BAEC;AAED,8BAEC;AAED,8CAIC;AAED,8BAMC;AAED,0CAKC;AAED,gEAYC;AA5DD,2CAAwD;AAGxD,SAAgB,UAAU,CAAC,KAAc;IACrC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,OAAO,KAAK,YAAY,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,CAAA;AACrE,CAAC;AAED,SAAS,OAAO,CAAI,KAAQ;IACxB,OAAO,OAAQ,KAA4B,CAAC,IAAI,KAAK,QAAQ,CAAA;AACjE,CAAC;AAED,SAAS,OAAO,CAAC,KAAY;IACzB,OAAQ,KAA4B,CAAC,IAAI,YAAY,IAAI,CAAA;AAC7D,CAAC;AAED,SAAgB,QAAQ,CAAC,KAAc;IACnC,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAA;AACtD,CAAC;AAED,SAAgB,SAAS,CAAC,KAAc;IACpC,OAAO,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,CAAA;AAChD,CAAC;AAED,SAAgB,iBAAiB,CAAC,UAAsB;IACpD,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,CAAA;IAE9F,OAAO,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAA;AAC/C,CAAC;AAED,SAAgB,SAAS,CAAC,UAAkB;IACxC,IAAI,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAA;IACjC,CAAC;IAAC,MAAM,CAAC;QACL,MAAM,IAAI,KAAK,CAAC,+BAA+B,UAAU,EAAE,CAAC,CAAA;IAChE,CAAC;AACL,CAAC;AAED,SAAgB,eAAe,CAAC,KAAU;IACtC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,OAAO,CAAA;IACxB,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAA;AACxB,CAAC;AAED,SAAgB,0BAA0B,CAAC,mBAAwC;IAC/E,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CAAC,sCAA0B,CAAC,CAAA;IAC/C,CAAC;IAED,MAAM,WAAW,GAAG,mBAAmB,CAAC,cAAc,CAAC,QAAQ,CAAsB,CAAA;IAErF,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,sCAA0B,CAAC,CAAA;IAC/C,CAAC;IAED,OAAO,WAAW,CAAC,KAAK,CAAA;AAC5B,CAAC;AAEM,MAAM,aAAa,GAAG,CAAC,GAAQ,EAAE,MAAM,GAAG,EAAE,EAAE,EAAE;IACnD,MAAM,SAAS,GAAQ,EAAE,CAAA;IAEzB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QAC3B,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAA;QAEtB,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAChB,GAAG,GAAG,GAAG,GAAG,GAAG,CAAA;QACnB,CAAC;QAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,IAAA,qBAAa,EAAC,KAAK,EAAE,MAAM,GAAG,GAAG,CAAC,CAAC,CAAA;QAChE,CAAC;aAAM,CAAC;YACJ,SAAS,CAAC,MAAM,GAAG,GAAG,CAAC,GAAG,KAAK,CAAA;QACnC,CAAC;IACL,CAAC,CAAC,CAAA;IAEF,OAAO,SAAS,CAAA;AACpB,CAAC,CAAA;AAlBY,QAAA,aAAa,iBAkBzB;AAED,MAAM,iBAAiB,GAAG,sBAAsB,CAAA;AACzC,MAAM,YAAY,GAAG,CAAC,gBAAkC,EAAE,UAAuB,EAAU,EAAE;IAChG,MAAM,MAAM,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;IAEnD,MAAM,KAAK,GAAa,EAAE,CAAA;IAE1B,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IAE7B,sCAAsC;IACtC,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;QACnB,UAAU,CAAC,OAAO;YACd,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;YAChE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAA;IAC7C,CAAC;IAED,iCAAiC;IACjC,iEAAiE;IACjE,oEAAoE;IACpE,IAAI,gBAAgB,EAAE,qBAAqB,EAAE,GAAG,EAAE,CAAC;QAC/C,MAAM,EAAE,UAAU,EAAE,GAAG,gBAAgB,EAAE,qBAAqB,EAAE,GAAG,CAAA;QACnE,MAAM,EAAE,SAAS,EAAE,GAAG,UAAU,IAAI,EAAE,CAAA;QAEtC,IAAI,SAAS,EAAE,CAAC;YACZ,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC,CAAA;QAC1G,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACb,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC,CAAA;QAC7G,CAAC;QAED,IAAI,UAAU,EAAE,QAAQ,EAAE,CAAC;YACvB,KAAK,CAAC,IAAI,CAAC,YAAY,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;QAClD,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,qFAAqF;QACrF,MAAM,EAAE,UAAU,EAAE,GAAG,gBAAgB,IAAI,EAAE,CAAA;QAC7C,IAAI,UAAU,EAAE,CAAC;YACb,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC,CAAA;QAC7G,CAAC;IACL,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAC1B,CAAC,CAAA;AAzCY,QAAA,YAAY,gBAyCxB"}