s2s/lib/templates/index.js.flow

Source To Source

// @flow
import path from 'path'
import fs from 'fs'
import chalk from 'chalk'
import ejs from 'ejs'
import slash from 'slash'
import type { Path, Template } from 'types'
import { formatText, trimAndFormatPath } from '../reporters'
import { isAlreadyExist, relativeFromCwd, getOutputPath } from '../utils'
import some from '../utils/some'

type CopyError = Error & { path: string, code: string }

function handleCopyError(err: CopyError): void {
  if (err.code === 'ENOENT') {
    const errorText = `${chalk.reset.inverse.bold.red(
      'TEMPLATE'
    )} ${trimAndFormatPath(err.path)}
          no such file or directory`
    console.log(errorText)
  } else {
    console.log(err.stack)
  }
}

const DEFAULT_TEMPLATES_DIR = 'templates'

function validate(template: Template, eventPath: Path) {
  if (typeof template.test === 'string' || Array.isArray(template.test)) {
    return some(eventPath, template.test)
  } else if (!template.test.test(eventPath)) {
    return false
  }

  return true
}

function handleTemplate(
  eventPath: Path,
  template: Template,
  templatesDir: string
) {
  if (!validate(template, eventPath)) {
    return
  }

  const outputPath = template.output
    ? getOutputPath(template.output, eventPath)
    : eventPath

  if (isAlreadyExist(outputPath)) {
    return
  }

  const templatePath = path.join(templatesDir, template.input)
  const str = fs.readFileSync(templatePath, 'utf8')

  const result = ejs.render(str, {
    FILENAME: path.parse(outputPath).name,
    DIRNAME: path.basename(path.dirname(outputPath)),
  })

  fs.writeFileSync(outputPath, result)

  console.log(
    formatText('TEMPLATE', relativeFromCwd(templatePath), slash(outputPath))
  )
}

export default function handleTemplates(
  eventPath: Path,
  templates: Array<Template> = [],
  templatesDir: string = DEFAULT_TEMPLATES_DIR
) {
  for (const template of templates) {
    try {
      handleTemplate(eventPath, template, templatesDir)
    } catch (error) {
      handleCopyError(error)
    }
  }
}