nsp/commands/gather.js

The Node Security (nodesecurity.io) command line interface

'use strict';

const Fs = require('fs');
const Path = require('path');

const Command = require('../lib/command');
const Wreck = require('../lib/wreck');

const internals = {};
internals.fetch = function (baseUrl, offset = 0, items = []) {

  return Wreck.get(`${baseUrl}/advisories?offset=${offset}`, { json: true }).then((res) => {

    if (res.data.total > res.data.count + res.data.offset) {
      return internals.fetch(baseUrl, res.data.offset + res.data.count, items.concat(res.data.results));
    }

    return items.concat(res.data.results);
  });
};

exports.command = 'gather';
exports.description = 'gathers current advisories into a json file for offline checks';

exports.builder = {
  path: {
    description: 'location to save advisories database',
    group: 'Locations:'
  }
};

exports.handler = Command.wrap('gather', (args) => {

  const path = args.path.endsWith('advisories.json') ? args.path : Path.join(args.path, 'advisories.json');

  return internals.fetch(args.baseUrl).then((items) => {

    Fs.writeFileSync(path, JSON.stringify(items, null, 2), 'utf8');
    return {
      message: 'Successfully saved advisories.json'
    };
  });
});