hygen/src/templates-resolver.js

The scalable code generator that saves you time.

// @flow

import type { RunnerConfig } from './types'
const L = require('lodash')
const fs = require('fs-extra')
const path = require('path')
module.exports = (config: RunnerConfig): RunnerConfig => {
  const { cwd, templates } = config
  const resolvedTemplates =
    L.find(
      [
        // $FlowFixMe
        process.env.HYGEN_TMPLS,
        path.join(cwd, '_templates')
      ],
      _ => fs.existsSync(_)
    ) || templates
  return { ...config, templates: resolvedTemplates }
}