AmazeeLabs/silverback-tools

import {Command, flags} from '@oclif/command';
import YAML from 'yaml';
import * as path from 'path';
import {YAMLError} from 'yaml/util';
import {readFile, writeFile} from '../utils/fs';
import {merge} from 'lodash';

export default class Yaml extends Command {
  static description = `
  Store YAML data into target file.
  Automatically changes output format based on file extension:
  .yml  -> YAML
  .yaml -> YAML
  .json -> formatted JSON

  If the target file already exists, the content is merged using lodash merge:
  https://lodash.com/docs/#merge
  `;

  static examples = ['$ svbt yaml test.yml "name: foo"'];

  static flags = {
    help: flags.help({char: 'h'}),
  };

  static args = [{name: 'file'}, {name: 'data'}];

  async run() {
    const {args} = this.parse(Yaml);
    try {
      const content = readFile(args.file);
      const data = merge(
        {},
        content ? YAML.parse(content) : {},
        YAML.parse(args.data)
      );
      const output =
        path.extname(args.file) === '.json'
          ? JSON.stringify(data, null, 2) + '\n'
          : YAML.stringify(data);
      writeFile(args.file, output);
    } catch (error) {
      if (error instanceof YAMLError) {
        this.warn(`Invalid YAML - ${error.message}:\n${error.source}`);
      }
    }
  }
}