@reproapp/node-sdk 0.0.26 → 0.0.28

package/dist/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.__reproTestHooks = exports.initRepro = exports.patchSendgridMail = exports.patchRedis = exports.patchAmqplib = exports.patchKafkaJs = exports.reproMongoosePlugin = exports.reproMiddleware = exports.__materializePendingTraceEventsForWorker = exports.isReproTracingEnabled = exports.initReproTracing = exports.disableReproTraceLogs = exports.enableReproTraceLogs = exports.setReproTraceLogsEnabled = exports.setDisabledTraceFiles = exports.setDisabledFunctionTypes = exports.setDisabledFunctionTraces = void 0;
+exports.__reproTestHooks = exports.initRepro = exports.patchSendgridMail = exports.patchMongoDb = exports.patchRedis = exports.patchAmqplib = exports.patchKafkaJs = exports.reproMongoosePlugin = exports.reproMiddleware = exports.__materializePendingTraceEventsForWorker = exports.isReproTracingEnabled = exports.initReproTracing = exports.disableReproTraceLogs = exports.enableReproTraceLogs = exports.setReproTraceLogsEnabled = exports.setDisabledTraceFiles = exports.setDisabledFunctionTypes = exports.setDisabledFunctionTraces = void 0;
 const mongoose = require("mongoose");
 const async_hooks_1 = require("async_hooks");
 const crypto_1 = require("crypto");
@@ -819,6 +819,57 @@ function buildRequestCaptureManifestPayload(scope, traceBatchTotal, closedAtMs)
 }
 const als = new async_hooks_1.AsyncLocalStorage();
 const getCtx = () => als.getStore() || {};
+function normalizeNativeMongoSuppressionOp(op) {
+    return String(op ?? '').trim();
+}
+function pushNativeMongoSuppression(collection, ops) {
+    const store = als.getStore();
+    if (!store)
+        return null;
+    const normalizedOps = Array.isArray(ops)
+        ? ops.map(normalizeNativeMongoSuppressionOp).filter(Boolean)
+        : [];
+    const token = {
+        collection: collection == null ? null : String(collection),
+        ops: normalizedOps.length ? new Set(normalizedOps) : undefined,
+        startedAt: Date.now(),
+    };
+    if (!Array.isArray(store.nativeMongoSuppressions)) {
+        store.nativeMongoSuppressions = [];
+    }
+    store.nativeMongoSuppressions.push(token);
+    return token;
+}
+function popNativeMongoSuppression(token) {
+    if (!token)
+        return;
+    const store = als.getStore();
+    const suppressions = store?.nativeMongoSuppressions;
+    if (!Array.isArray(suppressions) || suppressions.length === 0)
+        return;
+    const index = suppressions.indexOf(token);
+    if (index >= 0) {
+        suppressions.splice(index, 1);
+    }
+}
+function isNativeMongoSuppressed(collection, op) {
+    const suppressions = getCtx().nativeMongoSuppressions;
+    if (!Array.isArray(suppressions) || suppressions.length === 0)
+        return false;
+    const normalizedCollection = collection == null ? null : String(collection);
+    const normalizedOp = normalizeNativeMongoSuppressionOp(op);
+    const now = Date.now();
+    for (const token of suppressions) {
+        if (!token || now - token.startedAt > 60000)
+            continue;
+        if (token.collection && normalizedCollection && token.collection !== normalizedCollection)
+            continue;
+        if (token.ops && token.ops.size > 0 && !token.ops.has(normalizedOp))
+            continue;
+        return true;
+    }
+    return false;
+}
 function currentClockSkewMs() {
     const store = als.getStore();
     const skew = store?.clockSkewMs;
@@ -3946,6 +3997,7 @@ async function maybeCaptureDbValueAsync(params, sink) {
 }
 function reproMiddleware(cfg) {
     const masking = normalizeMaskingConfig(cfg.masking);
+    patchMongoDb(cfg);
     return async function (req, res, next) {
         const querySid = readRequestQueryParam(req, REPRO_QUERY_SESSION_PARAM);
         const queryAid = readRequestQueryParam(req, REPRO_QUERY_ACTION_PARAM);
@@ -4483,6 +4535,16 @@ exports.reproMiddleware = reproMiddleware;
 //   - ONLY schema middleware (pre/post) for specific ops
 //   - keeps your existing doc-diff hooks
 // ===================================================================
+function nativeMongoOpsForMongooseOp(op) {
+    if (op === 'findByIdAndUpdate')
+        return ['findOneAndUpdate'];
+    if (op === 'findByIdAndDelete' || op === 'findByIdAndRemove' || op === 'findOneAndRemove') {
+        return ['findOneAndDelete'];
+    }
+    if (op === 'findByIdAndReplace')
+        return ['findOneAndReplace'];
+    return [op];
+}
 function reproMongoosePlugin(cfg) {
     return function (schema) {
         // -------- pre/post save (unchanged) --------
@@ -4501,6 +4563,7 @@ function reproMongoosePlugin(cfg) {
                     ? this.modifiedPaths()
                     : [],
             };
+            this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_meta.collection, this.isNew ? ['insertOne'] : ['updateOne', 'replaceOne']);
             next();
         });
         schema.post('save', { document: true }, function () {
@@ -4534,6 +4597,11 @@ function reproMongoosePlugin(cfg) {
                 t: alignTimestamp(meta.t0 ?? Date.now()),
                 spanContext,
             });
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+        });
+        schema.post('save', { document: true }, function (error, _doc, next) {
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            next(error);
         });
         // -------- findOneAndUpdate capture (unchanged) --------
         schema.pre('findOneAndUpdate', function (next) {
@@ -4548,6 +4616,7 @@ function reproMongoosePlugin(cfg) {
                 this.__repro_spanContext = captureSpanContextFromTracer(this);
                 this.__repro_filter = filter;
                 this.__repro_update = this.getUpdate?.() ?? this._update ?? undefined;
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_collection, ['findOneAndUpdate']);
             }
             catch { }
             next();
@@ -4577,6 +4646,11 @@ function reproMongoosePlugin(cfg) {
                 t: alignTimestamp(this.__repro_t0 ?? Date.now()),
                 spanContext,
             });
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+        });
+        schema.post('findOneAndUpdate', function (error, _res, next) {
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            next(error);
         });
         // -------- deleteOne capture (unchanged) --------
         schema.pre('deleteOne', { document: false, query: true }, function (next) {
@@ -4589,6 +4663,7 @@ function reproMongoosePlugin(cfg) {
                 this.__repro_collection = resolveCollectionOrWarn(this, 'query');
                 this.__repro_filter = filter;
                 this.__repro_spanContext = captureSpanContextFromTracer(this);
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_collection, ['deleteOne']);
             }
             catch { }
             next();
@@ -4613,6 +4688,11 @@ function reproMongoosePlugin(cfg) {
                 t: alignTimestamp(this.__repro_t0 ?? Date.now()),
                 spanContext,
             });
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+        });
+        schema.post('deleteOne', { document: false, query: true }, function (error, _res, next) {
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            next(error);
         });
         // -------- NON-intrusive generic query telemetry via schema hooks -------
         const READ_OPS = [
@@ -4653,6 +4733,7 @@ function reproMongoosePlugin(cfg) {
                         rawProjection,
                         rawOptions,
                     };
+                    this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_qmeta.collection, nativeMongoOpsForMongooseOp(op));
                 }
                 catch {
                     this.__repro_qmeta = {
@@ -4661,6 +4742,7 @@ function reproMongoosePlugin(cfg) {
                         op,
                         spanContext: captureSpanContextFromTracer(this),
                     };
+                    this.__repro_native_mongo_suppression = pushNativeMongoSuppression('unknown', nativeMongoOpsForMongooseOp(op));
                 }
                 next();
             });
@@ -4684,6 +4766,11 @@ function reproMongoosePlugin(cfg) {
                     t: alignedNow(),
                     spanContext,
                 });
+                popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            });
+            schema.post(op, function (error, _res, next) {
+                popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+                next(error);
             });
         }
         READ_OPS.forEach(attachQueryHooks);
@@ -4698,6 +4785,7 @@ function reproMongoosePlugin(cfg) {
                     rawDocs,
                     spanContext: captureSpanContextFromTracer(this),
                 };
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_insert_meta.collection, ['insertMany']);
             }
             catch {
                 this.__repro_insert_meta = {
@@ -4705,6 +4793,7 @@ function reproMongoosePlugin(cfg) {
                     collection: 'unknown',
                     spanContext: captureSpanContextFromTracer(this),
                 };
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression('unknown', ['insertMany']);
             }
             next();
         });
@@ -4723,6 +4812,11 @@ function reproMongoosePlugin(cfg) {
                 t: alignedNow(),
                 spanContext,
             });
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+        });
+        schema.post('insertMany', { document: false, query: false }, function (error, _res, next) {
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            next(error);
         });
         schema.pre('bulkWrite', { document: false, query: false }, function (next, ops) {
             try {
@@ -4733,6 +4827,7 @@ function reproMongoosePlugin(cfg) {
                     rawOps,
                     spanContext: captureSpanContextFromTracer(this),
                 };
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_bulk_meta.collection, ['bulkWrite']);
             }
             catch {
                 this.__repro_bulk_meta = {
@@ -4740,6 +4835,7 @@ function reproMongoosePlugin(cfg) {
                     collection: 'unknown',
                     spanContext: captureSpanContextFromTracer(this),
                 };
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression('unknown', ['bulkWrite']);
             }
             next();
         });
@@ -4758,6 +4854,11 @@ function reproMongoosePlugin(cfg) {
                 t: alignedNow(),
                 spanContext,
             });
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+        });
+        schema.post('bulkWrite', { document: false, query: false }, function (error, _res, next) {
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            next(error);
         });
         // Aggregate middleware (non-intrusive)
         schema.pre('aggregate', function (next) {
@@ -4772,6 +4873,7 @@ function reproMongoosePlugin(cfg) {
                     spanContext: captureSpanContextFromTracer(this),
                     rawPipeline,
                 };
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression(this.__repro_aggmeta.collection, ['aggregate']);
             }
             catch {
                 this.__repro_aggmeta = {
@@ -4780,6 +4882,7 @@ function reproMongoosePlugin(cfg) {
                     pipeline: undefined,
                     spanContext: captureSpanContextFromTracer(this),
                 };
+                this.__repro_native_mongo_suppression = pushNativeMongoSuppression('unknown', ['aggregate']);
             }
             next();
         });
@@ -4798,6 +4901,11 @@ function reproMongoosePlugin(cfg) {
                 t: alignedNow(),
                 spanContext,
             });
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+        });
+        schema.post('aggregate', function (error, _res, next) {
+            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            next(error);
         });
     };
 }
@@ -4823,10 +4931,24 @@ function summarizeQueryResult(op, res) {
         }
         return summary;
     }
+    if (op === 'insertOne') {
+        const summary = {
+            insertedId: res?.insertedId ?? undefined,
+            acknowledged: res?.acknowledged ?? undefined,
+        };
+        if (typeof resultPreview !== 'undefined')
+            summary.result = resultPreview;
+        return summary;
+    }
     if (op === 'insertMany') {
         const summary = {};
         if (Array.isArray(res))
             summary.inserted = res.length;
+        else {
+            summary.inserted = res?.insertedCount ?? undefined;
+            summary.insertedIds = res?.insertedIds ?? undefined;
+            summary.acknowledged = res?.acknowledged ?? undefined;
+        }
         if (typeof resultPreview !== 'undefined')
             summary.result = resultPreview;
         return summary;
@@ -7409,6 +7531,465 @@ function patchRedis(cfg) {
     }
 }
 exports.patchRedis = patchRedis;
+const mongoDbPatchConfigs = [];
+let mongoDbRequireHookInstalled = false;
+const MONGODB_PROMISE_OPS = [
+    'insertOne',
+    'insertMany',
+    'updateOne',
+    'updateMany',
+    'replaceOne',
+    'deleteOne',
+    'deleteMany',
+    'findOne',
+    'countDocuments',
+    'estimatedDocumentCount',
+    'distinct',
+    'bulkWrite',
+    'findOneAndUpdate',
+    'findOneAndDelete',
+    'findOneAndReplace',
+];
+const MONGODB_CURSOR_OPS = ['find', 'aggregate'];
+const MONGODB_CURSOR_TERMINALS = ['toArray', 'next', 'forEach'];
+function resolveMongoDbRuntimeContext(cfg) {
+    const ctx = getCtx();
+    const fallback = cfg.resolveContext?.() ?? {};
+    return {
+        ctx,
+        sid: ctx.sid ?? fallback.sid ?? null,
+        aid: ctx.aid ?? fallback.aid ?? null,
+    };
+}
+function resolveMongoCollectionName(collection) {
+    const candidates = [
+        collection?.collectionName,
+        collection?.namespace?.collection,
+        collection?.s?.namespace?.collection,
+        collection?.s?.namespace?.collectionName,
+    ];
+    for (const candidate of candidates) {
+        if (typeof candidate === 'string' && candidate.trim())
+            return candidate;
+    }
+    const namespace = typeof collection?.namespace === 'string'
+        ? collection.namespace
+        : typeof collection?.s?.namespace === 'string'
+            ? collection.s.namespace
+            : typeof collection?.namespace?.toString === 'function'
+                ? collection.namespace.toString()
+                : typeof collection?.s?.namespace?.toString === 'function'
+                    ? collection.s.namespace.toString()
+                    : null;
+    if (namespace && namespace.includes('.')) {
+        return namespace.slice(namespace.indexOf('.') + 1) || namespace;
+    }
+    return namespace || 'unknown';
+}
+function resolveMongoDbName(collection) {
+    const candidates = [
+        collection?.dbName,
+        collection?.db?.databaseName,
+        collection?.s?.db?.databaseName,
+        collection?.namespace?.db,
+        collection?.s?.namespace?.db,
+    ];
+    for (const candidate of candidates) {
+        if (typeof candidate === 'string' && candidate.trim())
+            return candidate;
+    }
+    const namespace = typeof collection?.namespace === 'string'
+        ? collection.namespace
+        : typeof collection?.s?.namespace === 'string'
+            ? collection.s.namespace
+            : null;
+    if (namespace && namespace.includes('.')) {
+        return namespace.slice(0, namespace.indexOf('.')) || undefined;
+    }
+    return undefined;
+}
+function summarizeMongoSession(session) {
+    if (!session || typeof session !== 'object')
+        return session;
+    let inTransaction = undefined;
+    try {
+        inTransaction = typeof session.inTransaction === 'function' ? session.inTransaction() : undefined;
+    }
+    catch { }
+    return {
+        __kind: 'mongodb_client_session',
+        id: session.id ?? session.serverSession?.id ?? undefined,
+        hasEnded: session.hasEnded ?? undefined,
+        inTransaction,
+    };
+}
+function normalizeMongoDriverOptions(options) {
+    if (!options || typeof options !== 'object' || Array.isArray(options))
+        return options;
+    const normalized = {};
+    for (const key of Object.keys(options)) {
+        normalized[key] = key === 'session'
+            ? summarizeMongoSession(options[key])
+            : options[key];
+    }
+    return normalized;
+}
+function summarizeMongoError(error) {
+    return {
+        name: error?.name ?? undefined,
+        message: error?.message ?? String(error),
+        code: error?.code ?? undefined,
+        codeName: error?.codeName ?? undefined,
+    };
+}
+function extractMongoPkFromFilter(filter) {
+    if (filter && typeof filter === 'object' && Object.prototype.hasOwnProperty.call(filter, '_id')) {
+        return { _id: filter._id };
+    }
+    return undefined;
+}
+function extractMongoPkFromDoc(doc, result) {
+    const id = doc && typeof doc === 'object' ? doc._id : undefined;
+    const insertedId = result?.insertedId;
+    if (id !== undefined)
+        return { _id: id };
+    if (insertedId !== undefined)
+        return { _id: insertedId };
+    return undefined;
+}
+function buildMongoQueryPayload(op, args, dbName) {
+    const withDriver = (payload) => ({
+        driver: 'mongodb',
+        db: dbName,
+        ...payload,
+    });
+    if (op === 'insertOne') {
+        return withDriver({ doc: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    if (op === 'insertMany') {
+        return withDriver({ docs: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    if (op === 'updateOne' || op === 'updateMany') {
+        return withDriver({ filter: args[0], update: args[1], options: normalizeMongoDriverOptions(args[2]) });
+    }
+    if (op === 'replaceOne') {
+        return withDriver({ filter: args[0], replacement: args[1], options: normalizeMongoDriverOptions(args[2]) });
+    }
+    if (op === 'deleteOne' || op === 'deleteMany') {
+        return withDriver({ filter: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    if (op === 'findOne' || op === 'find' || op === 'countDocuments') {
+        return withDriver({ filter: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    if (op === 'estimatedDocumentCount') {
+        return withDriver({ options: normalizeMongoDriverOptions(args[0]) });
+    }
+    if (op === 'distinct') {
+        return withDriver({ key: args[0], filter: args[1], options: normalizeMongoDriverOptions(args[2]) });
+    }
+    if (op === 'bulkWrite') {
+        return withDriver({ ops: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    if (op === 'findOneAndUpdate') {
+        return withDriver({ filter: args[0], update: args[1], options: normalizeMongoDriverOptions(args[2]) });
+    }
+    if (op === 'findOneAndReplace') {
+        return withDriver({ filter: args[0], replacement: args[1], options: normalizeMongoDriverOptions(args[2]) });
+    }
+    if (op === 'findOneAndDelete') {
+        return withDriver({ filter: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    if (op === 'aggregate') {
+        return withDriver({ pipeline: args[0], options: normalizeMongoDriverOptions(args[1]) });
+    }
+    return withDriver({ args });
+}
+function buildMongoPk(op, args, result) {
+    if (op === 'insertOne')
+        return extractMongoPkFromDoc(args[0], result);
+    if (op === 'updateOne' ||
+        op === 'replaceOne' ||
+        op === 'deleteOne' ||
+        op === 'findOne' ||
+        op === 'findOneAndUpdate' ||
+        op === 'findOneAndDelete' ||
+        op === 'findOneAndReplace') {
+        return extractMongoPkFromFilter(args[0]);
+    }
+    return undefined;
+}
+function buildMongoAfter(op, args, result) {
+    if (op === 'insertOne')
+        return args[0] ?? null;
+    if (op === 'insertMany')
+        return args[0] ?? null;
+    if (op === 'findOne' || op === 'findOneAndUpdate' || op === 'findOneAndDelete' || op === 'findOneAndReplace') {
+        return result ?? null;
+    }
+    if (op === 'find' || op === 'aggregate') {
+        return Array.isArray(result) ? result : null;
+    }
+    return null;
+}
+function findMongoCallbackIndex(args) {
+    for (let i = args.length - 1; i >= 0; i -= 1) {
+        if (typeof args[i] === 'function')
+            return i;
+    }
+    return -1;
+}
+function patchMongoCursorInstance(cursor, meta) {
+    if (!cursor || typeof cursor !== 'object')
+        return cursor;
+    if (cursor.__repro_mongodb_cursor_patched)
+        return cursor;
+    try {
+        Object.defineProperty(cursor, '__repro_mongodb_cursor_patched', {
+            value: true,
+            configurable: true,
+        });
+    }
+    catch {
+        return cursor;
+    }
+    for (const methodName of MONGODB_CURSOR_TERMINALS) {
+        const original = cursor[methodName];
+        if (typeof original !== 'function')
+            continue;
+        cursor[methodName] = function reproPatchedMongoCursorTerminal(...callArgs) {
+            const t0 = Date.now();
+            let emitted = false;
+            let forEachDocs = null;
+            const emitSuccess = (value) => {
+                if (emitted)
+                    return;
+                emitted = true;
+                const terminalQueryArgs = [...meta.args];
+                const queryRaw = buildMongoQueryPayload(meta.op, terminalQueryArgs, meta.dbName);
+                queryRaw.cursorMethod = methodName;
+                emitDbQuery(meta.cfg, meta.sid, meta.aid ?? undefined, {
+                    type: 'mongo',
+                    collection: meta.collection,
+                    op: meta.op,
+                    pk: buildMongoPk(meta.op, meta.args, value),
+                    before: null,
+                    after: buildMongoAfter(meta.op, meta.args, forEachDocs ?? value),
+                    queryRaw,
+                    resultMetaRaw: forEachDocs ?? value,
+                    durMs: Date.now() - t0,
+                    t: alignedNow(),
+                    spanContext: meta.spanContext,
+                });
+            };
+            const emitFailure = (error) => {
+                if (emitted)
+                    return;
+                emitted = true;
+                const terminalQueryArgs = [...meta.args];
+                const queryRaw = buildMongoQueryPayload(meta.op, terminalQueryArgs, meta.dbName);
+                queryRaw.cursorMethod = methodName;
+                emitDbQuery(meta.cfg, meta.sid, meta.aid ?? undefined, {
+                    type: 'mongo',
+                    collection: meta.collection,
+                    op: meta.op,
+                    queryRaw,
+                    error: summarizeMongoError(error),
+                    durMs: Date.now() - t0,
+                    t: alignedNow(),
+                    spanContext: meta.spanContext,
+                });
+            };
+            const actualArgs = [...callArgs];
+            if (methodName === 'forEach' && typeof actualArgs[0] === 'function') {
+                forEachDocs = [];
+                const originalIterator = actualArgs[0];
+                actualArgs[0] = function reproMongoForEachIterator(doc, ...rest) {
+                    forEachDocs?.push(doc);
+                    return originalIterator.call(this, doc, ...rest);
+                };
+            }
+            try {
+                const value = original.apply(this, actualArgs);
+                if (value && typeof value.then === 'function') {
+                    return value.then((resolved) => {
+                        emitSuccess(resolved);
+                        return resolved;
+                    }, (err) => {
+                        emitFailure(err);
+                        throw err;
+                    });
+                }
+                emitSuccess(value);
+                return value;
+            }
+            catch (err) {
+                emitFailure(err);
+                throw err;
+            }
+        };
+    }
+    return cursor;
+}
+function patchMongoCollectionMethod(proto, op, cursorOp, cfg) {
+    if (!proto)
+        return;
+    const marker = `__repro_mongodb_${op}_patched`;
+    if (proto[marker])
+        return;
+    const original = proto[op];
+    if (typeof original !== 'function')
+        return;
+    proto[marker] = true;
+    proto[op] = function reproPatchedMongoCollectionMethod(...args) {
+        const collection = resolveMongoCollectionName(this);
+        if (isNativeMongoSuppressed(collection, op)) {
+            return original.apply(this, args);
+        }
+        const { sid, aid } = resolveMongoDbRuntimeContext(cfg);
+        if (!sid) {
+            return original.apply(this, args);
+        }
+        const spanContext = captureSpanContextFromTracer(this);
+        const t0 = Date.now();
+        const dbName = resolveMongoDbName(this);
+        const callbackIndex = cursorOp ? -1 : findMongoCallbackIndex(args);
+        const actualArgs = callbackIndex >= 0 ? [...args] : args;
+        let emitted = false;
+        const emitSuccess = (result) => {
+            if (emitted)
+                return;
+            emitted = true;
+            emitDbQuery(cfg, sid, aid ?? undefined, {
+                type: 'mongo',
+                collection,
+                op,
+                pk: buildMongoPk(op, args, result),
+                before: null,
+                after: buildMongoAfter(op, args, result),
+                queryRaw: buildMongoQueryPayload(op, args, dbName),
+                resultMetaRaw: result,
+                durMs: Date.now() - t0,
+                t: alignedNow(),
+                spanContext,
+            });
+        };
+        const emitFailure = (error) => {
+            if (emitted)
+                return;
+            emitted = true;
+            emitDbQuery(cfg, sid, aid ?? undefined, {
+                type: 'mongo',
+                collection,
+                op,
+                queryRaw: buildMongoQueryPayload(op, args, dbName),
+                error: summarizeMongoError(error),
+                durMs: Date.now() - t0,
+                t: alignedNow(),
+                spanContext,
+            });
+        };
+        if (callbackIndex >= 0) {
+            const originalCallback = args[callbackIndex];
+            actualArgs[callbackIndex] = function reproMongoCallback(err, result, ...rest) {
+                if (err)
+                    emitFailure(err);
+                else
+                    emitSuccess(result);
+                return originalCallback.call(this, err, result, ...rest);
+            };
+        }
+        try {
+            const result = original.apply(this, actualArgs);
+            if (cursorOp) {
+                return patchMongoCursorInstance(result, {
+                    cfg,
+                    sid,
+                    aid,
+                    collection,
+                    dbName,
+                    op,
+                    args,
+                    spanContext,
+                });
+            }
+            if (result && typeof result.then === 'function') {
+                return result.then((value) => {
+                    emitSuccess(value);
+                    return value;
+                }, (error) => {
+                    emitFailure(error);
+                    throw error;
+                });
+            }
+            if (callbackIndex < 0) {
+                emitSuccess(result);
+            }
+            return result;
+        }
+        catch (error) {
+            emitFailure(error);
+            throw error;
+        }
+    };
+}
+function patchMongoCollectionPrototype(proto, cfg) {
+    if (!proto)
+        return;
+    for (const op of MONGODB_PROMISE_OPS) {
+        patchMongoCollectionMethod(proto, op, false, cfg);
+    }
+    for (const op of MONGODB_CURSOR_OPS) {
+        patchMongoCollectionMethod(proto, op, true, cfg);
+    }
+}
+function patchMongoDbFromModuleExports(mongoModule, cfg) {
+    if (!mongoModule)
+        return;
+    const constructors = new Set();
+    const addCtor = (candidate) => {
+        if (typeof candidate === 'function' && candidate.prototype) {
+            constructors.add(candidate);
+        }
+    };
+    addCtor(mongoModule?.Collection);
+    addCtor(mongoModule?.default?.Collection);
+    for (const ctor of constructors) {
+        patchMongoCollectionPrototype(ctor.prototype, cfg);
+    }
+}
+function installMongoDbRequireHook() {
+    if (mongoDbRequireHookInstalled)
+        return;
+    mongoDbRequireHookInstalled = true;
+    try {
+        const rim = requireOptionalHostModule('require-in-the-middle');
+        const Hook = rim?.Hook ?? rim;
+        if (typeof Hook !== 'function')
+            return;
+        // eslint-disable-next-line no-new
+        new Hook(['mongodb'], (exports) => {
+            for (const cfg of mongoDbPatchConfigs) {
+                patchMongoDbFromModuleExports(exports, cfg);
+            }
+            return exports;
+        });
+    }
+    catch { }
+}
+function patchMongoDb(cfg) {
+    mongoDbPatchConfigs.push(cfg);
+    installMongoDbRequireHook();
+    const mongoModules = new Set();
+    for (const loadedMongoModule of getLoadedModuleExportsByPackage('mongodb')) {
+        mongoModules.add(loadedMongoModule);
+    }
+    if (!mongoModules.size)
+        return;
+    for (const mongoModule of mongoModules) {
+        patchMongoDbFromModuleExports(mongoModule, cfg);
+    }
+}
+exports.patchMongoDb = patchMongoDb;
 // ===================================================================
 // Auto init helpers
 // ===================================================================
@@ -7834,6 +8415,7 @@ async function initRepro(cfg) {
     patchKafkaJs(integrationCfg);
     patchAmqplib(integrationCfg);
     patchRedis(integrationCfg);
+    patchMongoDb(integrationCfg);
     patchSendgridMail(integrationCfg);
     patchMongooseIntegration(reproAutoMiddlewareConfig);
     await initializeRuntimePrivacyPolicyAtStartup(reproAutoMiddlewareConfig);
@@ -7881,6 +8463,8 @@ exports.__reproTestHooks = {
     patchAmqplibConnectionForTest: patchAmqplibConnection,
     patchAmqplibChannelForTest: patchAmqplibChannel,
     wrapAmqplibConsumeHandlerForTest: wrapAmqplibConsumeHandler,
+    patchMongoDbModuleForTest: patchMongoDbFromModuleExports,
+    patchMongoCollectionPrototypeForTest: patchMongoCollectionPrototype,
     chunkTraceEventsForTransportForTest: chunkTraceEventsForTransport,
     estimateTraceBatchSerializedBytesForTest: estimateTraceBatchSerializedBytes,
     sanitizeTraceValueForPrivacyForTest: sanitizeTraceValueForPrivacy,