@aws/lsp-codewhisperer 0.0.3

package/out/language-server/dependencyGraph/dependencyGraphFactory.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DependencyGraphFactory = void 0;
+const csharpDependencyGraph_1 = require("./csharpDependencyGraph");
+const languageMap = {
+    csharp: csharpDependencyGraph_1.CsharpDependencyGraph,
+};
+class DependencyGraphFactory {
+    static getDependencyGraphFromFileExtensions(fileName) {
+        return undefined;
+    }
+    static getDependencyGraph(document, workspace) {
+        switch (document.languageId.toLowerCase()) {
+            case 'csharp':
+                return new languageMap['csharp'](workspace);
+            default:
+                return this.getDependencyGraphFromFileExtensions(document.uri);
+        }
+    }
+}
+exports.DependencyGraphFactory = DependencyGraphFactory;
+//# sourceMappingURL=dependencyGraphFactory.js.map

package/out/language-server/dependencyGraph/dependencyGraphFactory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dependencyGraphFactory.js","sourceRoot":"","sources":["../../../src/language-server/dependencyGraph/dependencyGraphFactory.ts"],"names":[],"mappings":";;;AAEA,mEAA+D;AAE/D,MAAM,WAAW,GAAG;IAChB,MAAM,EAAE,6CAAqB;CACvB,CAAA;AAOV,MAAa,sBAAsB;IAC/B,MAAM,CAAC,oCAAoC,CAAiB,QAAgB;QACxE,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,MAAM,CAAC,kBAAkB,CAAiB,QAAsB,EAAE,SAA8B;QAC5F,QAAQ,QAAQ,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,CAAC;YACxC,KAAK,QAAQ;gBACT,OAAO,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAA;YAC/C;gBACI,OAAO,IAAI,CAAC,oCAAoC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;QACtE,CAAC;IACL,CAAC;CACJ;AAbD,wDAaC"}

package/out/language-server/languageDetection.d.ts

@@ -0,0 +1,3 @@
+import { TextDocument } from 'vscode-languageserver-textdocument';
+export type CodewhispererLanguage = 'java' | 'python' | 'jsx' | 'javascript' | 'typescript' | 'tsx' | 'csharp' | 'c' | 'cpp' | 'cpp' | 'go' | 'kotlin' | 'php' | 'ruby' | 'rust' | 'scala' | 'shell' | 'shell' | 'sql' | 'plaintext';
+export declare const getSupportedLanguageId: (textDocument: TextDocument | undefined) => CodewhispererLanguage | undefined;

package/out/language-server/languageDetection.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getSupportedLanguageId = void 0;
+// This will be extended as more language features
+// are integrated into the language server and clients.
+// See: https://microsoft.github.io/language-server-protocol/specifications/lsp/3.18/specification/#textDocumentItem
+const supportedFileTypes = ['c', 'cpp', 'csharp', 'javascript', 'python', 'typescript'];
+const supportedExtensions = {
+    '.c': 'c',
+    '.h': 'c',
+    '.cpp': 'cpp',
+    '.hpp': 'cpp',
+    '.cs': 'csharp',
+    '.js': 'javascript',
+    '.py': 'python',
+    '.ts': 'typescript',
+};
+const getSupportedLanguageId = (textDocument) => {
+    if (!textDocument) {
+        return;
+    }
+    const languageId = getCodeWhispererLanguageIdByTextDocumentLanguageId(textDocument.languageId);
+    if (languageId !== undefined) {
+        return languageId;
+    }
+    for (const extension in supportedExtensions) {
+        if (textDocument.uri.endsWith(extension)) {
+            return supportedExtensions[extension];
+        }
+    }
+};
+exports.getSupportedLanguageId = getSupportedLanguageId;
+/**
+ * Used to map different IDE values for TextDocument languageIds to CodeWhisperer languageIds.
+ * Examples of the CodeWhisperer defined language ids can be found in service-2.json, near "ProgrammingLanguageLanguageNameString"
+ * @param textDocumentLanguageId Value of the TextDocument's language id, provided by the IDE
+ * @returns Corresponding CodeWhisperer language id
+ */
+function getCodeWhispererLanguageIdByTextDocumentLanguageId(textDocumentLanguageId) {
+    if (textDocumentLanguageId === undefined) {
+        return undefined;
+    }
+    if (supportedFileTypes.includes(textDocumentLanguageId)) {
+        return textDocumentLanguageId;
+    }
+    // IDEs can identify a file's languageId using non-standardized values
+    // Eg: 'CSHARP', 'CSharp' => 'csharp'
+    // Try to map case-insensitive matches to increase the likelihood of supporting the file in an IDE.
+    for (const supportedFileType of supportedFileTypes) {
+        if (textDocumentLanguageId.toLowerCase() === supportedFileType.toLowerCase()) {
+            return supportedFileType;
+        }
+    }
+    return undefined;
+}
+//# sourceMappingURL=languageDetection.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"languageDetection.js","sourceRoot":"","sources":["../../src/language-server/languageDetection.ts"],"names":[],"mappings":";;;AAwBA,kDAAkD;AAClD,uDAAuD;AACvD,oHAAoH;AACpH,MAAM,kBAAkB,GAA4B,CAAC,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAA;AAChH,MAAM,mBAAmB,GAA6C;IAClE,IAAI,EAAE,GAAG;IACT,IAAI,EAAE,GAAG;IACT,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,KAAK,EAAE,QAAQ;IACf,KAAK,EAAE,YAAY;IACnB,KAAK,EAAE,QAAQ;IACf,KAAK,EAAE,YAAY;CACtB,CAAA;AAEM,MAAM,sBAAsB,GAAG,CAAC,YAAsC,EAAqC,EAAE;IAChH,IAAI,CAAC,YAAY,EAAE,CAAC;QAChB,OAAM;IACV,CAAC;IAED,MAAM,UAAU,GAAG,kDAAkD,CAAC,YAAY,CAAC,UAAU,CAAC,CAAA;IAC9F,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,UAAU,CAAA;IACrB,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,mBAAmB,EAAE,CAAC;QAC1C,IAAI,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACvC,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAA;QACzC,CAAC;IACL,CAAC;AACL,CAAC,CAAA;AAfY,QAAA,sBAAsB,0BAelC;AAED;;;;;GAKG;AACH,SAAS,kDAAkD,CACvD,sBAAsD;IAEtD,IAAI,sBAAsB,KAAK,SAAS,EAAE,CAAC;QACvC,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,IAAI,kBAAkB,CAAC,QAAQ,CAAC,sBAA+C,CAAC,EAAE,CAAC;QAC/E,OAAO,sBAA+C,CAAA;IAC1D,CAAC;IAED,sEAAsE;IACtE,qCAAqC;IACrC,mGAAmG;IACnG,KAAK,MAAM,iBAAiB,IAAI,kBAAkB,EAAE,CAAC;QACjD,IAAI,sBAAsB,CAAC,WAAW,EAAE,KAAK,iBAAiB,CAAC,WAAW,EAAE,EAAE,CAAC;YAC3E,OAAO,iBAA0C,CAAA;QACrD,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAA;AACpB,CAAC"}

package/out/language-server/mergeRightUtils.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Returns the longest overlap between the Suffix of firstString and Prefix of second string
+ * getPrefixSuffixOverlap("adwg31", "31ggrs") = "31"
+ */
+export declare function getPrefixSuffixOverlap(firstString: string, secondString: string): string;
+export declare function truncateOverlapWithRightContext(rightFileContent: string, suggestion: string): string;

package/out/language-server/mergeRightUtils.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.truncateOverlapWithRightContext = exports.getPrefixSuffixOverlap = void 0;
+/**
+ * Returns the longest overlap between the Suffix of firstString and Prefix of second string
+ * getPrefixSuffixOverlap("adwg31", "31ggrs") = "31"
+ */
+function getPrefixSuffixOverlap(firstString, secondString) {
+    let i = Math.min(firstString.length, secondString.length);
+    while (i > 0) {
+        if (secondString.slice(0, i) === firstString.slice(-i)) {
+            break;
+        }
+        i--;
+    }
+    return secondString.slice(0, i);
+}
+exports.getPrefixSuffixOverlap = getPrefixSuffixOverlap;
+function truncateOverlapWithRightContext(rightFileContent, suggestion) {
+    const trimmedSuggestion = suggestion.trim();
+    // limit of 5000 for right context matching
+    const rightContext = rightFileContent
+        .substring(0, 5000)
+        .replaceAll('\r\n', '\n')
+        .replace(/^[^\S\n]+/, ''); // remove leading tabs and whitespaces
+    const overlap = getPrefixSuffixOverlap(trimmedSuggestion, rightContext);
+    const overlapIndex = suggestion.lastIndexOf(overlap);
+    if (overlapIndex >= 0) {
+        const truncated = suggestion.slice(0, overlapIndex);
+        return truncated.trim().length ? truncated : '';
+    }
+    else {
+        return suggestion;
+    }
+}
+exports.truncateOverlapWithRightContext = truncateOverlapWithRightContext;
+//# sourceMappingURL=mergeRightUtils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"mergeRightUtils.js","sourceRoot":"","sources":["../../src/language-server/mergeRightUtils.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,SAAgB,sBAAsB,CAAC,WAAmB,EAAE,YAAoB;IAC5E,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC,CAAA;IACzD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACX,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACrD,MAAK;QACT,CAAC;QACD,CAAC,EAAE,CAAA;IACP,CAAC;IACD,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;AACnC,CAAC;AATD,wDASC;AAED,SAAgB,+BAA+B,CAAC,gBAAwB,EAAE,UAAkB;IACxF,MAAM,iBAAiB,GAAG,UAAU,CAAC,IAAI,EAAE,CAAA;IAC3C,2CAA2C;IAC3C,MAAM,YAAY,GAAG,gBAAgB;SAChC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC;SAClB,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC;SACxB,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAA,CAAC,sCAAsC;IACpE,MAAM,OAAO,GAAG,sBAAsB,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAA;IACvE,MAAM,YAAY,GAAG,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;IACpD,IAAI,YAAY,IAAI,CAAC,EAAE,CAAC;QACpB,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAA;QACnD,OAAO,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAA;IACnD,CAAC;SAAM,CAAC;QACJ,OAAO,UAAU,CAAA;IACrB,CAAC;AACL,CAAC;AAfD,0EAeC"}

package/out/language-server/mergeRightUtils.test.d.ts

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

package/out/language-server/mergeRightUtils.test.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const mergeRightUtils_1 = require("./mergeRightUtils");
+const testUtils_1 = require("./testUtils");
+const assert = require("assert");
+describe('Merge Right Utils', () => {
+    const HELLO_WORLD = `Console.WriteLine("Hello World!");`;
+    it('get prefix suffix overlap works as expected', () => {
+        const result = (0, mergeRightUtils_1.getPrefixSuffixOverlap)('adwg31', '31ggrs');
+        assert.deepEqual(result, '31');
+    });
+    it('should return empty suggestion when right context equals line content ', () => {
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)(HELLO_WORLD, HELLO_WORLD);
+        assert.deepEqual(result, '');
+    });
+    it('should return empty suggestion when right context equals file content', () => {
+        // Without trimStart, this test would fail because the function doesn't trim leading new line from right context
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)(testUtils_1.HELLO_WORLD_IN_CSHARP.trimStart(), testUtils_1.HELLO_WORLD_IN_CSHARP);
+        assert.deepEqual(result, '');
+    });
+    it('should not handle the case where right context fully matches suggestion but starts with a newline ', () => {
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)('\n' + testUtils_1.HELLO_WORLD_IN_CSHARP, testUtils_1.HELLO_WORLD_IN_CSHARP);
+        // Even though right context and suggestion are equal, the newline of right context doesn't get trimmed while the newline of suggestion gets trimmed
+        // As a result, we end up with no overlap
+        assert.deepEqual(result, testUtils_1.HELLO_WORLD_IN_CSHARP);
+    });
+    it('should return truncated suggestion when right context matches end of the suggestion', () => {
+        // File contents will be `nsole.WriteLine("Hello World!");`
+        // Suggestion will be the full HELLO_WORLD
+        // Final truncated result should be the first two letters of HELLO_WORLD
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)(HELLO_WORLD.substring(2), HELLO_WORLD);
+        assert.deepEqual(result, HELLO_WORLD.substring(0, 2));
+    });
+    it('should trim right-context tabs and whitespaces until first newline', () => {
+        const suggestion = '{\n            return a + b;\n        }';
+        const rightContent = '       \n        }\n\n    }\n}';
+        const expected_result = '{\n            return a + b;';
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)(rightContent, suggestion);
+        assert.deepEqual(result, expected_result);
+    });
+    it('should handle different line endings', () => {
+        const suggestion = '{\n            return a + b;\n        }';
+        const rightContent = '\r\n        }\r\n}\r\n}';
+        const expected_result = '{\n            return a + b;';
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)(rightContent, suggestion);
+        assert.deepEqual(result, expected_result);
+    });
+    it('should handle windows line endings for files', () => {
+        const result = (0, mergeRightUtils_1.truncateOverlapWithRightContext)(testUtils_1.HELLO_WORLD_WITH_WINDOWS_ENDING, testUtils_1.HELLO_WORLD_WITH_WINDOWS_ENDING.replaceAll('\r', ''));
+        assert.deepEqual(result, '');
+    });
+});
+//# sourceMappingURL=mergeRightUtils.test.js.map

package/out/language-server/mergeRightUtils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mergeRightUtils.test.js","sourceRoot":"","sources":["../../src/language-server/mergeRightUtils.test.ts"],"names":[],"mappings":";;AAAA,uDAA2F;AAC3F,2CAAoF;AACpF,iCAAiC;AAEjC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAC/B,MAAM,WAAW,GAAG,oCAAoC,CAAA;IAExD,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACnD,MAAM,MAAM,GAAG,IAAA,wCAAsB,EAAC,QAAQ,EAAE,QAAQ,CAAC,CAAA;QACzD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IAClC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;QAC9E,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAAC,WAAW,EAAE,WAAW,CAAC,CAAA;QACxE,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;IAChC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC7E,gHAAgH;QAChH,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAAC,iCAAqB,CAAC,SAAS,EAAE,EAAE,iCAAqB,CAAC,CAAA;QACxG,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;IAChC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oGAAoG,EAAE,GAAG,EAAE;QAC1G,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAAC,IAAI,GAAG,iCAAqB,EAAE,iCAAqB,CAAC,CAAA;QACnG,oJAAoJ;QACpJ,yCAAyC;QACzC,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,iCAAqB,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qFAAqF,EAAE,GAAG,EAAE;QAC3F,2DAA2D;QAC3D,0CAA0C;QAC1C,wEAAwE;QACxE,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAA;QAErF,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC1E,MAAM,UAAU,GAAG,yCAAyC,CAAA;QAC5D,MAAM,YAAY,GAAG,gCAAgC,CAAA;QACrD,MAAM,eAAe,GAAG,8BAA8B,CAAA;QACtD,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAAC,YAAY,EAAE,UAAU,CAAC,CAAA;QAExE,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,eAAe,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC5C,MAAM,UAAU,GAAG,yCAAyC,CAAA;QAC5D,MAAM,YAAY,GAAG,yBAAyB,CAAA;QAC9C,MAAM,eAAe,GAAG,8BAA8B,CAAA;QACtD,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAAC,YAAY,EAAE,UAAU,CAAC,CAAA;QAExE,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,eAAe,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACpD,MAAM,MAAM,GAAG,IAAA,iDAA+B,EAC1C,2CAA+B,EAC/B,2CAA+B,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC,CACvD,CAAA;QACD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;IAChC,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/proxy-server.d.ts

@@ -0,0 +1,2 @@
+export declare const CodeWhispererServerTokenProxy: import("@aws/language-server-runtimes").Server;
+export declare const CodeWhispererSecurityScanServerTokenProxy: import("@aws/language-server-runtimes").Server;

package/out/language-server/proxy-server.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeWhispererSecurityScanServerTokenProxy = exports.CodeWhispererServerTokenProxy = void 0;
+const codeWhispererSecurityScanServer_1 = require("./codeWhispererSecurityScanServer");
+const codeWhispererServer_1 = require("./codeWhispererServer");
+const codeWhispererService_1 = require("./codeWhispererService");
+exports.CodeWhispererServerTokenProxy = (0, codeWhispererServer_1.CodewhispererServerFactory)(credentialsProvider => {
+    let additionalAwsConfig = {};
+    const proxyUrl = process.env.HTTPS_PROXY ?? process.env.https_proxy;
+    if (proxyUrl) {
+        const { getProxyHttpAgent } = require('proxy-http-agent');
+        const proxyAgent = getProxyHttpAgent({
+            proxy: proxyUrl,
+        });
+        additionalAwsConfig = {
+            proxy: proxyAgent,
+        };
+    }
+    return new codeWhispererService_1.CodeWhispererServiceToken(credentialsProvider, additionalAwsConfig);
+});
+exports.CodeWhispererSecurityScanServerTokenProxy = (0, codeWhispererSecurityScanServer_1.SecurityScanServerToken)(credentialsProvider => {
+    let additionalAwsConfig = {};
+    const proxyUrl = process.env.HTTPS_PROXY ?? process.env.https_proxy;
+    if (proxyUrl) {
+        const { getProxyHttpAgent } = require('proxy-http-agent');
+        const proxyAgent = getProxyHttpAgent({
+            proxy: proxyUrl,
+        });
+        additionalAwsConfig = {
+            proxy: proxyAgent,
+        };
+    }
+    return new codeWhispererService_1.CodeWhispererServiceToken(credentialsProvider, additionalAwsConfig);
+});
+//# sourceMappingURL=proxy-server.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"proxy-server.js","sourceRoot":"","sources":["../../src/language-server/proxy-server.ts"],"names":[],"mappings":";;;AAAA,uFAA2E;AAC3E,+DAAkE;AAClE,iEAAkE;AAErD,QAAA,6BAA6B,GAAG,IAAA,gDAA0B,EAAC,mBAAmB,CAAC,EAAE;IAC1F,IAAI,mBAAmB,GAAG,EAAE,CAAA;IAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,CAAA;IAEnE,IAAI,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAA;QACzD,MAAM,UAAU,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,QAAQ;SAClB,CAAC,CAAA;QACF,mBAAmB,GAAG;YAClB,KAAK,EAAE,UAAU;SACpB,CAAA;IACL,CAAC;IACD,OAAO,IAAI,gDAAyB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAA;AAClF,CAAC,CAAC,CAAA;AAEW,QAAA,yCAAyC,GAAG,IAAA,yDAAuB,EAAC,mBAAmB,CAAC,EAAE;IACnG,IAAI,mBAAmB,GAAG,EAAE,CAAA;IAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,CAAA;IAEnE,IAAI,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAA;QACzD,MAAM,UAAU,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,QAAQ;SAClB,CAAC,CAAA;QACF,mBAAmB,GAAG;YAClB,KAAK,EAAE,UAAU;SACpB,CAAA;IACL,CAAC;IACD,OAAO,IAAI,gDAAyB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAA;AAClF,CAAC,CAAC,CAAA"}

package/out/language-server/securityScan/securityScanHandler.d.ts

@@ -0,0 +1,17 @@
+/// <reference types="node" />
+import { Workspace } from '@aws/language-server-runtimes/out/features';
+import { ArtifactMap, CreateUploadUrlResponse } from '../../client/token/codewhispererbearertokenclient';
+import { CodeWhispererServiceToken } from '../codeWhispererService';
+import { AggregatedCodeScanIssue, RawCodeScanIssue } from './types';
+export declare class SecurityScanHandler {
+    private client;
+    private workspace;
+    constructor(client: CodeWhispererServiceToken, workspace: Workspace);
+    getMd5(content: Buffer): string;
+    createCodeResourcePresignedUrlHandler(zipContent: Buffer): Promise<ArtifactMap>;
+    uploadArtifactToS3(zipBuffer: Buffer, resp: CreateUploadUrlResponse): Promise<import("got").Response<string>>;
+    createScanJob(artifactMap: ArtifactMap, languageName: string): Promise<import("aws-sdk/lib/request").PromiseResult<import("../../client/token/codewhispererbearertokenclient").StartCodeAnalysisResponse, import("aws-sdk").AWSError>>;
+    pollScanJobStatus(jobId: string): Promise<string>;
+    listScanResults(jobId: string, projectPath: string): Promise<AggregatedCodeScanIssue[]>;
+    mapToAggregatedList(codeScanIssueMap: Map<string, RawCodeScanIssue[]>, aggregatedCodeScanIssueList: AggregatedCodeScanIssue[], json: string, projectPath: string): void;
+}

package/out/language-server/securityScan/securityScanHandler.js

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SecurityScanHandler = void 0;
+const got_1 = require("got");
+const js_md5_1 = require("js-md5");
+const path = require("path");
+const commonUtil_1 = require("../dependencyGraph/commonUtil");
+class SecurityScanHandler {
+    constructor(client, workspace) {
+        this.client = client;
+        this.workspace = workspace;
+    }
+    getMd5(content) {
+        return js_md5_1.md5.base64(content);
+    }
+    async createCodeResourcePresignedUrlHandler(zipContent) {
+        const request = {
+            contentMd5: this.getMd5(zipContent),
+            artifactType: 'SourceCode',
+        };
+        const response = await this.client.createUploadUrl(request);
+        await this.uploadArtifactToS3(zipContent, response);
+        const artifactMap = {
+            SourceCode: response.uploadId,
+        };
+        return artifactMap;
+    }
+    async uploadArtifactToS3(zipBuffer, resp) {
+        const encryptionContext = `{"uploadId":"${resp.uploadId}"}`;
+        const md5Content = this.getMd5(zipBuffer);
+        const headersObj = resp.kmsKeyArn !== '' || resp.kmsKeyArn !== undefined
+            ? {
+                'Content-MD5': md5Content,
+                'x-amz-server-side-encryption': 'aws:kms',
+                'Content-Type': 'application/zip',
+                'x-amz-server-side-encryption-aws-kms-key-id': resp.kmsKeyArn,
+                'x-amz-server-side-encryption-context': Buffer.from(encryptionContext, 'utf8').toString('base64'),
+            }
+            : {
+                'Content-MD5': md5Content,
+                'x-amz-server-side-encryption': 'aws:kms',
+                'Content-Type': 'application/zip',
+                'x-amz-server-side-encryption-context': Buffer.from(encryptionContext, 'utf8').toString('base64'),
+            };
+        const response = await got_1.default.put(resp.uploadUrl, { body: zipBuffer, headers: headersObj });
+        return response;
+    }
+    async createScanJob(artifactMap, languageName) {
+        const req = {
+            artifacts: artifactMap,
+            programmingLanguage: {
+                languageName,
+            },
+        };
+        const resp = await this.client.startCodeAnalysis(req);
+        return resp;
+    }
+    async pollScanJobStatus(jobId) {
+        let status = 'Pending';
+        let timer = 0;
+        const codeScanJobPollingIntervalSeconds = 1;
+        const codeScanJobTimeoutSeconds = 50;
+        // eslint-disable-next-line no-constant-condition
+        while (true) {
+            const req = {
+                jobId: jobId,
+            };
+            const resp = await this.client.getCodeAnalysis(req);
+            if (resp.status !== 'Pending') {
+                status = resp.status;
+                break;
+            }
+            await (0, commonUtil_1.sleep)(codeScanJobPollingIntervalSeconds * 1000);
+            timer += codeScanJobPollingIntervalSeconds;
+            if (timer > codeScanJobTimeoutSeconds) {
+                throw new Error('Scan job timeout.');
+            }
+        }
+        return status;
+    }
+    async listScanResults(jobId, projectPath) {
+        const codeScanIssueMap = new Map();
+        const aggregatedCodeScanIssueList = [];
+        const request = {
+            jobId,
+            codeAnalysisFindingsSchema: 'codeanalysis/findings/1.0',
+        };
+        const response = await this.client.listCodeAnalysisFindings(request);
+        this.mapToAggregatedList(codeScanIssueMap, aggregatedCodeScanIssueList, response.codeAnalysisFindings, projectPath);
+        return aggregatedCodeScanIssueList;
+    }
+    mapToAggregatedList(codeScanIssueMap, aggregatedCodeScanIssueList, json, projectPath) {
+        const codeScanIssues = JSON.parse(json);
+        codeScanIssues.forEach(issue => {
+            if (codeScanIssueMap.has(issue.filePath)) {
+                const list = codeScanIssueMap.get(issue.filePath);
+                if (list === undefined) {
+                    codeScanIssueMap.set(issue.filePath, [issue]);
+                }
+                else {
+                    list.push(issue);
+                    codeScanIssueMap.set(issue.filePath, list);
+                }
+            }
+            else {
+                codeScanIssueMap.set(issue.filePath, [issue]);
+            }
+        });
+        codeScanIssueMap.forEach(async (issues, key) => {
+            const filePath = path.join(projectPath, '..', key);
+            const fileExists = await this.workspace.fs.exists(filePath);
+            if (fileExists) {
+                const aggregatedCodeScanIssue = {
+                    filePath: filePath,
+                    issues: issues.map(issue => {
+                        return {
+                            startLine: issue.startLine - 1 >= 0 ? issue.startLine - 1 : 0,
+                            endLine: issue.endLine,
+                            comment: `${issue.title.trim()}: ${issue.description.text.trim()}`,
+                            title: issue.title,
+                            description: issue.description,
+                            detectorId: issue.detectorId,
+                            detectorName: issue.detectorName,
+                            findingId: issue.findingId,
+                            ruleId: issue.ruleId,
+                            relatedVulnerabilities: issue.relatedVulnerabilities,
+                            severity: issue.severity,
+                            recommendation: issue.remediation.recommendation,
+                            suggestedFixes: issue.remediation.suggestedFixes,
+                        };
+                    }),
+                };
+                aggregatedCodeScanIssueList.push(aggregatedCodeScanIssue);
+            }
+        });
+    }
+}
+exports.SecurityScanHandler = SecurityScanHandler;
+//# sourceMappingURL=securityScanHandler.js.map

package/out/language-server/securityScan/securityScanHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"securityScanHandler.js","sourceRoot":"","sources":["../../../src/language-server/securityScan/securityScanHandler.ts"],"names":[],"mappings":";;;AACA,6BAAqB;AACrB,mCAA4B;AAC5B,6BAA4B;AAW5B,8DAAqD;AAGrD,MAAa,mBAAmB;IAG5B,YAAY,MAAiC,EAAE,SAAoB;QAC/D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC9B,CAAC;IAED,MAAM,CAAC,OAAe;QAClB,OAAO,YAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,qCAAqC,CAAC,UAAkB;QAC1D,MAAM,OAAO,GAA2B;YACpC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;YACnC,YAAY,EAAE,YAAY;SAC7B,CAAA;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;QAC3D,MAAM,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;QACnD,MAAM,WAAW,GAAgB;YAC7B,UAAU,EAAE,QAAQ,CAAC,QAAQ;SAChC,CAAA;QACD,OAAO,WAAW,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,SAAiB,EAAE,IAA6B;QACrE,MAAM,iBAAiB,GAAG,gBAAgB,IAAI,CAAC,QAAQ,IAAI,CAAA;QAC3D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;QACzC,MAAM,UAAU,GACZ,IAAI,CAAC,SAAS,KAAK,EAAE,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS;YACjD,CAAC,CAAC;gBACI,aAAa,EAAE,UAAU;gBACzB,8BAA8B,EAAE,SAAS;gBACzC,cAAc,EAAE,iBAAiB;gBACjC,6CAA6C,EAAE,IAAI,CAAC,SAAS;gBAC7D,sCAAsC,EAAE,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;aACpG;YACH,CAAC,CAAC;gBACI,aAAa,EAAE,UAAU;gBACzB,8BAA8B,EAAE,SAAS;gBACzC,cAAc,EAAE,iBAAiB;gBACjC,sCAAsC,EAAE,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;aACpG,CAAA;QACX,MAAM,QAAQ,GAAG,MAAM,aAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAA;QACxF,OAAO,QAAQ,CAAA;IACnB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,WAAwB,EAAE,YAAoB;QAC9D,MAAM,GAAG,GAA6B;YAClC,SAAS,EAAE,WAAW;YACtB,mBAAmB,EAAE;gBACjB,YAAY;aACf;SACJ,CAAA;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAA;QACrD,OAAO,IAAI,CAAA;IACf,CAAC;IACD,KAAK,CAAC,iBAAiB,CAAC,KAAa;QACjC,IAAI,MAAM,GAAG,SAAS,CAAA;QACtB,IAAI,KAAK,GAAG,CAAC,CAAA;QACb,MAAM,iCAAiC,GAAG,CAAC,CAAA;QAC3C,MAAM,yBAAyB,GAAG,EAAE,CAAA;QACpC,iDAAiD;QACjD,OAAO,IAAI,EAAE,CAAC;YACV,MAAM,GAAG,GAA2B;gBAChC,KAAK,EAAE,KAAK;aACf,CAAA;YACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;YACnD,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;gBACpB,MAAK;YACT,CAAC;YACD,MAAM,IAAA,kBAAK,EAAC,iCAAiC,GAAG,IAAI,CAAC,CAAA;YACrD,KAAK,IAAI,iCAAiC,CAAA;YAC1C,IAAI,KAAK,GAAG,yBAAyB,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAA;YACxC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,KAAa,EAAE,WAAmB;QACpD,MAAM,gBAAgB,GAAoC,IAAI,GAAG,EAAE,CAAA;QACnE,MAAM,2BAA2B,GAA8B,EAAE,CAAA;QACjE,MAAM,OAAO,GAAoC;YAC7C,KAAK;YACL,0BAA0B,EAAE,2BAA2B;SAC1D,CAAA;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAA;QAEpE,IAAI,CAAC,mBAAmB,CACpB,gBAAgB,EAChB,2BAA2B,EAC3B,QAAQ,CAAC,oBAAoB,EAC7B,WAAW,CACd,CAAA;QACD,OAAO,2BAA2B,CAAA;IACtC,CAAC;IAED,mBAAmB,CACf,gBAAiD,EACjD,2BAAsD,EACtD,IAAY,EACZ,WAAmB;QAEnB,MAAM,cAAc,GAAuB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC3D,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;YAC3B,IAAI,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvC,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;gBACjD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;oBACrB,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,CAAA;gBACjD,CAAC;qBAAM,CAAC;oBACJ,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;oBAChB,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;gBAC9C,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,CAAA;YACjD,CAAC;QACL,CAAC,CAAC,CAAA;QAEF,gBAAgB,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;YAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,GAAG,CAAC,CAAA;YAClD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;YAC3D,IAAI,UAAU,EAAE,CAAC;gBACb,MAAM,uBAAuB,GAAG;oBAC5B,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;wBACvB,OAAO;4BACH,SAAS,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;4BAC7D,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;4BAClE,KAAK,EAAE,KAAK,CAAC,KAAK;4BAClB,WAAW,EAAE,KAAK,CAAC,WAAW;4BAC9B,UAAU,EAAE,KAAK,CAAC,UAAU;4BAC5B,YAAY,EAAE,KAAK,CAAC,YAAY;4BAChC,SAAS,EAAE,KAAK,CAAC,SAAS;4BAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;4BACpB,sBAAsB,EAAE,KAAK,CAAC,sBAAsB;4BACpD,QAAQ,EAAE,KAAK,CAAC,QAAQ;4BACxB,cAAc,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc;4BAChD,cAAc,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc;yBACnD,CAAA;oBACL,CAAC,CAAC;iBACL,CAAA;gBACD,2BAA2B,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAA;YAC7D,CAAC;QACL,CAAC,CAAC,CAAA;IACN,CAAC;CACJ;AApJD,kDAoJC"}

package/out/language-server/securityScan/securityScanHandler.test.d.ts

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

package/out/language-server/securityScan/securityScanHandler.test.js

@@ -0,0 +1,170 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const aws_sdk_1 = require("aws-sdk");
+const got_1 = require("got");
+const Sinon = require("sinon");
+const ts_sinon_1 = require("ts-sinon");
+const securityScanHandler_1 = require("./securityScanHandler");
+const mockCodeScanFindings = JSON.stringify([
+    {
+        filePath: 'app.py',
+        startLine: 1,
+        endLine: 1,
+        title: 'title',
+        description: {
+            text: 'text',
+            markdown: 'markdown',
+        },
+        detectorId: 'detectorId',
+        detectorName: 'detectorName',
+        findingId: 'findingId',
+        relatedVulnerabilities: [],
+        severity: 'High',
+        remediation: {
+            recommendation: {
+                text: 'text',
+                url: 'url',
+            },
+            suggestedFixes: [],
+        },
+    },
+]);
+const mocked$Response = {
+    $response: {
+        hasNextPage: ts_sinon_1.default.mock(),
+        nextPage: ts_sinon_1.default.mock(),
+        data: undefined,
+        error: undefined,
+        requestId: '',
+        redirectCount: 0,
+        retryCount: 0,
+        httpResponse: new aws_sdk_1.HttpResponse(),
+    },
+};
+describe('securityScanHandler', () => {
+    let client;
+    let workspace;
+    let securityScanhandler;
+    beforeEach(async () => {
+        // Set up the server with a mock service
+        client = (0, ts_sinon_1.stubInterface)();
+        workspace = (0, ts_sinon_1.stubInterface)();
+        securityScanhandler = new securityScanHandler_1.SecurityScanHandler(client, workspace);
+    });
+    describe('Test createCodeResourcePresignedUrlHandler', () => {
+        let putStub;
+        beforeEach(async () => {
+            // mock default return value for createUploadUrl
+            client.createUploadUrl.resolves({
+                uploadId: 'dummy-upload-id',
+                uploadUrl: 'dummy-upload-url',
+                kmsKeyArn: 'ResourceArn',
+                ...mocked$Response,
+            });
+            putStub = Sinon.stub(got_1.default, 'put');
+        });
+        it('returns correct source code', async () => {
+            const expectedSourceCode = 'dummy-upload-id';
+            const res = await securityScanhandler.createCodeResourcePresignedUrlHandler(Buffer.from('dummy-data'));
+            ts_sinon_1.default.assert.callCount(putStub, 1);
+            assert.equal(res.SourceCode, expectedSourceCode);
+        });
+    });
+    describe('Test createScanJob', () => {
+        beforeEach(async () => {
+            // mock default return value for createCodeScan
+            client.startCodeAnalysis.returns(Promise.resolve({
+                jobId: 'dummy-job-id',
+                status: 'Pending',
+                ...mocked$Response,
+            }));
+        });
+        it('should create code scan', async () => {
+            const artifactMap = { SourceCode: 'dummy-upload-id' };
+            const requestParams = {
+                artifacts: { SourceCode: 'dummy-upload-id' },
+                programmingLanguage: { languageName: 'csharp' },
+            };
+            const res = await securityScanhandler.createScanJob(artifactMap, 'csharp');
+            ts_sinon_1.default.assert.calledOnceWithExactly(client.startCodeAnalysis, requestParams);
+            assert.equal(res.jobId, 'dummy-job-id');
+            assert.equal(res.status, 'Pending');
+        });
+    });
+    describe('Test pollScanJobStatus', () => {
+        beforeEach(async () => {
+            // mock default return value for getCodeAnalysis
+            client.getCodeAnalysis.resolves({
+                status: 'Pending',
+                ...mocked$Response,
+            });
+        });
+        it('should change job status from pending to completed', async () => {
+            client.getCodeAnalysis.onCall(0).resolves({
+                status: 'Pending',
+                ...mocked$Response,
+            });
+            client.getCodeAnalysis.onCall(1).resolves({
+                status: 'Completed',
+                ...mocked$Response,
+            });
+            const dummyJobId = 'dummy-job-id';
+            const requestParams = { jobId: dummyJobId };
+            const res = await securityScanhandler.pollScanJobStatus(dummyJobId);
+            ts_sinon_1.default.assert.calledWith(client.getCodeAnalysis, requestParams);
+            ts_sinon_1.default.assert.calledTwice(client.getCodeAnalysis);
+            assert.equal(res, 'Completed');
+        });
+        it('should change job status from pending to failed', async () => {
+            client.getCodeAnalysis.onCall(0).resolves({
+                status: 'Pending',
+                ...mocked$Response,
+            });
+            client.getCodeAnalysis.onCall(1).resolves({
+                status: 'Failed',
+                ...mocked$Response,
+            });
+            const dummyJobId = 'dummy-job-id';
+            const requestParams = { jobId: dummyJobId };
+            const res = await securityScanhandler.pollScanJobStatus(dummyJobId);
+            ts_sinon_1.default.assert.calledWith(client.getCodeAnalysis, requestParams);
+            ts_sinon_1.default.assert.calledTwice(client.getCodeAnalysis);
+            assert.equal(res, 'Failed');
+        });
+    });
+    describe('Test listScanResults', () => {
+        beforeEach(() => {
+            // mock default return value for listCodeAnalysisFindings
+            client.listCodeAnalysisFindings.resolves({
+                codeAnalysisFindings: mockCodeScanFindings,
+                ...mocked$Response,
+            });
+            workspace.fs.exists = ts_sinon_1.default.stub().resolves(true);
+        });
+        it('should return appropriate issues', async () => {
+            const dummyJobId = 'dummy-job-id';
+            const codeAnalysisFindingsSchema = 'codeanalysis/findings/1.0';
+            const dummyProjectPath = 'C:\\workspace\\workspaceFolder\\python3.7-plain-sam-app\\hello_world';
+            const requestParams = { jobId: dummyJobId, codeAnalysisFindingsSchema };
+            const aggregatedCodeScanIssueList = await securityScanhandler.listScanResults(dummyJobId, dummyProjectPath);
+            ts_sinon_1.default.assert.calledWith(client.listCodeAnalysisFindings, requestParams);
+            assert.equal(aggregatedCodeScanIssueList.length, 1);
+            assert.equal(aggregatedCodeScanIssueList[0].issues.length, 1);
+        });
+        it('should return zero issues', async () => {
+            client.listCodeAnalysisFindings.resolves({
+                codeAnalysisFindings: '[]',
+                ...mocked$Response,
+            });
+            const dummyJobId = 'dummy-job-id';
+            const codeAnalysisFindingsSchema = 'codeanalysis/findings/1.0';
+            const dummyProjectPath = 'C:\\workspace\\workspaceFolder\\python3.7-plain-sam-app\\hello_world';
+            const requestParams = { jobId: dummyJobId, codeAnalysisFindingsSchema };
+            const aggregatedCodeScanIssueList = await securityScanhandler.listScanResults(dummyJobId, dummyProjectPath);
+            ts_sinon_1.default.assert.calledWith(client.listCodeAnalysisFindings, requestParams);
+            assert.equal(aggregatedCodeScanIssueList.length, 0);
+        });
+    });
+});
+//# sourceMappingURL=securityScanHandler.test.js.map

package/out/language-server/securityScan/securityScanHandler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"securityScanHandler.test.js","sourceRoot":"","sources":["../../../src/language-server/securityScan/securityScanHandler.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,qCAAsC;AACtC,6BAAqB;AACrB,+BAA8B;AAC9B,uCAA2E;AAG3E,+DAA2D;AAG3D,MAAM,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC;IACxC;QACI,QAAQ,EAAE,QAAQ;QAClB,SAAS,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC;QACV,KAAK,EAAE,OAAO;QACd,WAAW,EAAE;YACT,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,UAAU;SACvB;QACD,UAAU,EAAE,YAAY;QACxB,YAAY,EAAE,cAAc;QAC5B,SAAS,EAAE,WAAW;QACtB,sBAAsB,EAAE,EAAE;QAC1B,QAAQ,EAAE,MAAM;QAChB,WAAW,EAAE;YACT,cAAc,EAAE;gBACZ,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,KAAK;aACb;YACD,cAAc,EAAE,EAAE;SACrB;KACuB;CAC/B,CAAC,CAAA;AAEF,MAAM,eAAe,GAAG;IACpB,SAAS,EAAE;QACP,WAAW,EAAE,kBAAK,CAAC,IAAI,EAAE;QACzB,QAAQ,EAAE,kBAAK,CAAC,IAAI,EAAE;QACtB,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,SAAS;QAChB,SAAS,EAAE,EAAE;QACb,aAAa,EAAE,CAAC;QAChB,UAAU,EAAE,CAAC;QACb,YAAY,EAAE,IAAI,sBAAY,EAAE;KACnC;CACJ,CAAA;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACjC,IAAI,MAAkD,CAAA;IACtD,IAAI,SAAqC,CAAA;IACzC,IAAI,mBAAwC,CAAA;IAC5C,UAAU,CAAC,KAAK,IAAI,EAAE;QAClB,wCAAwC;QACxC,MAAM,GAAG,IAAA,wBAAa,GAA6B,CAAA;QACnD,SAAS,GAAG,IAAA,wBAAa,GAAa,CAAA;QACtC,mBAAmB,GAAG,IAAI,yCAAmB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;IACpE,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACxD,IAAI,OAAwB,CAAA;QAC5B,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,gDAAgD;YAChD,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC;gBAC5B,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,kBAAkB;gBAC7B,SAAS,EAAE,aAAa;gBACxB,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,aAAG,EAAE,KAAK,CAAC,CAAA;QACpC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,kBAAkB,GAAG,iBAAiB,CAAA;YAC5C,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,qCAAqC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAA;YACtG,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;YAClC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAChC,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,+CAA+C;YAC/C,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAC5B,OAAO,CAAC,OAAO,CAAC;gBACZ,KAAK,EAAE,cAAc;gBACrB,MAAM,EAAE,SAAS;gBACjB,GAAG,eAAe;aACrB,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,WAAW,GAAG,EAAE,UAAU,EAAE,iBAAiB,EAAE,CAAA;YACrD,MAAM,aAAa,GAA6B;gBAC5C,SAAS,EAAE,EAAE,UAAU,EAAE,iBAAiB,EAAE;gBAC5C,mBAAmB,EAAE,EAAE,YAAY,EAAE,QAAQ,EAAE;aAClD,CAAA;YACD,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,aAAa,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAA;YAC1E,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,MAAM,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;YAC3E,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,cAAc,CAAC,CAAA;YACvC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;QACvC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACpC,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,gDAAgD;YAChD,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC;gBAC5B,MAAM,EAAE,SAAS;gBACjB,GAAG,eAAe;aACrB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACtC,MAAM,EAAE,SAAS;gBACjB,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACtC,MAAM,EAAE,WAAW;gBACnB,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,MAAM,UAAU,GAAG,cAAc,CAAA;YACjC,MAAM,aAAa,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,CAAA;YAC3C,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAA;YACnE,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,EAAE,aAAa,CAAC,CAAA;YAC9D,kBAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;YAChD,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,WAAW,CAAC,CAAA;QAClC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACtC,MAAM,EAAE,SAAS;gBACjB,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACtC,MAAM,EAAE,QAAQ;gBAChB,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,MAAM,UAAU,GAAG,cAAc,CAAA;YACjC,MAAM,aAAa,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,CAAA;YAC3C,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAA;YACnE,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,EAAE,aAAa,CAAC,CAAA;YAC9D,kBAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;YAChD,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAA;QAC/B,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAClC,UAAU,CAAC,GAAG,EAAE;YACZ,yDAAyD;YACzD,MAAM,CAAC,wBAAwB,CAAC,QAAQ,CAAC;gBACrC,oBAAoB,EAAE,oBAAoB;gBAC1C,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,SAAS,CAAC,EAAE,CAAC,MAAM,GAAG,kBAAK,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;QACrD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,UAAU,GAAG,cAAc,CAAA;YACjC,MAAM,0BAA0B,GAAG,2BAA2B,CAAA;YAC9D,MAAM,gBAAgB,GAAG,sEAAsE,CAAA;YAC/F,MAAM,aAAa,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,0BAA0B,EAAE,CAAA;YAEvE,MAAM,2BAA2B,GAAG,MAAM,mBAAmB,CAAC,eAAe,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAA;YAC3G,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,wBAAwB,EAAE,aAAa,CAAC,CAAA;YACvE,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;YACnD,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,CAAC,wBAAwB,CAAC,QAAQ,CAAC;gBACrC,oBAAoB,EAAE,IAAI;gBAC1B,GAAG,eAAe;aACrB,CAAC,CAAA;YACF,MAAM,UAAU,GAAG,cAAc,CAAA;YACjC,MAAM,0BAA0B,GAAG,2BAA2B,CAAA;YAC9D,MAAM,gBAAgB,GAAG,sEAAsE,CAAA;YAC/F,MAAM,aAAa,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,0BAA0B,EAAE,CAAA;YAEvE,MAAM,2BAA2B,GAAG,MAAM,mBAAmB,CAAC,eAAe,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAA;YAC3G,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,wBAAwB,EAAE,aAAa,CAAC,CAAA;YACvE,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACvD,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}