mongoe
Version:
MongoDB driver with relational functionalities
147 lines (146 loc) • 7.76 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator = (this && this.__generator) || function (thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (_) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.verifyDelete = exports.verifyUpdate = exports.verifyInsert = exports.verifyIntegrity = exports.registerRelations = exports.Policy = void 0;
var _1 = require(".");
var Policy;
(function (Policy) {
Policy[Policy["ByPass"] = 0] = "ByPass";
Policy[Policy["Delete"] = 1] = "Delete";
Policy[Policy["Reject"] = 2] = "Reject";
Policy[Policy["Nullify"] = 3] = "Nullify";
/*Unset,
Pull*/
})(Policy = exports.Policy || (exports.Policy = {}));
function registerRelations(relations, inputs) {
var _a, _b;
for (var _i = 0, _c = Object.entries(inputs); _i < _c.length; _i++) {
var _d = _c[_i], collection = _d[0], relation = _d[1];
if (!relations[collection])
relations[collection] = {
primaryKey: (_a = relation.primaryKey) !== null && _a !== void 0 ? _a : "_id",
foreignRefs: []
};
else if (relation.primaryKey)
relations[collection].primaryKey = relation.primaryKey;
for (var _e = 0, _f = Object.entries((_b = relation.foreignKeys) !== null && _b !== void 0 ? _b : {}); _e < _f.length; _e++) {
var _g = _f[_e], foreignKey = _g[0], _h = _g[1], foreignCollection = _h[0], policy = _h[1];
var foreignRef = [collection, foreignKey, policy];
if (!relations[foreignCollection])
relations[foreignCollection] = {
primaryKey: "_id",
foreignRefs: [foreignRef]
};
else
relations[foreignCollection].foreignRefs.push(foreignRef);
}
}
}
exports.registerRelations = registerRelations;
function verifyIntegrity(database) {
return __awaiter(this, void 0, void 0, function () {
return __generator(this, function (_a) {
return [2 /*return*/];
});
});
}
exports.verifyIntegrity = verifyIntegrity;
function verifyInsert(collection, docs) {
return __awaiter(this, void 0, void 0, function () {
return __generator(this, function (_a) {
return [2 /*return*/];
});
});
}
exports.verifyInsert = verifyInsert;
function verifyUpdate(collection, filter, update, options) {
return __awaiter(this, void 0, void 0, function () {
return __generator(this, function (_a) {
return [2 /*return*/];
});
});
}
exports.verifyUpdate = verifyUpdate;
function verifyDelete(collection, filter, deletedKeys, options) {
return __awaiter(this, void 0, void 0, function () {
var alreadyDeleted, _a, primaryKey, foreignRefs, keys, _i, foreignRefs_1, _b, foreignCollectionName, foreignKey, policy, foreignCollection, _c;
var _d, _e, _f, _g, _h;
return __generator(this, function (_j) {
switch (_j.label) {
case 0: return [2 /*return*/];
case 1:
keys = _j.sent();
alreadyDeleted.push.apply(alreadyDeleted, keys);
console.log("Want to remove", keys, "in", collection.name);
console.log("State of deletedKeys:", deletedKeys, "\n");
_i = 0, foreignRefs_1 = foreignRefs;
_j.label = 2;
case 2:
if (!(_i < foreignRefs_1.length)) return [3 /*break*/, 11];
_b = foreignRefs_1[_i], foreignCollectionName = _b[0], foreignKey = _b[1], policy = _b[2];
foreignCollection = collection.database.collection(foreignCollectionName);
return [4 /*yield*/, foreignCollection.countDocuments((_e = {}, _e[foreignKey] = { $in: keys }, _e))];
case 3:
if (!_j.sent()) return [3 /*break*/, 10];
_c = policy;
switch (_c) {
case Policy.ByPass: return [3 /*break*/, 4];
case Policy.Reject: return [3 /*break*/, 5];
case Policy.Delete: return [3 /*break*/, 6];
case Policy.Nullify: return [3 /*break*/, 8];
}
return [3 /*break*/, 10];
case 4: return [3 /*break*/, 10];
case 5: throw new Error("Mongoe: Deletion rejected by foreign key <" + foreignKey + "> on collection <" + foreignCollectionName + ">");
case 6: return [4 /*yield*/, foreignCollection.deleteMany((_f = {}, _f[foreignKey] = { $in: keys }, _f), undefined, deletedKeys)];
case 7:
_j.sent();
return [3 /*break*/, 10];
case 8: return [4 /*yield*/, foreignCollection.updateMany((_g = {}, _g[foreignKey] = { $in: keys }, _g), { $set: (_h = {}, _h[foreignKey] = null, _h) })];
case 9:
_j.sent();
_j.label = 10;
case 10:
_i++;
return [3 /*break*/, 2];
case 11: return [2 /*return*/];
}
});
});
}
exports.verifyDelete = verifyDelete;