@aws/lsp-codewhisperer 0.0.3

package/out/language-server/dependencyGraph/csharpDependencyGraph.test.js

@@ -0,0 +1,341 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const path = require("path");
+const Sinon = require("sinon");
+const csharpDependencyGraph_1 = require("./csharpDependencyGraph");
+describe('test CsharpDependencyGraph', () => {
+    let csharpDependencyGraph;
+    const mockedGetWorkspaceFolder = Sinon.mock();
+    const projectPathUri = path.resolve(path.join(__dirname, 'sampleWs'));
+    const tempDirPath = path.resolve('\\Temp');
+    const mockedFs = {
+        copy: Sinon.mock(),
+        exists: Sinon.mock(),
+        getFileSize: Sinon.mock(),
+        getTempDirPath: () => tempDirPath,
+        readdir: Sinon.stub(),
+        readFile: Sinon.stub(),
+        isFile: Sinon.mock(),
+        remove: Sinon.mock(),
+    };
+    const mockedWorkspace = {
+        getTextDocument: Sinon.mock(),
+        getWorkspaceFolder: mockedGetWorkspaceFolder,
+        fs: mockedFs,
+    };
+    before(() => {
+        csharpDependencyGraph = new csharpDependencyGraph_1.CsharpDependencyGraph(mockedWorkspace);
+    });
+    beforeEach(() => {
+        mockedGetWorkspaceFolder.reset();
+        mockedGetWorkspaceFolder.returns(undefined);
+        csharpDependencyGraph = new csharpDependencyGraph_1.CsharpDependencyGraph(mockedWorkspace);
+    });
+    describe('Test getPayloadSizeLimitInBytes', () => {
+        it('should return correct payload size', () => {
+            const expectedPayloadSize = Math.pow(2, 20);
+            assert.equal(csharpDependencyGraph.getPayloadSizeLimitInBytes(), expectedPayloadSize);
+        });
+    });
+    describe('Test getProjectName', () => {
+        beforeEach(() => {
+            mockedFs.isFile.reset();
+            mockedFs.isFile.resolves(false);
+        });
+        it('should return current folder name for given file path outside workspace folder', async () => {
+            mockedFs.isFile.resolves(true);
+            assert.equal(await csharpDependencyGraph.getProjectName(path.join(projectPathUri, 'sample.cs')), 'sampleWs');
+        });
+        it('should return correct project name for given forlder path within workspace folder', async () => {
+            const expectedProjectName = 'sampleWs';
+            mockedGetWorkspaceFolder.returns({ uri: projectPathUri, name: expectedProjectName });
+            assert.equal(await csharpDependencyGraph.getProjectName(path.join(projectPathUri, 'src')), expectedProjectName);
+        });
+    });
+    describe('Test getProjectPath', () => {
+        beforeEach(() => {
+            mockedFs.isFile.reset();
+            mockedFs.isFile.resolves(false);
+        });
+        it('should not find workspace folder path for given path', async () => {
+            const fileFolderPath = path.resolve('\\workspace\\src');
+            const fileUri = path.join(fileFolderPath, 'sample.cs');
+            mockedFs.isFile.resolves(true);
+            const response = await csharpDependencyGraph.getProjectPath(fileUri);
+            assert.equal(response, fileFolderPath);
+        });
+        it('should return project name for within project file', async () => {
+            const expectedProjectName = 'sampleWs';
+            mockedGetWorkspaceFolder.returns({ uri: projectPathUri, name: expectedProjectName });
+            assert.equal(await csharpDependencyGraph.getProjectPath(path.join(projectPathUri, 'src')), projectPathUri);
+            mockedGetWorkspaceFolder.reset();
+        });
+    });
+    describe('Test getReadableSizeLimit', () => {
+        it('should return size 1 MB', () => {
+            assert.equal(csharpDependencyGraph.getReadableSizeLimit(), '1MB');
+        });
+    });
+    describe('Test exceedsSizeLimit', () => {
+        it('should not exceed size limit', () => {
+            const size = 100;
+            const expectedResult = false;
+            assert.equal(csharpDependencyGraph.exceedsSizeLimit(size), expectedResult);
+        });
+        it('should exceed size limit', () => {
+            const size = Math.pow(2, 21);
+            const expectedResult = true;
+            assert.equal(csharpDependencyGraph.exceedsSizeLimit(size), expectedResult);
+        });
+    });
+    describe('Test createNamespaceFilenameMapper', () => {
+        beforeEach(() => {
+            mockedFs.readdir.reset();
+            mockedFs.readFile.reset();
+            mockedFs.readdir.resolves([
+                {
+                    isFile: () => false,
+                    isDirectory: () => true,
+                    name: 'src',
+                    path: projectPathUri,
+                },
+                {
+                    isFile: () => true,
+                    isDirectory: () => false,
+                    name: 'sample.cs',
+                    path: path.join(projectPathUri, 'src'),
+                },
+                {
+                    isFile: () => true,
+                    isDirectory: () => false,
+                    name: 'model.cs',
+                    path: path.join(projectPathUri, 'src'),
+                },
+            ]);
+        });
+        it('should create the map with namespace to filepath mapping', async () => {
+            csharpDependencyGraph.namespaceToFilepathDirectory = new Map();
+            mockedFs.readFile.callsFake(async (filePath) => {
+                if (filePath === path.join(projectPathUri, 'src', 'sample.cs')) {
+                    return `namespace Amazon.Cw.Utils.Sample {}`;
+                }
+                if (filePath === path.join(projectPathUri, 'src', 'model.cs')) {
+                    return `namespace Amazon.Cw.Model {}`;
+                }
+            });
+            await csharpDependencyGraph.createNamespaceFilenameMapper(projectPathUri);
+            assert.deepEqual(csharpDependencyGraph.namespaceToFilepathDirectory, new Map([
+                ['Amazon.Cw.Model', new Set([path.join(projectPathUri, 'src', 'model.cs')])],
+                ['Amazon.Cw.Utils.Sample', new Set([path.join(projectPathUri, 'src', 'sample.cs')])],
+            ]));
+        });
+        it('should create empty map', async () => {
+            csharpDependencyGraph.namespaceToFilepathDirectory = new Map();
+            mockedFs.readFile.callsFake(async (filePath) => {
+                if (filePath === path.join(projectPathUri, 'src', 'sample.cs')) {
+                    return `using static Amazon.Cw.Sample;`;
+                }
+                if (filePath === path.join(projectPathUri, 'src', 'model.cs')) {
+                    return ``;
+                }
+            });
+            await csharpDependencyGraph.createNamespaceFilenameMapper(projectPathUri);
+            assert.deepEqual(csharpDependencyGraph.namespaceToFilepathDirectory, new Map([]));
+        });
+    });
+    describe('Test searchDependency', () => {
+        beforeEach(() => {
+            mockedFs.getFileSize.reset();
+            mockedFs.readFile.reset();
+            csharpDependencyGraph.namespaceToFilepathDirectory = new Map([
+                ['Amazon.Cw.Model', new Set([path.join(projectPathUri, 'src', 'model.cs')])],
+                ['Amazon.Cw.Utils.Sample', new Set([path.join(projectPathUri, 'src', 'sample.cs')])],
+                [
+                    'Amazon.Cw.Props',
+                    new Set([
+                        path.join(projectPathUri, 'src', 'interface', 'model.cs'),
+                        path.join(projectPathUri, 'src', 'interface', 'recommendations.cs'),
+                    ]),
+                ],
+            ]);
+        });
+        it('should return source file only for no dependencies', async () => {
+            mockedFs.readFile.resolves(`var total = 5 + 4;`);
+            mockedFs.getFileSize.resolves({ size: 1000 });
+            const filePath = path.join(projectPathUri, 'main.cs');
+            const pickedSourceFiles = await csharpDependencyGraph.searchDependency(filePath);
+            assert.deepEqual(pickedSourceFiles, new Set([filePath]));
+        });
+        it("should return source file with its dependecies' file path", async () => {
+            mockedFs.getFileSize.atLeast(1).resolves({ size: 1000 });
+            mockedFs.readFile.onFirstCall().resolves(`using Amazon.Cw.Props;
+        var total = 5 + 4;`);
+            const pickedSourceFiles = await csharpDependencyGraph.searchDependency(path.join(projectPathUri, 'main.cs'));
+            assert.deepEqual(pickedSourceFiles, new Set([
+                path.join(projectPathUri, 'main.cs'),
+                path.join(projectPathUri, 'src', 'interface', 'recommendations.cs'),
+                path.join(projectPathUri, 'src', 'interface', 'model.cs'),
+            ]));
+        });
+        it("should return source file with its dependecies' until it reaches to payload size limit", async () => {
+            mockedFs.getFileSize.onFirstCall().resolves({ size: Math.pow(2, 19) });
+            mockedFs.getFileSize.onSecondCall().resolves({ size: Math.pow(2, 19) });
+            mockedFs.readFile.onFirstCall().resolves(`using Amazon.Cw.Props;
+        var total = 5 + 4;`);
+            const pickedSourceFiles = await csharpDependencyGraph.searchDependency(path.join(projectPathUri, 'main.cs'));
+            assert.deepEqual(pickedSourceFiles, new Set([
+                path.join(projectPathUri, 'main.cs'),
+                path.join(projectPathUri, 'src', 'interface', 'model.cs'),
+            ]));
+        });
+    });
+    describe('Test traverseDir', () => {
+        beforeEach(() => {
+            mockedFs.getFileSize.reset();
+            mockedFs.readFile.reset();
+            mockedFs.readdir.reset();
+            csharpDependencyGraph.namespaceToFilepathDirectory = new Map([
+                ['Amazon.Cw.Model', new Set([path.join(projectPathUri, 'src', 'model.cs')])],
+                ['Amazon.Cw.Utils.Sample', new Set([path.join(projectPathUri, 'src', 'sample.cs')])],
+                [
+                    'Amazon.Cw.Props',
+                    new Set([
+                        path.join(projectPathUri, 'src', 'interface', 'scan.cs'),
+                        path.join(projectPathUri, 'src', 'interface', 'recommendations.cs'),
+                    ]),
+                ],
+            ]);
+            mockedFs.readdir.resolves([
+                {
+                    isFile: () => false,
+                    isDirectory: () => true,
+                    name: 'src',
+                    path: projectPathUri,
+                },
+                {
+                    isFile: () => true,
+                    isDirectory: () => false,
+                    name: 'sample.cs',
+                    path: path.join(projectPathUri, 'src'),
+                },
+                {
+                    isFile: () => true,
+                    isDirectory: () => false,
+                    name: 'model.cs',
+                    path: path.join(projectPathUri, 'src'),
+                },
+                {
+                    isFile: () => true,
+                    isDirectory: () => false,
+                    name: 'scan.cs',
+                    path: path.join(projectPathUri, 'src', 'props'),
+                },
+                {
+                    isFile: () => true,
+                    isDirectory: () => false,
+                    name: 'recommendation.cs',
+                    path: path.join(projectPathUri, 'src', 'props'),
+                },
+            ]);
+        });
+        it('should return without traversing due to payload size limit reached', async () => {
+            mockedFs.getFileSize.resolves({ size: Math.pow(2, 20) });
+            await csharpDependencyGraph.searchDependency(path.join(projectPathUri, 'main.cs'));
+            await csharpDependencyGraph.traverseDir(projectPathUri);
+            assert.equal(mockedFs.readFile.calledWith(projectPathUri), false);
+        });
+        it('should traverse through files until it reaches payload size limit', async () => {
+            mockedFs.getFileSize.atLeast(1).resolves({ size: Math.pow(2, 19) });
+            await csharpDependencyGraph.traverseDir(projectPathUri);
+            assert.strictEqual(csharpDependencyGraph.isProjectTruncated, true);
+        });
+        it('should traverse through all files', async () => {
+            mockedFs.getFileSize.atLeast(1).resolves({ size: Math.pow(2, 10) });
+            await csharpDependencyGraph.traverseDir(projectPathUri);
+            assert.strictEqual(csharpDependencyGraph.isProjectTruncated, false);
+        });
+    });
+    describe('Test getDependencies', () => {
+        beforeEach(() => {
+            csharpDependencyGraph.namespaceToFilepathDirectory = new Map([
+                ['Amazon.Cw.Model', new Set([path.join(projectPathUri, 'src', 'model.cs')])],
+                ['Amazon.Cw.Utils.Sample', new Set([path.join(projectPathUri, 'src', 'sample.cs')])],
+                [
+                    'Amazon.Cw.Props',
+                    new Set([
+                        path.join(projectPathUri, 'src', 'interface', 'scan.cs'),
+                        path.join(projectPathUri, 'src', 'interface', 'recommendations.cs'),
+                    ]),
+                ],
+            ]);
+        });
+        it('should return file paths for given imports', () => {
+            const expectedResponse = new Set([
+                path.join(projectPathUri, 'src', 'model.cs'),
+                path.join(projectPathUri, 'src', 'sample.cs'),
+            ]);
+            assert.deepEqual(csharpDependencyGraph.getDependencies(['Amazon.Cw.Utils.Sample', 'Amazon.Cw.Model']), expectedResponse);
+        });
+        it('should return empty list', () => {
+            assert.deepEqual(csharpDependencyGraph.getDependencies(['Amazon.Cw']), new Set());
+        });
+    });
+    describe('Test readImports', () => {
+        it('should return list of imports ', () => {
+            const content = `
+global using static Amazon.Cw.Utils;
+using Amazon.Cw.Model;
+using Proto = Amazon.Cw.Prototypes;
+namespace Amazon.Toolkit.Demo {
+  class Main {
+    pulbic print() {
+
+    }
+  }
+}
+        `;
+            const response = csharpDependencyGraph.readImports(content);
+            const expectedImports = ['Amazon.Cw.Utils', 'Amazon.Cw.Model', 'Amazon.Cw.Prototypes'];
+            assert.deepEqual(response, expectedImports);
+        });
+        it('should return empty list', () => {
+            const content = `
+namespace Amazon.Toolkit.Demo {
+  class Main {
+    pulbic print() {
+    }
+  }
+}
+        `;
+            const response = csharpDependencyGraph.readImports(content);
+            assert.deepEqual(response, []);
+        });
+    });
+    describe('Test generateTruncation', () => {
+        before(() => {
+            Sinon.stub(Date, 'now').returns(111111111);
+        });
+        it('should call zip dir', async () => {
+            const zipSize = Math.pow(2, 19);
+            const zipFileBuffer = 'dummy-zip-data';
+            mockedFs.getFileSize.atLeast(1).resolves({ size: zipSize });
+            csharpDependencyGraph.createZip = Sinon.stub().returns({
+                zipFileBuffer,
+                zipFileSize: zipSize,
+            });
+            const expectedResult = {
+                rootDir: path.join(tempDirPath, 'codewhisperer_scan_111111111'),
+                zipFileBuffer,
+                scannedFiles: new Set([path.join(projectPathUri, 'main.cs')]),
+                srcPayloadSizeInBytes: 0,
+                zipFileSizeInBytes: zipSize,
+                buildPayloadSizeInBytes: 0,
+                lines: 0,
+            };
+            const trucation = await csharpDependencyGraph.generateTruncation(path.join(projectPathUri, 'main.cs'));
+            assert.deepEqual(trucation, expectedResult);
+        });
+    });
+});
+//# sourceMappingURL=csharpDependencyGraph.test.js.map

package/out/language-server/dependencyGraph/csharpDependencyGraph.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"csharpDependencyGraph.test.js","sourceRoot":"","sources":["../../../src/language-server/dependencyGraph/csharpDependencyGraph.test.ts"],"names":[],"mappings":";;AAAA,iCAAgC;AAChC,6BAA4B;AAC5B,+BAA8B;AAC9B,mEAA+D;AAE/D,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IACxC,IAAI,qBAA4C,CAAA;IAChD,MAAM,wBAAwB,GAAG,KAAK,CAAC,IAAI,EAAE,CAAA;IAC7C,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAA;IACrE,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;IAC1C,MAAM,QAAQ,GAAG;QACb,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;QAClB,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE;QACpB,WAAW,EAAE,KAAK,CAAC,IAAI,EAAE;QACzB,cAAc,EAAE,GAAG,EAAE,CAAC,WAAW;QACjC,OAAO,EAAE,KAAK,CAAC,IAAI,EAAE;QACrB,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE;QACtB,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE;QACpB,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE;KACvB,CAAA;IACD,MAAM,eAAe,GAAG;QACpB,eAAe,EAAE,KAAK,CAAC,IAAI,EAAE;QAC7B,kBAAkB,EAAE,wBAAwB;QAC5C,EAAE,EAAE,QAAQ;KACf,CAAA;IACD,MAAM,CAAC,GAAG,EAAE;QACR,qBAAqB,GAAG,IAAI,6CAAqB,CAAC,eAAe,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IACF,UAAU,CAAC,GAAG,EAAE;QACZ,wBAAwB,CAAC,KAAK,EAAE,CAAA;QAChC,wBAAwB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAE3C,qBAAqB,GAAG,IAAI,6CAAqB,CAAC,eAAe,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC1C,MAAM,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;YAC3C,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,0BAA0B,EAAE,EAAE,mBAAmB,CAAC,CAAA;QACzF,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACjC,UAAU,CAAC,GAAG,EAAE;YACZ,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;YACvB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC5F,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,KAAK,CAAC,MAAM,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,EAAE,UAAU,CAAC,CAAA;QAChH,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,mBAAmB,GAAG,UAAU,CAAA;YACtC,wBAAwB,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,cAAc,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAA;YACpF,MAAM,CAAC,KAAK,CACR,MAAM,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,EAC5E,mBAAmB,CACtB,CAAA;QACL,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACjC,UAAU,CAAC,GAAG,EAAE;YACZ,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;YACvB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAA;YACvD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,CAAA;YACtD,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;YAE9B,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;YACpE,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAA;QAC1C,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,mBAAmB,GAAG,UAAU,CAAA;YAEtC,wBAAwB,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,cAAc,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAA;YACpF,MAAM,CAAC,KAAK,CAAC,MAAM,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAA;YAC1G,wBAAwB,CAAC,KAAK,EAAE,CAAA;QACpC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YAC/B,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,oBAAoB,EAAE,EAAE,KAAK,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACpC,MAAM,IAAI,GAAG,GAAG,CAAA;YAChB,MAAM,cAAc,GAAG,KAAK,CAAA;YAC5B,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,cAAc,CAAC,CAAA;QAC9E,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAChC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;YAC5B,MAAM,cAAc,GAAG,IAAI,CAAA;YAC3B,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,cAAc,CAAC,CAAA;QAC9E,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAChD,UAAU,CAAC,GAAG,EAAE;YACZ,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;YACxB,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;YACzB,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,KAAK;oBACnB,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI;oBACvB,IAAI,EAAE,KAAK;oBACX,IAAI,EAAE,cAAc;iBACvB;gBACD;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;oBAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;oBACxB,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC;iBACzC;gBACD;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;oBAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;oBACxB,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC;iBACzC;aACJ,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACtE,qBAAqB,CAAC,4BAA4B,GAAG,IAAI,GAAG,EAAuB,CAAA;YACnF,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAC,QAAQ,EAAC,EAAE;gBACzC,IAAI,QAAQ,KAAK,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;oBAC7D,OAAO,qCAAqC,CAAA;gBAChD,CAAC;gBACD,IAAI,QAAQ,KAAK,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC;oBAC5D,OAAO,8BAA8B,CAAA;gBACzC,CAAC;YACL,CAAC,CAAC,CAAA;YACF,MAAM,qBAAqB,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAA;YACzE,MAAM,CAAC,SAAS,CACZ,qBAAqB,CAAC,4BAA4B,EAClD,IAAI,GAAG,CAAC;gBACJ,CAAC,iBAAiB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;gBAC5E,CAAC,wBAAwB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;aACvF,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACrC,qBAAqB,CAAC,4BAA4B,GAAG,IAAI,GAAG,EAAuB,CAAA;YACnF,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAC,QAAQ,EAAC,EAAE;gBACzC,IAAI,QAAQ,KAAK,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;oBAC7D,OAAO,gCAAgC,CAAA;gBAC3C,CAAC;gBACD,IAAI,QAAQ,KAAK,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC;oBAC5D,OAAO,EAAE,CAAA;gBACb,CAAC;YACL,CAAC,CAAC,CAAA;YACF,MAAM,qBAAqB,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAA;YACzE,MAAM,CAAC,SAAS,CAAC,qBAAqB,CAAC,4BAA4B,EAAE,IAAI,GAAG,CAAC,EAAE,CAAC,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACnC,UAAU,CAAC,GAAG,EAAE;YACZ,QAAQ,CAAC,WAAW,CAAC,KAAK,EAAE,CAAA;YAC5B,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;YAEzB,qBAAqB,CAAC,4BAA4B,GAAG,IAAI,GAAG,CAAC;gBACzD,CAAC,iBAAiB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;gBAC5E,CAAC,wBAAwB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;gBACpF;oBACI,iBAAiB;oBACjB,IAAI,GAAG,CAAC;wBACJ,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,CAAC;wBACzD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,oBAAoB,CAAC;qBACtE,CAAC;iBACL;aACJ,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAChE,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAA;YAChD,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAA;YAErD,MAAM,iBAAiB,GAAG,MAAM,qBAAqB,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAA;YAChF,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;QAC5D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACvE,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAExD,QAAQ,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,QAAQ,CACpC;2BACW,CACd,CAAA;YACD,MAAM,iBAAiB,GAAG,MAAM,qBAAqB,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAA;YAC5G,MAAM,CAAC,SAAS,CACZ,iBAAiB,EACjB,IAAI,GAAG,CAAC;gBACJ,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC;gBACpC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,oBAAoB,CAAC;gBACnE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,CAAC;aAC5D,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;YACpG,QAAQ,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;YACtE,QAAQ,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;YAEvE,QAAQ,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,QAAQ,CACpC;2BACW,CACd,CAAA;YACD,MAAM,iBAAiB,GAAG,MAAM,qBAAqB,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAA;YAC5G,MAAM,CAAC,SAAS,CACZ,iBAAiB,EACjB,IAAI,GAAG,CAAC;gBACJ,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC;gBACpC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,CAAC;aAC5D,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC9B,UAAU,CAAC,GAAG,EAAE;YACZ,QAAQ,CAAC,WAAW,CAAC,KAAK,EAAE,CAAA;YAC5B,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;YACzB,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;YACxB,qBAAqB,CAAC,4BAA4B,GAAG,IAAI,GAAG,CAAC;gBACzD,CAAC,iBAAiB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;gBAC5E,CAAC,wBAAwB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;gBACpF;oBACI,iBAAiB;oBACjB,IAAI,GAAG,CAAC;wBACJ,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,CAAC;wBACxD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,oBAAoB,CAAC;qBACtE,CAAC;iBACL;aACJ,CAAC,CAAA;YACF,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,KAAK;oBACnB,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI;oBACvB,IAAI,EAAE,KAAK;oBACX,IAAI,EAAE,cAAc;iBACvB;gBACD;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;oBAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;oBACxB,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC;iBACzC;gBACD;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;oBAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;oBACxB,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC;iBACzC;gBACD;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;oBAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;oBACxB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,OAAO,CAAC;iBAClD;gBACD;oBACI,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;oBAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;oBACxB,IAAI,EAAE,mBAAmB;oBACzB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,OAAO,CAAC;iBAClD;aACJ,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAChF,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;YACxD,MAAM,qBAAqB,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAA;YAClF,MAAM,qBAAqB,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;YACvD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YAC/E,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;YACnE,MAAM,qBAAqB,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;YACvD,MAAM,CAAC,WAAW,CAAC,qBAAqB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAA;QACtE,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YAC/C,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;YACnE,MAAM,qBAAqB,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;YACvD,MAAM,CAAC,WAAW,CAAC,qBAAqB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAClC,UAAU,CAAC,GAAG,EAAE;YACZ,qBAAqB,CAAC,4BAA4B,GAAG,IAAI,GAAG,CAAC;gBACzD,CAAC,iBAAiB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;gBAC5E,CAAC,wBAAwB,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;gBACpF;oBACI,iBAAiB;oBACjB,IAAI,GAAG,CAAC;wBACJ,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,CAAC;wBACxD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,EAAE,oBAAoB,CAAC;qBACtE,CAAC;iBACL;aACJ,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;gBAC7B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC;gBAC5C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,WAAW,CAAC;aAChD,CAAC,CAAA;YACF,MAAM,CAAC,SAAS,CACZ,qBAAqB,CAAC,eAAe,CAAC,CAAC,wBAAwB,EAAE,iBAAiB,CAAC,CAAC,EACpF,gBAAgB,CACnB,CAAA;QACL,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAChC,MAAM,CAAC,SAAS,CAAC,qBAAqB,CAAC,eAAe,CAAC,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,GAAG,EAAE,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG;;;;;;;;;;;SAWnB,CAAA;YACG,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;YAC3D,MAAM,eAAe,GAAG,CAAC,iBAAiB,EAAE,iBAAiB,EAAE,sBAAsB,CAAC,CAAA;YACtF,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAA;QAC/C,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAChC,MAAM,OAAO,GAAG;;;;;;;SAOnB,CAAA;YACG,MAAM,QAAQ,GAAG,qBAAqB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;YAC3D,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAA;QAClC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,GAAG,EAAE;YACR,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;YAC/B,MAAM,aAAa,GAAG,gBAAgB,CAAA;YACtC,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YAC3D,qBAAqB,CAAC,SAAS,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC;gBACnD,aAAa;gBACb,WAAW,EAAE,OAAO;aACvB,CAAC,CAAA;YACF,MAAM,cAAc,GAAG;gBACnB,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,8BAA8B,CAAC;gBAC/D,aAAa;gBACb,YAAY,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAC;gBAC7D,qBAAqB,EAAE,CAAC;gBACxB,kBAAkB,EAAE,OAAO;gBAC3B,uBAAuB,EAAE,CAAC;gBAC1B,KAAK,EAAE,CAAC;aACX,CAAA;YAED,MAAM,SAAS,GAAG,MAAM,qBAAqB,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAA;YAEtG,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE,cAAc,CAAC,CAAA;QAC/C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

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

@@ -0,0 +1,125 @@
+/// <reference types="node" />
+import { Workspace } from '@aws/language-server-runtimes/out/features';
+export interface Truncation {
+    rootDir: string;
+    zipFileBuffer: Buffer;
+    scannedFiles: Set<string>;
+    srcPayloadSizeInBytes: number;
+    buildPayloadSizeInBytes: number;
+    zipFileSizeInBytes: number;
+    lines: number;
+}
+/**
+ * Create a dependency graph to select all the dependent files associated with current active file.
+ * @param uri
+ * @returns
+ */
+export declare abstract class DependencyGraph {
+    workspace: Workspace;
+    protected _sysPaths: Set<string>;
+    protected _parsedStatements: Set<string>;
+    protected _pickedSourceFiles: Set<string>;
+    protected _fetchedDirs: Set<string>;
+    protected _totalSize: number;
+    protected _tmpDir: string;
+    protected _truncDir: string;
+    protected _totalLines: number;
+    private _isProjectTruncated;
+    constructor(workspace: Workspace);
+    /**
+     * Retrun project name if given uri is within project. If not then return last folder name where file is present.
+     * @param uri file path to get project name
+     * @returns project folder name
+     */
+    getProjectName(uri: string): Promise<string>;
+    /**
+     * Retrun project path if given uri is within project. If not then return current directory path where file is present.
+     * @param uri file path to get project path
+     * @returns project path uri if found within workspace otherwise current directory path of input uri
+     */
+    getProjectPath(uri: string): Promise<string>;
+    /**
+     * Retrun readable size limit value in MB or KB
+     */
+    getReadableSizeLimit(): string;
+    /**
+     * checks if the given size value exceeds the payload size limit or not
+     */
+    exceedsSizeLimit(size: number): boolean;
+    /**
+     * get value of _isProjectTruncated
+     */
+    get isProjectTruncated(): boolean;
+    /**
+     * set value for _isProjectTruncated
+     */
+    set isProjectTruncated(value: boolean);
+    /**
+     * copy list of file to temp dir
+     */
+    protected copyFilesToTmpDir(files: Set<string> | string[], dir: string): Promise<void>;
+    /**
+     * copy input file in temp dir (destDir) with same relative path of srcFile in destDir.
+     * @param srcFilePath file path to be copied
+     * @param destDir destination directory path
+     */
+    protected copyFileToTmp(srcFilePath: string, destDir: string): Promise<void>;
+    /**
+     * create a zip dir buffer of the given dir.
+     * @param dir directory path to create zip of the directory
+     * @returns zipped dir buffer object and its size in bytes
+     */
+    createZip(dir: string): {
+        zipFileBuffer: Buffer;
+        zipFileSize: number;
+    };
+    /**
+     * delete directory along with its sub-directories and files if it exists
+     * @param dir directory path to remove
+     */
+    protected removeDir(dir: string): Promise<void>;
+    /**
+     * get or create truncation directory path
+     * @returns truncation directory path
+     */
+    protected getTruncDirPath(): string;
+    /**
+     * Get total size of list of files
+     * @param files list of file paths to find size
+     * @returns total size of the input list of files
+     */
+    protected getFilesTotalSize(files: string[]): Promise<number>;
+    /**
+     * remove all copied files from temp directory
+     */
+    protected removeTmpFiles(truncation: Truncation): Promise<void>;
+    /**
+     * This method will traverse throw the input file and its dependecy files to creates a list of files for the scan.
+     * If the list does not exceeds the payload size limit then it will scan all the remaining files to add them into the list
+     * until it reaches to the payload size limit. Then, it copies all the selected files to temp directory, creates a zip buffer.
+     * @param uri file path for which truncation being created
+     * @returns Truncation object
+     */
+    abstract generateTruncation(uri: string): Promise<Truncation>;
+    /**
+     * Search for all the dependecies for the given input file. Store the input filepath along with
+     * the dependent filepaths until it does not exceeds the payload size limit.
+     * @param uri file path to seach dependency of the file
+     * @returns set of file paths found as dependecy of input uri
+     */
+    abstract searchDependency(uri: string): Promise<Set<string>>;
+    /**
+     * Traverse all the Csharp files in the given workspace and add each file along with its dependencies to
+     * the list of files selected for security scan.
+     */
+    abstract traverseDir(dirPath: string): void;
+    /**
+     * Create a dependency file path set for given list of namespace.
+     * @param imports list of import strings
+     */
+    abstract getDependencies(imports: string[]): void;
+    /**
+     * Get payload size limit in bytes for the given Language.
+     */
+    abstract getPayloadSizeLimitInBytes(): number;
+}

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

@@ -0,0 +1,152 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DependencyGraph = void 0;
+const admZip = require("adm-zip");
+const path = require("path");
+const CodeWhispererConstants = require("./constants");
+/**
+ * Create a dependency graph to select all the dependent files associated with current active file.
+ * @param uri
+ * @returns
+ */
+class DependencyGraph {
+    constructor(workspace) {
+        this._sysPaths = new Set();
+        this._parsedStatements = new Set();
+        this._pickedSourceFiles = new Set();
+        this._fetchedDirs = new Set();
+        this._totalSize = 0;
+        this._tmpDir = '';
+        this._truncDir = '';
+        this._totalLines = 0;
+        this._isProjectTruncated = false;
+        this.workspace = workspace;
+        this._tmpDir = this.workspace.fs.getTempDirPath();
+    }
+    /**
+     * Retrun project name if given uri is within project. If not then return last folder name where file is present.
+     * @param uri file path to get project name
+     * @returns project folder name
+     */
+    async getProjectName(uri) {
+        const projectPath = await this.getProjectPath(uri);
+        return path.basename(projectPath);
+    }
+    /**
+     * Retrun project path if given uri is within project. If not then return current directory path where file is present.
+     * @param uri file path to get project path
+     * @returns project path uri if found within workspace otherwise current directory path of input uri
+     */
+    async getProjectPath(uri) {
+        const workspaceFolder = this.workspace.getWorkspaceFolder(uri);
+        if (!workspaceFolder) {
+            return (await this.workspace.fs.isFile(uri)) ? path.dirname(uri) : uri;
+        }
+        return workspaceFolder.uri;
+    }
+    /**
+     * Retrun readable size limit value in MB or KB
+     */
+    getReadableSizeLimit() {
+        const totalBytesInMB = Math.pow(2, 20);
+        const totalBytesInKB = Math.pow(2, 10);
+        if (this.getPayloadSizeLimitInBytes() >= totalBytesInMB) {
+            return `${this.getPayloadSizeLimitInBytes() / totalBytesInMB}MB`;
+        }
+        else {
+            return `${this.getPayloadSizeLimitInBytes() / totalBytesInKB}KB`;
+        }
+    }
+    /**
+     * checks if the given size value exceeds the payload size limit or not
+     */
+    exceedsSizeLimit(size) {
+        return size >= this.getPayloadSizeLimitInBytes();
+    }
+    /**
+     * get value of _isProjectTruncated
+     */
+    get isProjectTruncated() {
+        return this._isProjectTruncated;
+    }
+    /**
+     * set value for _isProjectTruncated
+     */
+    set isProjectTruncated(value) {
+        this._isProjectTruncated = value;
+    }
+    /**
+     * copy list of file to temp dir
+     */
+    async copyFilesToTmpDir(files, dir) {
+        // Convert Set to an array for compatibility with Promise.all
+        const fileArray = Array.from(files);
+        // Use Promise.all to asynchronously copy all files
+        await Promise.all(fileArray.map(async (filePath) => {
+            await this.copyFileToTmp(filePath, dir);
+        }));
+    }
+    /**
+     * copy input file in temp dir (destDir) with same relative path of srcFile in destDir.
+     * @param srcFilePath file path to be copied
+     * @param destDir destination directory path
+     */
+    async copyFileToTmp(srcFilePath, destDir) {
+        const sourceWorkspacePath = await this.getProjectPath(srcFilePath);
+        const fileRelativePath = path.relative(sourceWorkspacePath, srcFilePath);
+        const destinationFileAbsolutePath = path.join(destDir, fileRelativePath);
+        await this.workspace.fs.copy(srcFilePath, destinationFileAbsolutePath);
+    }
+    /**
+     * create a zip dir buffer of the given dir.
+     * @param dir directory path to create zip of the directory
+     * @returns zipped dir buffer object and its size in bytes
+     */
+    createZip(dir) {
+        const zip = new admZip();
+        zip.addLocalFolder(dir);
+        // writeZip uses `fs` under the hood and it wouldn't work in browsers
+        // Instead of writeZip to write to disk.
+        // The zip buffer can then be uploaded to s3
+        const zipBuffer = zip.toBuffer();
+        return { zipFileBuffer: zipBuffer, zipFileSize: zipBuffer.byteLength };
+    }
+    /**
+     * delete directory along with its sub-directories and files if it exists
+     * @param dir directory path to remove
+     */
+    async removeDir(dir) {
+        if (await this.workspace.fs.exists(dir)) {
+            await this.workspace.fs.remove(dir);
+        }
+    }
+    /**
+     * get or create truncation directory path
+     * @returns truncation directory path
+     */
+    getTruncDirPath() {
+        if (this._truncDir === '') {
+            this._truncDir = path.join(this._tmpDir, CodeWhispererConstants.codeScanTruncDirPrefix + '_' + Date.now().toString());
+        }
+        return this._truncDir;
+    }
+    /**
+     * Get total size of list of files
+     * @param files list of file paths to find size
+     * @returns total size of the input list of files
+     */
+    async getFilesTotalSize(files) {
+        const fileStatsPromises = files.map(file => this.workspace.fs.getFileSize(file));
+        const fileStats = await Promise.all(fileStatsPromises);
+        const totalSize = fileStats.reduce((accumulator, { size }) => accumulator + size, 0);
+        return totalSize;
+    }
+    /**
+     * remove all copied files from temp directory
+     */
+    async removeTmpFiles(truncation) {
+        await this.removeDir(truncation.rootDir);
+    }
+}
+exports.DependencyGraph = DependencyGraph;
+//# sourceMappingURL=dependencyGraph.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"dependencyGraph.js","sourceRoot":"","sources":["../../../src/language-server/dependencyGraph/dependencyGraph.ts"],"names":[],"mappings":";;;AACA,kCAAiC;AACjC,6BAA4B;AAC5B,sDAAqD;AAYrD;;;;GAIG;AACH,MAAsB,eAAe;IAajC,YAAY,SAAoB;QAXtB,cAAS,GAAgB,IAAI,GAAG,EAAU,CAAA;QAC1C,sBAAiB,GAAgB,IAAI,GAAG,EAAU,CAAA;QAClD,uBAAkB,GAAgB,IAAI,GAAG,EAAU,CAAA;QACnD,iBAAY,GAAgB,IAAI,GAAG,EAAU,CAAA;QAC7C,eAAU,GAAG,CAAC,CAAA;QACd,YAAO,GAAG,EAAE,CAAA;QACZ,cAAS,GAAG,EAAE,CAAA;QACd,gBAAW,GAAG,CAAC,CAAA;QAEjB,wBAAmB,GAAG,KAAK,CAAA;QAG/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;QAC1B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,cAAc,EAAE,CAAA;IACrD,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,cAAc,CAAC,GAAW;QACnC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAA;QAClD,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAA;IACrC,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,cAAc,CAAC,GAAW;QACnC,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAE9D,IAAI,CAAC,eAAe,EAAE,CAAC;YACnB,OAAO,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAA;QAC1E,CAAC;QACD,OAAO,eAAe,CAAC,GAAG,CAAA;IAC9B,CAAC;IAED;;OAEG;IACI,oBAAoB;QACvB,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACtC,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACtC,IAAI,IAAI,CAAC,0BAA0B,EAAE,IAAI,cAAc,EAAE,CAAC;YACtD,OAAO,GAAG,IAAI,CAAC,0BAA0B,EAAE,GAAG,cAAc,IAAI,CAAA;QACpE,CAAC;aAAM,CAAC;YACJ,OAAO,GAAG,IAAI,CAAC,0BAA0B,EAAE,GAAG,cAAc,IAAI,CAAA;QACpE,CAAC;IACL,CAAC;IAED;;OAEG;IACI,gBAAgB,CAAC,IAAY;QAChC,OAAO,IAAI,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAA;IACpD,CAAC;IAED;;OAEG;IACH,IAAI,kBAAkB;QAClB,OAAO,IAAI,CAAC,mBAAmB,CAAA;IACnC,CAAC;IAED;;OAEG;IACH,IAAI,kBAAkB,CAAC,KAAc;QACjC,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAA;IACpC,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,iBAAiB,CAAC,KAA6B,EAAE,GAAW;QACxE,6DAA6D;QAC7D,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAEnC,mDAAmD;QACnD,MAAM,OAAO,CAAC,GAAG,CACb,SAAS,CAAC,GAAG,CAAC,KAAK,EAAC,QAAQ,EAAC,EAAE;YAC3B,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAA;QAC3C,CAAC,CAAC,CACL,CAAA;IACL,CAAC;IAED;;;;OAIG;IACO,KAAK,CAAC,aAAa,CAAC,WAAmB,EAAE,OAAe;QAC9D,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAA;QAClE,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAA;QACxE,MAAM,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAA;QACxE,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAAA;IAC1E,CAAC;IAED;;;;OAIG;IACH,SAAS,CAAC,GAAW;QACjB,MAAM,GAAG,GAAG,IAAI,MAAM,EAAE,CAAA;QACxB,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,CAAA;QAEvB,qEAAqE;QACrE,wCAAwC;QACxC,4CAA4C;QAC5C,MAAM,SAAS,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAA;QAEhC,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,CAAC,UAAU,EAAE,CAAA;IAC1E,CAAC;IAED;;;OAGG;IACO,KAAK,CAAC,SAAS,CAAC,GAAW;QACjC,IAAI,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;QACvC,CAAC;IACL,CAAC;IAED;;;OAGG;IACO,eAAe;QACrB,IAAI,IAAI,CAAC,SAAS,KAAK,EAAE,EAAE,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,IAAI,CACtB,IAAI,CAAC,OAAO,EACZ,sBAAsB,CAAC,sBAAsB,GAAG,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAC9E,CAAA;QACL,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAA;IACzB,CAAC;IAED;;;;OAIG;IACO,KAAK,CAAC,iBAAiB,CAAC,KAAe;QAC7C,MAAM,iBAAiB,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAA;QAChF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;QACtD,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,WAAW,GAAG,IAAI,EAAE,CAAC,CAAC,CAAA;QACpF,OAAO,SAAS,CAAA;IACpB,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,cAAc,CAAC,UAAsB;QACjD,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAA;IAC5C,CAAC;CAmCJ;AApMD,0CAoMC"}

package/out/language-server/dependencyGraph/dependencyGraphFactory.d.ts

@@ -0,0 +1,15 @@
+import { Workspace } from '@aws/language-server-runtimes/out/features';
+import { TextDocument } from 'vscode-languageserver-textdocument';
+import { CsharpDependencyGraph } from './csharpDependencyGraph';
+declare const languageMap: {
+    readonly csharp: typeof CsharpDependencyGraph;
+};
+type LanguageMap = typeof languageMap;
+type Keys = keyof LanguageMap;
+type Tuples<T> = T extends Keys ? [T, InstanceType<LanguageMap[T]>] : never;
+type ClassType<A extends Keys> = Extract<Tuples<Keys>, [A, any]>[1];
+export declare class DependencyGraphFactory {
+    static getDependencyGraphFromFileExtensions<K extends Keys>(fileName: string): ClassType<K>;
+    static getDependencyGraph<K extends Keys>(document: TextDocument, workspace: Required<Workspace>): ClassType<K>;
+}
+export {};