ua-parser2/js/bin/add.js

An improved user agent parser.

#!/usr/bin/env node

/*!
 * Add new test cases from a txt file containing User-Agent strings to
 * a json based test cases file.
 *
 * If no testcases file is given, the default in "test_resources/tests.json"
 * is choosen for appending the test cases.
 */

'use strict'

const fs = require('fs')
const path = require('path')
const cmd = require('commander')
const { SplitLine, through } = require('streamss')
const helper = require('../test/lib/helper')
const parser = require('../')()

const
  pwd = process.cwd(),
  config = {
    version: '0.0.2',
    params: ['ua', 'engine', 'os', 'device'], // / params to print on console
    testsFile: path.resolve(__dirname, '../../test_resources/tests.json') // / default tests file
  }

// / the program
cmd
  .version(config.version)
  .option('-u, --useragents <file>', 'Add User-Agents from <file>')
  .option('-t, --tests <file>', 'to tests file <file>')
  .option('-c, --console', 'output details to console')
  .parse(process.argv)

if (cmd.tests) {
  config.testsFile = path.resolve(pwd, cmd.tests)
} else {
  console.log('\n' +
    '    appending tests to "' + path.relative(pwd, config.testsFile) + '"' +
    '\n')
}

if (!cmd.useragents || /^-/.test(cmd.useragents)) {
  console.error('\n' +
    '    specify user-agents file with "-u <file>".\n' +
    '    exiting...' +
    '\n')
  process.exit(1)
}

config.uaFile = path.resolve(pwd, cmd.useragents)

/*
 * parse a single user-agent and write the result to the stream
 */
function parse (txt, encoding, done) {
  let res
  const out = []

  res = parser.parse(txt.toString())
  res = helper.compact.strip(res)

  if (cmd.console) {
    out.push('== ' + res.string)
    config.params.forEach(function (p) {
      out.push(p + ': ' + JSON.stringify(res[p]))
    })
    console.log(out.join('\n') + '\n')
  }

  config.params.forEach(function (p) {
    if (res[p] && res[p].debug) { delete (res[p].debug) }
  })
  this.push(JSON.stringify(res) + '\n') // jshint ignore:line

  done()
}

/*
 * the pipe - appending new parse results to the tests output
 */
fs.createReadStream(config.uaFile, { encoding: 'utf8' })
  .pipe(new SplitLine({ chomp: true }))
  .pipe(through(parse))
  .pipe(fs.createWriteStream(config.testsFile, { flags: 'a', encoding: 'utf8' }))