hapi.js

/**
 * Exposes HAPI integration for the mock server
 */
var smocks = require('./lib');
var Hapi = require('hapi');
var _ = require('lodash');
var _inputs = {
  boolean: require('./lib/admin/api/input-plugins/checkbox'),
  text: require('./lib/admin/api/input-plugins/text'),
  select: require('./lib/admin/api/input-plugins/select'),
  multiselect: require('./lib/admin/api/input-plugins/multiselect')
};


module.exports = {
  toPlugin: function(pluginOptions, smocksOptions) {
    pluginOptions = pluginOptions || {};
    smocksOptions = smocksOptions || {};
    smocks.initOptions = smocksOptions;

    smocks._sanityCheckRoutes();
    options = smocks._sanitizeOptions(smocksOptions);

    var register = function (server, options, next) {
      function _next (err) {
        if (err) {
          next(err);
        } else {
          configServer(server);
          next();
        }
      }

      if (pluginOptions.onRegister) {
        pluginOptions.onRegister(server, options, _next);
      } else {
        _next();
      }
    };
    return register;
  },

  start: function(hapiOptions, smocksOptions) {
    if (!smocks.id()) {
      throw new Error('You must set an id value for the smocks instance... smocks.id("my-project")');
    }

    hapiOptions = hapiOptions || {};
    var hapiServerOptions = hapiOptions.server;
    var hapiConnectionOptions = hapiOptions.connection;
    if (!hapiServerOptions && !hapiConnectionOptions) {
      hapiConnectionOptions = hapiOptions;
    }
    smocksOptions = smocks._sanitizeOptions(smocksOptions || {});

    smocks.initOptions = smocksOptions;
    smocks._sanityCheckRoutes();

    if (!hapiConnectionOptions.routes) {
      hapiConnectionOptions.routes = { cors: true };
    }

    var server = new Hapi.Server(hapiServerOptions);
    server.connection(hapiConnectionOptions);

    configServer(server);
    server.start(function(err) {
      if (err) {
        console.error(err.message);
        process.exit(1);
      }
    });
    console.log('started smocks server on ' + hapiConnectionOptions.port + '.  visit http://localhost:' + hapiConnectionOptions.port + '/_admin to configure');

    return {
      server: server,
      start: function(options) {
        self.start(options);
      }
    };
  }
};


function wrapReply(request, reply, plugins) {
  var rtn = function() {
    var response = reply.apply(this, arguments);
    if (smocks.state.onResponse) {
      smocks.state.onResponse(request, response);
    }
    _.each(plugins, function(plugin) {
      if (plugin.onResponse) {
        plugin.onResponse(request, response);
      }
    });
    return response;
  };
  _.each(['continue', 'file', 'view', 'close', 'proxy', 'redirect'], function(key) {
    rtn[key] = function() {
      reply[key].apply(reply, arguments);
    };
  });
  return rtn;
}

function configServer(server) {
  // set the input types on the smocks object
  smocks.input = function(type, options) {
    _inputs[type] = options;
  };
  smocks.inputs = {
    get: function() {
      return _inputs;
    }
  };

  var _routes = smocks.routes.get();
  var _plugins = smocks.plugins.get();

  _.each(_routes, function(route) {
    if (route.hasVariants()) {

      var connection = server;

      if (route.connection()) {
        connection = server.select(route.connection());
      }
      connection.route({
        method: route.method(),
        path: route.path(),
        config: route.config(),
        handler: function(request, reply) {
          if (!smocks.state.isInitialized(request)) {
            _.each(_routes, function(route) {
              route.resetRouteVariant(request);
              route.resetSelectedInput(request);
            });
            smocks.plugins.resetInput(request);
            smocks.state.onInitialized && smocks.state.onInitialized(request);
          }

          if (smocks.state.onRequest) {
            smocks.state.onRequest(request, reply);
          }

          var pluginIndex = 0;
          function handlePlugins() {
            var plugin = _plugins[pluginIndex++];
            if (plugin) {
              if (plugin.onRequest) {
                plugin.onRequest.call(smocks._executionContext(request, route, plugin), request, reply, handlePlugins);
              } else {
                handlePlugins();
              }
            } else {
              reply = wrapReply(request, reply, _plugins);
              route._handleRequest.call(route, request, reply);
            }
          }

          handlePlugins();
        }
      });
    }
  }, this);

  require('./lib/admin')(server, smocks);
}