strapi/lib/restart.js

'use strict';

/**
 * Module dependencies
 */

// Node.js core.
const cluster = require('cluster');

// Public node modules.
const _ = require('lodash');
const async = require('async');

/**
 * Programmatically restart the server
 * (useful for the Studio)
 */

module.exports = cb => {
  const self = this;

  console.log();

  // Update the Strapi status (might be used
  // by the core or some hooks).
  this.reloading = true;

  // Async module loader to rebuild a
  // dictionary of the application.
  async.auto({

    // Rebuild the dictionaries.
    dictionaries: cb => {
      self.on('hook:_config:reloaded', () => {
        self.on('hook:_api:reloaded', () => cb());
        self.hooks._api.reload();
      });

      self.hooks._config.reload();
    }
  },

  // Callback.
  err => {

    // Just in case there is an error.
    if (err) {
      self.log.error('Impossible to reload the server');
      self.log.error('Please restart the server manually');
      self.stop();
    }

    // Tell the application the framework is reloading
    // (might be used by some hooks).
    self.reloading = true;

    // Run adapters installation
    if (cluster.isMaster) {
      self.hooks.waterline.installation();
    }

    // Install new adapters
    strapi.after('hook:waterline:installed', () => {
      self.log.warn('Application is restarting...');
      console.log();

      // Teardown Waterline adapters and
      // reload the Waterline ORM.
      self.after('hook:waterline:reloaded', () => {
        self.after('hook:router:reloaded', () => {
          process.nextTick(() => cb());

          // Update `strapi` status.
          self.reloaded = true;
          self.reloading = false;

          // Finally inform the developer everything seems ok.
          if (cluster.isMaster && _.isPlainObject(strapi.config.reload) && !_.isEmpty(strapi.config.reload) && strapi.config.reload.workers < 1) {
            self.log.info('Application\'s dictionnary updated');
            self.log.warn('You still need to restart your server to fully enjoy changes...');
          }

          self.once('restart:done', function () {
            strapi.log.info('Application successfully restarted');
          });

          if (cluster.isMaster) {
            _.forEach(cluster.workers, worker => worker.on('message', () => self.emit('restart:done')));
          }

          // Kill every worker processes.
          _.forEach(cluster.workers, () => process.kill(process.pid, 'SIGHUP'));
        });

        // Reloading the router.
        self.hooks.router.reload();
      });

      // Reloading the ORM.
      self.hooks.waterline.reload();
    });
  });
};