mos/lib/cli.js

#!/usr/bin/env node
'use strict'

process.on('unhandledRejection', function (reason, p) {
  console.log('Possibly Unhandled Rejection at: Promise ', p, ' reason: ',
    reason)
})

const resolve = require('resolve')
const cwd = process.cwd()

let localCLI
try {
  localCLI = resolve.sync('mos/lib/cli', { basedir: cwd })
} catch (err) {
  localCLI = __filename
}

if (localCLI && localCLI !== __filename) {
  console.log('Using local install of mos')
  require(localCLI)
  return // eslint-disable-line
}

const updateNotifier = require('update-notifier')
const meow = require('meow')
const processFiles = require('./process-files')
const normalizeNewline = require('normalize-newline')
const relative = require('relative')
const normalizePath = require('normalize-path')
const chalk = require('chalk')
const fs = require('fs')
const path = require('path')
const test = require('tape')
const tapDiff = require('@zkochan/tap-diff')
const readPkgUp = require('@zkochan/read-pkg-up')
const mos = require('.')
const initCustomPlugins = require('./init-custom-plugins')

const cli = meow([
  'Usage',
  '  mos [test] [files]',
  '',
  'Options',
  ' --init             Add mos to your project',
  ' --help, -h         Display usage',
  ' --version, -v      Display version',
  ' --tap              Generate TAP output when testing markdown files',
  '',
  ' -x="<exclude-pattern>"',
  '                    Exclude pattern',
  '',
  'Examples',
  '  mos',
  '  mos test',
  '  mos test README.md',
  '  mos --init',
  '  mos --init README.md',
  '',
  'Tips:',
  '  Add `mos test` to your `scripts.test` property in `package.json`',
].join('\n'), {
  alias: {
    help: 'h',
    version: 'v',
  },
})

updateNotifier({pkg: cli.pkg}).notify()

if (cli.flags.init) {
  require('mos-init')()
  return // eslint-disable-line
}

const highlightPath = chalk.bgBlack.yellow

const isTest = ~['test', 't'].indexOf((cli.input[0] || '').toLowerCase())

readPkgUp({ cwd })
  .then(result => {
    const pkgPlugins = Object
      .keys(result.pkg.dependencies || {})
      .concat(Object.keys(result.pkg.devDependencies || {}))
      .filter(dep => dep.indexOf('mos-plugin-') === 0)
      .map(dep => resolve.sync(dep, { basedir: cwd }))
      .map(normalizePath)
      .map(require)

    const processor = initCustomPlugins(mos())
      .use(pkgPlugins)

    const mdExtensions = ['markdown', 'mdown', 'mkdn', 'mkd', 'md']
    const files = cli.input[isTest ? 1 : 0]
    const pattern = files
      ? path.resolve(cwd, files)
      : path.resolve(cwd, `{/**/,/}*.{${mdExtensions.join()}}`)
    const ignorePattern = cli.flags.x
      ? path.resolve(cwd, cli.flags.x)
      : null

    if (isTest) {
      if (cli.flags.tap !== true) {
        test.createStream()
          .pipe(tapDiff())
          .pipe(process.stdout)
      }
      test('markdown', t => {
        processFiles({
          processor,
          pattern,
          ignorePattern,
          afterEachRender (opts) {
            const relativePath = normalizePath(getRelativePath(opts.filePath))
            t.equal(normalizeNewline(opts.newMD), normalizeNewline(opts.currentMD), relativePath)
          },
        })
        .then(() => t.end())
        .catch(err => { throw err })
      })
    } else {
      processFiles({
        processor,
        pattern,
        ignorePattern,
        afterEachRender (opts) {
          if (normalizeNewline(opts.newMD) !== normalizeNewline(opts.currentMD)) {
            fs.writeFileSync(opts.filePath, opts.newMD, {
              encoding: 'utf8',
            })
            const relativePath = normalizePath(getRelativePath(opts.filePath))
            console.log('updated', highlightPath(relativePath))
          }
        },
      })
      .catch(err => { throw err })
    }
  })
  .catch(err => { throw err })

function getRelativePath (filePath) {
  return relative(cwd, filePath)
}