swagger-merger/lib/merger_yaml.js

Merge multiple swagger files into a swagger file, support JSON/YAML.

/**
 * Created by WindomZ on 17-4-11.
 */
'use strict'

const fs = require('fs')
const path = require('path')

const fmtconv = require('fmtconv')

const mergeJSON = require('./merge_json')

function mergerYAML (param) {
  // read file
  let doc = '' + fs.readFileSync(param.input, 'utf8')

  switch (path.extname(param.input).toLowerCase()) {
    case '.json':
      break
    default:
      // parse to JSON
      doc = fmtconv.stringYAML2JSON(doc, true)
      break
  }

  // merge to JSON
  const obj = mergeJSON(param, param.dir, param.input, doc)

  let dump
  if (param.output) {
    switch (path.extname(param.output).toLowerCase()) {
      case '.json':
        // parse to JSON
        dump = fmtconv.stringJSON2JSON(obj, param.compact)
        break
      default:
        // parse to YAML
        dump = fmtconv.stringJSON2YAML(obj, param.compact)
        break
    }

    // write to file
    if (dump) {
      fs.writeFileSync(param.output, dump, null)
    }
  }

  return obj
}

module.exports = mergerYAML