entrypoint.js

#!/usr/bin/env node
'use strict';

const fs = require('fs');
const path = require('path');
const url = require('url');
const yaml = require('yaml');
const fetch = require('node-fetch');
const logger = require('./logger');
const mkdirp = require('mkdirp');

const processor = require('./processor.js');

var argv = require('yargs')
    .usage('simplify [generate]|template [options]')
    .describe('openapi', 'OpenAPI 3.0 spec YAML')
    .string('openapi')
    .alias('i', 'openapi')
    .describe('openapi', 'contains openapi.yaml specs files')
    .string('output')
    .alias('o', 'output')
    .describe('output', 'output directory')
    .default('output', '.')
    .string('project')
    .alias('p', 'project')
    .describe('project', 'project Id')
    .string('account')
    .alias('a', 'account')
    .describe('account', 'account Id')
    .boolean('verbose')
    .default('verbose', false)
    .describe('verbose', 'Increase verbosity')
    .alias('v', 'verbose')
    .boolean('merge')
    .default('merge', false)
    .describe('merge', 'Auto merge files')
    .boolean('diff')
    .default('diff', false)
    .describe('diff', 'Generate diff file')
    .string('ignores')
    .default('ignores', false)
    .alias('n', 'ignores')
    .describe('ignores', 'eg: keep-your-code.js;keep-your-data.json')
    .demandOption(['o'])
    .demandCommand(0)
    .argv;

if (argv._[0] === 'generate') {
    console.log(` - The command '${argv._[0]}' is now supported by default. You can ignore it for your short hand.`)
}
const templatePath = require("simplify-templates")
let configFile = path.join(path.join(templatePath), 'config-openapi.json');
let config = yaml.parse(fs.readFileSync(configFile, 'utf8'), { prettyErrors: true });
let defName = path.resolve(path.join(argv.openapi || 'openapi.yaml'));
const sampleName = path.join(__dirname, 'templates', (argv.input || 'petsample') + '.yaml')
const outputYAML = path.resolve(argv.output, 'openapi.yaml')
mkdirp(path.resolve(argv.output)).then(function () {
    if (argv._.length && argv._[0] === 'template') {
        console.log("╓───────────────────────────────────────────────────────────────╖")
        console.log("║               Simplify Framework  - OpenAPI                   ║")
        console.log("╙───────────────────────────────────────────────────────────────╜")
        console.log(` - Sample definition ${outputYAML}`);
        fs.writeFileSync(outputYAML, fs.readFileSync(sampleName, 'utf8'), 'utf8')
        process.exit(0)
    } else {
        console.log("╓───────────────────────────────────────────────────────────────╖")
        console.log("║               Simplify Framework  - OpenAPI                   ║")
        console.log("╙───────────────────────────────────────────────────────────────╜")
        console.log(` - OpenAPI definition ${defName}`);
        runCommandLine()
    }
}, function (err) {
    console.error(`${err}`)
})

function mergeArrays(arrObj, moreArrObj) {
    if (!moreArrObj) return arrObj
    moreArrObj.forEach(function (tmp, i) {
        if (typeof tmp === 'object') {
            if (Array.isArray(tmp)) {
                arrObj[i] = arrObj[i] || []
                arrObj[i] = mergeArrays(arrObj[i], tmp)
            } else {
                arrObj[i] = arrObj[i] || {}
                arrObj[i] = mergeObjects(arrObj[i], tmp)
            }
        } else {
            arrObj[i] = tmp
        }
    })
    return arrObj
}

function mergeObjects(obj, moreObj) {
    if (!moreObj) return obj
    Object.keys(moreObj).map(function (k) {
        if (Array.isArray(moreObj[k])) {
            obj[k] = obj[k] || []
            obj[k] = mergeArrays(obj[k], moreObj[k])
        } else if (typeof moreObj[k] == 'object') {
            obj[k] = obj[k] || {}
            obj[k] = mergeObjects(obj[k], moreObj[k])
        } else {
            obj[k] = moreObj[k]
        }
    })
    return obj
}

function runCommandLine() {
    config.generator = templatePath
    if (argv.verbose) {
        config.defaults.verbose = true;
        logger.debug('Loaded configuration.');
    }
    if (argv.diff) {
        config.defaults.diff = true;
    }
    if (argv.merge) {
        config.defaults.merge = true;
    }
    if (argv.project) {
        config.defaults.project = argv.project;
    }
    if (argv.account) {
        config.defaults.account = argv.account;
    }
    if (argv.ignores) {
        config.defaults.ignores = argv.ignores;
    }
    if (argv.zip) {
        processor.fileFunctions.createFile = zipFile;
        processor.fileFunctions.rimraf = nop;
        processor.fileFunctions.mkdirp = nop;
        processor.fileFunctions.mkdirp.sync = nop;
    }
    config.outputDir = argv.output;
    config.defaults.source = defName;
    config.defaults.flat = true;

    let up = url.parse(defName);
    if (up.protocol && up.protocol.startsWith('http')) {
        fetch(defName)
            .then(function (res) {
                return res.text();
            }).then(function (body) {
                main(body);
            }).catch(function (err) {
                console.error(err.message);
            });
    }
    else {
        let o = {}
        let arrFiles = defName.split('+')
        arrFiles.forEach(function (f) {
            let s = fs.readFileSync(f.trim(), 'utf8');
            o = mergeObjects(o, yaml.parse(s, { prettyErrors: true }))
        })
        main(o);
    }
}

function nop(arg, callback) { if (callback) callback(null, true); return true; }

function despatch(obj, config, callback) {
    processor.main(obj, config, callback);
}

function main(o) {
    if (o && o.openapi) {
        despatch(o, config, function (err) {
            console.log(` - Automatic code merge is ${config.defaults.merge ? 'on (use option --merge=false to turn off)' : 'off (use option --merge to turn on)'}`)
            console.log(` - Diff file generation is ${config.defaults.diff ? 'on (automatic turn on if --merge=false)' : 'off (use option --diff to turn on)'}`)
            console.log(` - Finish code generation ${!err ? `with NO error. See ${argv.output=="./"?"current folder":argv.output} for your code!` : err}`);
        });
    }
    else {
        console.error('Unrecognised OpenAPI 3.0 version');
    }
}