t32k/stylestats

#!/usr/bin/env node

const fs = require('fs');
const chalk = require('chalk');
const program = require('commander');
const StyleStats = require('../lib/stylestats');
const Format = require('../lib/format');
const util = require('../lib/util');
const prettify = require('../lib/prettify');

program
  .version(require('../package.json').version)
  .usage('[options] <file ...>')
  .option('-c, --config <path>', 'set configurations')
  .option('-f, --format <format>', 'set the output format <json|csv>')
  .option('-p, --prettify', 'prettify raw data')
  .option('-n, --number', 'show only numeral metrics')
  .option('-m, --mobile', 'set the mobile user agent')
  .parse(process.argv);

if (program.args.length === 0) {
  console.log(chalk.red('\n No input file specified.'));
  program.help();
}

// Config
const config = {
  requestOptions: {
    headers: {}
  }
};
const MOBILE_UA = 'Mozilla/5.0 (iPhone; CPU iPhone OS 10_0 like Mac OS X) AppleWebKit/602.1.38 (KHTML, like Gecko) Version/10.0 Mobile/14A5297c Safari/602.1';
const numberConfig = {
  published: false,
  paths: false,
  mostIdentifierSelector: false,
  lowestCohesionSelector: false,
  uniqueFontFamilies: false,
  uniqueFontSizes: false,
  uniqueColors: false,
  propertiesCount: false
};
let userConfig = {};

if (program.mobile) {
  config.requestOptions.headers['User-Agent'] = MOBILE_UA;
}
if (program.number) {
  Object.assign(config, numberConfig);
}
if (program.config && util.isFile(program.config)) {
  const configString = fs.readFileSync(program.config, {
    encoding: 'utf8'
  });
  try {
    userConfig = JSON.parse(configString);
  } catch (err) {
    throw err;
  }
} else if (util.isObject(program.config)) {
  userConfig = config;
}
Object.assign(config, userConfig);

// Parse
const stats = new StyleStats(program.args, config);
stats.parse()
  .then(result => {
    if (program.prettify || !program.format) {
      result = prettify(result);
    }
    const format = new Format(result);
    // Other formants
    switch (program.format) {
      case 'json':
        console.log(format.toJSON());
        break;
      case 'csv':
        format.toCSV().then(csv => console.log(csv));
        break;
      default:
        console.log(` StyleStats!
${format.toTable()}`);
        break;
    }
  })
  .catch(err => console.log(chalk.red(` [ERROR] ${err.message}`)));