iframe-proxy | Sunbelt Computer Software

167 lines (146 loc) · 3.68 KB
var common      = exports;
var path        = require('path');
var async       = require('async');
var _           = require('lodash');
var util        = require('util');
var querystring = require('querystring');
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);
			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);
			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();
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
