tingodb/lib/updater.js

Embedded Node.js database upward compatible with MongoDB

var _ = require('lodash');

function ensurePath(obj,k,cb) {
	var path = k.split(".");
	var t = null;
	if (path.length==1)
		t = obj;
	else {
		var l = obj;
		for (var i=0; i<path.length-1; i++) {
			var p = path[i];
			if (!l[p])
				l[p] = {};
			l = l[p];
		}
		t = l;
		k = path[i];
	}
	cb(t,k);
}

function applySet(obj,$set) {
	_.each($set, function (v,k) {
		_.set(obj,k,v);
	});
}

function applyUnset(obj,$unset) {
	_.each($unset, function (v,k) {
		_.unset(obj,k);
	});
}

function applyInc(obj, $inc) {
	_.each($inc, function (v, k) {
		ensurePath(obj,k, function (t,k) {
			if (!t[k]) t[k] = 0;
			if (!_.isFinite(t[k]))
				throw new Error("Cannot apply $inc modifier to non-number");
			t[k] += v;
		});
	});
}

function applyPush(obj,$push) {
	_.each($push, function (v,k) {
		ensurePath(obj,k, function (t,k) {
			if (!t[k]) {
				t[k] = v.$each?v.$each:[v];
			} else {
				if (!_.isArray(t[k]))
					throw new Error("Cannot apply $push/$pushAll modifier to non-array");

				if (v.$each) {
					t[k] = t[k].concat(v.$each);
				} else
					t[k].push(v);
			}
		});
	});
}

function applyPop(obj,$op) {
	_.each($op, function (v,k) {
		var path = k.split("."); var t = obj;
		for (var i=0; i<path.length-1 && t[path[i]]; t=t[path[i++]]); k = path[i];
		if (t==null || t[k]==null) return;
		if (_.isArray(t[k])) {
			if (v>=0)
				t[k]=t[k].slice(0,-1);
			else if (v==-1)
				t[k]=t[k].slice(1);
		} else throw new Error("Cannot apply $pop modifier to non-array");
	});
}

function applyPull(obj,$op,tdb) {
	_.each($op, function (v,k) {
		var path = k.split("."); var t = obj;
		for (var i=0; i<path.length-1 && t[path[i]]; t=t[path[i++]]); k = path[i];
		if (t==null || t[k]==null) return;
		if (_.isArray(t[k])) {
			var qt = tdb.Finder.matcher({v:v});
			var matcher = new Function("obj", "return " + (qt.native()) );
			t[k] = _.reject(t[k], function (obj) { return matcher({v:obj}); });
		} else throw new Error("Cannot apply $pull/$pullAll modifier to non-array");
	});
}

function applyPullAll(obj,$op) {
	_.each($op, function (v,k) {
		var path = k.split("."); var t = obj;
		for (var i=0; i<path.length-1 && t[path[i]]; t=t[path[i++]]); k = path[i];
		if (t==null || t[k]==null) return;
		if (_.isArray(t[k])) {
			t[k] = _.difference(t[k], v);
		} else throw new Error("Cannot apply $pull/$pullAll modifier to non-array");
	});
}

function applyRename(obj,$op) {
	_.each($op, function (v,k) {
		var path = k.split("."); var t = obj;
		for (var i=0; i<path.length-1 && t[path[i]]; t=t[path[i++]]); k = path[i];
		if (t==null || t[k]==null) return;
		ensurePath(obj,v, function (t1,k1) {
			t1[k1] = t[k];
			delete t[k];
		});
	});
}

function applyAddToSet(obj,$op) {
	_.each($op, function (v,k) {
		ensurePath(obj,k, function (t,k) {
			if (!t[k]) {
				t[k] = v.$each?v.$each:[v];
			} else {
				if (!_.isArray(t[k]))
					throw new Error("Cannot apply $addToSet modifier to non-array");

				if (v.$each) {
					t[k] = _.union(t[k], v.$each);
				} else {
					if (!_.includes(t[k],v))
						t[k].push(v);
				}
			}
		});
	});
}

function applyPushAll(obj,$pushAll) {
	_.each($pushAll, function (v,k) {
		ensurePath(obj,k, function (t,k) {
			if (!t[k]) {
				t[k] = v;
			} else {
				if (!_.isArray(t[k]))
					throw new Error("Cannot apply $push/$pushAll modifier to non-array");

				t[k] = t[k].concat(v);
			}
		});
	});
}

function updater(op,tdb) {
	this.hasAtomic = function () {
		return _.findKey(op, function (v, k) { return k[0] === "$"; }) != null;
	};

	this.update = function ($doc,upsert) {
		if (op.$set)
			applySet($doc,op.$set);
		if (op.$unset)
			applyUnset($doc,op.$unset);
		if (op.$inc)
			applyInc($doc, op.$inc);
		if (op.$push)
			applyPush($doc, op.$push);
		if (op.$pushAll)
			applyPushAll($doc, op.$pushAll);
		if (op.$addToSet)
			applyAddToSet($doc, op.$addToSet);
		if (op.$pop)
			applyPop($doc, op.$pop);
		if (op.$pull)
			applyPull($doc, op.$pull, tdb);
		if (op.$pullAll)
			applyPullAll($doc, op.$pullAll);
		if (op.$rename)
			applyRename($doc, op.$rename);
		if (upsert && op.$setOnInsert)
			applySet($doc,op.$setOnInsert);
	};
}

module.exports = updater;