avifors/src/commands/Generator.js

A MDE tool that generates code from a YAML definition of your app domain model.

import fs from 'fs'
import path from 'path'
import mkdirp from 'mkdirp'
import chalk from 'chalk'

export default class Generator {
  constructor(avifors, yamlHelper) {
    this.avifors = avifors
    this.yamlHelper = yamlHelper
  }

  generate(model) {
    this._writeGenerators(model)
    this._writeAutoGenerators(model)
    console.log(chalk.bold.green("Generation completed."))
  }

  _writeGenerators(model) {
    model.forEach(item => {
      const generator = this.avifors.getGenerator(item.type)[0]
      generator.outputs(item.arguments)
        .map(i => i(item.arguments))
        .map(i => ({
          contents: this.avifors.nunjucks.render(i.template, item.arguments),
          ...i
        }))
        .forEach(i => this._writeFile(i.path, i.contents))
    })
  }

  _writeAutoGenerators(model) {
    this.avifors.autoGenerators
      .forEach(autoGenerator => autoGenerator
        .map(i => ({
          contents: this.avifors.nunjucks.render(i.template, i.variables),
          ...i
        }))
        .forEach(i => this._writeFile(i.path, i.contents))
      )
  }

  _writeFile(filePath, contents) {
    try {
      let dirPath = path.dirname(filePath)
      try { fs.statSync(dirPath) }
      catch(e) { mkdirp.sync(dirPath) }

      fs.writeFileSync(filePath, contents, { flag: 'w+' })
    } catch (e) {
      throw `Could not write file ${filePath}`
    }
  }
}