@cto.ai/ops/lib/commands/init.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const tslib_1 = require("tslib");
const sdk_1 = require("@cto.ai/sdk");
const fs = tslib_1.__importStar(require("fs-extra"));
const path = tslib_1.__importStar(require("path"));
const yaml = tslib_1.__importStar(require("yaml"));
const base_1 = tslib_1.__importStar(require("../base"));
const asyncPipe_1 = require("../utils/asyncPipe");
const CustomErrors_1 = require("../errors/CustomErrors");
const opConfig_1 = require("../constants/opConfig");
const utils_1 = require("../utils");
class Init extends base_1.default {
    constructor() {
        super(...arguments);
        this.questions = [];
        this.srcDir = path.resolve(__dirname, '../templates/');
        this.destDir = path.resolve(process.cwd());
        this.initPrompts = {
            [utils_1.appendSuffix(opConfig_1.COMMAND, 'Name')]: {
                type: 'input',
                name: utils_1.appendSuffix(opConfig_1.COMMAND, 'Name'),
                message: `\n Provide a name for your new command ${sdk_1.ux.colors.reset.green('→')}\n${sdk_1.ux.colors.reset(sdk_1.ux.colors.secondary('Names must be lowercase'))}\n\n🏷  ${sdk_1.ux.colors.white('Name:')}`,
                afterMessage: sdk_1.ux.colors.reset.green('✓'),
                afterMessageAppend: sdk_1.ux.colors.reset(' added!'),
                validate: this._validateName,
                transformer: input => sdk_1.ux.colors.cyan(input.toLocaleLowerCase()),
                filter: input => input.toLowerCase(),
            },
            [utils_1.appendSuffix(opConfig_1.COMMAND, 'Description')]: {
                type: 'input',
                name: utils_1.appendSuffix(opConfig_1.COMMAND, 'Description'),
                message: `\nProvide a description ${sdk_1.ux.colors.reset.green('→')}  \n📝 ${sdk_1.ux.colors.white('Description:')}`,
                afterMessage: sdk_1.ux.colors.reset.green('✓'),
                afterMessageAppend: sdk_1.ux.colors.reset(' added!'),
                validate: this._validateDescription,
            },
            [utils_1.appendSuffix(opConfig_1.WORKFLOW, 'Name')]: {
                type: 'input',
                name: utils_1.appendSuffix(opConfig_1.WORKFLOW, 'Name'),
                message: `\n Provide a name for your new workflow ${sdk_1.ux.colors.reset.green('→')}\n${sdk_1.ux.colors.reset(sdk_1.ux.colors.secondary('Names must be lowercase'))}\n\n🏷  ${sdk_1.ux.colors.white('Name:')}`,
                afterMessage: sdk_1.ux.colors.reset.green('✓'),
                afterMessageAppend: sdk_1.ux.colors.reset(' added!'),
                validate: this._validateName,
                transformer: input => sdk_1.ux.colors.cyan(input.toLocaleLowerCase()),
                filter: input => input.toLowerCase(),
            },
            [utils_1.appendSuffix(opConfig_1.WORKFLOW, 'Description')]: {
                type: 'input',
                name: utils_1.appendSuffix(opConfig_1.WORKFLOW, 'Description'),
                message: `\nProvide a description ${sdk_1.ux.colors.reset.green('→')}\n\n📝 ${sdk_1.ux.colors.white('Description:')}`,
                afterMessage: sdk_1.ux.colors.reset.green('✓'),
                afterMessageAppend: sdk_1.ux.colors.reset(' added!'),
                validate: this._validateDescription,
            },
        };
        this.determineTemplate = async (prompts) => {
            const { templates } = await sdk_1.ux.prompt({
                type: 'checkbox',
                name: 'templates',
                message: `\n What type of op would you like to create ${sdk_1.ux.colors.reset.green('→')}`,
                choices: [
                    {
                        name: `${utils_1.titleCase(opConfig_1.COMMAND)} - A template for building commands which can be distributed via The Ops Platform.`,
                        value: opConfig_1.COMMAND,
                    },
                    {
                        name: `${utils_1.titleCase(opConfig_1.WORKFLOW)} - A template for combining many commands into a workflow which can be distributed via The Ops Platform.`,
                        value: opConfig_1.WORKFLOW,
                    },
                ],
                afterMessage: `${sdk_1.ux.colors.reset.green('✓')}`,
                validate: input => input.length != 0,
            });
            return { prompts, templates };
        };
        this.determineQuestions = ({ prompts, templates, }) => {
            // Filters initPrompts based on the templates selected in determineTemplate
            const removeIfNotSelectedTemplate = ([key, _val]) => {
                return key.includes(templates[0]) || key.includes(templates[1]);
            };
            const questions = Object.entries(prompts)
                .filter(removeIfNotSelectedTemplate)
                .map(([_key, question]) => question);
            return { questions, templates };
        };
        this.askQuestions = async ({ questions, templates, }) => {
            const answers = await sdk_1.ux.prompt(questions);
            return { answers, templates };
        };
        this.determineInitPaths = ({ answers, templates, }) => {
            const initParams = Object.assign({}, answers, { templates });
            const { name } = this.getNameAndDescription(initParams);
            const sharedDir = `${this.srcDir}/shared`;
            const destDir = `${this.destDir}/${name}`;
            const initPaths = { sharedDir, destDir };
            return { initPaths, initParams };
        };
        this.copyTemplateFiles = async ({ initPaths, initParams, }) => {
            try {
                const { templates } = initParams;
                const { destDir, sharedDir } = initPaths;
                await fs.ensureDir(destDir);
                // copies op files if selected
                if (templates.includes(opConfig_1.COMMAND)) {
                    await fs.copy(`${this.srcDir}/${opConfig_1.COMMAND}`, destDir);
                }
                // copies shared files
                await fs.copy(sharedDir, destDir);
                return { initPaths, initParams };
            }
            catch (err) {
                this.debug('%O', err);
                throw new CustomErrors_1.CopyTemplateFilesError(err);
            }
        };
        this.customizePackageJson = async ({ initPaths, initParams, }) => {
            try {
                const { destDir, sharedDir } = initPaths;
                const { name, description } = this.getNameAndDescription(initParams);
                const packageObj = JSON.parse(fs.readFileSync(`${sharedDir}/package.json`, 'utf8'));
                packageObj.name = name;
                packageObj.description = description;
                const newPackageString = JSON.stringify(packageObj, null, 2);
                fs.writeFileSync(`${destDir}/package.json`, newPackageString);
                return { initPaths, initParams };
            }
            catch (err) {
                this.debug('%O', err);
                throw new CustomErrors_1.CouldNotInitializeOp(err);
            }
        };
        this.customizeYaml = async ({ initPaths, initParams, }) => {
            try {
                const { destDir } = initPaths;
                // Parse YAML as document so we can work with comments
                const opsYamlDoc = yaml.parseDocument(fs.readFileSync(`${destDir}/ops.yml`, 'utf8'));
                await this.customizeOpsYaml(initParams, opsYamlDoc);
                await this.customizeWorkflowYaml(initParams, opsYamlDoc);
                // Process each root level section of the YAML file & add comments
                Object.keys(opConfig_1.HELP_COMMENTS).forEach(rootKey => {
                    this.addHelpCommentsFor(rootKey, opsYamlDoc);
                });
                // Get the YAML file as string
                const newOpsString = opsYamlDoc.toString();
                fs.writeFileSync(`${destDir}/ops.yml`, newOpsString);
                return { initPaths, initParams };
            }
            catch (err) {
                this.debug('%O', err);
                throw new CustomErrors_1.CouldNotInitializeOp(err);
            }
        };
        // The `yaml` library has a pretty bad API for handling comments
        // More: https://eemeli.org/yaml/#comments'
        // TODO: Review type checking for yamlDoc (yaml.ast.Document) & remove tsignores
        this.addHelpCommentsFor = (key, yamlDoc) => {
            const docContents = yamlDoc.contents;
            const docContentsItems = docContents.items;
            const configItem = docContentsItems.find(item => {
                if (!item || !item.key)
                    return;
                const itemKey = item.key;
                return itemKey.value === key;
            });
            // Simple config fields (`version`)
            if (configItem &&
                configItem.value &&
                configItem.value.type === opConfig_1.YAML_TYPE_STRING &&
                opConfig_1.HELP_COMMENTS[key]) {
                configItem.comment = ` ${opConfig_1.HELP_COMMENTS[key]}`;
            }
            // Config fields with nested values (`ops`, `workflows`)
            if (configItem &&
                configItem.value &&
                configItem.value.type === opConfig_1.YAML_TYPE_SEQUENCE) {
                // @ts-ignore
                yamlDoc.getIn([key, 0]).items.map(configItem => {
                    const comment = opConfig_1.HELP_COMMENTS[key][configItem.key];
                    if (comment)
                        configItem.comment = ` ${opConfig_1.HELP_COMMENTS[key][configItem.key]}`;
                });
            }
        };
        this.customizeOpsYaml = async (initParams, yamlDoc) => {
            const { templates, commandName, commandDescription } = initParams;
            if (!templates.includes(opConfig_1.COMMAND)) {
                // @ts-ignore
                yamlDoc.delete('ops');
                return;
            }
            // @ts-ignore
            yamlDoc.getIn(['ops', 0]).set('name', commandName);
            // @ts-ignore
            yamlDoc.getIn(['ops', 0]).set('description', commandDescription);
        };
        this.customizeWorkflowYaml = async (initParams, yamlDoc) => {
            const { templates, workflowName, workflowDescription } = initParams;
            if (!templates.includes(opConfig_1.WORKFLOW)) {
                // @ts-ignore
                yamlDoc.delete('workflows');
                return;
            }
            // @ts-ignore
            yamlDoc.getIn(['workflows', 0]).set('name', workflowName);
            // @ts-ignore
            yamlDoc.getIn(['workflows', 0]).set('description', workflowDescription);
        };
        this.logMessages = async ({ initPaths, initParams, }) => {
            const { destDir } = initPaths;
            const { templates } = initParams;
            const { name } = this.getNameAndDescription(initParams);
            this.logSuccessMessage(templates);
            fs.readdirSync(`${destDir}`).forEach((file) => {
                let callout = '';
                if (file.indexOf('index.js') > -1) {
                    callout = `${sdk_1.ux.colors.green('←')} ${sdk_1.ux.colors.white('Start developing here!')}`;
                }
                let msg = sdk_1.ux.colors.italic(`${path.relative(this.destDir, process.cwd())}/${name}/${file} ${callout}`);
                this.log(`📁 .${msg}`);
            });
            if (templates.includes(opConfig_1.COMMAND)) {
                this.logCommandMessage(initParams);
            }
            if (templates.includes(opConfig_1.WORKFLOW)) {
                this.logWorkflowMessage(initParams);
            }
            return { initPaths, initParams };
        };
        this.logCommandMessage = (initParams) => {
            const { commandName } = initParams;
            this.log(`\n🚀 To test your ${opConfig_1.COMMAND} run: ${sdk_1.ux.colors.green('$')} ${sdk_1.ux.colors.callOutCyan(`ops run ${commandName}`)}`);
        };
        this.logWorkflowMessage = (initParams) => {
            const { workflowName } = initParams;
            const { name } = this.getNameAndDescription(initParams);
            this.log(`\n🚀 To test your ${opConfig_1.WORKFLOW} run: ${sdk_1.ux.colors.green('$')} ${sdk_1.ux.colors.callOutCyan(`cd ${name} && npm install && ops run ${workflowName}`)}`);
        };
        this.logSuccessMessage = (templates) => {
            const successMessageBoth = `\n🎉 Success! Your ${opConfig_1.COMMAND} and ${opConfig_1.WORKFLOW} template Ops are ready to start coding... \n`;
            const getSuccessMessage = (opType) => `\n🎉 Success! Your ${opType} template Op is ready to start coding... \n`;
            if (templates.includes(opConfig_1.COMMAND) && templates.includes(opConfig_1.WORKFLOW)) {
                return this.log(successMessageBoth);
            }
            const opType = templates.includes(opConfig_1.COMMAND) ? opConfig_1.COMMAND : opConfig_1.WORKFLOW;
            return this.log(getSuccessMessage(opType));
        };
        this.sendAnalytics = async ({ initPaths, initParams, }) => {
            try {
                const { destDir } = initPaths;
                const { templates } = initParams;
                const { name, description } = this.getNameAndDescription(initParams);
                this.services.analytics.track({
                    userId: this.user.email,
                    teamId: this.team.id,
                    event: 'Ops CLI Init',
                    properties: {
                        email: this.user.email,
                        username: this.user.username,
                        path: destDir,
                        name,
                        description,
                        templates,
                    },
                }, this.accessToken);
            }
            catch (err) {
                this.debug('%O', err);
                throw new CustomErrors_1.AnalyticsError(err);
            }
        };
        this.getNameAndDescription = (initParams) => {
            return {
                name: initParams.commandName || initParams.workflowName,
                description: initParams.commandDescription || initParams.workflowDescription,
            };
        };
    }
    _validateName(input) {
        if (input === '')
            return 'You need name your op before you can continue';
        if (!input.match('^[a-z0-9_-]*$')) {
            return 'Sorry, please name the Op using only numbers, letters, -, or _';
        }
        return true;
    }
    _validateDescription(input) {
        if (input === '')
            return 'You need to provide a description of your op before continuing';
        return true;
    }
    async run() {
        this.parse(Init);
        try {
            await this.isLoggedIn();
            const initPipeline = asyncPipe_1.asyncPipe(this.determineTemplate, this.determineQuestions, this.askQuestions, this.determineInitPaths, this.copyTemplateFiles, this.customizePackageJson, this.customizeYaml, this.logMessages, this.sendAnalytics);
            await initPipeline(this.initPrompts);
        }
        catch (err) {
            this.debug('%O', err);
            this.config.runHook('error', { err, accessToken: this.accessToken });
        }
    }
}
Init.description = 'Easily create a new op.';
Init.flags = {
    help: base_1.flags.help({ char: 'h' }),
};
exports.default = Init;