iframe-proxy | Sunbelt Computer Software

172 lines (150 loc) · 4.08 KB
var common      = exports;
var path        = require('path');
var async       = require('async');
var _           = require('lodash');
var util        = require('util');
var querystring = require('querystring');
var Semver      = require('semver');
var ORM         = require('../');
common.ORM = ORM;
common.protocol = function () {
  return process.env.ORM_PROTOCOL;
common.isTravis = function() {
  return Boolean(process.env.CI);
common.createConnection = function(opts, cb) {
  ORM.connect(this.getConnectionString(opts), cb);
common.hasConfig = function (proto) {
  var config;
  if (common.isTravis()) return 'found';
    config = require("./config");
  } catch (ex) {
    return 'not-found';
  return (config.hasOwnProperty(proto) ? 'found' : 'not-defined');
common.getConfig = function () {
  if (common.isTravis()) {
    switch (this.protocol()) {
      case 'mysql':
        return { user: "root", host: "localhost", database: "orm_test" };
      case 'postgres':
      case 'redshift':
        return { user: "postgres", host: "localhost", database: "orm_test" };
      case 'sqlite':
        return {};
      case 'mongodb':
        return { host: "localhost", database: "test" };
      default:
        throw new Error("Unknown protocol");
    var config = require("./config")[this.protocol()];
    if (typeof config == "string") {
      config = require("url").parse(config);
    if (config.hasOwnProperty("auth")) {
      if (config.auth.indexOf(":") >= 0) {
        config.user = config.auth.substr(0, config.auth.indexOf(":"));
        config.password = config.auth.substr(config.auth.indexOf(":") + 1);
      } else {
        config.user = config.auth;
        config.password = "";
    if (config.hostname) {
      config.host = config.hostname;
    return config;
common.getConnectionString = function (opts) {
  var config   = this.getConfig();
  var protocol = this.protocol();
  var query;
  _.defaults(config, {
    user     : { postgres: 'postgres', redshift: 'postgres', mongodb: '' }[protocol] || 'root',
    database : { mongodb:  'test'     }[protocol] || 'orm_test',
    password : '',
    host     : 'localhost',
    pathname : '',
    query    : {}
  _.merge(config, opts || {});
  query = querystring.stringify(config.query);
  switch (protocol) {
    case 'mysql':
    case 'postgres':
    case 'redshift':
    case 'mongodb':
      if (common.isTravis()) {
        if (protocol == 'redshift') protocol = 'postgres';
        return util.format("%s://%s@%s/%s?%s",
          protocol, config.user, config.host, config.database, query
      } else {
        return util.format("%s://%s:%s@%s/%s?%s",
          protocol, config.user, config.password,
          config.host, config.database, query
        ).replace(':@','@');
    case 'sqlite':
      return util.format("%s://%s?%s", protocol, config.pathname, query);
    default:
      throw new Error("Unknown protocol " + protocol);
common.retry = function (before, run, until, done, args) {
  if (typeof until === "number") {
    var countDown = until;
    until = function (err) {
      if (err && --countDown > 0) return false;
      return true;
  if (typeof args === "undefined") args = [];
  var handler = function (err) {
    if (until(err)) return done.apply(this, arguments);
    return runNext();
  args.push(handler);
  var runCurrent = function () {
    if (run.length == args.length) {
      return run.apply(this, args);
    } else {
      run.apply(this, args);
      handler();
  var runNext = function () {
      if (before.length > 0) {
        before(function (err) {
          if (until(err)) return done(err);
          return runCurrent();
        });
      } else {
        before();
        runCurrent();
    catch (e) {
      handler(e);
  if (before.length > 0) {
    before(function (err) {
      if (err) return done(err);
      runNext();
    before();
    runNext();
common.nodeVersion = function () {
  return new Semver(process.versions.node);
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

common.js

Latest commit

History

common.js

File metadata and controls
