mingo
Version:
MongoDB query language for in-memory objects
200 lines (199 loc) • 6.34 kB
JavaScript
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var predicates_exports = {};
__export(predicates_exports, {
$all: () => $all,
$elemMatch: () => $elemMatch,
$eq: () => $eq,
$gt: () => $gt,
$gte: () => $gte,
$in: () => $in,
$lt: () => $lt,
$lte: () => $lte,
$mod: () => $mod,
$ne: () => $ne,
$nin: () => $nin,
$regex: () => $regex,
$size: () => $size,
$type: () => $type,
processExpression: () => processExpression,
processQuery: () => processQuery
});
module.exports = __toCommonJS(predicates_exports);
var import_core = require("../core");
var import_internal = require("../query/_internal");
var import_util = require("../util");
function processQuery(selector, value, options, predicate) {
const opts = { unwrapArray: true };
const depth = Math.max(1, selector.split(".").length - 1);
return (o) => {
const lhs = (0, import_util.resolve)(o, selector, opts);
return predicate(lhs, value, { ...options, depth });
};
}
function processExpression(obj, expr, options, predicate) {
const args = (0, import_core.computeValue)(obj, expr, null, options);
return predicate(...args);
}
function $eq(a, b, options) {
if ((0, import_util.isEqual)(a, b)) return true;
if ((0, import_util.isNil)(a) && (0, import_util.isNil)(b)) return true;
if ((0, import_util.isArray)(a)) {
return a.some((v) => (0, import_util.isEqual)(v, b)) || (0, import_util.flatten)(a, options?.depth).some((v) => (0, import_util.isEqual)(v, b));
}
return false;
}
function $ne(a, b, options) {
return !$eq(a, b, options);
}
function $in(a, b, options) {
if ((0, import_util.isNil)(a)) return b.some((v) => v === null);
return (0, import_util.intersection)([(0, import_util.ensureArray)(a), b], options?.hashFunction).length > 0;
}
function $nin(a, b, options) {
return !$in(a, b, options);
}
function $lt(a, b, _options) {
return compare(a, b, (x, y) => (0, import_util.compare)(x, y) < 0);
}
function $lte(a, b, _options) {
return compare(a, b, (x, y) => (0, import_util.compare)(x, y) <= 0);
}
function $gt(a, b, _options) {
return compare(a, b, (x, y) => (0, import_util.compare)(x, y) > 0);
}
function $gte(a, b, _options) {
return compare(a, b, (x, y) => (0, import_util.compare)(x, y) >= 0);
}
function $mod(a, b, _options) {
return (0, import_util.ensureArray)(a).some(
((x) => b.length === 2 && x % b[0] === b[1])
);
}
function $regex(a, b, options) {
const lhs = (0, import_util.ensureArray)(a);
const match = (x) => (0, import_util.isString)(x) && (0, import_util.truthy)(b.exec(x), options?.useStrictMode);
return lhs.some(match) || (0, import_util.flatten)(lhs, 1).some(match);
}
function $all(values, queries, options) {
if (!(0, import_util.isArray)(values) || !(0, import_util.isArray)(queries) || !values.length || !queries.length) {
return false;
}
let matched = true;
for (const query of queries) {
if (!matched) break;
if ((0, import_util.isObject)(query) && Object.keys(query).includes("$elemMatch")) {
matched = $elemMatch(values, query["$elemMatch"], options);
} else if ((0, import_util.isRegExp)(query)) {
matched = values.some((s) => typeof s === "string" && query.test(s));
} else {
matched = values.some((v) => (0, import_util.isEqual)(query, v));
}
}
return matched;
}
function $size(a, b, _options) {
return Array.isArray(a) && a.length === b;
}
function isNonBooleanOperator(name) {
return (0, import_util.isOperator)(name) && ["$and", "$or", "$nor"].indexOf(name) === -1;
}
function $elemMatch(a, b, options) {
if ((0, import_util.isArray)(a) && !(0, import_util.isEmpty)(a)) {
let format = (x) => x;
let criteria = b;
if (Object.keys(b).every(isNonBooleanOperator)) {
criteria = { temp: b };
format = (x) => ({ temp: x });
}
const query = new import_internal.QueryImpl(criteria, options);
for (let i = 0, len = a.length; i < len; i++) {
if (query.test(format(a[i]))) {
return true;
}
}
}
return false;
}
const isNull = (a) => a === null;
const compareFuncs = {
array: import_util.isArray,
boolean: import_util.isBoolean,
bool: import_util.isBoolean,
date: import_util.isDate,
number: import_util.isNumber,
int: import_util.isNumber,
long: import_util.isNumber,
double: import_util.isNumber,
decimal: import_util.isNumber,
null: isNull,
object: import_util.isObject,
regexp: import_util.isRegExp,
regex: import_util.isRegExp,
string: import_util.isString,
// added for completeness
undefined: import_util.isNil,
// deprecated
// Mongo identifiers
1: import_util.isNumber,
//double
2: import_util.isString,
3: import_util.isObject,
4: import_util.isArray,
6: import_util.isNil,
// deprecated
8: import_util.isBoolean,
9: import_util.isDate,
10: isNull,
11: import_util.isRegExp,
16: import_util.isNumber,
//int
18: import_util.isNumber,
//long
19: import_util.isNumber
//decimal
};
function compareType(a, b, _) {
const f = compareFuncs[b];
return f ? f(a) : false;
}
function $type(a, b, options) {
return (0, import_util.isArray)(b) ? b.findIndex((t) => compareType(a, t, options)) >= 0 : compareType(a, b, options);
}
function compare(a, b, f) {
return (0, import_util.ensureArray)(a).some((x) => (0, import_util.typeOf)(x) === (0, import_util.typeOf)(b) && f(x, b));
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
$all,
$elemMatch,
$eq,
$gt,
$gte,
$in,
$lt,
$lte,
$mod,
$ne,
$nin,
$regex,
$size,
$type,
processExpression,
processQuery
});