var safeRequire = require('../utils').safeRequire; /** * Module dependencies */ var cradle = safeRequire('cradle'); /** * Private functions for internal use */ function CradleAdapter(client) { this._models = {}; this.client = client; } function createdbif(client, callback) { client.exists(function (err, exists) { if(err) callback(err); if (!exists) { client.create(function() { callback(); }); } else { callback(); } }); } function naturalize(data, model) { data.nature = model; //TODO: maybe this is not a really good idea if(data.date) data.date = data.date.toString(); return data; } function idealize(data) { data.id = data._id; return data; } function stringify(data) { return data ? data.toString() : data } function errorHandler(callback, func) { return function(err, res) { if (err) { console.log('cradle', err); callback(err); } else { if(func) { func(res, function(res) { callback(null, res); }); } else { callback(null, res); } } } }; function synchronize(functions, args, callback) { if(functions.length === 0) callback(); if(functions.length > 0 && args.length === functions.length) { functions[0](args[0][0], args[0][1], function(err, res) { if(err) callback(err); functions.splice(0, 1); args.splice(0, 1); synchronize(functions, args, callback); }); } }; function applyFilter(filter) { if (typeof filter.where === 'function') { return filter.where; } var keys = Object.keys(filter.where); return function (obj) { var pass = true; keys.forEach(function (key) { if (!test(filter.where[key], obj[key])) { pass = false; } }); return pass; } function test(example, value) { if (typeof value === 'string' && example && example.constructor.name === 'RegExp') { return value.match(example); } // not strict equality return example == value; } } function numerically(a, b) { return a[this[0]] - b[this[0]]; } function literally(a, b) { return a[this[0]] > b[this[0]]; } function filtering(res, model, filter, instance) { if(model) { if(filter == null) filter = {}; if(filter.where == null) filter.where = {}; filter.where.nature = model; } // do we need some filtration? if (filter.where) { res = res ? res.filter(applyFilter(filter)) : res; } // do we need some sorting? if (filter.order) { var props = instance[model].properties; var allNumeric = true; var orders = filter.order; var reverse = false; if (typeof filter.order === "string") { orders = [filter.order]; } orders.forEach(function (key, i) { var m = key.match(/\s+(A|DE)SC$/i); if (m) { key = key.replace(/\s+(A|DE)SC/i, ''); if (m[1] === 'DE') reverse = true; } orders[i] = key; if (props[key].type.name !== 'Number') { allNumeric = false; } }); if (allNumeric) { res = res.sort(numerically.bind(orders)); } else { res = res.sort(literally.bind(orders)); } if (reverse) res = res.reverse(); } return res; } /** * Connection/Disconnection */ exports.initialize = function(schema, callback) { if (!cradle) return; // when using cradle if we dont wait for the schema to be connected, the models fails to load correctly. schema.waitForConnect = true; if (!schema.settings.url) { var host = schema.settings.host || 'localhost'; var port = schema.settings.port || '5984'; var options = schema.settings.options || { cache: true, raw: false }; if (schema.settings.username) { options.auth = {}; options.auth.username = schema.settings.username; if (schema.settings.password) { options.auth.password = schema.settings.password; } } var database = schema.settings.database || 'loopback-data'; schema.settings.host = host; schema.settings.port = port; schema.settings.database = database; schema.settings.options = options; } schema.client = new(cradle.Connection)(schema.settings.host, schema.settings.port,schema.settings.options).database(schema.settings.database); createdbif( schema.client, errorHandler(callback, function() { schema.adapter = new CradleAdapter(schema.client); process.nextTick(callback); })); }; CradleAdapter.prototype.disconnect = function() { }; /** * Write methods */ CradleAdapter.prototype.define = function(descr) { this._models[descr.model.modelName] = descr; }; CradleAdapter.prototype.create = function(model, data, callback) { this.client.save( stringify(data.id), naturalize(data, model), errorHandler(callback, function(res, cb) { cb(res.id); }) ); }; CradleAdapter.prototype.save = function(model, data, callback) { this.client.save( stringify(data.id), naturalize(data, model), errorHandler(callback) ) }; CradleAdapter.prototype.updateAttributes = function(model, id, data, callback) { this.client.merge( stringify(id), data, errorHandler(callback, function(doc, cb) { cb(idealize(doc)); }) ); }; CradleAdapter.prototype.updateOrCreate = function(model, data, callback) { this.client.get( stringify(data.id), function (err, doc) { if(err) { this.create(model, data, callback); } else { this.updateAttributes(model, data.id, data, callback); } }.bind(this) ) }; /** * Read methods */ CradleAdapter.prototype.exists = function(model, id, callback) { this.client.get( stringify(id), errorHandler(callback, function(doc, cb) { cb(!!doc); }) ); }; CradleAdapter.prototype.find = function(model, id, callback) { this.client.get( stringify(id), errorHandler(callback, function(doc, cb) { cb(idealize(doc)); }) ); }; CradleAdapter.prototype.count = function(model, callback, where) { this.models( model, {where: where}, callback, function(docs, cb) { cb(docs.length); } ); }; CradleAdapter.prototype.models = function(model, filter, callback, func) { var limit = 200; var skip = 0; if (filter != null) { limit = filter.limit || limit; skip = filter.skip ||skip; } var self = this; self.client.save('_design/'+model, { views : { all : { map : 'function(doc) { if (doc.nature == "'+model+'") { emit(doc._id, doc); } }' } } }, function() { self.client.view(model+'/all', {include_docs:true, limit:limit, skip:skip}, errorHandler(callback, function(res, cb) { var docs = res.map(function(doc) { return idealize(doc); }); var filtered = filtering(docs, model, filter, this._models) func ? func(filtered, cb) : cb(filtered); }.bind(self))); }); }; CradleAdapter.prototype.all = function(model, filter, callback) { this.models( model, filter, callback ); }; /** * Detroy methods */ CradleAdapter.prototype.destroy = function(model, id, callback) { this.client.remove( stringify(id), function (err, doc) { callback(err); } ); }; CradleAdapter.prototype.destroyAll = function(model, callback) { this.models( model, null, callback, function(docs, cb) { var docIds = docs.map(function(doc) { return doc.id; }); this.client.get(docIds, function(err, res) { if(err) cb(err); var funcs = res.map(function(doc) { return this.client.remove.bind(this.client); }.bind(this)); var args = res.map(function(doc) { return [doc._id, doc._rev]; }); synchronize(funcs, args, cb); }.bind(this)); }.bind(this) ); };