add basic test support for prettyhtml cli and api

Author: StarpTech

packages/prettyhtml/cli.js

@@ -0,0 +1,76 @@
+const meow = require('meow')
+
+function args(prettierConfig = {}) {
+  return meow(
+    `
+    Usage: prettyhtml [<glob> ...] [options ...],
+
+    Options:
+
+    --tab-width       Specify the number of spaces per indentation-level
+    --print-width     Specify the maximum line length
+    --use-tabs        Use tabs for indentation
+    --single-quote    Use single instead of double quotes
+    --use-prettier    Use prettier to format embedded content
+    --wrapAttributes  Force to wrap attributes (when it has multiple)
+    --sortAttributes  Sort attributes alphabetically
+    --stdin           Specify the standard stream as source (for pipe mode)
+    --quiet           Do not output anything for a file which has no warnings or errors
+    --silent          Do not output messages without fatal set to true
+
+    Examples
+      $ prettyhtml *.html
+      $ prettyhtml *.html !example.html
+      $ echo "<custom foo='bat'></custom>" | prettyhtml --stdin
+      $ echo "<custom foo='bat'></custom>" --stdin ./test.html
+    `,
+    {
+      autoHelp: true,
+      autoVersion: true,
+      flags: {
+        tabWidth: {
+          type: 'number',
+          default: prettierConfig.tabWidth || 2
+        },
+        printWidth: {
+          type: 'number',
+          default: prettierConfig.printWidth || 80
+        },
+        useTabs: {
+          type: 'boolean',
+          default: prettierConfig.useTabs || false
+        },
+        singleQuote: {
+          type: 'boolean',
+          default: false
+        },
+        usePrettier: {
+          type: 'boolean',
+          default: true
+        },
+        wrapAttributes: {
+          type: 'boolean',
+          default: false
+        },
+        sortAttributes: {
+          type: 'boolean',
+          default: false
+        },
+        stdin: {
+          type: 'boolean',
+          default: false
+        },
+        quiet: {
+          type: 'boolean',
+          default: false
+        },
+        silent: {
+          type: 'boolean',
+          default: false
+        }
+      }
+    }
+  )
+}
+
+module.exports = args

packages/prettyhtml/cli/args.js

@@ -0,0 +1,51 @@
+#!/usr/bin/env node
+
+'use strict'
+
+const PassThrough = require('stream').PassThrough
+const notifier = require('update-notifier')
+const engine = require('unified-engine')
+const unified = require('unified')
+const { basename } = require('path')
+const pack = require('./../package')
+const prettier = require('prettier')
+const args = require('./args')
+const { transform, processor } = require('./processor')
+
+const extensions = ['html']
+
+notifier({ pkg: pack }).notify()
+
+const prettierConfig = prettier.resolveConfig.sync(process.cwd()) || {}
+const cli = args(prettierConfig)
+
+const settings = {
+  processor: unified(),
+  extensions: extensions,
+  configTransform: transform,
+  streamError: new PassThrough(), // sink errors
+  rcName: '.prettyhtmlrc',
+  packageField: 'prettyhtml',
+  ignoreName: '.prettyhtmlignore',
+  frail: false,
+  defaultConfig: transform({ prettierConfig, flags: cli.flags })
+}
+
+const processResult = processor({ flags: cli.flags })
+
+if (cli.flags.stdin === false) {
+  if (cli.input.length === 0) {
+    cli.showHelp()
+  } else {
+    settings.files = cli.input
+    settings.output = true // Whether to overwrite the input files
+    settings.out = false // Whether to write the processed file to streamOut
+
+    engine(settings, processResult)
+  }
+} else {
+  if (cli.input.length !== 0) {
+    settings.output = basename(cli.input[0])
+  }
+  engine(settings, processResult)
+}

packages/prettyhtml/cli/index.js

@@ -0,0 +1,62 @@
+const parse = require('@starptech/rehype-webparser')
+const stringify = require('@starptech/prettyhtml-formatter/stringify')
+const sortAttributes = require('@starptech/prettyhtml-sort-attributes')
+const format = require('@starptech/prettyhtml-formatter')
+const report = require('vfile-reporter')
+
+function processor({ flags }) {
+  return function processResult(err, code, result) {
+    const out = report(err || result.files, {
+      quiet: flags.quiet,
+      silent: flags.silent
+    })
+
+    if (out) {
+      console.error(out)
+    }
+
+    process.exit(code)
+  }
+}
+
+function transform({ prettierConfig, flags }) {
+  const plugins = [
+    [
+      parse,
+      {
+        ignoreFirstLf: false,
+        decodeEntities: false,
+        selfClosingCustomElements: true,
+        selfClosingElements: true
+      }
+    ],
+    [
+      format,
+      {
+        tabWidth: flags.tabWidth,
+        useTabs: flags.useTabs,
+        singleQuote: flags.singleQuote,
+        usePrettier: flags.usePrettier,
+        prettier: prettierConfig
+      }
+    ]
+  ]
+
+  if (flags.sortAttributes || prettierConfig.sortAttributes) {
+    plugins.push([sortAttributes, {}])
+  }
+
+  plugins.push([
+    stringify,
+    {
+      tabWidth: flags.tabWidth,
+      printWidth: flags.printWidth,
+      singleQuote: flags.singleQuote,
+      wrapAttributes: flags.wrapAttributes
+    }
+  ])
+
+  return { plugins }
+}
+
+module.exports = { transform, processor }