generator-convenia/route/index.js

const Generator = require('yeoman-generator')
const yosay = require('yosay')
const utils = require('../app/utils');


module.exports = class extends Generator {
  constructor (args, opts) {
    super(args, opts)

    this.modules = this.config.getAll().modules;
  }

  initializing () {
    this.log(yosay('Iniciando generator de rotas'))
  }

  prompting () {
    const restrictedNames = this.modules.join('|')
    return this.prompt([{
      type: 'input',
      name: 'routeName',
      message : `Escolha o nome da rota (Rotas existentes: ${restrictedNames}):`,
      default: 'Teste'
    }]).then(answers => {
      this.routeName = answers.routeName

      const modules = this.config.getAll().modules;

      if (modules.indexOf(this.routeName) > 0) {
        this.routeName = [this.routeName, +new Date()].join('')
      }

      this.lrouteName = this.routeName.toLowerCase();
      modules.push(this.lrouteName)

      this.config.set({ modules })
      this.config.save()
    })
  }

  writing() {
    const routerPath = this.destinationPath('src/router/index.js')
    const modulePath = this.destinationPath('src/vuex/modules.js')
    const baseName = utils.getBaseName(this.lrouteName)
    const relativePath = utils.getRelativePath(this.lrouteName, 'app', '')

    // Copy the route and vuex-module template
    this.fs.copyTpl(
      this.templatePath('route.vue'),
      this.destinationPath(`src/app/${this.lrouteName}/main.vue`),
      {
        lmoduleName: this.lrouteName,
        moduleName: this.routeName
      }
    )

    this.fs.copy(
      this.templatePath('vuex'),
      this.destinationPath(`src/app/${this.lrouteName}/vuex`)
    )

    // Create the vuex module import declaration
    const importString = `import ${this.lrouteName} from '../app/${this.lrouteName}/vuex'`
    const importDeclaration = (utils.readString(importString)).body[0]

    const exportSpecifier = {
      type: 'ExportSpecifier',
      local: { type: 'Identifier', name: this.lrouteName },
      exported: { type: 'Identifier', name: this.lrouteName }
    }
    const tree = utils.read(modulePath)

    let position = 0

    for (const item of tree.body) {
      if (item.type === 'ImportDeclaration') { position += 1; }
      else if (item.type === 'ExportNamedDeclaration') {
        console.log(item.specifiers)
        item.specifiers.push(exportSpecifier)
      }
    }

    // Insert the module after the last import
    tree.body.splice(position, 0, importDeclaration)

    utils.write(modulePath, tree)
  }
}