loopback-datasource-juggler/lib/railway.js

216 lines
5.9 KiB
JavaScript
Raw Normal View History

2012-03-27 14:22:24 +00:00
var fs = require('fs');
var path = require('path');
2012-10-28 23:21:51 +00:00
var Schema = require('./schema').Schema;
2012-03-27 14:22:24 +00:00
var existsSync = fs.existsSync || path.existsSync;
2012-10-28 23:21:51 +00:00
if (global.railway) {
railway.orm._schemas = [];
}
2012-03-27 14:22:24 +00:00
2012-05-02 22:15:09 +00:00
module.exports = function init(root) {
2012-10-29 14:11:43 +00:00
var railway, app, models;
2012-10-28 23:21:51 +00:00
if (typeof root !== 'object' || root.constructor.name !== 'Railway') {
railway = global.railway;
app = global.app;
2012-10-29 14:11:43 +00:00
models = app.models;
2012-10-28 23:21:51 +00:00
} else {
railway = root;
app = railway.app;
root = railway.root;
2012-10-29 14:11:43 +00:00
models = railway.models;
2012-10-28 23:21:51 +00:00
}
2012-05-02 22:15:09 +00:00
2012-10-28 23:21:51 +00:00
railway.orm._schemas = [];
2012-03-27 19:48:23 +00:00
2012-10-28 23:21:51 +00:00
var confFile = (root || app.root) + '/config/database';
var config;
if (existsSync(confFile + '.json')) {
try {
config = JSON.parse(fs.readFileSync(confFile + '.json', 'utf-8'))[app.set('env')];
} catch (e) {
console.log('Could not parse config/database.json');
throw e;
}
} else if (existsSync(confFile + '.yml')) {
try {
config = railway.utils.readYaml(confFile + '.yml')[app.set('env')];
} catch (e) {
console.log('Could not parse config/database.yml');
throw e;
}
2012-10-28 23:21:51 +00:00
config = {};
}
// when driver name started with point - look for driver in app root (relative path)
if (config.driver && config.driver.match(/^\./)) {
config.driver = path.join(app.root, config.driver);
2012-03-27 19:48:23 +00:00
}
2012-03-27 14:22:24 +00:00
2012-10-28 23:21:51 +00:00
var schema = new Schema(config && config.driver || 'memory', config);
schema.log = log;
railway.orm._schemas.push(schema);
2012-03-27 14:22:24 +00:00
var context = prepareContext(models, railway, app, schema);
2012-03-27 14:22:24 +00:00
// run schema first
2012-05-02 22:15:09 +00:00
var schemaFile = (root || app.root) + '/db/schema.';
if (existsSync(schemaFile + 'js')) {
2012-03-27 14:22:24 +00:00
schemaFile += 'js';
} else if (existsSync(schemaFile + 'coffee')) {
2012-03-27 14:22:24 +00:00
schemaFile += 'coffee';
2012-03-27 19:48:23 +00:00
} else {
schemaFile = false;
}
if (schemaFile) {
var code = fs.readFileSync(schemaFile).toString();
2012-10-28 23:21:51 +00:00
if (schemaFile.match(/\.coffee$/)) {
code = require('coffee-script').compile(code);
}
var fn = new Function('context', 'require', 'with(context){(function(){' + code + '})()}');
fn(context, require);
2012-03-27 14:22:24 +00:00
}
// and freeze schemas
railway.orm._schemas.forEach(function (schema) {
schema.freeze();
});
// check validations and display warning
var displayWarning = false;
2012-10-29 14:11:43 +00:00
Object.keys(models).forEach(function (model) {
var Model = models[model];
if (Model._validations) {
displayWarning = true;
}
});
if (displayWarning) {
var $ = railway.utils.stylize.$;
require('util').puts($('WARNING:').bold.red + ' ' + $('I can see that you\'ve added validation to db/schema.js. However schema.js file is only used to describe database schema. Therefore validations configured in db/schema.js will be ignored.\nFor business logic (incl. validations) please create models as separate .js files here: app/models/*.js').yellow);
}
2012-03-27 14:22:24 +00:00
function log(str, startTime) {
2012-04-07 13:43:15 +00:00
var $ = railway.utils.stylize.$;
2012-03-27 14:22:24 +00:00
var m = Date.now() - startTime;
2012-04-07 13:43:15 +00:00
railway.utils.debug(str + $(' [' + (m < 10 ? m : $(m).red) + ' ms]').bold);
2012-03-27 14:22:24 +00:00
app.emit('app-event', {
type: 'query',
param: str,
time: m
});
}
function prepareContext(models, railway, app, defSchema, done) {
2012-03-27 14:22:24 +00:00
var ctx = {app: app},
_models = {},
2012-03-27 14:22:24 +00:00
settings = {},
cname,
schema,
wait = connected = 0,
nonJugglingSchema = false;
done = done || function () {};
/**
* Multiple schemas support
* example:
* schema('redis', {url:'...'}, function () {
* describe models using redis connection
* ...
* });
* schema(function () {
* describe models stored in memory
* ...
* });
*/
ctx.schema = function () {
var name = argument('string');
var opts = argument('object') || {};
var def = argument('function') || function () {};
schema = new Schema(name || opts.driver || 'memory', opts);
railway.orm._schemas.push(schema);
wait += 1;
ctx.gotSchema = true;
schema.on('log', log);
schema.on('connected', function () {
if (wait === ++connected) done();
});
def();
schema = false;
};
/**
* Use custom schema driver
*/
ctx.customSchema = function () {
var def = argument('function') || function () {};
nonJugglingSchema = true;
def();
Object.keys(ctx.exports).forEach(function (m) {
ctx.define(m, ctx.exports[m]);
});
nonJugglingSchema = false;
};
ctx.exports = {};
ctx.module = { exports: ctx.exports };
/**
* Define a class in current schema
*/
ctx.describe = ctx.define = function (className, callback) {
var m;
cname = className;
2012-10-29 14:11:43 +00:00
_models[cname] = {};
2012-03-27 14:22:24 +00:00
settings[cname] = {};
if (nonJugglingSchema) {
m = callback;
} else {
callback && callback();
2012-10-29 14:11:43 +00:00
m = (schema || defSchema).define(className, _models[cname], settings[cname]);
2012-03-27 14:22:24 +00:00
}
if (global.railway) {
global[cname] = m;
}
2012-10-29 14:11:43 +00:00
return models[cname] = ctx[cname] = m;
2012-03-27 14:22:24 +00:00
};
/**
* Define a property in current class
*/
ctx.property = function (name, type, params) {
if (!params) params = {};
if (typeof type !== 'function' && typeof type === 'object') {
params = type;
type = String;
}
params.type = type || String;
2012-10-29 14:11:43 +00:00
_models[cname][name] = params;
2012-03-27 14:22:24 +00:00
};
/**
* Set custom table name for current class
* @param name - name of table
*/
ctx.setTableName = function (name) {
if (cname) settings[cname].table = name;
};
ctx.Text = Schema.Text;
return ctx;
function argument(type) {
var r;
[].forEach.call(arguments.callee.caller.arguments, function (a) {
if (!r && typeof a === type) r = a;
});
return r;
}
}
2012-05-02 22:15:09 +00:00
};