talk/bin/cli-assets

#!/usr/bin/env node

/**
 * Module dependencies.
 */

const util = require('./util');
const program = require('commander');
const parseDuration = require('ms');
const Table = require('cli-table');
const AssetModel = require('../models/asset');
const CommentModel = require('../models/comment');
const AssetsService = require('../services/assets');
const mongoose = require('../services/mongoose');
const scraper = require('../services/scraper');
const inquirer = require('inquirer');

// Register the shutdown criteria.
util.onshutdown([
  () => mongoose.disconnect()
]);

/**
 * Lists all the assets registered in the database.
 */
async function listAssets() {
  try {
    let assets = await AssetModel.find({}).sort({'created_at': 1});

    let table = new Table({
      head: [
        'ID',
        'Title',
        'URL'
      ]
    });

    assets.forEach((asset) => {
      table.push([
        asset.id,
        asset.title ? asset.title : '',
        asset.url ? asset.url : ''
      ]);
    });

    console.log(table.toString());
    util.shutdown();
  } catch (e) {
    console.error(e);
    util.shutdown(1);
  }
}

async function refreshAssets(ageString) {
  try {
    const now = new Date().getTime();
    const ageMs = parseDuration(ageString);
    const age = new Date(now - ageMs);

    let assets = await AssetModel.find({
      $or: [
        {
          scraped: {
            $lte: age
          }
        },
        {
          scraped: null
        }
      ]
    });

    // Queue all the assets for scraping.
    await Promise.all(assets.map(scraper.create));

    console.log('Assets were queued to be scraped');
    util.shutdown();
  } catch (e) {
    console.error(e);
    util.shutdown(1);
  }
}

async function updateURL(assetID, assetURL) {
  try {
    await AssetsService.updateURL(assetID, assetURL);

    console.log(`Asset ${assetID} was updated to have url ${assetURL}.`);
    util.shutdown();
  } catch (e) {
    console.error(e);
    util.shutdown(1);
  }
}

async function merge(srcID, dstID) {
  try {

    // Grab the assets...
    let [srcAsset, dstAsset] = await AssetsService.findByIDs([srcID, dstID]);
    if (!srcAsset || !dstAsset) {
      throw new Error('Not all assets indicated by id exist, cannot merge');
    }

    // Count the affected resources...
    let srcCommentCount = await CommentModel.find({asset_id: srcID}).count();

    console.log(`Now going to update ${srcCommentCount} comments and delete the source Asset[${srcID}].`);

    let {confirm} = await inquirer.prompt([
      {
        type: 'confirm',
        name: 'confirm',
        message: 'Proceed with merge',
        default: false
      }
    ]);

    if (confirm) {

      // Perform the merge!
      await AssetsService.merge(srcID, dstID);
    } else {
      console.warn('Aborting merge');
    }

    util.shutdown(0);
  } catch (e) {
    console.error(e);
    util.shutdown(1);
  }
}

//==============================================================================
// Setting up the program command line arguments.
//==============================================================================

program
  .command('list')
  .description('list all the assets in the database')
  .action(listAssets);

program
  .command('refresh <age>')
  .description('queues the assets that exceed the age requested')
  .action(refreshAssets);

program
  .command('update-url <assetID> <url>')
  .description('update the URL of an asset')
  .action(updateURL);

program
  .command('merge <srcID> <dstID>')
  .description('merges two assets together by moving comments from src to dst and deleting the src asset')
  .action(merge);

program.parse(process.argv);

// If there is no command listed, output help.
if (!process.argv.slice(2).length) {
  program.outputHelp();
  util.shutdown();
}