@reproapp/node-sdk 0.0.30 → 0.0.32

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reproapp/node-sdk",
-  "version": "0.0.30",
+  "version": "0.0.32",
   "description": "Repro Nest SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -12,7 +12,7 @@
     "build": "tsc -p tsconfig.json",
     "dev": "tsc -p tsconfig.json --watch --preserveWatchOutput",
     "prepublishOnly": "npm run build",
-    "test": "npm run build && node test/unawaited.test.js && node test/integration-unawaited.js && node test/request-flush-timing.test.js && node test/request-capture-background.test.js && node test/init-auto-http-server.test.js && node test/client-performance-regression.test.js && node test/request-manifest-priority.test.js && node test/runtime-no-double-invoke.test.js && node test/express-trace-http-args.test.js && node test/trace-batch-size.test.js && node test/sdk-background-priority.test.js && node test/ingest-queue-retention.test.js && node test/ingest-high-priority-retention.test.js && node test/ingest-critical-retention.test.js && node test/ingest-critical-retry-retention.test.js && node test/ingest-timeout.test.js && node -r ./tracer/register test/promise-map.test.js && node test/disable-subtree.test.js && node test/circular-capture.test.js && node test/wrap-plugin-arrow-args.test.js && node test/wrap-plugin-super-call.test.js && node test/function-arg-identity.test.js && node test/privacy-runtime-policy.test.js && node test/runtime-privacy-materialization.test.js && node test/kafka-runtime-privacy-policy.test.js && node test/amqplib-integration.test.js && node test/mongodb-integration.test.js",
+    "test": "npm run build && node test/unawaited.test.js && node test/integration-unawaited.js && node test/request-flush-timing.test.js && node test/request-capture-background.test.js && node test/init-auto-http-server.test.js && node test/client-performance-regression.test.js && node test/request-manifest-priority.test.js && node test/runtime-no-double-invoke.test.js && node test/express-trace-http-args.test.js && node test/trace-batch-size.test.js && node test/sdk-background-priority.test.js && node test/ingest-queue-retention.test.js && node test/ingest-high-priority-retention.test.js && node test/ingest-critical-retention.test.js && node test/ingest-critical-retry-retention.test.js && node test/ingest-timeout.test.js && node -r ./tracer/register test/promise-map.test.js && node test/disable-subtree.test.js && node test/circular-capture.test.js && node test/wrap-plugin-arrow-args.test.js && node test/wrap-plugin-super-call.test.js && node test/function-arg-identity.test.js && node test/privacy-runtime-policy.test.js && node test/runtime-privacy-materialization.test.js && node test/kafka-runtime-privacy-policy.test.js && node test/amqplib-integration.test.js && node test/mongodb-integration.test.js && node test/mongodb-esm-import-hook.test.js && node test/mongodb-suppression-cleanup.test.js",
     "test:containers": "npm run build && node test/runtime-no-double-invoke.testcontainers.test.js && node test/wrap-plugin-super-call.testcontainers.test.js && node test/function-arg-identity.testcontainers.test.js"
   },
   "peerDependencies": {

package/src/index.ts

@@ -5,6 +5,7 @@ import { AsyncLocalStorage } from 'async_hooks';
 import { randomUUID } from 'crypto';
 import { createRequire } from 'module';
 import { join } from 'path';
+import { pathToFileURL } from 'url';
 import {
     configureIngestQueueDrain,
     enqueueIngestEntries,
@@ -1162,6 +1163,7 @@ type NativeMongoSuppression = {
     collection?: string | null;
     ops?: Set<string>;
     startedAt: number;
+    ownerStore?: Ctx;
 };
 
 type RequestCaptureScopeState = {
@@ -1235,6 +1237,7 @@ function pushNativeMongoSuppression(collection: any, ops?: string[] | readonly s
         collection: collection == null ? null : String(collection),
         ops: normalizedOps.length ? new Set(normalizedOps) : undefined,
         startedAt: Date.now(),
+        ownerStore: store,
     };
     if (!Array.isArray(store.nativeMongoSuppressions)) {
         store.nativeMongoSuppressions = [];
@@ -1245,12 +1248,22 @@ function pushNativeMongoSuppression(collection: any, ops?: string[] | readonly s
 
 function popNativeMongoSuppression(token: NativeMongoSuppression | null | undefined): void {
     if (!token) return;
-    const store = als.getStore();
+    const store = token.ownerStore ?? 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);
+    if (Array.isArray(suppressions) && suppressions.length > 0) {
+        const index = suppressions.indexOf(token);
+        if (index >= 0) {
+            suppressions.splice(index, 1);
+        }
+    }
+    token.ownerStore = undefined;
+}
+
+function releaseNativeMongoSuppression(target: any): void {
+    const token = target?.__repro_native_mongo_suppression;
+    popNativeMongoSuppression(token);
+    try { delete target.__repro_native_mongo_suppression; } catch {
+        try { target.__repro_native_mongo_suppression = undefined; } catch {}
     }
 }
 
@@ -1260,8 +1273,12 @@ function isNativeMongoSuppressed(collection: any, op: string): boolean {
     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;
+    for (let i = suppressions.length - 1; i >= 0; i -= 1) {
+        const token = suppressions[i];
+        if (!token || now - token.startedAt > 60000) {
+            suppressions.splice(i, 1);
+            continue;
+        }
         if (token.collection && normalizedCollection && token.collection !== normalizedCollection) continue;
         if (token.ops && token.ops.size > 0 && !token.ops.has(normalizedOp)) continue;
         return true;
@@ -5571,41 +5588,44 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
         });
 
         schema.post('save', { document: true }, function () {
-            const { sid, aid } = getCtx();
-            if (!sid || !aid) return;
-            if ((this as any).$isSubdocument) return;
-
-            const meta = (this as any).__repro_meta || {};
-            const before = null;
-            const after = this;
-            const collection = meta.collection || resolveCollectionOrWarn(this, 'doc');
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-            if (!shouldCaptureDbSpan(spanContext)) return;
-            const modifiedPaths = Array.isArray(meta.modifiedPaths) ? meta.modifiedPaths : [];
-
-            const query = meta.wasNew
-                ? { op: 'insertOne', doc: after }
-                : {
-                    filter: { _id: this._id },
-                    update: buildModifiedPathUpdate(this, modifiedPaths),
-                    options: { upsert: false },
-                };
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid || !aid) return;
+                if ((this as any).$isSubdocument) return;
 
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection,
-                pk: { _id: (this as any)._id },
-                before,
-                after,
-                op: meta.wasNew ? 'insert' : 'update',
-                query,
-                t: alignTimestamp(meta.t0 ?? Date.now()),
-                spanContext,
-            });
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+                const meta = (this as any).__repro_meta || {};
+                const before = null;
+                const after = this;
+                const collection = meta.collection || resolveCollectionOrWarn(this, 'doc');
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                if (!shouldCaptureDbSpan(spanContext)) return;
+                const modifiedPaths = Array.isArray(meta.modifiedPaths) ? meta.modifiedPaths : [];
+
+                const query = meta.wasNew
+                    ? { op: 'insertOne', doc: after }
+                    : {
+                        filter: { _id: this._id },
+                        update: buildModifiedPathUpdate(this, modifiedPaths),
+                        options: { upsert: false },
+                    };
+
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection,
+                    pk: { _id: (this as any)._id },
+                    before,
+                    after,
+                    op: meta.wasNew ? 'insert' : 'update',
+                    query,
+                    t: alignTimestamp(meta.t0 ?? Date.now()),
+                    spanContext,
+                });
+            } finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
 
         schema.post('save', { document: true }, function (error: any, _doc: any, next: Function) {
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
 
@@ -5630,35 +5650,38 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
         });
 
         schema.post<Query<any, any>>('findOneAndUpdate', function (res: any) {
-            const { sid, aid } = getCtx();
-            if (!sid || !aid) return;
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid || !aid) return;
 
-            const before = null;
-            const after = res ?? null;
-            const collection = (this as any).__repro_collection || resolveCollectionOrWarn(this, 'query');
-            const spanContext = (this as any).__repro_spanContext || captureSpanContextFromTracer(this);
-            if (!shouldCaptureDbSpan(spanContext)) return;
-            const filter = (this as any).__repro_filter;
-            const pk = after?._id ?? (filter && typeof filter === 'object' ? (filter as any)._id : undefined);
+                const before = null;
+                const after = res ?? null;
+                const collection = (this as any).__repro_collection || resolveCollectionOrWarn(this, 'query');
+                const spanContext = (this as any).__repro_spanContext || captureSpanContextFromTracer(this);
+                if (!shouldCaptureDbSpan(spanContext)) return;
+                const filter = (this as any).__repro_filter;
+                const pk = after?._id ?? (filter && typeof filter === 'object' ? (filter as any)._id : undefined);
 
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection,
-                pk: { _id: pk },
-                before,
-                after,
-                op: 'update',
-                queryRaw: {
-                    filter,
-                    update: (this as any).__repro_update,
-                },
-                t: alignTimestamp((this as any).__repro_t0 ?? Date.now()),
-                spanContext,
-            });
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection,
+                    pk: { _id: pk },
+                    before,
+                    after,
+                    op: 'update',
+                    queryRaw: {
+                        filter,
+                        update: (this as any).__repro_update,
+                    },
+                    t: alignTimestamp((this as any).__repro_t0 ?? Date.now()),
+                    spanContext,
+                });
+            } finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
 
         schema.post<Query<any, any>>('findOneAndUpdate', function (this: any, error: any, _res: any, next: Function) {
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         } as any);
 
@@ -5680,27 +5703,30 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
         });
 
         schema.post<Query<any, any>>('deleteOne', { document: false, query: true }, function () {
-            const { sid, aid } = getCtx(); if (!sid || !aid) return;
-            const before = null;
-            const collection = (this as any).__repro_collection || resolveCollectionOrWarn(this, 'query');
-            const filter = (this as any).__repro_filter ?? undefined;
-            const spanContext = (this as any).__repro_spanContext || captureSpanContextFromTracer(this);
-            if (!shouldCaptureDbSpan(spanContext)) return;
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection,
-                pk: filter && typeof filter === 'object' && '_id' in filter ? { _id: (filter as any)._id } : undefined,
-                before,
-                after: null,
-                op: 'delete',
-                queryRaw: { filter },
-                t: alignTimestamp((this as any).__repro_t0 ?? Date.now()),
-                spanContext,
-            });
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx(); if (!sid || !aid) return;
+                const before = null;
+                const collection = (this as any).__repro_collection || resolveCollectionOrWarn(this, 'query');
+                const filter = (this as any).__repro_filter ?? undefined;
+                const spanContext = (this as any).__repro_spanContext || captureSpanContextFromTracer(this);
+                if (!shouldCaptureDbSpan(spanContext)) return;
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection,
+                    pk: filter && typeof filter === 'object' && '_id' in filter ? { _id: (filter as any)._id } : undefined,
+                    before,
+                    after: null,
+                    op: 'delete',
+                    queryRaw: { filter },
+                    t: alignTimestamp((this as any).__repro_t0 ?? Date.now()),
+                    spanContext,
+                });
+            } finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
 
         schema.post<Query<any, any>>('deleteOne', { document: false, query: true }, function (this: any, error: any, _res: any, next: Function) {
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         } as any);
 
@@ -5765,31 +5791,34 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
             });
 
             schema.post(op as any, function (this: any, res: any) {
-                const { sid, aid } = getCtx();
-                if (!sid) return;
+                try {
+                    const { sid, aid } = getCtx();
+                    if (!sid) return;
 
-                const meta = (this as any).__repro_qmeta || { t0: Date.now(), collection: 'unknown', op };
-                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                    const meta = (this as any).__repro_qmeta || { t0: Date.now(), collection: 'unknown', op };
+                    const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
 
-                emitDbQuery(cfg, sid, aid, {
-                    collection: meta.collection,
-                    op,
-                    queryRaw: {
-                        filter: meta.rawFilter,
-                        update: meta.rawUpdate,
-                        projection: meta.rawProjection,
-                        options: meta.rawOptions,
-                    },
-                    resultMetaRaw: res,
-                    durMs: Date.now() - meta.t0,
-                    t: alignedNow(),
-                    spanContext,
-                });
-                popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+                    emitDbQuery(cfg, sid, aid, {
+                        collection: meta.collection,
+                        op,
+                        queryRaw: {
+                            filter: meta.rawFilter,
+                            update: meta.rawUpdate,
+                            projection: meta.rawProjection,
+                            options: meta.rawOptions,
+                        },
+                        resultMetaRaw: res,
+                        durMs: Date.now() - meta.t0,
+                        t: alignedNow(),
+                        spanContext,
+                    });
+                } finally {
+                    releaseNativeMongoSuppression(this);
+                }
             });
 
             schema.post(op as any, function (this: any, error: any, _res: any, next: Function) {
-                popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+                releaseNativeMongoSuppression(this);
                 next(error);
             } as any);
         }
@@ -5823,25 +5852,28 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
         } as any);
 
         schema.post<any>('insertMany' as any, { document: false, query: false } as any, function (this: any, docs: any[]) {
-            const { sid, aid } = getCtx();
-            if (!sid) return;
-            const meta = (this as any).__repro_insert_meta || { t0: Date.now(), collection: 'unknown' };
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection: meta.collection,
-                op: 'insertMany',
-                queryRaw: { docs: meta.rawDocs ?? undefined },
-                resultMetaRaw: docs,
-                durMs: Date.now() - meta.t0,
-                t: alignedNow(),
-                spanContext,
-            });
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid) return;
+                const meta = (this as any).__repro_insert_meta || { t0: Date.now(), collection: 'unknown' };
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection: meta.collection,
+                    op: 'insertMany',
+                    queryRaw: { docs: meta.rawDocs ?? undefined },
+                    resultMetaRaw: docs,
+                    durMs: Date.now() - meta.t0,
+                    t: alignedNow(),
+                    spanContext,
+                });
+            } finally {
+                releaseNativeMongoSuppression(this);
+            }
         } as any);
 
         schema.post<any>('insertMany' as any, { document: false, query: false } as any, function (this: any, error: any, _res: any, next: Function) {
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         } as any);
 
@@ -5870,25 +5902,28 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
         } as any);
 
         schema.post<any>('bulkWrite' as any, { document: false, query: false } as any, function (this: any, res: any) {
-            const { sid, aid } = getCtx();
-            if (!sid) return;
-            const meta = (this as any).__repro_bulk_meta || { t0: Date.now(), collection: 'unknown' };
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-
-            emitDbQuery(cfg, sid, aid, {
-                collection: meta.collection,
-                op: 'bulkWrite',
-                queryRaw: { ops: meta.rawOps ?? undefined },
-                resultMetaRaw: res,
-                durMs: Date.now() - meta.t0,
-                t: alignedNow(),
-                spanContext,
-            });
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid) return;
+                const meta = (this as any).__repro_bulk_meta || { t0: Date.now(), collection: 'unknown' };
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+
+                emitDbQuery(cfg, sid, aid, {
+                    collection: meta.collection,
+                    op: 'bulkWrite',
+                    queryRaw: { ops: meta.rawOps ?? undefined },
+                    resultMetaRaw: res,
+                    durMs: Date.now() - meta.t0,
+                    t: alignedNow(),
+                    spanContext,
+                });
+            } finally {
+                releaseNativeMongoSuppression(this);
+            }
         } as any);
 
         schema.post<any>('bulkWrite' as any, { document: false, query: false } as any, function (this: any, error: any, _res: any, next: Function) {
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         } as any);
 
@@ -5923,26 +5958,29 @@ export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
         });
 
         schema.post('aggregate', function (this: any, res: any[]) {
-            const { sid, aid } = getCtx();
-            if (!sid) return;
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid) return;
 
-            const meta = (this as any).__repro_aggmeta || { t0: Date.now(), collection: 'unknown' };
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                const meta = (this as any).__repro_aggmeta || { t0: Date.now(), collection: 'unknown' };
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
 
-            emitDbQuery(cfg, sid, aid, {
-                collection: meta.collection,
-                op: 'aggregate',
-                queryRaw: { pipeline: meta.rawPipeline },
-                resultMetaRaw: res,
-                durMs: Date.now() - meta.t0,
-                t: alignedNow(),
-                spanContext,
-            });
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+                emitDbQuery(cfg, sid, aid, {
+                    collection: meta.collection,
+                    op: 'aggregate',
+                    queryRaw: { pipeline: meta.rawPipeline },
+                    resultMetaRaw: res,
+                    durMs: Date.now() - meta.t0,
+                    t: alignedNow(),
+                    spanContext,
+                });
+            } finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
 
         schema.post('aggregate', function (this: any, error: any, _res: any, next: Function) {
-            popNativeMongoSuppression((this as any).__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         } as any);
     };
@@ -8810,6 +8848,8 @@ export type MongoDbPatchConfig = ReproMiddlewareConfig & {
 
 const mongoDbPatchConfigs: MongoDbPatchConfig[] = [];
 let mongoDbRequireHookInstalled = false;
+let mongoDbImportHookInstalled = false;
+let mongoDbImportHookReady: Promise<void> | null = null;
 
 const MONGODB_PROMISE_OPS = [
     'insertOne',
@@ -8832,6 +8872,12 @@ const MONGODB_PROMISE_OPS = [
 const MONGODB_CURSOR_OPS = ['find', 'aggregate'] as const;
 const MONGODB_CURSOR_TERMINALS = ['toArray', 'next', 'forEach'] as const;
 
+function registerMongoDbPatchConfig(cfg: MongoDbPatchConfig): void {
+    if (!mongoDbPatchConfigs.includes(cfg)) {
+        mongoDbPatchConfigs.push(cfg);
+    }
+}
+
 function resolveMongoDbRuntimeContext(cfg: MongoDbPatchConfig) {
     const ctx = getCtx();
     const fallback = cfg.resolveContext?.() ?? {};
@@ -8842,6 +8888,18 @@ function resolveMongoDbRuntimeContext(cfg: MongoDbPatchConfig) {
     };
 }
 
+function resolveMongoDbCaptureContext(): { cfg: MongoDbPatchConfig; sid: string; aid: string | null } | null {
+    for (let i = mongoDbPatchConfigs.length - 1; i >= 0; i -= 1) {
+        const cfg = mongoDbPatchConfigs[i];
+        if (!cfg) continue;
+        const { sid, aid } = resolveMongoDbRuntimeContext(cfg);
+        if (sid) {
+            return { cfg, sid, aid };
+        }
+    }
+    return null;
+}
+
 function resolveMongoCollectionName(collection: any): string {
     const candidates = [
         collection?.collectionName,
@@ -9142,10 +9200,11 @@ function patchMongoCollectionMethod(proto: any, op: string, cursorOp: boolean, c
             return original.apply(this, args);
         }
 
-        const { sid, aid } = resolveMongoDbRuntimeContext(cfg);
-        if (!sid) {
+        const captureContext = resolveMongoDbCaptureContext();
+        if (!captureContext) {
             return original.apply(this, args);
         }
+        const { cfg: activeCfg, sid, aid } = captureContext;
 
         const spanContext = captureSpanContextFromTracer(this);
         const t0 = Date.now();
@@ -9157,7 +9216,7 @@ function patchMongoCollectionMethod(proto: any, op: string, cursorOp: boolean, c
         const emitSuccess = (result: any) => {
             if (emitted) return;
             emitted = true;
-            emitDbQuery(cfg, sid, aid ?? undefined, {
+            emitDbQuery(activeCfg, sid, aid ?? undefined, {
                 type: 'mongo',
                 collection,
                 op,
@@ -9174,7 +9233,7 @@ function patchMongoCollectionMethod(proto: any, op: string, cursorOp: boolean, c
         const emitFailure = (error: any) => {
             if (emitted) return;
             emitted = true;
-            emitDbQuery(cfg, sid, aid ?? undefined, {
+            emitDbQuery(activeCfg, sid, aid ?? undefined, {
                 type: 'mongo',
                 collection,
                 op,
@@ -9199,7 +9258,7 @@ function patchMongoCollectionMethod(proto: any, op: string, cursorOp: boolean, c
             const result = original.apply(this, actualArgs);
             if (cursorOp) {
                 return patchMongoCursorInstance(result, {
-                    cfg,
+                    cfg: activeCfg,
                     sid,
                     aid,
                     collection,
@@ -9283,6 +9342,7 @@ function patchMongoClientPrototype(proto: any, cfg: MongoDbPatchConfig): void {
 }
 
 function patchMongoDbFromModuleExports(mongoModule: any, cfg: MongoDbPatchConfig): void {
+    registerMongoDbPatchConfig(cfg);
     if (!mongoModule) return;
     const constructors = new Set<any>();
     const addCtor = (candidate: any) => {
@@ -9343,9 +9403,50 @@ function installMongoDbRequireHook(): void {
     } catch {}
 }
 
+function installMongoDbImportHook(): Promise<void> {
+    if (mongoDbImportHookInstalled) return mongoDbImportHookReady ?? Promise.resolve();
+    mongoDbImportHookInstalled = true;
+
+    mongoDbImportHookReady = (async () => {
+        try {
+            const nodeModule = requireOptionalHostModule('node:module') ?? requireOptionalHostModule('module');
+            if (typeof nodeModule?.register !== 'function') return;
+
+            const iitm = requireOptionalHostModule('import-in-the-middle');
+            const Hook = iitm?.Hook ?? iitm;
+            if (typeof Hook !== 'function') return;
+
+            let waitForAllMessagesAcknowledged: undefined | (() => Promise<void>);
+            let registerOptions: any = { data: { include: ['mongodb'] } };
+            if (typeof iitm?.createAddHookMessageChannel === 'function') {
+                const channel = iitm.createAddHookMessageChannel();
+                registerOptions = channel.registerOptions;
+                waitForAllMessagesAcknowledged = channel.waitForAllMessagesAcknowledged;
+            }
+
+            nodeModule.register('import-in-the-middle/hook.mjs', pathToFileURL(__filename), registerOptions);
+
+            // eslint-disable-next-line no-new
+            new Hook(['mongodb'], (exports: any) => {
+                for (const cfg of mongoDbPatchConfigs) {
+                    patchMongoDbFromModuleExports(exports, cfg);
+                }
+                return exports;
+            });
+
+            if (waitForAllMessagesAcknowledged) {
+                await waitForAllMessagesAcknowledged();
+            }
+        } catch {}
+    })();
+
+    return mongoDbImportHookReady;
+}
+
 export function patchMongoDb(cfg: MongoDbPatchConfig): void {
-    mongoDbPatchConfigs.push(cfg);
+    registerMongoDbPatchConfig(cfg);
     installMongoDbRequireHook();
+    void installMongoDbImportHook();
 
     const mongoModules = new Set<any>();
     for (const loadedMongoModule of getLoadedModuleExportsByPackage('mongodb')) {
@@ -9867,6 +9968,7 @@ export async function initRepro(cfg: ReproInitConfig): Promise<void> {
     patchAmqplib(integrationCfg);
     patchRedis(integrationCfg);
     patchMongoDb(integrationCfg);
+    await installMongoDbImportHook();
     patchSendgridMail(integrationCfg);
     patchMongooseIntegration(reproAutoMiddlewareConfig);
 
@@ -9901,6 +10003,16 @@ export const __reproTestHooks = {
         lastClientActivityAt = 0;
         sdkBackgroundQuietUntil = 0;
     },
+    runWithRequestContextForTest<T>(ctx: Partial<Ctx>, fn: () => T): T {
+        return als.run({
+            ...ctx,
+            excludedSpanIds: ctx.excludedSpanIds ?? new Set<string>(),
+        }, fn);
+    },
+    getNativeMongoSuppressionCountForTest(): number {
+        const suppressions = getCtx().nativeMongoSuppressions;
+        return Array.isArray(suppressions) ? suppressions.length : 0;
+    },
     getRuntimePrivacyPolicyForTest(cfg: ReproMiddlewareConfig): NormalizedRuntimePrivacyPolicy | null {
         return getRuntimePrivacyState(cfg).policy;
     },