@aws/lsp-codewhisperer 0.0.9 → 0.0.10

package/out/language-server/netTransform/transformHandler.js

@@ -0,0 +1,315 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TransformHandler = void 0;
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const fs = require("fs");
+const got_1 = require("got");
+const uuid_1 = require("uuid");
+const artifactManager_1 = require("./artifactManager");
+const converter_1 = require("./converter");
+const models_1 = require("./models");
+const validation = require("./validation");
+const path = require("path");
+const AdmZip = require("adm-zip");
+const workspaceFolderName = 'artifactWorkspace';
+class TransformHandler {
+    client;
+    workspace;
+    logging;
+    constructor(client, workspace, logging) {
+        this.client = client;
+        this.workspace = workspace;
+        this.logging = logging;
+    }
+    async startTransformation(userInputrequest) {
+        var unsupportedProjects = [];
+        const isProject = validation.isProject(userInputrequest);
+        const containsUnsupportedViews = await validation.checkForUnsupportedViews(userInputrequest, isProject);
+        /*
+        if (isProject) {
+            let isValid = validation.validateProject(userInputrequest)
+            if (!isValid) {
+                return {
+                    Error: 'NotSupported',
+                    IsSupported: false,
+                    ContainsUnsupportedViews: containsUnsupportedViews,
+                } as StartTransformResponse
+            }
+        } else {
+            unsupportedProjects = validation.validateSolution(userInputrequest)
+        }
+*/
+        const artifactManager = new artifactManager_1.ArtifactManager(this.workspace, this.logging, this.getWorkspacePath(userInputrequest.SolutionRootPath));
+        try {
+            const payloadFilePath = await this.zipCodeAsync(userInputrequest, artifactManager);
+            this.logging.log('payload path: ' + payloadFilePath);
+            const uploadId = await this.preTransformationUploadCode(payloadFilePath);
+            const request = (0, converter_1.getCWStartTransformRequest)(userInputrequest, uploadId);
+            this.logging.log('send request to start transform api: ' + JSON.stringify(request));
+            const response = await this.client.codeModernizerStartCodeTransformation(request);
+            this.logging.log('response start transform api: ' + JSON.stringify(response));
+            return (0, converter_1.getCWStartTransformResponse)(response, uploadId, payloadFilePath, unsupportedProjects, containsUnsupportedViews);
+        }
+        catch (error) {
+            const errorMessage = error.message ?? 'Error in StartTransformation API call';
+            this.logging.log(errorMessage);
+            throw new Error(errorMessage);
+        }
+        finally {
+            artifactManager.cleanup();
+        }
+    }
+    async preTransformationUploadCode(payloadFilePath) {
+        try {
+            const uploadId = await this.uploadPayloadAsync(payloadFilePath);
+            this.logging.log('artifact successfully uploaded. upload tracking id: ' + uploadId);
+            return uploadId;
+        }
+        catch (error) {
+            const errorMessage = error.message ?? 'Failed to upload zip file';
+            throw new Error(errorMessage);
+        }
+    }
+    async uploadPayloadAsync(payloadFileName) {
+        const sha256 = artifactManager_1.ArtifactManager.getSha256(payloadFileName);
+        let response;
+        try {
+            response = await this.client.codeModernizerCreateUploadUrl({
+                contentChecksum: sha256,
+                contentChecksumType: 'SHA_256',
+                uploadIntent: 'TRANSFORMATION',
+            });
+        }
+        catch (e) {
+            const errorMessage = e.message ?? 'Error in CreateUploadUrl API call';
+            this.logging.log('Error when creating Upload url: ' + errorMessage);
+            throw new Error(errorMessage);
+        }
+        try {
+            await this.uploadArtifactToS3Async(payloadFileName, response, sha256);
+        }
+        catch (e) {
+            const errorMessage = e.message ?? 'Error in uploadArtifactToS3 call';
+            this.logging.log('Error when calling uploadArtifactToS3Async: ' + errorMessage);
+            throw new Error(errorMessage);
+        }
+        return response.uploadId;
+    }
+    async zipCodeAsync(request, artifactManager) {
+        try {
+            return await artifactManager.createZip(request);
+        }
+        catch (e) {
+            this.logging.log('cause:' + e);
+        }
+        return '';
+    }
+    async uploadArtifactToS3Async(fileName, resp, sha256) {
+        const headersObj = this.getHeadersObj(sha256, resp.kmsKeyArn);
+        try {
+            const response = await got_1.default.put(resp.uploadUrl, {
+                body: fs.readFileSync(fileName),
+                headers: headersObj,
+            });
+            this.logging.log(`CodeTransform: Response from S3 Upload = ${response.statusCode}`);
+        }
+        catch (e) {
+            const errorMessage = e.message ?? 'Error in S3 UploadZip API call';
+            this.logging.log(errorMessage);
+        }
+    }
+    getHeadersObj(sha256, kmsKeyArn) {
+        let headersObj = {};
+        if (kmsKeyArn === undefined || kmsKeyArn.length === 0) {
+            headersObj = {
+                'x-amz-checksum-sha256': sha256,
+                'Content-Type': 'application/zip',
+            };
+        }
+        else {
+            headersObj = {
+                'x-amz-checksum-sha256': sha256,
+                'Content-Type': 'application/zip',
+                'x-amz-server-side-encryption': 'aws:kms',
+                'x-amz-server-side-encryption-aws-kms-key-id': kmsKeyArn,
+            };
+        }
+        return headersObj;
+    }
+    async getTransformation(request) {
+        try {
+            const getCodeTransformationRequest = {
+                transformationJobId: request.TransformationJobId,
+            };
+            this.logging.log('send request to get transform api: ' + JSON.stringify(getCodeTransformationRequest));
+            const response = await this.client.codeModernizerGetCodeTransformation(getCodeTransformationRequest);
+            this.logging.log('response received from get transform api: ' + JSON.stringify(response));
+            return {
+                TransformationJob: response.transformationJob,
+            };
+        }
+        catch (e) {
+            const errorMessage = e.message ?? 'Error in GetTransformation API call';
+            this.logging.log('Error: ' + errorMessage);
+            return {
+                TransformationJob: { status: 'FAILED' },
+            };
+        }
+    }
+    async getTransformationPlan(request) {
+        const getCodeTransformationPlanRequest = {
+            transformationJobId: request.TransformationJobId,
+        };
+        this.logging.log('send request to get transform plan api: ' + JSON.stringify(getCodeTransformationPlanRequest));
+        const response = await this.client.codeModernizerGetCodeTransformationPlan(getCodeTransformationPlanRequest);
+        this.logging.log('received response from get transform plan api: ' + JSON.stringify(response));
+        return {
+            TransformationPlan: response.transformationPlan,
+        };
+    }
+    async cancelTransformation(request) {
+        try {
+            const stopCodeTransformationRequest = {
+                transformationJobId: request.TransformationJobId,
+            };
+            this.logging.log('send request to cancel transform plan api: ' + JSON.stringify(stopCodeTransformationRequest));
+            const response = await this.client.codeModernizerStopCodeTransformation(stopCodeTransformationRequest);
+            this.logging.log('received response from cancel transform plan api: ' + JSON.stringify(response));
+            let status;
+            switch (response.transformationStatus) {
+                case 'STOPPED':
+                    status = models_1.CancellationJobStatus.SUCCESSFULLY_CANCELLED;
+                    break;
+                default:
+                    status = models_1.CancellationJobStatus.FAILED_TO_CANCEL;
+                    break;
+            }
+            return {
+                TransformationJobStatus: status,
+            };
+        }
+        catch (e) {
+            const errorMessage = e.message ?? 'Error in CancelTransformation API call';
+            this.logging.log('Error: ' + errorMessage);
+            return {
+                TransformationJobStatus: models_1.CancellationJobStatus.FAILED_TO_CANCEL,
+            };
+        }
+    }
+    async sleep(duration = 0) {
+        return new Promise(r => setTimeout(r, Math.max(duration, 0)));
+    }
+    async pollTransformation(request, validExitStatus, failureStates) {
+        let timer = 0;
+        const getCodeTransformationRequest = {
+            transformationJobId: request.TransformationJobId,
+        };
+        this.logging.log('poll : send request to get transform  api: ' + JSON.stringify(getCodeTransformationRequest));
+        let response = await this.client.codeModernizerGetCodeTransformation(getCodeTransformationRequest);
+        this.logging.log('poll : received response from get transform  api: ' + JSON.stringify(response));
+        let status = response?.transformationJob?.status ?? 'NOT_FOUND';
+        this.logging.log('validExitStatus here are : ' + validExitStatus);
+        this.logging.log('failureStatus here are : ' + failureStates);
+        while (status != 'Timed_out' && !failureStates.includes(status)) {
+            try {
+                const apiStartTime = Date.now();
+                const getCodeTransformationRequest = {
+                    transformationJobId: request.TransformationJobId,
+                };
+                this.logging.log('poll : send request to get transform  api: ' + JSON.stringify(getCodeTransformationRequest));
+                response = await this.client.codeModernizerGetCodeTransformation(getCodeTransformationRequest);
+                this.logging.log('poll : received response from get transform  api: ' + JSON.stringify(response));
+                this.logging.log('poll : job status here : ' + response.transformationJob.status);
+                if (response.transformationJob?.status) {
+                    this.logging.log('status is included in validExitSTatus for poll ' +
+                        validExitStatus.includes(response.transformationJob.status));
+                }
+                if (validExitStatus.includes(status)) {
+                    this.logging.log('returning status as : ' + status);
+                    break;
+                }
+                status = response.transformationJob.status;
+                await this.sleep(10 * 1000);
+                timer += 10;
+                this.logging.log('current polling timer ' + timer);
+                if (timer > 24 * 3600 * 1000) {
+                    status = 'Timed_out';
+                    break;
+                }
+            }
+            catch (e) {
+                const errorMessage = e.message ?? 'Error in GetTransformation API call';
+                this.logging.log('CodeTransformation: GetTransformation error = ' + errorMessage);
+                status = 'FAILED';
+                break;
+            }
+        }
+        this.logging.log('poll : returning response from server : ' + JSON.stringify(response));
+        return {
+            TransformationJob: response.transformationJob,
+        };
+    }
+    async downloadExportResultArchive(cwStreamingClient, exportId, saveToDir) {
+        let result;
+        try {
+            result = await cwStreamingClient.exportResultArchive({
+                exportId,
+                exportIntent: codewhisperer_streaming_1.ExportIntent.TRANSFORMATION,
+            });
+            const buffer = [];
+            this.logging.log('artifact downloaded successfully.');
+            if (result.body === undefined) {
+                throw new Error('Empty response from CodeWhisperer Streaming service.');
+            }
+            for await (const chunk of result.body) {
+                if (chunk.binaryPayloadEvent) {
+                    const chunkData = chunk.binaryPayloadEvent;
+                    if (chunkData.bytes) {
+                        buffer.push(chunkData.bytes);
+                    }
+                }
+            }
+            const pathContainingArchive = await this.archivePathGenerator(exportId, buffer, saveToDir);
+            this.logging.log('pathContainingArchive :' + pathContainingArchive);
+            return {
+                PathTosave: pathContainingArchive,
+            };
+        }
+        catch (error) {
+            const errorMessage = error.message ?? 'Failed to download the artifacts';
+            return {
+                Error: errorMessage,
+            };
+        }
+    }
+    async archivePathGenerator(exportId, buffer, saveToDir) {
+        const tempDir = path.join(saveToDir, exportId);
+        const pathToArchive = path.join(tempDir, 'ExportResultsArchive.zip');
+        await this.directoryExists(tempDir);
+        await fs.writeFileSync(pathToArchive, Buffer.concat(buffer));
+        let pathContainingArchive = '';
+        pathContainingArchive = path.dirname(pathToArchive);
+        const zip = new AdmZip(pathToArchive);
+        zip.extractAllTo(pathContainingArchive);
+        return pathContainingArchive;
+    }
+    async directoryExists(directoryPath) {
+        try {
+            await fs.accessSync(directoryPath);
+        }
+        catch (error) {
+            // Directory doesn't exist, create it
+            await fs.mkdirSync(directoryPath, { recursive: true });
+        }
+    }
+    getWorkspacePath(solutionRootPath) {
+        const randomPath = (0, uuid_1.v4)().substring(0, 8);
+        const workspacePath = path.join(solutionRootPath, workspaceFolderName, randomPath);
+        if (!fs.existsSync(workspacePath)) {
+            fs.mkdirSync(workspacePath, { recursive: true });
+        }
+        return workspacePath;
+    }
+}
+exports.TransformHandler = TransformHandler;
+//# sourceMappingURL=transformHandler.js.map

package/out/language-server/netTransform/transformHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transformHandler.js","sourceRoot":"","sources":["../../../src/language-server/netTransform/transformHandler.ts"],"names":[],"mappings":";;;AAAA,2EAAoF;AAEpF,yBAAwB;AACxB,6BAAqB;AACrB,+BAAmC;AAOnC,uDAAmD;AACnD,2CAAqF;AACrF,qCAYiB;AACjB,2CAA0C;AAC1C,6BAA6B;AAC7B,kCAAkC;AAOlC,MAAM,mBAAmB,GAAG,mBAAmB,CAAA;AAE/C,MAAa,gBAAgB;IACjB,MAAM,CAA2B;IACjC,SAAS,CAAW;IACpB,OAAO,CAAS;IACxB,YAAY,MAAiC,EAAE,SAAoB,EAAE,OAAgB;QACjF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;IAC1B,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,gBAAuC;QAC7D,IAAI,mBAAmB,GAAa,EAAE,CAAA;QACtC,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAA;QACxD,MAAM,wBAAwB,GAAG,MAAM,UAAU,CAAC,wBAAwB,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAA;QACvG;;;;;;;;;;;;;EAaN;QACM,MAAM,eAAe,GAAG,IAAI,iCAAe,CACvC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAC3D,CAAA;QACD,IAAI;YACA,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,eAAe,CAAC,CAAA;YAClF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,eAAe,CAAC,CAAA;YAEpD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,eAAe,CAAC,CAAA;YACxE,MAAM,OAAO,GAAG,IAAA,sCAA0B,EAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAA;YACtE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,uCAAuC,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;YACnF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,qCAAqC,CAAC,OAAO,CAAC,CAAA;YACjF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gCAAgC,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;YAC7E,OAAO,IAAA,uCAA2B,EAC9B,QAAQ,EACR,QAAQ,EACR,eAAe,EACf,mBAAmB,EACnB,wBAAwB,CAC3B,CAAA;SACJ;QAAC,OAAO,KAAK,EAAE;YACZ,MAAM,YAAY,GAAI,KAAe,CAAC,OAAO,IAAI,uCAAuC,CAAA;YACxF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAA;YAC9B,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAA;SAChC;gBAAS;YACN,eAAe,CAAC,OAAO,EAAE,CAAA;SAC5B;IACL,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,eAAuB;QACrD,IAAI;YACA,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAA;YAC/D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sDAAsD,GAAG,QAAQ,CAAC,CAAA;YACnF,OAAO,QAAQ,CAAA;SAClB;QAAC,OAAO,KAAK,EAAE;YACZ,MAAM,YAAY,GAAI,KAAe,CAAC,OAAO,IAAI,2BAA2B,CAAA;YAC5E,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAA;SAChC;IACL,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,eAAuB;QAC5C,MAAM,MAAM,GAAG,iCAAe,CAAC,SAAS,CAAC,eAAe,CAAC,CAAA;QACzD,IAAI,QAAiC,CAAA;QACrC,IAAI;YACA,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,6BAA6B,CAAC;gBACvD,eAAe,EAAE,MAAM;gBACvB,mBAAmB,EAAE,SAAS;gBAC9B,YAAY,EAAE,gBAAgB;aACjC,CAAC,CAAA;SACL;QAAC,OAAO,CAAM,EAAE;YACb,MAAM,YAAY,GAAI,CAAW,CAAC,OAAO,IAAI,mCAAmC,CAAA;YAChF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kCAAkC,GAAG,YAAY,CAAC,CAAA;YACnE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAA;SAChC;QAED,IAAI;YACA,MAAM,IAAI,CAAC,uBAAuB,CAAC,eAAe,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAA;SACxE;QAAC,OAAO,CAAM,EAAE;YACb,MAAM,YAAY,GAAI,CAAW,CAAC,OAAO,IAAI,kCAAkC,CAAA;YAC/E,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,8CAA8C,GAAG,YAAY,CAAC,CAAA;YAC/E,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAA;SAChC;QACD,OAAO,QAAQ,CAAC,QAAQ,CAAA;IAC5B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAA8B,EAAE,eAAgC;QAC/E,IAAI;YACA,OAAO,MAAM,eAAe,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;SAClD;QAAC,OAAO,CAAM,EAAE;YACb,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAA;SACjC;QACD,OAAO,EAAE,CAAA;IACb,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,QAAgB,EAAE,IAA6B,EAAE,MAAc;QACzF,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;QAC7D,IAAI;YACA,MAAM,QAAQ,GAAG,MAAM,aAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE;gBAC3C,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC;gBAC/B,OAAO,EAAE,UAAU;aACtB,CAAC,CAAA;YAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,4CAA4C,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAA;SACtF;QAAC,OAAO,CAAM,EAAE;YACb,MAAM,YAAY,GAAI,CAAW,CAAC,OAAO,IAAI,gCAAgC,CAAA;YAE7E,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAA;SACjC;IACL,CAAC;IAED,aAAa,CAAC,MAAc,EAAE,SAA6B;QACvD,IAAI,UAAU,GAAG,EAAE,CAAA;QACnB,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;YACnD,UAAU,GAAG;gBACT,uBAAuB,EAAE,MAAM;gBAC/B,cAAc,EAAE,iBAAiB;aACpC,CAAA;SACJ;aAAM;YACH,UAAU,GAAG;gBACT,uBAAuB,EAAE,MAAM;gBAC/B,cAAc,EAAE,iBAAiB;gBACjC,8BAA8B,EAAE,SAAS;gBACzC,6CAA6C,EAAE,SAAS;aAC3D,CAAA;SACJ;QACD,OAAO,UAAU,CAAA;IACrB,CAAC;IACD,KAAK,CAAC,iBAAiB,CAAC,OAA4B;QAChD,IAAI;YACA,MAAM,4BAA4B,GAAG;gBACjC,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;aACvB,CAAA;YAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,qCAAqC,GAAG,IAAI,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC,CAAA;YACtG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,mCAAmC,CAAC,4BAA4B,CAAC,CAAA;YACpG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,4CAA4C,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;YACzF,OAAO;gBACH,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;aACxB,CAAA;SAC5B;QAAC,OAAO,CAAM,EAAE;YACb,MAAM,YAAY,GAAI,CAAW,CAAC,OAAO,IAAI,qCAAqC,CAAA;YAClF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,GAAG,YAAY,CAAC,CAAA;YAE1C,OAAO;gBACH,iBAAiB,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE;aAClB,CAAA;SAC5B;IACL,CAAC;IACD,KAAK,CAAC,qBAAqB,CAAC,OAAgC;QACxD,MAAM,gCAAgC,GAAG;YACrC,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;SACvB,CAAA;QAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,0CAA0C,GAAG,IAAI,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC,CAAA;QAC/G,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,uCAAuC,CAAC,gCAAgC,CAAC,CAAA;QAC5G,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,iDAAiD,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;QAC9F,OAAO;YACH,kBAAkB,EAAE,QAAQ,CAAC,kBAAkB;SACtB,CAAA;IACjC,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,OAA+B;QACtD,IAAI;YACA,MAAM,6BAA6B,GAAG;gBAClC,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;aACtB,CAAA;YAC9B,IAAI,CAAC,OAAO,CAAC,GAAG,CACZ,6CAA6C,GAAG,IAAI,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAChG,CAAA;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,oCAAoC,CAAC,6BAA6B,CAAC,CAAA;YACtG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,oDAAoD,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;YACjG,IAAI,MAA6B,CAAA;YACjC,QAAQ,QAAQ,CAAC,oBAAoB,EAAE;gBACnC,KAAK,SAAS;oBACV,MAAM,GAAG,8BAAqB,CAAC,sBAAsB,CAAA;oBACrD,MAAK;gBACT;oBACI,MAAM,GAAG,8BAAqB,CAAC,gBAAgB,CAAA;oBAC/C,MAAK;aACZ;YACD,OAAO;gBACH,uBAAuB,EAAE,MAAM;aACP,CAAA;SAC/B;QAAC,OAAO,CAAM,EAAE;YACb,MAAM,YAAY,GAAI,CAAW,CAAC,OAAO,IAAI,wCAAwC,CAAA;YACrF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,GAAG,YAAY,CAAC,CAAA;YAC1C,OAAO;gBACH,uBAAuB,EAAE,8BAAqB,CAAC,gBAAgB;aACvC,CAAA;SAC/B;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC;QACpB,OAAO,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;IACjE,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAA4B,EAAE,eAAyB,EAAE,aAAuB;QACrG,IAAI,KAAK,GAAG,CAAC,CAAA;QAEb,MAAM,4BAA4B,GAAG;YACjC,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;SACvB,CAAA;QAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,6CAA6C,GAAG,IAAI,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC,CAAA;QAC9G,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,mCAAmC,CAAC,4BAA4B,CAAC,CAAA;QAClG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,oDAAoD,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;QACjG,IAAI,MAAM,GAAG,QAAQ,EAAE,iBAAiB,EAAE,MAAM,IAAI,WAAW,CAAA;QAE/D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,eAAe,CAAC,CAAA;QACjE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,GAAG,aAAa,CAAC,CAAA;QAE7D,OAAO,MAAM,IAAI,WAAW,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;YAC7D,IAAI;gBACA,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;gBAE/B,MAAM,4BAA4B,GAAG;oBACjC,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;iBACvB,CAAA;gBAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CACZ,6CAA6C,GAAG,IAAI,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAC/F,CAAA;gBACD,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,mCAAmC,CAAC,4BAA4B,CAAC,CAAA;gBAC9F,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,oDAAoD,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;gBACjG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,GAAG,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAA;gBAEjF,IAAI,QAAQ,CAAC,iBAAiB,EAAE,MAAM,EAAE;oBACpC,IAAI,CAAC,OAAO,CAAC,GAAG,CACZ,iDAAiD;wBAC7C,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAClE,CAAA;iBACJ;gBAED,IAAI,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;oBAClC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,MAAM,CAAC,CAAA;oBACnD,MAAK;iBACR;gBAED,MAAM,GAAG,QAAQ,CAAC,iBAAiB,CAAC,MAAO,CAAA;gBAC3C,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,IAAI,CAAC,CAAA;gBAC3B,KAAK,IAAI,EAAE,CAAA;gBACX,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,KAAK,CAAC,CAAA;gBAElD,IAAI,KAAK,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE;oBAC1B,MAAM,GAAG,WAAW,CAAA;oBACpB,MAAK;iBACR;aACJ;YAAC,OAAO,CAAM,EAAE;gBACb,MAAM,YAAY,GAAI,CAAW,CAAC,OAAO,IAAI,qCAAqC,CAAA;gBAClF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gDAAgD,GAAG,YAAY,CAAC,CAAA;gBACjF,MAAM,GAAG,QAAQ,CAAA;gBACjB,MAAK;aACR;SACJ;QACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,0CAA0C,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;QACvF,OAAO;YACH,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;SACxB,CAAA;IAC7B,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,iBAAyC,EAAE,QAAgB,EAAE,SAAiB;QAC5G,IAAI,MAAM,CAAA;QACV,IAAI;YACA,MAAM,GAAG,MAAM,iBAAiB,CAAC,mBAAmB,CAAC;gBACjD,QAAQ;gBACR,YAAY,EAAE,sCAAY,CAAC,cAAc;aAC5C,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,EAAE,CAAA;YACjB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAA;YAErD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE;gBAC3B,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAA;aAC1E;YAED,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,IAAI,EAAE;gBACnC,IAAI,KAAK,CAAC,kBAAkB,EAAE;oBAC1B,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAA;oBAC1C,IAAI,SAAS,CAAC,KAAK,EAAE;wBACjB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;qBAC/B;iBACJ;aACJ;YACD,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,CAAA;YAC1F,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,GAAG,qBAAqB,CAAC,CAAA;YACnE,OAAO;gBACH,UAAU,EAAE,qBAAqB;aACP,CAAA;SACjC;QAAC,OAAO,KAAK,EAAE;YACZ,MAAM,YAAY,GAAI,KAAe,CAAC,OAAO,IAAI,kCAAkC,CAAA;YACnF,OAAO;gBACH,KAAK,EAAE,YAAY;aACO,CAAA;SACjC;IACL,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,QAAgB,EAAE,MAAoB,EAAE,SAAiB;QAChF,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,0BAA0B,CAAC,CAAA;QACpE,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;QACnC,MAAM,EAAE,CAAC,aAAa,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;QAC5D,IAAI,qBAAqB,GAAG,EAAE,CAAA;QAC9B,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;QACnD,MAAM,GAAG,GAAG,IAAI,MAAM,CAAC,aAAa,CAAC,CAAA;QACrC,GAAG,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAA;QACvC,OAAO,qBAAqB,CAAA;IAChC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,aAAkB;QACpC,IAAI;YACA,MAAM,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,CAAA;SACrC;QAAC,OAAO,KAAK,EAAE;YACZ,qCAAqC;YACrC,MAAM,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;SACzD;IACL,CAAC;IAED,gBAAgB,CAAC,gBAAwB;QACrC,MAAM,UAAU,GAAG,IAAA,SAAM,GAAE,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,mBAAmB,EAAE,UAAU,CAAC,CAAA;QAClF,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE;YAC/B,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;SACnD;QACD,OAAO,aAAa,CAAA;IACxB,CAAC;CACJ;AA1UD,4CA0UC"}

package/out/language-server/netTransform/validation.d.ts

@@ -0,0 +1,9 @@
+import { StartTransformRequest } from './models';
+export declare function isProject(userInputrequest: StartTransformRequest): boolean;
+export declare function isSolution(userInputrequest: StartTransformRequest): boolean;
+export declare function validateProject(userInputrequest: StartTransformRequest): boolean;
+export declare function validateSolution(userInputrequest: StartTransformRequest): string[];
+export declare function checkForUnsupportedViews(userInputRequest: StartTransformRequest, isProject: boolean): Promise<boolean>;
+export declare function findFilesWithExtension(userInputRequest: StartTransformRequest, isProject: boolean, ext?: string): string[];
+export declare function readFile(filePath: string): Promise<string>;
+export declare function parseAndCheckUnsupportedComponents(htmlString: string): boolean;

package/out/language-server/netTransform/validation.js

@@ -0,0 +1,79 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseAndCheckUnsupportedComponents = exports.readFile = exports.findFilesWithExtension = exports.checkForUnsupportedViews = exports.validateSolution = exports.validateProject = exports.isSolution = exports.isProject = void 0;
+const fs = require("fs");
+const SupportedProjects_1 = require("./resources/SupportedProjects");
+function isProject(userInputrequest) {
+    return userInputrequest.SelectedProjectPath.endsWith('.csproj');
+}
+exports.isProject = isProject;
+function isSolution(userInputrequest) {
+    return userInputrequest.SelectedProjectPath.endsWith('.sln');
+}
+exports.isSolution = isSolution;
+function validateProject(userInputrequest) {
+    var selectedProject = userInputrequest.ProjectMetadata.find(project => project.ProjectPath == userInputrequest.SelectedProjectPath);
+    if (selectedProject)
+        return SupportedProjects_1.supportedProjects.includes(selectedProject.ProjectType);
+    return false;
+}
+exports.validateProject = validateProject;
+function validateSolution(userInputrequest) {
+    return userInputrequest.ProjectMetadata.filter(project => !SupportedProjects_1.supportedProjects.includes(project.ProjectType)).map(project => project.ProjectPath);
+}
+exports.validateSolution = validateSolution;
+async function checkForUnsupportedViews(userInputRequest, isProject) {
+    var containsUnsupportedComponents = false;
+    var cshtmlFiles = findFilesWithExtension(userInputRequest, isProject);
+    for (const file of cshtmlFiles) {
+        let htmlString = await readFile(file);
+        containsUnsupportedComponents = parseAndCheckUnsupportedComponents(htmlString);
+        if (containsUnsupportedComponents)
+            return containsUnsupportedComponents;
+    }
+    return containsUnsupportedComponents;
+}
+exports.checkForUnsupportedViews = checkForUnsupportedViews;
+function findFilesWithExtension(userInputRequest, isProject, ext = '.cshtml') {
+    var cshtmlFiles = [];
+    var projectMetaData;
+    if (isProject) {
+        let projectPath = userInputRequest.SelectedProjectPath;
+        projectMetaData = userInputRequest.ProjectMetadata.filter(projectMd => projectMd.ProjectPath == projectPath);
+    }
+    else {
+        projectMetaData = userInputRequest.ProjectMetadata;
+    }
+    projectMetaData.forEach(projectMd => {
+        projectMd.SourceCodeFilePaths.forEach(sourceCode => {
+            if (sourceCode.endsWith('.cshtml')) {
+                cshtmlFiles.push(sourceCode);
+            }
+        });
+    });
+    return cshtmlFiles;
+}
+exports.findFilesWithExtension = findFilesWithExtension;
+async function readFile(filePath) {
+    try {
+        let fileContents = await fs.promises.readFile(filePath, 'utf-8');
+        return fileContents;
+    }
+    catch (err) {
+        console.error(`Error reading/parsing file ${filePath}: ${err}`);
+        return '';
+    }
+}
+exports.readFile = readFile;
+function parseAndCheckUnsupportedComponents(htmlString) {
+    var containsUnsupportedComponents = false;
+    SupportedProjects_1.unsupportedViewComponents.forEach(component => {
+        if (htmlString.includes(component)) {
+            containsUnsupportedComponents = true;
+            return containsUnsupportedComponents;
+        }
+    });
+    return containsUnsupportedComponents;
+}
+exports.parseAndCheckUnsupportedComponents = parseAndCheckUnsupportedComponents;
+//# sourceMappingURL=validation.js.map

package/out/language-server/netTransform/validation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../../src/language-server/netTransform/validation.ts"],"names":[],"mappings":";;;AAAA,yBAAwB;AAExB,qEAA4F;AAE5F,SAAgB,SAAS,CAAC,gBAAuC;IAC7D,OAAO,gBAAgB,CAAC,mBAAmB,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAA;AACnE,CAAC;AAFD,8BAEC;AAED,SAAgB,UAAU,CAAC,gBAAuC;IAC9D,OAAO,gBAAgB,CAAC,mBAAmB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;AAChE,CAAC;AAFD,gCAEC;AAED,SAAgB,eAAe,CAAC,gBAAuC;IACnE,IAAI,eAAe,GAAG,gBAAgB,CAAC,eAAe,CAAC,IAAI,CACvD,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,WAAW,IAAI,gBAAgB,CAAC,mBAAmB,CACzE,CAAA;IACD,IAAI,eAAe;QAAE,OAAO,qCAAiB,CAAC,QAAQ,CAAC,eAAe,CAAC,WAAW,CAAC,CAAA;IACnF,OAAO,KAAK,CAAA;AAChB,CAAC;AAND,0CAMC;AAED,SAAgB,gBAAgB,CAAC,gBAAuC;IACpE,OAAO,gBAAgB,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,qCAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAC3G,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,WAAW,CACjC,CAAA;AACL,CAAC;AAJD,4CAIC;AAEM,KAAK,UAAU,wBAAwB,CAAC,gBAAuC,EAAE,SAAkB;IACtG,IAAI,6BAA6B,GAAY,KAAK,CAAA;IAClD,IAAI,WAAW,GAAa,sBAAsB,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAA;IAC/E,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE;QAC5B,IAAI,UAAU,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAA;QACrC,6BAA6B,GAAG,kCAAkC,CAAC,UAAU,CAAC,CAAA;QAC9E,IAAI,6BAA6B;YAAE,OAAO,6BAA6B,CAAA;KAC1E;IACD,OAAO,6BAA6B,CAAA;AACxC,CAAC;AATD,4DASC;AAED,SAAgB,sBAAsB,CAClC,gBAAuC,EACvC,SAAkB,EAClB,MAAc,SAAS;IAEvB,IAAI,WAAW,GAAa,EAAE,CAAA;IAC9B,IAAI,eAA2C,CAAA;IAC/C,IAAI,SAAS,EAAE;QACX,IAAI,WAAW,GAAG,gBAAgB,CAAC,mBAAmB,CAAA;QACtD,eAAe,GAAG,gBAAgB,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,WAAW,IAAI,WAAW,CAAC,CAAA;KAC/G;SAAM;QACH,eAAe,GAAG,gBAAgB,CAAC,eAAe,CAAA;KACrD;IACD,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;QAChC,SAAS,CAAC,mBAAmB,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;YAC/C,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;gBAChC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;aAC/B;QACL,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,OAAO,WAAW,CAAA;AACtB,CAAC;AArBD,wDAqBC;AAEM,KAAK,UAAU,QAAQ,CAAC,QAAgB;IAC3C,IAAI;QACA,IAAI,YAAY,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,YAAY,CAAA;KACtB;IAAC,OAAO,GAAG,EAAE;QACV,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,KAAK,GAAG,EAAE,CAAC,CAAA;QAC/D,OAAO,EAAE,CAAA;KACZ;AACL,CAAC;AARD,4BAQC;AAED,SAAgB,kCAAkC,CAAC,UAAkB;IACjE,IAAI,6BAA6B,GAAG,KAAK,CAAA;IAEzC,6CAAyB,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;QAC1C,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;YAChC,6BAA6B,GAAG,IAAI,CAAA;YACpC,OAAO,6BAA6B,CAAA;SACvC;IACL,CAAC,CAAC,CAAA;IACF,OAAO,6BAA6B,CAAA;AACxC,CAAC;AAVD,gFAUC"}

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

@@ -1,12 +1,11 @@
-import { Server, CredentialsProvider } from '@aws/language-server-runtimes/server-interface';
+import { CredentialsProvider, Server } from '@aws/language-server-runtimes/server-interface';
 import { CodeWhispererServiceToken } from './codeWhispererService';
+export declare const validStatesForGettingPlan: string[];
+export declare const validStatesForComplete: string[];
+export declare const failureStates: string[];
 /**
  *
  * @param createService Inject service instance based on credentials provider.
  * @returns  NetTransform server
  */
-export declare const NetTransformServerFactory: (createService: (credentialsProvider: CredentialsProvider) => CodeWhispererServiceToken) => Server;
-/**
- * Default NetTransformServer using Token authentication.
- */
-export declare const NetTransformServer: Server;
+export declare const QNetTransformServerToken: (service: (credentialsProvider: CredentialsProvider) => CodeWhispererServiceToken) => Server;

package/out/language-server/netTransformServer.js

@@ -1,27 +1,161 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.NetTransformServer = exports.NetTransformServerFactory = void 0;
-const codeWhispererService_1 = require("./codeWhispererService");
+exports.QNetTransformServerToken = exports.failureStates = exports.validStatesForComplete = exports.validStatesForGettingPlan = void 0;
+const codewhispererStreamingClient_1 = require("../client/streamingClient/codewhispererStreamingClient");
+const metrics_1 = require("./netTransform/metrics");
+const transformHandler_1 = require("./netTransform/transformHandler");
+exports.validStatesForGettingPlan = ['COMPLETED', 'PARTIALLY_COMPLETED', 'PLANNED', 'TRANSFORMING', 'TRANSFORMED'];
+exports.validStatesForComplete = ['COMPLETED'];
+exports.failureStates = ['FAILED', 'STOPPING', 'STOPPED', 'REJECTED'];
+const StartTransformCommand = 'aws/qNetTransform/startTransform';
+const GetTransformCommand = 'aws/qNetTransform/getTransform';
+const PollTransformCommand = 'aws/qNetTransform/pollTransform';
+const PollTransformForPlanCommand = 'aws/qNetTransform/pollTransformForPlan';
+const GetTransformPlanCommand = 'aws/qNetTransform/getTransformPlan';
+const CancelTransformCommand = 'aws/qNetTransform/cancelTransform';
+const DownloadArtifactsCommand = 'aws/qNetTransform/downloadArtifacts';
 /**
  *
  * @param createService Inject service instance based on credentials provider.
  * @returns  NetTransform server
  */
-const NetTransformServerFactory = createService => ({ credentialsProvider, lsp, workspace, telemetry, logging }) => {
-    const service = createService(credentialsProvider);
+const QNetTransformServerToken = (service) => ({ credentialsProvider, workspace, logging, lsp, telemetry }) => {
+    const codewhispererclient = service(credentialsProvider);
+    const transformHandler = new transformHandler_1.TransformHandler(codewhispererclient, workspace, logging);
+    const runTransformCommand = async (params, _token) => {
+        try {
+            switch (params.command) {
+                case StartTransformCommand: {
+                    const request = params;
+                    logging.log('Calling startTransformRequest: ' + request.SolutionRootPath);
+                    const response = await transformHandler.startTransformation(request);
+                    (0, metrics_1.emitTransformationJobStartedTelemetry)(telemetry, response);
+                    return response;
+                }
+                case GetTransformCommand: {
+                    const request = params;
+                    logging.log('Calling getTransform request with job Id: ' + request.TransformationJobId);
+                    const response = await transformHandler.getTransformation(request);
+                    (0, metrics_1.emitTransformationJobReceivedTelemetry)(telemetry, response);
+                    return response;
+                }
+                case PollTransformCommand: {
+                    const request = params;
+                    logging.log('Calling pollTransform request with job Id: ' + request.TransformationJobId);
+                    const response = await transformHandler.pollTransformation(request, exports.validStatesForComplete, exports.failureStates);
+                    logging.log('Transformation job for job Id' +
+                        request.TransformationJobId +
+                        ' is ' +
+                        JSON.stringify(response));
+                    (0, metrics_1.emitTransformationJobPolledTelemetry)(telemetry, response);
+                    return response;
+                }
+                case PollTransformForPlanCommand: {
+                    const request = params;
+                    logging.log('Calling pollTransformForPlan request with job Id: ' + request.TransformationJobId);
+                    const response = await transformHandler.pollTransformation(request, exports.validStatesForGettingPlan, exports.failureStates);
+                    logging.log('Transformation job for job Id' +
+                        request.TransformationJobId +
+                        ' is ' +
+                        JSON.stringify(response));
+                    (0, metrics_1.emitTransformationJobPolledForPlanTelemetry)(telemetry, response);
+                    return response;
+                }
+                case GetTransformPlanCommand: {
+                    const request = params;
+                    logging.log('Calling getTransformPlan request with job Id: ' + request.TransformationJobId);
+                    const response = await transformHandler.getTransformationPlan(request);
+                    logging.log('Transformation plan for job Id' +
+                        request.TransformationJobId +
+                        ' is ' +
+                        JSON.stringify(response));
+                    (0, metrics_1.emitTransformationPlanReceivedTelemetry)(telemetry, response, request.TransformationJobId);
+                    return response;
+                }
+                case CancelTransformCommand: {
+                    const request = params;
+                    logging.log('request job ID: ' + request.TransformationJobId);
+                    const response = await transformHandler.cancelTransformation(request);
+                    (0, metrics_1.emitTransformationJobCancelledTelemetry)(telemetry, response, request.TransformationJobId);
+                    return response;
+                }
+                case DownloadArtifactsCommand: {
+                    const request = params;
+                    const cwStreamingClientInstance = new codewhispererStreamingClient_1.StreamingClient();
+                    const cwStreamingClient = await cwStreamingClientInstance.getStreamingClient(credentialsProvider);
+                    logging.log('Calling Download Archive  with job Id: ' + request.TransformationJobId);
+                    const response = await transformHandler.downloadExportResultArchive(cwStreamingClient, request.TransformationJobId, request.SolutionRootPath);
+                    (0, metrics_1.emitTransformationJobArtifactsDownloadedTelemetry)(telemetry, response, request.TransformationJobId);
+                    return response;
+                }
+            }
+            return;
+        }
+        catch (e) {
+            logging.log('Server side error while executing transformer Command ' + e);
+            switch (params.command) {
+                case StartTransformCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationJobStartedFailure)(telemetry, request, e);
+                    break;
+                }
+                case GetTransformCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationJobReceivedFailure)(telemetry, request, e);
+                    break;
+                }
+                case PollTransformCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationJobPolledFailure)(telemetry, request, e);
+                    break;
+                }
+                case PollTransformForPlanCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationJobPolledForPlanFailure)(telemetry, request, e);
+                    break;
+                }
+                case GetTransformPlanCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationPlanReceivedFailure)(telemetry, request, e);
+                    break;
+                }
+                case CancelTransformCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationJobCancelledFailure)(telemetry, request, e);
+                    break;
+                }
+                case DownloadArtifactsCommand: {
+                    const request = params;
+                    (0, metrics_1.emitTransformationJobArtifactsDownloadedFailure)(telemetry, request, e);
+                    break;
+                }
+            }
+        }
+    };
     const onExecuteCommandHandler = async (params, _token) => {
-        //Placeholder for logic
+        logging.log(params.command);
+        return runTransformCommand(params, _token);
     };
-    // Do the thing
-    lsp.onExecuteCommand(onExecuteCommandHandler);
-    // Disposable
-    return () => {
-        // Do nothing
+    const onInitializeHandler = () => {
+        return {
+            capabilities: {
+                executeCommandProvider: {
+                    commands: [
+                        StartTransformCommand,
+                        GetTransformCommand,
+                        PollTransformCommand,
+                        PollTransformForPlanCommand,
+                        GetTransformPlanCommand,
+                        CancelTransformCommand,
+                        DownloadArtifactsCommand,
+                    ],
+                },
+            },
+        };
     };
+    lsp.addInitializer(onInitializeHandler);
+    lsp.onExecuteCommand(onExecuteCommandHandler);
+    return () => { };
 };
-exports.NetTransformServerFactory = NetTransformServerFactory;
-/**
- * Default NetTransformServer using Token authentication.
- */
-exports.NetTransformServer = (0, exports.NetTransformServerFactory)(credentialsProvider => new codeWhispererService_1.CodeWhispererServiceToken(credentialsProvider, {}));
+exports.QNetTransformServerToken = QNetTransformServerToken;
 //# sourceMappingURL=netTransformServer.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"netTransformServer.js","sourceRoot":"","sources":["../../src/language-server/netTransformServer.ts"],"names":[],"mappings":";;;AAMA,iEAAkE;AAElE;;;;GAIG;AACI,MAAM,yBAAyB,GAGlC,aAAa,CAAC,EAAE,CAChB,CAAC,EAAE,mBAAmB,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE;IAC5D,MAAM,OAAO,GAAG,aAAa,CAAC,mBAAmB,CAAC,CAAA;IAClD,MAAM,uBAAuB,GAAG,KAAK,EACjC,MAA4B,EAC5B,MAAyB,EACb,EAAE;QACd,uBAAuB;IAC3B,CAAC,CAAA;IAED,eAAe;IACf,GAAG,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAA;IAE7C,aAAa;IACb,OAAO,GAAG,EAAE;QACR,aAAa;IACjB,CAAC,CAAA;AACL,CAAC,CAAA;AApBQ,QAAA,yBAAyB,6BAoBjC;AAEL;;GAEG;AACU,QAAA,kBAAkB,GAAG,IAAA,iCAAyB,EACvD,mBAAmB,CAAC,EAAE,CAAC,IAAI,gDAAyB,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAChF,CAAA"}
+{"version":3,"file":"netTransformServer.js","sourceRoot":"","sources":["../../src/language-server/netTransformServer.ts"],"names":[],"mappings":";;;AAMA,yGAAwF;AAExF,oDAe+B;AAQ/B,sEAAkE;AAErD,QAAA,yBAAyB,GAAG,CAAC,WAAW,EAAE,qBAAqB,EAAE,SAAS,EAAE,cAAc,EAAE,aAAa,CAAC,CAAA;AAC1G,QAAA,sBAAsB,GAAG,CAAC,WAAW,CAAC,CAAA;AACtC,QAAA,aAAa,GAAG,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,CAAC,CAAA;AAC1E,MAAM,qBAAqB,GAAG,kCAAkC,CAAA;AAChE,MAAM,mBAAmB,GAAG,gCAAgC,CAAA;AAC5D,MAAM,oBAAoB,GAAG,iCAAiC,CAAA;AAC9D,MAAM,2BAA2B,GAAG,wCAAwC,CAAA;AAC5E,MAAM,uBAAuB,GAAG,oCAAoC,CAAA;AACpE,MAAM,sBAAsB,GAAG,mCAAmC,CAAA;AAClE,MAAM,wBAAwB,GAAG,qCAAqC,CAAA;AACtE;;;;GAIG;AACI,MAAM,wBAAwB,GACjC,CAAC,OAAgF,EAAU,EAAE,CAC7F,CAAC,EAAE,mBAAmB,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE;IAC5D,MAAM,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;IACxD,MAAM,gBAAgB,GAAG,IAAI,mCAAgB,CAAC,mBAAmB,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;IACtF,MAAM,mBAAmB,GAAG,KAAK,EAAE,MAA4B,EAAE,MAAyB,EAAE,EAAE;QAC1F,IAAI;YACA,QAAQ,MAAM,CAAC,OAAO,EAAE;gBACpB,KAAK,qBAAqB,CAAC,CAAC;oBACxB,MAAM,OAAO,GAAG,MAA+B,CAAA;oBAC/C,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAA;oBACzE,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAA;oBACpE,IAAA,+CAAqC,EAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;oBAC1D,OAAO,QAAQ,CAAA;iBAClB;gBACD,KAAK,mBAAmB,CAAC,CAAC;oBACtB,MAAM,OAAO,GAAG,MAA6B,CAAA;oBAC7C,OAAO,CAAC,GAAG,CAAC,4CAA4C,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACvF,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAA;oBAClE,IAAA,gDAAsC,EAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;oBAC3D,OAAO,QAAQ,CAAA;iBAClB;gBACD,KAAK,oBAAoB,CAAC,CAAC;oBACvB,MAAM,OAAO,GAAG,MAA6B,CAAA;oBAC7C,OAAO,CAAC,GAAG,CAAC,6CAA6C,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACxF,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,kBAAkB,CACtD,OAAO,EACP,8BAAsB,EACtB,qBAAa,CAChB,CAAA;oBACD,OAAO,CAAC,GAAG,CACP,+BAA+B;wBAC3B,OAAO,CAAC,mBAAmB;wBAC3B,MAAM;wBACN,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAC/B,CAAA;oBACD,IAAA,8CAAoC,EAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;oBACzD,OAAO,QAAQ,CAAA;iBAClB;gBACD,KAAK,2BAA2B,CAAC,CAAC;oBAC9B,MAAM,OAAO,GAAG,MAA6B,CAAA;oBAC7C,OAAO,CAAC,GAAG,CAAC,oDAAoD,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBAC/F,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,kBAAkB,CACtD,OAAO,EACP,iCAAyB,EACzB,qBAAa,CAChB,CAAA;oBACD,OAAO,CAAC,GAAG,CACP,+BAA+B;wBAC3B,OAAO,CAAC,mBAAmB;wBAC3B,MAAM;wBACN,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAC/B,CAAA;oBACD,IAAA,qDAA2C,EAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;oBAChE,OAAO,QAAQ,CAAA;iBAClB;gBACD,KAAK,uBAAuB,CAAC,CAAC;oBAC1B,MAAM,OAAO,GAAG,MAAiC,CAAA;oBACjD,OAAO,CAAC,GAAG,CAAC,gDAAgD,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBAC3F,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAA;oBACtE,OAAO,CAAC,GAAG,CACP,gCAAgC;wBAC5B,OAAO,CAAC,mBAAmB;wBAC3B,MAAM;wBACN,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAC/B,CAAA;oBACD,IAAA,iDAAuC,EAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACzF,OAAO,QAAQ,CAAA;iBAClB;gBACD,KAAK,sBAAsB,CAAC,CAAC;oBACzB,MAAM,OAAO,GAAG,MAAgC,CAAA;oBAChD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBAC7D,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;oBACrE,IAAA,iDAAuC,EAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACzF,OAAO,QAAQ,CAAA;iBAClB;gBACD,KAAK,wBAAwB,CAAC,CAAC;oBAC3B,MAAM,OAAO,GAAG,MAAkC,CAAA;oBAClD,MAAM,yBAAyB,GAAG,IAAI,8CAAe,EAAE,CAAA;oBACvD,MAAM,iBAAiB,GACnB,MAAM,yBAAyB,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAA;oBAC3E,OAAO,CAAC,GAAG,CAAC,yCAAyC,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACpF,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,2BAA2B,CAC/D,iBAAiB,EACjB,OAAO,CAAC,mBAAmB,EAC3B,OAAO,CAAC,gBAAgB,CAC3B,CAAA;oBACD,IAAA,2DAAiD,EAC7C,SAAS,EACT,QAAQ,EACR,OAAO,CAAC,mBAAmB,CAC9B,CAAA;oBACD,OAAO,QAAQ,CAAA;iBAClB;aACJ;YACD,OAAM;SACT;QAAC,OAAO,CAAM,EAAE;YACb,OAAO,CAAC,GAAG,CAAC,wDAAwD,GAAG,CAAC,CAAC,CAAA;YAEzE,QAAQ,MAAM,CAAC,OAAO,EAAE;gBACpB,KAAK,qBAAqB,CAAC,CAAC;oBACxB,MAAM,OAAO,GAAG,MAA+B,CAAA;oBAC/C,IAAA,6CAAmC,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBAC1D,MAAK;iBACR;gBACD,KAAK,mBAAmB,CAAC,CAAC;oBACtB,MAAM,OAAO,GAAG,MAA6B,CAAA;oBAC7C,IAAA,8CAAoC,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBAC3D,MAAK;iBACR;gBACD,KAAK,oBAAoB,CAAC,CAAC;oBACvB,MAAM,OAAO,GAAG,MAA6B,CAAA;oBAC7C,IAAA,4CAAkC,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBACzD,MAAK;iBACR;gBACD,KAAK,2BAA2B,CAAC,CAAC;oBAC9B,MAAM,OAAO,GAAG,MAA6B,CAAA;oBAC7C,IAAA,mDAAyC,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBAChE,MAAK;iBACR;gBACD,KAAK,uBAAuB,CAAC,CAAC;oBAC1B,MAAM,OAAO,GAAG,MAAiC,CAAA;oBACjD,IAAA,+CAAqC,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBAC5D,MAAK;iBACR;gBACD,KAAK,sBAAsB,CAAC,CAAC;oBACzB,MAAM,OAAO,GAAG,MAAgC,CAAA;oBAChD,IAAA,+CAAqC,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBAC5D,MAAK;iBACR;gBACD,KAAK,wBAAwB,CAAC,CAAC;oBAC3B,MAAM,OAAO,GAAG,MAAkC,CAAA;oBAClD,IAAA,yDAA+C,EAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;oBACtE,MAAK;iBACR;aACJ;SACJ;IACL,CAAC,CAAA;IAED,MAAM,uBAAuB,GAAG,KAAK,EACjC,MAA4B,EAC5B,MAAyB,EACb,EAAE;QACd,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAC3B,OAAO,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC9C,CAAC,CAAA;IACD,MAAM,mBAAmB,GAAG,GAAG,EAAE;QAC7B,OAAO;YACH,YAAY,EAAE;gBACV,sBAAsB,EAAE;oBACpB,QAAQ,EAAE;wBACN,qBAAqB;wBACrB,mBAAmB;wBACnB,oBAAoB;wBACpB,2BAA2B;wBAC3B,uBAAuB;wBACvB,sBAAsB;wBACtB,wBAAwB;qBAC3B;iBACJ;aACJ;SACJ,CAAA;IACL,CAAC,CAAA;IACD,GAAG,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAA;IACvC,GAAG,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAA;IAE7C,OAAO,GAAG,EAAE,GAAE,CAAC,CAAA;AACnB,CAAC,CAAA;AAvKQ,QAAA,wBAAwB,4BAuKhC"}

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

@@ -1,4 +1,5 @@
 export declare const CodeWhispererServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
 export declare const CodeWhispererServerIAMProxy: import("@aws/language-server-runtimes/server-interface").Server;
 export declare const CodeWhispererSecurityScanServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
+export declare const QNetTransformServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
 export declare const QChatServerProxy: import("@aws/language-server-runtimes/server-interface").Server;