ultimate/lib/db/mongoose/connect.js

Dependency library for `ultimate-seed`.

/*
 * ultimate.db.mongoose.connect
 */

'use strict';

var url = require('url'),
    util = require('util');

var _ = require('lodash'),
    mongoose = require('mongoose');

var debug = require('debug')('ultimate.db.mongoose');

var _client = null;

/**
 * Connect to MongoDB.
 *
 * @param {Object} mongoConfig Mongo config.
 * @param {Object} cb Callback.
 * @return {Client}
 */
function connect(mongoConfig, cb) {
  var mongoUrl = url.parse(process.env.MONGOHQ_URL || process.env.MONGOLAB_URI || '');

  if (mongoUrl.href) {
    mongoConfig.host = mongoUrl.hostname;
    mongoConfig.port = parseInt(mongoUrl.port, 10);
    mongoConfig.username = mongoUrl.auth.split(':')[0];
    mongoConfig.password = mongoUrl.auth.split(':')[1];
    mongoConfig.db = mongoUrl.pathname.slice(1);
  }

  // Override with Docker container link environment variable if detected.
  _.assign(mongoConfig, _.reduce(process.env, function (result, val, key) {
    var matched = key.match(/^MONGODB_PORT_(\d+)_TCP_ADDR$/);
    if (matched) {
      result.push({
        host: val,
        port: +matched[1]
      });
    }
    return result;
  }, [])[0]);

  if (!_.isFunction(cb)) {
    cb = function (err) {
      if (err) { console.error(err); }
    };
  }

  if (!_.isObject(mongoConfig)) {
    return cb('mongoConfig must be an object.');
  }

  // Connect
  var uri = util.format(
    'mongodb://%s:%s/%s',
    mongoConfig.host || 'localhost',
    mongoConfig.port || 27017,
    mongoConfig.db || 'ultimate-seed'
  );
  if (mongoConfig.username && mongoConfig.password) {
    _client = mongoose.connect(uri, {
      user: mongoConfig.username || null,
      pass: mongoConfig.password || null,
    });
  } else {
    _client = mongoose.connect(uri);
  }

  // Events
  mongoose.connection.once('open', function () {
    debug('Connected to MongoDB:'.cyan);
    debug(uri.bold);
    return cb();
  });
  mongoose.connection.on('error', function (err) {
    debug('Connection error:'.cyan);
    debug(err);
    return cb(err);
  });

  return _client;
}

// Public API
exports = module.exports = connect;

Object.defineProperty(exports, '_client', {
  get: function () {
    return _client;
  }
});