index.ts

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

// import dependencies
import * as fs from 'fs';
import * as readline from 'readline';
import { ArgumentParser } from 'argparse';
import { build, generate } from './src/SwaggerDataGen';

const INPUT_ARG = 'swagger-input';
const OUTPUT_ARG = 'json-output';

// grab expected user input
const parser = new ArgumentParser({
  addHelp: true,
  description: 'Swagger Data Generator generates mock data from Swagger files.'
});
let args: string[];

/**
 * saveOutput - Verify output path and save file
 *
 * @param generatedData {Object} - containes the key-value pairs of definition and its created data
 */
function saveOutput(generatedData: any) {
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
  });

  function writeData(yes: boolean) {
    if (yes) {
      fs.writeSync(fs.openSync(args['json-output'], 'w'),
        JSON.stringify(generatedData, null, '\t'));
    } else {
      rl.write('...Aborting\n');
    }
    rl.close();
    process.stdin.pause();
  }

  if (args['force-yes']) {
    writeData(true);
  } else {
    rl.question('content in ' + args['json-output'] + ' will be overwritten. continue? (y or n): ',
      function handleAnswer(answer) {
        writeData(answer === 'y' || answer === 'Y');
      });
  }
}

parser.addArgument(['-y'], { help: 'Always overwrite output file (do not ask to overwrite)', action: 'storeTrue', dest: 'force-yes' });
parser.addArgument([INPUT_ARG], { help: 'Input Swagger file' });
parser.addArgument([OUTPUT_ARG], { help: 'Output file for generated mock data' });
args = parser.parseArgs();

build(args[INPUT_ARG])
  .then(api => saveOutput(generate(api)));