gridsome/lib/app/codegen/index.js

A JAMstack framework for building blazing fast websites with Vue.js

const path = require('path')
const fs = require('fs-extra')
const genIcons = require('./icons')
const genConfig = require('./config')
const genRoutes = require('./routes')
const genPlugins = require('./plugins')
const genConstants = require('./constants')
const { BOOTSTRAP_FULL } = require('../../utils/constants')

// TODO: let plugins add generated files

class Codegen {
  constructor (app) {
    this.app = app

    this.files = {
      'icons.js': genIcons,
      'config.js': genConfig,
      'routes.js': genRoutes,
      'constants.js': genConstants,
      'plugins-server.js': () => genPlugins(app, true),
      'plugins-client.js': () => genPlugins(app, false),
      'now.js': () => `export default ${app.store.lastUpdate}`
    }

    app.hooks.bootstrap.tapPromise(
      {
        name: 'GridsomeCodegen',
        label: 'Generate temporary code',
        phase: BOOTSTRAP_FULL
      },
      () => this.generate()
    )
  }

  async generate (filename = null, ...args) {
    const outputDir = this.app.config.tmpDir

    const outputFile = async (filename, ...args) => {
      const filepath = path.join(outputDir, filename)
      const content = await this.files[filename](this.app, ...args)

      if (await fs.exists(filepath)) {
        await fs.remove(filepath)
      }

      await fs.outputFile(filepath, content)
    }

    if (filename) {
      await outputFile(filename, ...args)
    } else {
      await fs.remove(outputDir)

      for (const filename in this.files) {
        await outputFile(filename)
      }
    }
  }
}

module.exports = Codegen