mongoose-killer 0.1.0

package/dist/esm/populate.js

@@ -0,0 +1,180 @@
+export const normalizePopulateArg = (arg) => {
+    if (!arg)
+        return [];
+    if (typeof arg === "string") {
+        // Mongoose accepts "a b c" → three populates.
+        return arg
+            .split(/\s+/)
+            .filter(Boolean)
+            .map((path) => ({ path }));
+    }
+    if (Array.isArray(arg)) {
+        return arg.flatMap(normalizePopulateArg);
+    }
+    if (typeof arg === "object") {
+        const { path, select, match, populate } = arg;
+        return [
+            {
+                path,
+                select,
+                match,
+                populate: populate ? normalizePopulateArg(populate) : undefined,
+            },
+        ];
+    }
+    return [];
+};
+export const runPopulate = async (docs, specs, opts) => {
+    for (const spec of specs) {
+        await applySpec(docs, spec, opts);
+    }
+    return docs;
+};
+const applySpec = async (docs, spec, opts) => {
+    const cfg = opts.ownerModel.populates[spec.path];
+    if (!cfg) {
+        // No mapping configured for this path → nothing to do. Matches
+        // mongoose's behavior of silently skipping unknown refs.
+        return;
+    }
+    const refModelName = opts.ownerModel.collectionToModelName.get(cfg.collection);
+    if (!refModelName)
+        return;
+    const refModel = opts.ownerModel.getModelByName(refModelName);
+    const isArrayPath = spec.path.includes(".");
+    if (isArrayPath) {
+        await populateArrayPath(docs, spec, refModel, opts);
+    }
+    else {
+        await populateScalarPath(docs, spec, refModel, opts);
+    }
+};
+const populateScalarPath = async (docs, spec, refModel, opts) => {
+    const ids = uniqueIds(docs.map((d) => d?.[spec.path]).filter(isPresent));
+    if (ids.length === 0)
+        return;
+    const filter = { _id: { $in: ids } };
+    if (spec.match)
+        Object.assign(filter, spec.match);
+    const findOpts = {};
+    if (spec.select)
+        findOpts.projection = normalizeProjection(spec.select);
+    if (opts.session)
+        findOpts.session = opts.session;
+    const fetched = await refModel.collection.find(filter, findOpts).toArray();
+    if (spec.populate) {
+        await runPopulate(fetched, spec.populate, {
+            ownerModel: refModel,
+            session: opts.session,
+        });
+    }
+    const byId = new Map();
+    for (const f of fetched)
+        byId.set(idKey(f._id), f);
+    // Mongoose semantics: if a match filter is present, refs that don't
+    // satisfy it get nulled out (the parent doc still exists, but the
+    // populated field becomes null). Without a match, leave non-matches
+    // alone (the raw ref id stays, matching mongoose's behavior when a
+    // referenced doc has been deleted).
+    const nullifyMisses = !!spec.match;
+    for (const d of docs) {
+        const raw = d?.[spec.path];
+        if (!isPresent(raw))
+            continue;
+        const hit = byId.get(idKey(raw));
+        if (hit)
+            d[spec.path] = hit;
+        else if (nullifyMisses)
+            d[spec.path] = null;
+    }
+};
+// "tags._id" — parent has an array of subdocs, each with an _id
+// field that's a ref. After populate, each subdoc's _id is replaced
+// with the full referenced doc.
+const populateArrayPath = async (docs, spec, refModel, opts) => {
+    const [parent, child] = spec.path.split(".");
+    const allIds = [];
+    for (const d of docs) {
+        const arr = d?.[parent];
+        if (!Array.isArray(arr))
+            continue;
+        for (const sub of arr) {
+            const v = sub?.[child];
+            if (isPresent(v))
+                allIds.push(v);
+        }
+    }
+    const ids = uniqueIds(allIds);
+    if (ids.length === 0)
+        return;
+    const filter = { _id: { $in: ids } };
+    if (spec.match)
+        Object.assign(filter, spec.match);
+    const findOpts = {};
+    if (spec.select)
+        findOpts.projection = normalizeProjection(spec.select);
+    if (opts.session)
+        findOpts.session = opts.session;
+    const fetched = await refModel.collection.find(filter, findOpts).toArray();
+    if (spec.populate) {
+        await runPopulate(fetched, spec.populate, {
+            ownerModel: refModel,
+            session: opts.session,
+        });
+    }
+    const byId = new Map();
+    for (const f of fetched)
+        byId.set(idKey(f._id), f);
+    const nullifyMisses = !!spec.match;
+    for (const d of docs) {
+        const arr = d?.[parent];
+        if (!Array.isArray(arr))
+            continue;
+        for (const sub of arr) {
+            const raw = sub?.[child];
+            if (!isPresent(raw))
+                continue;
+            const hit = byId.get(idKey(raw));
+            if (hit)
+                sub[child] = hit;
+            else if (nullifyMisses)
+                sub[child] = null;
+        }
+    }
+};
+const isPresent = (v) => v !== null && v !== undefined;
+// ObjectId values aren't === equal even when they represent the same
+// 24 hex chars, so key by their string form.
+const idKey = (v) => {
+    if (v == null)
+        return "";
+    if (typeof v === "string")
+        return v;
+    if (typeof v.toHexString === "function")
+        return v.toHexString();
+    return String(v);
+};
+const uniqueIds = (ids) => {
+    const seen = new Set();
+    const out = [];
+    for (const id of ids) {
+        const k = idKey(id);
+        if (seen.has(k))
+            continue;
+        seen.add(k);
+        out.push(id);
+    }
+    return out;
+};
+const normalizeProjection = (sel) => {
+    if (typeof sel !== "string")
+        return sel;
+    const out = {};
+    for (const tok of sel.split(/\s+/).filter(Boolean)) {
+        if (tok.startsWith("-"))
+            out[tok.slice(1)] = 0;
+        else
+            out[tok] = 1;
+    }
+    return out;
+};

package/dist/esm/types.js

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -0,0 +1,74 @@
+{
+  "name": "mongoose-killer",
+  "version": "0.1.0",
+  "description": "In a fight of Mongoose vs. Viper, only 1 in 10 Vipers come out on top. This is the 10th Viper. Viper is a drop-in Mongoose replacement built for speed.",
+  "license": "MIT",
+  "author": "Ben Thayer",
+  "keywords": [
+    "mongodb",
+    "mongoose",
+    "odm",
+    "driver",
+    "schema"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/benthayer/viper.git"
+  },
+  "bugs": {
+    "url": "https://github.com/benthayer/viper/issues"
+  },
+  "homepage": "https://github.com/benthayer/viper#readme",
+  "main": "./dist/cjs/index.js",
+  "module": "./dist/esm/index.js",
+  "types": "./dist/cjs/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/cjs/index.d.ts",
+      "import": "./dist/esm/index.js",
+      "require": "./dist/cjs/index.js",
+      "default": "./dist/cjs/index.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "scripts": {
+    "build": "rm -rf dist && yarn build:cjs && yarn build:esm && yarn build:esm:pkgjson",
+    "build:cjs": "tsc -p tsconfig.cjs.json",
+    "build:esm": "tsc -p tsconfig.esm.json",
+    "build:esm:pkgjson": "node -e \"require('fs').writeFileSync('dist/esm/package.json', JSON.stringify({type:'module'}) + '\\n')\"",
+    "prepare": "yarn build",
+    "prepublishOnly": "yarn build && yarn test",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:real": "MONGOOSE_BACKEND=real vitest run",
+    "test:fake": "MONGOOSE_BACKEND=fake vitest run",
+    "test:both": "MONGOOSE_BACKEND=real vitest run && MONGOOSE_BACKEND=fake vitest run",
+    "typecheck": "tsc --noEmit"
+  },
+  "peerDependencies": {
+    "mongodb": "^6.0.0"
+  },
+  "dependencies": {
+    "bson": "^6.8.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "mongodb": "^6.9.0",
+    "mongoose": "^8.7.0",
+    "tsx": "^4.19.0",
+    "typescript": "^5.5.0",
+    "vitest": "^2.1.0"
+  },
+  "packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
+}