kata-cli/components/bots/deployment.ts

Kata AI Command Line Tools

import { IHelper } from "interfaces/main";
import { JsonObject } from "merapi";
const Table = require("cli-table");
import inquirer = require("inquirer");

export default class Deployment {
    constructor(private helper: IHelper, private api: any) {
    }

    public async create(versionType?: string) {
        const projectId = this.helper.getProjectId();
        const { response: { body: project } } = await this.helper.toPromise(
            this.api.projectApi, this.api.projectApi.projectsProjectIdGet,
            projectId
        );
        // TODO: {page:1, limit:1}
        let botRevision: string;
        let nluRevision: string;
        let cmsRevision: string;

        try {
            const { response: { body: data } } = await this.helper.toPromise(
                this.api.projectApi,
                this.api.projectApi.projectsProjectIdBotGet, projectId
            );

            if (data.revision) {
                botRevision = data.revision;
            }
        } catch (e) {
            console.error("Error");
            console.log(this.helper.wrapError(e));
        }

        try {
            const { response: {body: data} } = await this.helper.toPromise(this.api.projectApi,
                this.api.projectApi.projectsProjectIdNluRevisionsGet, projectId);

          if (data.data[0] && data.data[0].revision) {
              nluRevision = data.data[0].revision;
          }
        } catch (e) {
            console.error("Error");
            console.log(this.helper.wrapError(e));
        }

        try {
            const { response: {body: data} } = await this.helper.toPromise(this.api.projectApi, 
                this.api.projectApi.projectsProjectIdCmsGet, projectId);
            if (data.revision) {
                cmsRevision = data.revision;
            } 
        } catch (e) {
            console.error("Error");
            console.log(this.helper.wrapError(e));
        }

        let targetVersion;

        try {
            // get previous deployment version
            const { response: { body: latestDeployment } } = await this.helper.toPromise(this.api.projectApi,
                this.api.projectApi.projectsProjectIdDeploymentGet, projectId);
            const prevVersion = latestDeployment.version;
            let [major, minor, patch] = prevVersion.split(".");

            if (versionType === "major") {
                major = Number(major) + 1;
                minor = 0;
                patch = 0;
            } else if (versionType === "minor") {
                minor = Number(minor) + 1;
                patch = 0;
            } else {
                patch = Number(patch) + 1;
            }
            targetVersion = `${major}.${minor}.${patch}`;
            
        } catch (e) {
            targetVersion = "0.0.1";
        }

        try {
            const postBody = {
                version: targetVersion,
                botRevision,
                nluRevision,
                cmsRevision,
                modules: (null as any),
            };

            const { response: { body } } = await this.helper.toPromise(
                this.api.deploymentApi, this.api.deploymentApi.projectsProjectIdDeploymentVersionsPost,
                postBody, projectId,
            );


            console.log(`Bot Revision: ${botRevision.substring(0, 7)}`);
            console.log(`NLU Revision: ${nluRevision.substring(0, 7)}`);
            console.log(`CMS Revision: ${cmsRevision.substring(0, 7)}`);
            console.log(`Succesfully create Deployment to version ${targetVersion}`);
        } catch (e) {
            console.error("Error");
            console.log(this.helper.wrapError(e));
        }
    }

    public async list(options: JsonObject) {
        const projectId = this.helper.getProjectId();

        try {
            const { response: { body } } = await this.helper.toPromise(
                this.api.deploymentApi, this.api.deploymentApi.projectsProjectIdDeploymentVersionsGet, projectId, {}
            );

            if (body && body.data) {
                const table = new Table({
                    head: ["Deployment Name", "Deployment Version", "Bot Revision"],
                    colWidths: [30, 30, 42]
                });
                body.data.forEach((deployment: JsonObject) => {
                    table.push([deployment.name, deployment.version, deployment.botRevision]);
                });

                console.log(table.toString());
            }
        } catch (e) {
            console.error("Error");
            console.log(this.helper.wrapError(e));
        }
    }

    public async rollback(version: string) {
        try {
            const projectId = this.helper.getProjectId();
            const author = this.helper.getProp("current_login");

            const { response: { body } } = await this.helper.toPromise(
                this.api.deploymentApi, this.api.deploymentApi.projectsProjectIdDeploymentVersionsGet, projectId, { limit: 1000000, page: 1}
            );

            if (body && body.data) {
                const deployments: object[] = body.data;
                const versionData: any = deployments.find((deployment: any) => deployment.version === version);
                if (!versionData) {
                    throw new Error("Version not found");
                }

                let { changelog, confirm } = await inquirer.prompt<any>([
                    {
                        type: "text",
                        name: "changelog",
                        message: "Changelog:",
                    },
                    {
                        type: "confirm",
                        name: "confirm",
                        message: `IMPORTANT: Existing NL training data will also be rolled back to version ${versionData.version}`,
                        default: true,
                    }
                ]);

                const requestBody = {
                    author,
                    changelog,
                    version: versionData.version
                }

                if (confirm) {
                    const { response } = await this.helper.toPromise(
                        this.api.deploymentApi, this.api.deploymentApi.deploymentsDeploymentIdRollbackPost, versionData.id, requestBody
                    );
    
                    if (response && response.body) {
                        console.log(`Successfully rolled back to version ${versionData.version}`)
                    } else {
                        console.log(`Error when trying to rollback to version ${versionData.version}`)
                    }
                }
            }
        } catch (e) {
            console.log(this.helper.wrapError(e));
        }
    }
}