talk/bin/cli-serve

#!/usr/bin/env node

const program = require('./commander');
const app = require('../app');
const debug = require('debug')('talk:cli:serve');
const errors = require('../errors');
const {createServer} = require('http');
const scraper = require('../services/scraper');
const mailer = require('../services/mailer');
const MigrationService = require('../services/migration');
const SetupService = require('../services/setup');
const kue = require('../services/kue');
const mongoose = require('../services/mongoose');
const util = require('./util');
const {createSubscriptionManager} = require('../graph/subscriptions');
const {
  PORT
} = require('../config');

/**
* Get port from environment and store in Express.
*/

const port = normalizePort(PORT);
app.set('port', port);

/**
* Create HTTP server.
*/
const server = createServer(app);

/**
 * Event listener for HTTP server "error" event.
 */
function onError(error) {
  if (error.syscall !== 'listen') {
    throw error;
  }

  let bind = typeof port === 'string'
    ? `Pipe ${port}`
    : `Port ${port}`;

  // handle specific listen errors with friendly messages
  switch (error.code) {
  case 'EACCES':
    console.error(`${bind} requires elevated privileges`);
    break;
  case 'EADDRINUSE':
    console.error(`${bind} is already in use`);
    break;
  }

  throw error;
}

/**
 * Normalize a port into a number, string, or false.
 */

function normalizePort(val) {
  let port = parseInt(val, 10);

  if (isNaN(port)) {

    // named pipe
    return val;
  }

  if (port >= 0) {

    // port number
    return port;
  }

  return false;
}

/**
 * Event listener for HTTP server "listening" event.
 */

function onListening() {
  let addr = server.address();
  let bind = typeof addr === 'string'
    ? `pipe ${addr}`
    : `port ${addr.port}`;
  console.log(`API Server Listening on ${bind}`);
}

/**
 * Start the app.
 */
async function startApp(program) {

  try {

    // Check to see if the application is installed. If the application
    // has been installed, then it will throw errors.ErrSettingsNotInit, this
    // just means we don't have to check that the migrations have run.
    await SetupService.isAvailable();

    debug('setup is currently available, migrations not being checked');

  } catch (e) {

    // Check the error.
    switch (e) {
    case errors.ErrInstallLock, errors.ErrSettingsInit:

      debug('setup is not currently available, migrations now being checked');

      // The error was expected, just continue.
      break;
    default:

      // The error was not expected, throw the error!
      throw e;
    }

    // Now try and check the migration status.
    try {

      // Verify that the minimum migration version is met.
      await MigrationService.verify();

    } catch (e) {
      console.error(e);
      process.exit(1);
    }

    debug('migrations do not have to be run');
  }

  /**
  * Listen on provided port, on all network interfaces.
  */
  server.listen(port, () => {

    // Mount the websocket server if requested.
    if (program.websockets) {
      console.log(`Websocket Server Listening on ${port}`);

      // Mount the subscriptions server on the application server.
      createSubscriptionManager(server);
    }
  });
  server.on('error', onError);
  server.on('listening', onListening);
}

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

program
  .option('-j, --jobs', 'enable job processing on this thread')
  .option('-w, --websockets', 'enable the websocket (subscriptions) handler on this thread')
  .parse(process.argv);

// Start the application serving.
startApp(program);

// Enable job processing on the thread if enabled.
if (program.jobs) {

  // Start the scraper processor.
  scraper.process();

  // Start the mail processor.
  mailer.process();
}

// Define a safe shutdown function to call in the event we need to shutdown
// because the node hooks are below which will interrupt the shutdown process.
// Shutdown the mongoose connection, the app server, and the scraper.
util.onshutdown([
  () => program.jobs ? kue.Task.shutdown() : null,
  () => mongoose.disconnect(),
  () => server.close()
]);