mongo-dsn/lib/DSN.js

A DSN Object for MongoDB

// Generated by CoffeeScript 1.7.1
var DSN, DSNOptions, EventEmitter, ObjUtils, fs,
  __hasProp = {}.hasOwnProperty,
  __extends = function(child, parent) { for (var key in parent) { if (__hasProp.call(parent, key)) child[key] = parent[key]; } function ctor() { this.constructor = child; } ctor.prototype = parent.prototype; child.prototype = new ctor(); child.__super__ = parent.prototype; return child; };

fs = require('fs');

ObjUtils = require('obj-utils');

DSNOptions = require('./DSNOptions');

EventEmitter = require('events').EventEmitter;

DSN = (function(_super) {
  __extends(DSN, _super);

  DSN.prototype.__dsn = null;

  function DSN(dsn) {
    if (dsn) {
      this.setDSN((dsn instanceof String ? this.parseDSNString(dsn) : dsn));
    }
  }

  DSN.prototype.parseDSNString = function(string) {
    var dsnParams, pass, protoDSN;
    if ((dsnParams = string.match(/^(mongodb\:\/\/)?(.+:?.?@)?([a-z0-9\.]+)+(:[a-zA-Z0-9]{4,5})?\,?([a-zA-Z0-9\.\,:]*)?(\/\w+)?\??([a-zA-Z0-9\_=&\.]*)?$/)) !== null) {
      return protoDSN = {
        protocol: dsnParams[1] ? dsnParams[1].split(':').shift() : null,
        username: dsnParams[2] ? (pass = dsnParams[2].replace('@', '').split(':')).shift() : null,
        password: pass && pass.length ? "" + pass : null,
        host: dsnParams[3] || null,
        port: dsnParams[4] ? parseInt(dsnParams[4].split(':').pop()) : null,
        replicas: dsnParams[5] ? dsnParams[5].split(',').map(function(v, k) {
          var port;
          return {
            host: (port = v.split(':')).shift(),
            port: parseInt(port.shift())
          };
        }) : null,
        database: dsnParams[6] ? dsnParams[6].split('/').pop() : null,
        options: dsnParams[7] ? new DSNOptions(dsnParams[7]) : null
      };
    }
    return null;
  };

  DSN.prototype.setOptions = function(options) {
    return this.__dsn != null ? this.__dsn : this.__dsn = {};
  };

  DSN.prototype.getOptions = function() {
    var _ref;
    return ((_ref = this.__dsn) != null ? _ref.options : void 0) || null;
  };

  DSN.prototype.setDSN = function(dsn) {
    var e;
    if (ObjUtils.isOfType(dsn, String)) {
      dsn = this.parseDSNString(dsn);
    }
    try {
      if (this.validate(dsn)) {
        return this.__dsn = dsn;
      }
    } catch (_error) {
      e = _error;
      throw Error(e);
    }
  };

  DSN.prototype.getDSN = function() {
    return this.__dsn || null;
  };

  DSN.prototype.validate = function(dsn) {
    var e, key, oTypes, options, value;
    oTypes = {
      protocol: {
        type: String,
        required: false,
        restrict: /^mongodb+$/
      },
      username: {
        type: String,
        required: false
      },
      password: {
        type: String,
        required: false
      },
      host: {
        type: String,
        required: true,
        restrict: /^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])|(([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-]*[a-zA-Z0-9])\.)*([A-Za-z0-9]|[A-Za-z0-9][A-Za-z0-9\-]*[A-Za-z0-9])$/
      },
      port: {
        type: Number,
        required: false,
        restrict: /^[0-9]{4,5}$/
      },
      replicas: {
        type: Array,
        required: false
      },
      database: {
        type: String,
        required: false
      },
      options: {
        type: DSNOptions,
        required: false
      }
    };
    for (key in oTypes) {
      value = oTypes[key];
      if (dsn[key] != null) {
        if (!value.type) {
          this.emit('error', "Validator for DSN::'" + key + "' was missing type param");
        }
        if (!(ObjUtils.isOfType(dsn[key], value.type))) {
          this.emit('error', "" + key + " was expected to be " + (Util.getFunctionName(value.type)) + ". Type was '" + (typeof dsn[key]) + "'");
        }
        if (value.restrict && !(("" + dsn[key]).match(value.restrict))) {
          this.emit('error', "" + key + " was malformed");
        }
      } else if (value.required) {
        return this.emit('error', "" + key + " was required but not defined");
      }
    }
    if (options) {
      try {
        options = new DSNOptions(options);
      } catch (_error) {
        e = _error;
        this.emit('error', e);
      }
    }
    return true;
  };

  DSN.prototype.toJSON = function() {
    return this.__dsn;
  };

  DSN.prototype.toDSN = function() {
    var userPass;
    userPass = "" + (this.__dsn.username || '') + (this.__dsn.username && this.__dsn.password ? ':' + this.__dsn.password : '');
    return "" + (this.__dsn.protocol || 'mongodb') + "://" + userPass + (userPass.length ? '@' : '') + this.__dsn.host + ":" + (this.__dsn.port || '27017') + "/" + (this.__dsn.database || '') + (this.__dsn.options ? '?' + this.__dsn.options : '');
  };

  DSN.prototype.toString = function() {
    return this.toDSN();
  };

  return DSN;

})(EventEmitter);

DSN.loadConfig = function(path, callback) {
  var config, data, dsn, e, env;
  try {
    data = fs.readFileSync(path);
  } catch (_error) {
    e = _error;
  }
  try {
    config = JSON.parse(data);
  } catch (_error) {
    e = _error;
  }
  try {
    dsn = new DSN(config.hasOwnProperty((env = process.env.NODE_ENV || 'development')) ? config[env] : config);
  } catch (_error) {
    e = _error;
  }
  return callback != null ? callback.apply(this, e != null ? [e, null] : [null, dsn]) : void 0;
};

module.exports = DSN;