@aws/lsp-codewhisperer 0.0.14 → 0.0.16

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

@@ -0,0 +1,85 @@
+"use strict";
+// Partial 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 filesystemUtilities_1 = require("./filesystemUtilities");
+describe('filesystemUtilities', function () {
+    describe('getFileDistance', function () {
+        let fileA;
+        let fileB;
+        it('distance 0', function () {
+            fileA = 'foo/bar/a.java';
+            fileB = 'foo/bar/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 0);
+        });
+        it('root distance 0', function () {
+            fileA = 'a.txt';
+            fileB = 'b.txt';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 0);
+        });
+        it('distance 0 with whitespace', function () {
+            fileA = 'foo/b%20ar/a.java';
+            fileB = 'foo/b%20ar/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 0);
+        });
+        it('distance 1', function () {
+            fileA = 'foo/bar/a.java';
+            fileB = 'foo/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 1);
+        });
+        it('distance 3', function () {
+            fileA = 'foo/bar/a.java';
+            fileB = 'lzz/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 3);
+        });
+        it('distance 4', function () {
+            fileA = 'foo/bar/a.java';
+            fileB = 'lzz/baz/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 4);
+        });
+        it('another distance 4', function () {
+            fileA = 'foo/a.py';
+            fileB = 'foo/foo/foo/foo/foo/b.py';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 4);
+        });
+        it('distance 5', function () {
+            fileA = 'foo/bar/a.java';
+            fileB = 'lzz/baz/zoo/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 5);
+        });
+        it('distance 6', function () {
+            fileA = 'foo/zoo/a.java';
+            fileB = 'bar/baz/bee/bww/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 6);
+        });
+        it('distance 6 with whitespaces', function () {
+            fileA = 'foo/zo%20o/a.java';
+            fileB = 'bar/ba%20z/bee/bww/b.java';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 6);
+        });
+        it('backslash distance 1', function () {
+            fileA = 'C:\\FOO\\BAR\\BAZ\\A.TXT';
+            fileB = 'C:\\FOO\\BAR\\B.TXT';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 1);
+        });
+        it('backslash distnace 3', function () {
+            fileA = 'C:\\FOO\\BAR\\BAZ\\LOO\\WOW\\A.txt';
+            fileB = 'C:\\FOO\\BAR\\B.txt';
+            const actual = (0, filesystemUtilities_1.getFileDistance)(fileA, fileB);
+            assert.strictEqual(actual, 3);
+        });
+    });
+});
+//# sourceMappingURL=filesystemUtilities.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"filesystemUtilities.test.js","sourceRoot":"","sources":["../../../src/language-server/utilities/filesystemUtilities.test.ts"],"names":[],"mappings":";AAAA,2DAA2D;AAC3D,oJAAoJ;;AAEpJ,iCAAgC;AAChC,+DAAuD;AAEvD,QAAQ,CAAC,qBAAqB,EAAE;IAC5B,QAAQ,CAAC,iBAAiB,EAAE;QACxB,IAAI,KAAa,CAAA;QACjB,IAAI,KAAa,CAAA;QAEjB,EAAE,CAAC,YAAY,EAAE;YACb,KAAK,GAAG,gBAAgB,CAAA;YACxB,KAAK,GAAG,gBAAgB,CAAA;YACxB,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iBAAiB,EAAE;YAClB,KAAK,GAAG,OAAO,CAAA;YACf,KAAK,GAAG,OAAO,CAAA;YACf,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,4BAA4B,EAAE;YAC7B,KAAK,GAAG,mBAAmB,CAAA;YAC3B,KAAK,GAAG,mBAAmB,CAAA;YAC3B,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,YAAY,EAAE;YACb,KAAK,GAAG,gBAAgB,CAAA;YACxB,KAAK,GAAG,YAAY,CAAA;YACpB,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,YAAY,EAAE;YACb,KAAK,GAAG,gBAAgB,CAAA;YACxB,KAAK,GAAG,YAAY,CAAA;YACpB,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,YAAY,EAAE;YACb,KAAK,GAAG,gBAAgB,CAAA;YACxB,KAAK,GAAG,gBAAgB,CAAA;YACxB,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oBAAoB,EAAE;YACrB,KAAK,GAAG,UAAU,CAAA;YAClB,KAAK,GAAG,0BAA0B,CAAA;YAClC,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,YAAY,EAAE;YACb,KAAK,GAAG,gBAAgB,CAAA;YACxB,KAAK,GAAG,oBAAoB,CAAA;YAC5B,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,YAAY,EAAE;YACb,KAAK,GAAG,gBAAgB,CAAA;YACxB,KAAK,GAAG,wBAAwB,CAAA;YAChC,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6BAA6B,EAAE;YAC9B,KAAK,GAAG,mBAAmB,CAAA;YAC3B,KAAK,GAAG,2BAA2B,CAAA;YACnC,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sBAAsB,EAAE;YACvB,KAAK,GAAG,0BAA0B,CAAA;YAClC,KAAK,GAAG,qBAAqB,CAAA;YAC7B,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sBAAsB,EAAE;YACvB,KAAK,GAAG,oCAAoC,CAAA;YAC5C,KAAK,GAAG,qBAAqB,CAAA;YAC7B,MAAM,MAAM,GAAG,IAAA,qCAAe,EAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/{pathUtils.d.ts → utilities/pathUtils.d.ts}

@@ -1,3 +1,5 @@
+/** Matches Windows drive letter ("C:"). */
+export declare const driveLetterRegex: RegExp;
 /**
  * Returns `true` if path `p` is a descendant of directory `d` (or if they are
  * identical).
@@ -15,3 +17,11 @@ export declare function isInDirectory(d: string, p: string): boolean;
  */
 export declare function normalizeSeparator(p: string): string;
 export declare function isUncPath(path: string): boolean;
+/**
+ * Normalizes path `p`:
+ * - Replaces backslashes "\\" with "/".
+ * - Removes redundant path separators (except initial double-slash for UNC-style paths).
+ * - Uppercases drive-letter (Windows).
+ * - ...and returns the result of `path.normalize()`.
+ */
+export declare function normalize(p: string): string;

package/out/language-server/{pathUtils.js → utilities/pathUtils.js}

@@ -1,7 +1,17 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isUncPath = exports.normalizeSeparator = exports.isInDirectory = void 0;
+exports.driveLetterRegex = void 0;
+exports.isInDirectory = isInDirectory;
+exports.normalizeSeparator = normalizeSeparator;
+exports.isUncPath = isUncPath;
+exports.normalize = normalize;
+// In order for this to work in non-NodeJS environments, servers must be built with bundler including `path-browserify` and `os-browserify` modules.
 const os = require("os");
+const path = require("path");
+// Partial port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/9d8ddbd85f4533e539a58e76f7c46883d8e50a79/packages/core/src/shared/utilities/pathUtils.ts
+/** Matches Windows drive letter ("C:"). */
+exports.driveLetterRegex = /^[a-zA-Z]\:/;
 /**
  * Returns `true` if path `p` is a descendant of directory `d` (or if they are
  * identical).
@@ -32,7 +42,6 @@ function isInDirectory(d, p) {
             : value === containedPathPieces[index];
     });
 }
-exports.isInDirectory = isInDirectory;
 /**
  * - Replaces backslashes "\\" with "/"
  * - Removes redundant path separators (except initial double-slash for UNC-style paths).
@@ -44,9 +53,27 @@ function normalizeSeparator(p) {
     }
     return normalized;
 }
-exports.normalizeSeparator = normalizeSeparator;
 function isUncPath(path) {
     return /^\s*[\/\\]{2}[^\/\\]+/.test(path);
 }
-exports.isUncPath = isUncPath;
+/**
+ * Normalizes path `p`:
+ * - Replaces backslashes "\\" with "/".
+ * - Removes redundant path separators (except initial double-slash for UNC-style paths).
+ * - Uppercases drive-letter (Windows).
+ * - ...and returns the result of `path.normalize()`.
+ */
+function normalize(p) {
+    if (!p || p.length === 0) {
+        return p;
+    }
+    const firstChar = p.substring(0, 1);
+    if (exports.driveLetterRegex.test(p.substring(0, 2))) {
+        return normalizeSeparator(path.normalize(firstChar.toUpperCase() + p.substring(1)));
+    }
+    if (isUncPath(p)) {
+        return normalizeSeparator(p);
+    }
+    return normalizeSeparator(path.normalize(p));
+}
 //# sourceMappingURL=pathUtils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"pathUtils.js","sourceRoot":"","sources":["../../../src/language-server/utilities/pathUtils.ts"],"names":[],"mappings":";;;AAoBA,sCAsBC;AAMD,gDAMC;AAED,8BAEC;AASD,8BAYC;AA/ED,oJAAoJ;AACpJ,yBAAwB;AACxB,6BAA4B;AAE5B,2DAA2D;AAC3D,0IAA0I;AAE1I,2CAA2C;AAC9B,QAAA,gBAAgB,GAAG,aAAa,CAAA;AAE7C;;;;;;;;;GASG;AACH,SAAgB,aAAa,CAAC,CAAS,EAAE,CAAS;IAC9C,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;QACvB,OAAO,IAAI,CAAA;IACf,CAAC;IACD,MAAM,eAAe,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IACxD,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAE5D,IAAI,eAAe,CAAC,MAAM,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC;QACtD,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,6DAA6D;IAC7D,OAAO,eAAe,CAAC,MAAM,GAAG,CAAC,IAAI,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;QACtF,eAAe,CAAC,GAAG,EAAE,CAAA;IACzB,CAAC;IACD,MAAM,eAAe,GAAG,EAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAA;IAEjD,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QAC1C,OAAO,eAAe;YAClB,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,mBAAmB,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE;YAClE,CAAC,CAAC,KAAK,KAAK,mBAAmB,CAAC,KAAK,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;AACN,CAAC;AAED;;;GAGG;AACH,SAAgB,kBAAkB,CAAC,CAAS;IACxC,MAAM,UAAU,GAAG,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC,CAAA;IAC7C,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QACf,OAAO,GAAG,GAAG,UAAU,CAAA;IAC3B,CAAC;IACD,OAAO,UAAU,CAAA;AACrB,CAAC;AAED,SAAgB,SAAS,CAAC,IAAY;IAClC,OAAO,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;AAC7C,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,SAAS,CAAC,CAAS;IAC/B,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,CAAC,CAAA;IACZ,CAAC;IACD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IACnC,IAAI,wBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,OAAO,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IACvF,CAAC;IACD,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QACf,OAAO,kBAAkB,CAAC,CAAC,CAAC,CAAA;IAChC,CAAC;IACD,OAAO,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAA;AAChD,CAAC"}

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

@@ -0,0 +1,43 @@
+"use strict";
+// Partial port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/24840fda8559a3e3ace3517ad9844db76680dc50/packages/core/src/test/shared/utilities/pathUtils.test.ts
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const pathUtils_1 = require("./pathUtils");
+describe('pathUtils', async function () {
+    it('normalizeSeparator()', function () {
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('a/b/c'), 'a/b/c');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('a\\b\\c'), 'a/b/c');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('a\\\\b\\c\\/\\'), 'a/b/c/');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('/a\\\\b\\c\\/\\/'), '/a/b/c/');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('//\\\\\\\\/\\//'), '/');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('a\\b\\c'), 'a/b/c');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('//////'), '/');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('//UNC///////path'), '//UNC/path');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('\\\\UNC\\path'), '//UNC/path');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('/'), '/');
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)(''), '');
+        // Preserves double-slash at start (UNC path).
+        assert.strictEqual((0, pathUtils_1.normalizeSeparator)('\\\\codebuild\\tmp\\output\\js-manifest-in-root\\'), '//codebuild/tmp/output/js-manifest-in-root/');
+    });
+    it('normalize()', function () {
+        assert.strictEqual((0, pathUtils_1.normalize)('../../FOO/BAR'), '../../FOO/BAR');
+        assert.strictEqual((0, pathUtils_1.normalize)('c:\\foo\\bar.txt'), 'C:/foo/bar.txt');
+        assert.strictEqual((0, pathUtils_1.normalize)('C:\\foo\\bar.txt'), 'C:/foo/bar.txt');
+        assert.strictEqual((0, pathUtils_1.normalize)('c:/foo/bar.txt'), 'C:/foo/bar.txt');
+        assert.strictEqual((0, pathUtils_1.normalize)('c:/foo'), 'C:/foo');
+        assert.strictEqual((0, pathUtils_1.normalize)('/foo/bar.txt'), '/foo/bar.txt');
+        assert.strictEqual((0, pathUtils_1.normalize)('/foo/bar'), '/foo/bar');
+        assert.strictEqual((0, pathUtils_1.normalize)('\\foo/bar\\'), '/foo/bar/');
+        assert.strictEqual((0, pathUtils_1.normalize)('/foo/'), '/foo/');
+        assert.strictEqual((0, pathUtils_1.normalize)('//////'), '/');
+        assert.strictEqual((0, pathUtils_1.normalize)('//UNC///////path'), '//UNC/path');
+        assert.strictEqual((0, pathUtils_1.normalize)('\\\\UNC\\path'), '//UNC/path');
+        assert.strictEqual((0, pathUtils_1.normalize)('/'), '/');
+        assert.strictEqual((0, pathUtils_1.normalize)(''), '');
+        assert.strictEqual((0, pathUtils_1.normalize)('a/b/c'), 'a/b/c');
+        // Preserves double-slash at start (UNC path).
+        assert.strictEqual((0, pathUtils_1.normalize)('\\\\codebuild\\tmp\\output\\js-manifest-in-root\\'), '//codebuild/tmp/output/js-manifest-in-root/');
+    });
+});
+//# sourceMappingURL=pathUtils.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"pathUtils.test.js","sourceRoot":"","sources":["../../../src/language-server/utilities/pathUtils.test.ts"],"names":[],"mappings":";AAAA,2DAA2D;AAC3D,oJAAoJ;;AAEpJ,iCAAgC;AAChC,2CAA2D;AAE3D,QAAQ,CAAC,WAAW,EAAE,KAAK;IACvB,EAAE,CAAC,sBAAsB,EAAE;QACvB,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAA;QACxD,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAA;QAC1D,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,CAAA;QAClE,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,kBAAkB,CAAC,EAAE,SAAS,CAAC,CAAA;QACrE,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,iBAAiB,CAAC,EAAE,GAAG,CAAC,CAAA;QAC9D,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAA;QAC1D,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAA;QACrD,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,kBAAkB,CAAC,EAAE,YAAY,CAAC,CAAA;QACxE,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,eAAe,CAAC,EAAE,YAAY,CAAC,CAAA;QACrE,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,GAAG,CAAC,EAAE,GAAG,CAAC,CAAA;QAChD,MAAM,CAAC,WAAW,CAAC,IAAA,8BAAkB,EAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;QAE9C,8CAA8C;QAC9C,MAAM,CAAC,WAAW,CACd,IAAA,8BAAkB,EAAC,mDAAmD,CAAC,EACvE,6CAA6C,CAChD,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,aAAa,EAAE;QACd,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,eAAe,CAAC,EAAE,eAAe,CAAC,CAAA;QAC/D,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,kBAAkB,CAAC,EAAE,gBAAgB,CAAC,CAAA;QACnE,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,kBAAkB,CAAC,EAAE,gBAAgB,CAAC,CAAA;QACnE,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,gBAAgB,CAAC,EAAE,gBAAgB,CAAC,CAAA;QACjE,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAA;QACjD,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,cAAc,CAAC,EAAE,cAAc,CAAC,CAAA;QAC7D,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,UAAU,CAAC,EAAE,UAAU,CAAC,CAAA;QACrD,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,aAAa,CAAC,EAAE,WAAW,CAAC,CAAA;QACzD,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAA;QAC/C,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAA;QAC5C,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,kBAAkB,CAAC,EAAE,YAAY,CAAC,CAAA;QAC/D,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,eAAe,CAAC,EAAE,YAAY,CAAC,CAAA;QAC5D,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,GAAG,CAAC,EAAE,GAAG,CAAC,CAAA;QACvC,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;QACrC,MAAM,CAAC,WAAW,CAAC,IAAA,qBAAS,EAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAA;QAE/C,8CAA8C;QAC9C,MAAM,CAAC,WAAW,CACd,IAAA,qBAAS,EAAC,mDAAmD,CAAC,EAC9D,6CAA6C,CAChD,CAAA;IACL,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

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

@@ -0,0 +1,13 @@
+import { TextDocument } from '@aws/language-server-runtimes/server-interface';
+export interface utgLanguageConfig {
+    extension: string;
+    testFilenamePattern: RegExp;
+    functionExtractionPattern: RegExp;
+    classExtractionPattern: RegExp;
+    importStatementRegExp: RegExp;
+}
+export declare const utgLanguageConfigs: Record<string, utgLanguageConfig>;
+export declare function isTestFile(filePath: string, languageConfig: {
+    languageId: TextDocument['languageId'];
+    fileContent?: string;
+}): boolean;

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

@@ -0,0 +1,48 @@
+"use strict";
+// Partial port of implementation in AWS Toolkit for VSCode
+// https://github.com/aws/aws-toolkit-vscode/blob/9d8ddbd85f4533e539a58e76f7c46883d8e50a79/packages/core/src/codewhisperer/util/supplementalContext/codeParsingUtil.ts
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.utgLanguageConfigs = void 0;
+exports.isTestFile = isTestFile;
+const path = require("path");
+const pathUtils_1 = require("../pathUtils");
+exports.utgLanguageConfigs = {
+    // Java regexes are not working efficiently for class or function extraction
+    java: {
+        extension: '.java',
+        testFilenamePattern: /(?:Test([^/\\]+)\.java|([^/\\]+)Test\.java|([^/\\]+)Tests\.java)$/,
+        functionExtractionPattern: /(?:(?:public|private|protected)\s+)(?:static\s+)?(?:[\w<>]+\s+)?(\w+)\s*\([^)]*\)\s*(?:(?:throws\s+\w+)?\s*)[{;]/gm, // TODO: Doesn't work for generice <T> T functions.
+        classExtractionPattern: /(?<=^|\n)\s*public\s+class\s+(\w+)/gm, // TODO: Verify these.
+        importStatementRegExp: /import .*\.([a-zA-Z0-9]+);/,
+    },
+    python: {
+        extension: '.py',
+        testFilenamePattern: /(?:test_([^/\\]+)\.py|([^/\\]+)_test\.py)$/,
+        functionExtractionPattern: /def\s+([a-zA-Z_][a-zA-Z0-9_]*)\s*\(/g, // Worked fine
+        classExtractionPattern: /^class\s+(\w+)\s*:/gm,
+        importStatementRegExp: /from (.*) import.*/,
+    },
+};
+function isTestFile(filePath, languageConfig) {
+    const normalizedFilePath = (0, pathUtils_1.normalize)(filePath);
+    const pathContainsTest = normalizedFilePath.includes('tests/') ||
+        normalizedFilePath.includes('test/') ||
+        normalizedFilePath.includes('tst/');
+    const fileNameMatchTestPatterns = isTestFileByName(normalizedFilePath, languageConfig.languageId);
+    if (pathContainsTest || fileNameMatchTestPatterns) {
+        return true;
+    }
+    return false;
+}
+function isTestFileByName(filePath, language) {
+    const languageConfig = exports.utgLanguageConfigs[language];
+    if (!languageConfig) {
+        // We have enabled the support only for python and Java for this check
+        // as we depend on Regex for this validation.
+        return false;
+    }
+    const testFilenamePattern = languageConfig.testFilenamePattern;
+    const filename = path.basename(filePath);
+    return testFilenamePattern.test(filename);
+}
+//# sourceMappingURL=codeParsingUtil.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"codeParsingUtil.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/codeParsingUtil.ts"],"names":[],"mappings":";AAAA,2DAA2D;AAC3D,sKAAsK;;;AAiCtK,gCAmBC;AAlDD,6BAA4B;AAC5B,4CAAwC;AAW3B,QAAA,kBAAkB,GAAsC;IACjE,4EAA4E;IAC5E,IAAI,EAAE;QACF,SAAS,EAAE,OAAO;QAClB,mBAAmB,EAAE,mEAAmE;QACxF,yBAAyB,EACrB,oHAAoH,EAAE,mDAAmD;QAC7K,sBAAsB,EAAE,sCAAsC,EAAE,sBAAsB;QACtF,qBAAqB,EAAE,4BAA4B;KACtD;IACD,MAAM,EAAE;QACJ,SAAS,EAAE,KAAK;QAChB,mBAAmB,EAAE,4CAA4C;QACjE,yBAAyB,EAAE,sCAAsC,EAAE,cAAc;QACjF,sBAAsB,EAAE,sBAAsB;QAC9C,qBAAqB,EAAE,oBAAoB;KAC9C;CACJ,CAAA;AAED,SAAgB,UAAU,CACtB,QAAgB,EAChB,cAGC;IAED,MAAM,kBAAkB,GAAG,IAAA,qBAAS,EAAC,QAAQ,CAAC,CAAA;IAC9C,MAAM,gBAAgB,GAClB,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACrC,kBAAkB,CAAC,QAAQ,CAAC,OAAO,CAAC;QACpC,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IACvC,MAAM,yBAAyB,GAAG,gBAAgB,CAAC,kBAAkB,EAAE,cAAc,CAAC,UAAU,CAAC,CAAA;IAEjG,IAAI,gBAAgB,IAAI,yBAAyB,EAAE,CAAC;QAChD,OAAO,IAAI,CAAA;IACf,CAAC;IAED,OAAO,KAAK,CAAA;AAChB,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,QAAoC;IAC5E,MAAM,cAAc,GAAG,0BAAkB,CAAC,QAAQ,CAAC,CAAA;IACnD,IAAI,CAAC,cAAc,EAAE,CAAC;QAClB,sEAAsE;QACtE,6CAA6C;QAC7C,OAAO,KAAK,CAAA;IAChB,CAAC;IACD,MAAM,mBAAmB,GAAG,cAAc,CAAC,mBAAmB,CAAA;IAE9D,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;IAExC,OAAO,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;AAC7C,CAAC"}

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

@@ -0,0 +1,5 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

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

@@ -0,0 +1,49 @@
+"use strict";
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+const codeParsingUtil_1 = require("./codeParsingUtil");
+const assert = require("assert");
+describe('isTestFile', () => {
+    it('should return true if the file name matches the test filename pattern - Java', async () => {
+        const filePaths = ['/path/to/MyClassTest.java', '/path/to/TestMyClass.java', '/path/to/MyClassTests.java'];
+        const language = 'java';
+        for (const filePath of filePaths) {
+            const result = await (0, codeParsingUtil_1.isTestFile)(filePath, { languageId: language });
+            assert.strictEqual(result, true);
+        }
+    });
+    it('should return false if the file name does not match the test filename pattern - Java', async () => {
+        const filePaths = ['/path/to/MyClass.java', '/path/to/MyClass_test.java', '/path/to/test_MyClass.java'];
+        const language = 'java';
+        for (const filePath of filePaths) {
+            const result = await (0, codeParsingUtil_1.isTestFile)(filePath, { languageId: language });
+            assert.strictEqual(result, false);
+        }
+    });
+    it('should return true if the file name does not match the test filename pattern - Python', async () => {
+        const filePaths = ['/path/to/util_test.py', '/path/to/test_util.py'];
+        const language = 'python';
+        for (const filePath of filePaths) {
+            const result = await (0, codeParsingUtil_1.isTestFile)(filePath, { languageId: language });
+            assert.strictEqual(result, true);
+        }
+    });
+    it('should return false if the file name does not match the test filename pattern - Python', async () => {
+        const filePaths = ['/path/to/util.py', '/path/to/utilTest.java', '/path/to/Testutil.java'];
+        const language = 'python';
+        for (const filePath of filePaths) {
+            const result = await (0, codeParsingUtil_1.isTestFile)(filePath, { languageId: language });
+            assert.strictEqual(result, false);
+        }
+    });
+    it('should return false if the language is not supported', async () => {
+        const filePath = '/path/to/MyClass.cpp';
+        const language = 'c++';
+        const result = await (0, codeParsingUtil_1.isTestFile)(filePath, { languageId: language });
+        assert.strictEqual(result, false);
+    });
+});
+//# sourceMappingURL=codeParsingUtil.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"codeParsingUtil.test.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/codeParsingUtil.test.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAEH,uDAA8C;AAC9C,iCAAgC;AAEhC,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,SAAS,GAAG,CAAC,2BAA2B,EAAE,2BAA2B,EAAE,4BAA4B,CAAC,CAAA;QAC1G,MAAM,QAAQ,GAAG,MAAM,CAAA;QAEvB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAA,4BAAU,EAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;YACnE,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;QACpC,CAAC;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QAClG,MAAM,SAAS,GAAG,CAAC,uBAAuB,EAAE,4BAA4B,EAAE,4BAA4B,CAAC,CAAA;QACvG,MAAM,QAAQ,GAAG,MAAM,CAAA;QAEvB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAA,4BAAU,EAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;YACnE,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;QACrC,CAAC;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACnG,MAAM,SAAS,GAAG,CAAC,uBAAuB,EAAE,uBAAuB,CAAC,CAAA;QACpE,MAAM,QAAQ,GAAG,QAAQ,CAAA;QAEzB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAA,4BAAU,EAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;YACnE,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;QACpC,CAAC;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;QACpG,MAAM,SAAS,GAAG,CAAC,kBAAkB,EAAE,wBAAwB,EAAE,wBAAwB,CAAC,CAAA;QAC1F,MAAM,QAAQ,GAAG,QAAQ,CAAA;QAEzB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAA,4BAAU,EAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;YACnE,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;QACrC,CAAC;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,sBAAsB,CAAA;QACvC,MAAM,QAAQ,GAAG,KAAK,CAAA;QACtB,MAAM,MAAM,GAAG,MAAM,IAAA,4BAAU,EAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;QACnE,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

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

@@ -0,0 +1,18 @@
+import { CodeWhispererSupplementalContext } from '../../models/model';
+import { CancellationToken, Position, TextDocument, Workspace } from '@aws/language-server-runtimes/server-interface';
+type CrossFileSupportedLanguage = 'java' | 'python' | 'javascript' | 'typescript' | 'javascriptreact' | 'typescriptreact';
+export declare const supportedLanguageToDialects: Readonly<Record<CrossFileSupportedLanguage, Set<string>>>;
+interface Chunk {
+    fileName: string;
+    content: string;
+    nextContent: string;
+    score?: number;
+}
+export declare function fetchSupplementalContextForSrc(document: TextDocument, position: Position, workspace: Workspace, cancellationToken: CancellationToken): Promise<Pick<CodeWhispererSupplementalContext, 'supplementalContextItems' | 'strategy'> | undefined>;
+export declare function splitFileToChunks(document: TextDocument, chunkSize: number): Chunk[];
+/**
+ * This function will return relevant cross files sorted by file distance for the given editor file
+ * by referencing open files, imported files and same package files.
+ */
+export declare function getCrossFileCandidates(document: TextDocument, workspace: Workspace): Promise<TextDocument[]>;
+export {};

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

@@ -0,0 +1,205 @@
+"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/crossFileContextUtil.ts
+// Implementation is converted to work with LSP TextDocument instead of vscode APIs.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.supportedLanguageToDialects = void 0;
+exports.fetchSupplementalContextForSrc = fetchSupplementalContextForSrc;
+exports.splitFileToChunks = splitFileToChunks;
+exports.getCrossFileCandidates = getCrossFileCandidates;
+const path = require("path");
+const rankBm25_1 = require("./rankBm25");
+const constants_1 = require("../../models/constants");
+const codeParsingUtil_1 = require("./codeParsingUtil");
+const filesystemUtilities_1 = require("../filesystemUtilities");
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const supplementalContextUtil_1 = require("./supplementalContextUtil");
+// TODO: ugly, can we make it prettier? like we have to manually type 'java', 'javascriptreact' which is error prone
+// TODO: Move to another config file or constants file
+// Supported language to its corresponding file ext
+exports.supportedLanguageToDialects = {
+    java: new Set(['.java']),
+    python: new Set(['.py']),
+    javascript: new Set(['.js', '.jsx']),
+    javascriptreact: new Set(['.js', '.jsx']),
+    typescript: new Set(['.ts', '.tsx']),
+    typescriptreact: new Set(['.ts', '.tsx']),
+};
+function isCrossFileSupported(languageId) {
+    return Object.keys(exports.supportedLanguageToDialects).includes(languageId);
+}
+async function fetchSupplementalContextForSrc(document, position, workspace, cancellationToken) {
+    const shouldProceed = shouldFetchCrossFileContext(document.languageId);
+    if (!shouldProceed) {
+        return shouldProceed === undefined
+            ? undefined
+            : {
+                supplementalContextItems: [],
+                strategy: 'Empty',
+            };
+    }
+    const codeChunksCalculated = constants_1.crossFileContextConfig.numberOfChunkToFetch;
+    // Step 1: Get relevant cross files to refer
+    const relevantCrossFileCandidates = await getCrossFileCandidates(document, workspace);
+    throwIfCancelled(cancellationToken);
+    // Step 2: Split files to chunks with upper bound on chunkCount
+    // We restrict the total number of chunks to improve on latency.
+    // Chunk linking is required as we want to pass the next chunk value for matched chunk.
+    let chunkList = [];
+    for (const relevantFile of relevantCrossFileCandidates) {
+        throwIfCancelled(cancellationToken);
+        const chunks = splitFileToChunks(relevantFile, constants_1.crossFileContextConfig.numberOfLinesEachChunk);
+        const linkedChunks = linkChunks(chunks);
+        chunkList.push(...linkedChunks);
+        if (chunkList.length >= codeChunksCalculated) {
+            break;
+        }
+    }
+    // it's required since chunkList.push(...) is likely giving us a list of size > 60
+    chunkList = chunkList.slice(0, codeChunksCalculated);
+    // Step 3: Generate Input chunk (10 lines left of cursor position)
+    // and Find Best K chunks w.r.t input chunk using BM25
+    const inputChunk = getInputChunk(document, position, constants_1.crossFileContextConfig.numberOfLinesEachChunk);
+    const bestChunks = findBestKChunkMatches(inputChunk, chunkList, constants_1.crossFileContextConfig.topK);
+    throwIfCancelled(cancellationToken);
+    // Step 4: Transform best chunks to supplemental contexts
+    const supplementalContexts = [];
+    for (const chunk of bestChunks) {
+        throwIfCancelled(cancellationToken);
+        supplementalContexts.push({
+            filePath: chunk.fileName,
+            content: chunk.nextContent,
+            score: chunk.score,
+        });
+    }
+    // DO NOT send code chunk with empty content
+    return {
+        supplementalContextItems: supplementalContexts.filter(item => item.content.trim().length !== 0),
+        strategy: 'OpenTabs_BM25',
+    };
+}
+function findBestKChunkMatches(chunkInput, chunkReferences, k) {
+    const chunkContentList = chunkReferences.map(chunk => chunk.content);
+    //performBM25Scoring returns the output in a sorted order (descending of scores)
+    const top3 = new rankBm25_1.BM25Okapi(chunkContentList).topN(chunkInput.content, constants_1.crossFileContextConfig.topK);
+    return top3.map(doc => {
+        // reference to the original metadata since BM25.top3 will sort the result
+        const chunkIndex = doc.index;
+        const chunkReference = chunkReferences[chunkIndex];
+        return {
+            content: chunkReference.content,
+            fileName: chunkReference.fileName,
+            nextContent: chunkReference.nextContent,
+            score: doc.score,
+        };
+    });
+}
+/* This extract 10 lines to the left of the cursor from trigger file.
+ * This will be the inputquery to bm25 matching against list of cross-file chunks
+ */
+function getInputChunk(document, cursorPosition, chunkSize) {
+    const startLine = Math.max(cursorPosition.line - chunkSize, 0);
+    const endLine = Math.max(cursorPosition.line - 1, 0);
+    const endCharacter = document.getText(server_interface_1.Range.create(endLine, 0, endLine + 1, 0)).trimRight().length;
+    const inputChunkContent = document
+        .getText(server_interface_1.Range.create(startLine, 0, endLine, endCharacter))
+        .replaceAll('\r\n', '\n');
+    const inputChunk = { fileName: path.basename(document.uri), content: inputChunkContent, nextContent: '' };
+    return inputChunk;
+}
+/**
+ * Note: UserGroup is not used in VsCode implementation, keeping param for reference, but not implemented.
+ *
+ * Util to decide if we need to fetch crossfile context since CodeWhisperer CrossFile Context feature is gated by userGroup and language level
+ * @param languageId: LSP TextDocument language Identifier
+ * @param userGroup: CodeWhisperer user group settings, refer to userGroupUtil.ts
+ * @returns specifically returning undefined if the langueage is not supported,
+ * otherwise true/false depending on if the language is fully supported or not belonging to the user group
+ */
+function shouldFetchCrossFileContext(languageId, _userGroup) {
+    if (!isCrossFileSupported(languageId)) {
+        return undefined;
+    }
+    return true;
+}
+/**
+ * This linking is required from science experimentations to pass the next contnet chunk
+ * when a given chunk context passes the match in BM25.
+ * Special handling is needed for last(its next points to its own) and first chunk
+ */
+function linkChunks(chunks) {
+    const updatedChunks = [];
+    // This additional chunk is needed to create a next pointer to chunk 0.
+    const firstChunk = chunks[0];
+    const firstChunkSubContent = firstChunk.content.split('\n').slice(0, 3).join('\n').trimEnd();
+    const newFirstChunk = {
+        fileName: firstChunk.fileName,
+        content: firstChunkSubContent,
+        nextContent: firstChunk.content,
+    };
+    updatedChunks.push(newFirstChunk);
+    const n = chunks.length;
+    for (let i = 0; i < n; i++) {
+        const chunk = chunks[i];
+        const nextChunk = i < n - 1 ? chunks[i + 1] : chunk;
+        chunk.nextContent = nextChunk.content;
+        updatedChunks.push(chunk);
+    }
+    return updatedChunks;
+}
+function splitFileToChunks(document, chunkSize) {
+    const chunks = [];
+    const fileContent = document.getText().trimEnd().replaceAll('\r\n', '\n');
+    const lines = fileContent.split('\n');
+    for (let i = 0; i < lines.length; i += chunkSize) {
+        const chunkContent = lines.slice(i, Math.min(i + chunkSize, lines.length)).join('\n');
+        const chunk = { fileName: new URL(document.uri).pathname, content: chunkContent.trimEnd(), nextContent: '' };
+        chunks.push(chunk);
+    }
+    return chunks;
+}
+/**
+ * This function will return relevant cross files sorted by file distance for the given editor file
+ * by referencing open files, imported files and same package files.
+ */
+async function getCrossFileCandidates(document, workspace) {
+    const targetFile = document.uri;
+    const language = document.languageId;
+    const dialects = exports.supportedLanguageToDialects[language];
+    /**
+     * Consider a file which
+     * 1. is different from the target
+     * 2. has the same file extension or it's one of the dialect of target file (e.g .js vs. .jsx)
+     * 3. is not a test file
+     *
+     * Porting note: this function relies of Workspace feature to get all documents,
+     * managed by this language server, instead of VSCode `vscode.window` API as VSCode toolkit does.
+     */
+    const unsortedCandidates = await workspace.getAllTextDocuments();
+    return unsortedCandidates
+        .filter((candidateFile) => {
+        return !!(targetFile !== candidateFile.uri &&
+            (path.extname(targetFile) === path.extname(candidateFile.uri) ||
+                (dialects && dialects.has(path.extname(candidateFile.uri)))) &&
+            !(0, codeParsingUtil_1.isTestFile)(new URL(candidateFile.uri).pathname, { languageId: language }));
+    })
+        .map((candidate) => {
+        return {
+            file: candidate,
+            // PORT_TODO: port and test getFileDistance to work with LSP's URIs
+            fileDistance: (0, filesystemUtilities_1.getFileDistance)(targetFile, candidate.uri),
+        };
+    })
+        .sort((file1, file2) => {
+        return file1.fileDistance - file2.fileDistance;
+    })
+        .map((fileToDistance) => {
+        return fileToDistance.file;
+    });
+}
+function throwIfCancelled(token) {
+    if (token.isCancellationRequested) {
+        throw new supplementalContextUtil_1.CancellationError();
+    }
+}
+//# sourceMappingURL=crossFileContextUtil.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"crossFileContextUtil.js","sourceRoot":"","sources":["../../../../src/language-server/utilities/supplementalContextUtil/crossFileContextUtil.ts"],"names":[],"mappings":";AAAA,mDAAmD;AACnD,2KAA2K;AAC3K,oFAAoF;;;AAgDpF,wEAgEC;AAkFD,8CAYC;AAWD,wDAqCC;AA5PD,6BAA4B;AAC5B,yCAAoD;AACpD,sDAA+D;AAC/D,uDAA8C;AAC9C,gEAAwD;AAExD,qFAMuD;AACvD,uEAA6D;AAU7D,oHAAoH;AACpH,sDAAsD;AACtD,mDAAmD;AACtC,QAAA,2BAA2B,GAA8D;IAClG,IAAI,EAAE,IAAI,GAAG,CAAS,CAAC,OAAO,CAAC,CAAC;IAChC,MAAM,EAAE,IAAI,GAAG,CAAS,CAAC,KAAK,CAAC,CAAC;IAChC,UAAU,EAAE,IAAI,GAAG,CAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC5C,eAAe,EAAE,IAAI,GAAG,CAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACjD,UAAU,EAAE,IAAI,GAAG,CAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC5C,eAAe,EAAE,IAAI,GAAG,CAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;CACpD,CAAA;AAED,SAAS,oBAAoB,CAAC,UAAkB;IAC5C,OAAO,MAAM,CAAC,IAAI,CAAC,mCAA2B,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;AACxE,CAAC;AASM,KAAK,UAAU,8BAA8B,CAChD,QAAsB,EACtB,QAAkB,EAClB,SAAoB,EACpB,iBAAoC;IAEpC,MAAM,aAAa,GAAG,2BAA2B,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;IAEtE,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,OAAO,aAAa,KAAK,SAAS;YAC9B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC;gBACI,wBAAwB,EAAE,EAAE;gBAC5B,QAAQ,EAAE,OAAO;aACpB,CAAA;IACX,CAAC;IAED,MAAM,oBAAoB,GAAG,kCAAsB,CAAC,oBAAoB,CAAA;IAExE,4CAA4C;IAC5C,MAAM,2BAA2B,GAAG,MAAM,sBAAsB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAA;IAErF,gBAAgB,CAAC,iBAAiB,CAAC,CAAA;IAEnC,+DAA+D;IAC/D,gEAAgE;IAChE,uFAAuF;IACvF,IAAI,SAAS,GAAY,EAAE,CAAA;IAC3B,KAAK,MAAM,YAAY,IAAI,2BAA2B,EAAE,CAAC;QACrD,gBAAgB,CAAC,iBAAiB,CAAC,CAAA;QACnC,MAAM,MAAM,GAAY,iBAAiB,CAAC,YAAY,EAAE,kCAAsB,CAAC,sBAAsB,CAAC,CAAA;QACtG,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,CAAC,CAAA;QACvC,SAAS,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAA;QAC/B,IAAI,SAAS,CAAC,MAAM,IAAI,oBAAoB,EAAE,CAAC;YAC3C,MAAK;QACT,CAAC;IACL,CAAC;IAED,kFAAkF;IAClF,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAA;IAEpD,kEAAkE;IAClE,sDAAsD;IACtD,MAAM,UAAU,GAAU,aAAa,CAAC,QAAQ,EAAE,QAAQ,EAAE,kCAAsB,CAAC,sBAAsB,CAAC,CAAA;IAC1G,MAAM,UAAU,GAAY,qBAAqB,CAAC,UAAU,EAAE,SAAS,EAAE,kCAAsB,CAAC,IAAI,CAAC,CAAA;IACrG,gBAAgB,CAAC,iBAAiB,CAAC,CAAA;IAEnC,yDAAyD;IACzD,MAAM,oBAAoB,GAA2C,EAAE,CAAA;IACvE,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QAC7B,gBAAgB,CAAC,iBAAiB,CAAC,CAAA;QAEnC,oBAAoB,CAAC,IAAI,CAAC;YACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,OAAO,EAAE,KAAK,CAAC,WAAW;YAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;SACrB,CAAC,CAAA;IACN,CAAC;IAED,4CAA4C;IAC5C,OAAO;QACH,wBAAwB,EAAE,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,CAAC;QAC/F,QAAQ,EAAE,eAAe;KAC5B,CAAA;AACL,CAAC;AAED,SAAS,qBAAqB,CAAC,UAAiB,EAAE,eAAwB,EAAE,CAAS;IACjF,MAAM,gBAAgB,GAAG,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IAEpE,gFAAgF;IAChF,MAAM,IAAI,GAAmB,IAAI,oBAAS,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,kCAAsB,CAAC,IAAI,CAAC,CAAA;IAElH,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QAClB,0EAA0E;QAC1E,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAA;QAC5B,MAAM,cAAc,GAAG,eAAe,CAAC,UAAU,CAAC,CAAA;QAClD,OAAO;YACH,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,QAAQ,EAAE,cAAc,CAAC,QAAQ;YACjC,WAAW,EAAE,cAAc,CAAC,WAAW;YACvC,KAAK,EAAE,GAAG,CAAC,KAAK;SACnB,CAAA;IACL,CAAC,CAAC,CAAA;AACN,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,QAAsB,EAAE,cAAwB,EAAE,SAAiB;IACtF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,GAAG,SAAS,EAAE,CAAC,CAAC,CAAA;IAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC,CAAC,CAAA;IACpD,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,wBAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,MAAM,CAAA;IAClG,MAAM,iBAAiB,GAAG,QAAQ;SAC7B,OAAO,CAAC,wBAAK,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;SAC1D,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IAC7B,MAAM,UAAU,GAAU,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,EAAE,CAAA;IAChH,OAAO,UAAU,CAAA;AACrB,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,2BAA2B,CAAC,UAAsC,EAAE,UAAgB;IACzF,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC;QACpC,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,OAAO,IAAI,CAAA;AACf,CAAC;AAED;;;;GAIG;AACH,SAAS,UAAU,CAAC,MAAe;IAC/B,MAAM,aAAa,GAAY,EAAE,CAAA;IAEjC,uEAAuE;IACvE,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,CAAA;IAC5B,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,CAAA;IAC5F,MAAM,aAAa,GAAG;QAClB,QAAQ,EAAE,UAAU,CAAC,QAAQ;QAC7B,OAAO,EAAE,oBAAoB;QAC7B,WAAW,EAAE,UAAU,CAAC,OAAO;KAClC,CAAA;IACD,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IAEjC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAA;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAA;QACvB,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;QAEnD,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,OAAO,CAAA;QACrC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IAC7B,CAAC;IAED,OAAO,aAAa,CAAA;AACxB,CAAC;AAED,SAAgB,iBAAiB,CAAC,QAAsB,EAAE,SAAiB;IACvE,MAAM,MAAM,GAAY,EAAE,CAAA;IAE1B,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IACzE,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;QAC/C,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACrF,MAAM,KAAK,GAAG,EAAE,QAAQ,EAAE,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAA;QAC5G,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACtB,CAAC;IACD,OAAO,MAAM,CAAA;AACjB,CAAC;AAOD;;;GAGG;AACI,KAAK,UAAU,sBAAsB,CAAC,QAAsB,EAAE,SAAoB;IACrF,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAA;IAC/B,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAwC,CAAA;IAClE,MAAM,QAAQ,GAAG,mCAA2B,CAAC,QAAQ,CAAC,CAAA;IAEtD;;;;;;;;OAQG;IACH,MAAM,kBAAkB,GAAG,MAAM,SAAS,CAAC,mBAAmB,EAAE,CAAA;IAChE,OAAO,kBAAkB;SACpB,MAAM,CAAC,CAAC,aAA2B,EAAE,EAAE;QACpC,OAAO,CAAC,CAAC,CACL,UAAU,KAAK,aAAa,CAAC,GAAG;YAChC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC;gBACzD,CAAC,QAAQ,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAChE,CAAC,IAAA,4BAAU,EAAC,IAAI,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAC7E,CAAA;IACL,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,SAAuB,EAAgB,EAAE;QAC3C,OAAO;YACH,IAAI,EAAE,SAAS;YACf,mEAAmE;YACnE,YAAY,EAAE,IAAA,qCAAe,EAAC,UAAU,EAAE,SAAS,CAAC,GAAG,CAAC;SAC3D,CAAA;IACL,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,KAAmB,EAAE,KAAmB,EAAE,EAAE;QAC/C,OAAO,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAA;IAClD,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,cAA4B,EAAE,EAAE;QAClC,OAAO,cAAc,CAAC,IAAI,CAAA;IAC9B,CAAC,CAAC,CAAA;AACV,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAwB;IAC9C,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;QAChC,MAAM,IAAI,2CAAiB,EAAE,CAAA;IACjC,CAAC;AACL,CAAC"}