mdx-deck/cli.js

#!/usr/bin/env node
const path = require('path')
const meow = require('meow')
const findup = require('find-up')
const open = require('react-dev-utils/openBrowser')
const chalk = require('chalk')
const remark = {
  emoji: require('remark-emoji'),
  unwrapImages: require('remark-unwrap-images'),
}
const pkg = require('./package.json')

const config = require('pkg-conf').sync('mdx-deck')
const log = require('./lib/log')

const cli = meow(
  `
  ${chalk.gray('Usage')}

    $ ${chalk.magenta('mdx-deck deck.mdx')}

    $ ${chalk.magenta('mdx-deck build deck.mdx')}

    $ ${chalk.magenta('mdx-deck pdf deck.mdx')}

    $ ${chalk.magenta('mdx-deck screenshot deck.mdx')}

  ${chalk.gray('Options')}

      --webpack     Path to webpack config file

    ${chalk.gray('Dev server options')}

      -h --host     Dev server host
      -p --port     Dev server port
      --hot-port    Dev server hot reload port
      --no-open     Prevent from opening in default browser

    ${chalk.gray('Build options')}

      -d --out-dir  Output directory for exporting
      --no-html     Disable static HTML rendering

    ${chalk.gray('Export options')}

      --out-file    Filename for screenshot or PDF export
      --width       Width in pixels
      --height      Height in pixels

    ${chalk.gray('PDF options')}

      --no-sandbox  Disable Puppeteer sandbox

`,
  {
    description: chalk.magenta('[mdx-deck] ') + chalk.gray(pkg.description),
    flags: {
      port: {
        type: 'string',
        alias: 'p',
      },
      host: {
        type: 'string',
        alias: 'h',
      },
      hotPort: {
        type: 'string',
      },
      open: {
        type: 'boolean',
        alias: 'o',
        default: true,
      },
      outDir: {
        type: 'string',
        alias: 'd',
      },
      outFile: {
        type: 'string',
      },
      html: {
        type: 'boolean',
        default: true,
      },
      webpack: {
        type: 'string',
      },
      sandbox: {
        type: 'boolean',
        default: true,
      },
    },
  }
)

const [cmd, file] = cli.input
const doc = file || cmd

if (!doc) cli.showHelp(0)

const opts = Object.assign(
  {
    dirname: path.dirname(path.resolve(doc)),
    globals: {
      FILENAME: JSON.stringify(path.resolve(doc)),
    },
    host: 'localhost',
    port: 8080,
    outDir: 'dist',
  },
  config,
  cli.flags
)

opts.outDir = path.resolve(opts.outDir)
if (opts.webpack) {
  opts.webpack = require(path.resolve(opts.webpack))
} else {
  const webpackConfig = findup.sync('webpack.config.js', { cwd: opts.dirname })
  if (webpackConfig) opts.webpack = require(webpackConfig)
}

let dev

switch (cmd) {
  case 'build':
    log('building')
    const build = require('./lib/build')
    build(opts)
      .then(res => {
        log('done')
        process.exit(0)
      })
      .catch(err => {
        log.error(err)
        process.exit(1)
      })
    break
  case 'pdf':
    log('exporting to PDF')
    const pdf = require('./lib/pdf')
    dev = require('./lib/dev')
    dev(opts)
      .then(({ server }) => {
        log('rendering PDF')
        pdf(opts)
          .then(filename => {
            server.close()
            log('done', filename)
            process.exit(0)
          })
          .catch(err => {
            log.error(err)
            process.exit(1)
          })
      })
      .catch(err => {
        log.error(err)
        process.exit(1)
      })
    break
  case 'screenshot':
    log('exporting to PNG')
    const screenshot = require('./lib/screenshot')
    dev = require('./lib/dev')
    dev(opts)
      .then(({ server }) => {
        log('rendering screenshot')
        screenshot(opts)
          .then(filename => {
            server.close()
            log('done', filename)
            process.exit(0)
          })
          .catch(err => {
            log.error(err)
            process.exit(1)
          })
      })
      .catch(err => {
        log.error(err)
        process.exit(1)
      })
    break
  case 'dev':
  default:
    log('starting dev server')
    dev = require('./lib/dev')
    dev(opts)
      .then(res => {
        const { address, port } = res.server.address()
        const url = 'http://' + address + ':' + res.port
        if (opts.open) open(url)
        log('listening on', chalk.magenta(url))
      })
      .catch(err => {
        log.error(err)
        process.exit(1)
      })
    break
}