@reproapp/node-sdk 0.0.28 → 0.0.30

package/dist/index.js

@@ -295,6 +295,7 @@ const REQUEST_START_HEADER = 'x-bug-request-start';
 const REPRO_QUERY_SESSION_PARAM = '__repro_sid';
 const REPRO_QUERY_ACTION_PARAM = '__repro_aid';
 const REPRO_QUERY_START_PARAM = '__repro_start';
+const REPRO_REQUEST_MIDDLEWARE_RAN = Symbol.for('__repro_request_middleware_ran');
 const KAFKA_HEADER_TRACE_ID = 'x-repro-trace-id';
 const KAFKA_HEADER_SPAN_ID = 'x-repro-span-id';
 const KAFKA_HEADER_PARENT_SPAN_ID = 'x-repro-parent-span-id';
@@ -1028,10 +1029,6 @@ function scheduleTraceMaterializerIdleShutdown() {
             catch { }
         }
     }, TRACE_MATERIALIZE_WORKER_IDLE_SHUTDOWN_MS);
-    try {
-        traceMaterializerIdleTimer.unref?.();
-    }
-    catch { }
 }
 function rejectTraceMaterializerPending(error) {
     for (const [taskId, pending] of traceMaterializerPending.entries()) {
@@ -1373,11 +1370,43 @@ function waitForSessionDrain(sessionId) {
     if (remaining <= 0)
         return Promise.resolve();
     return new Promise(resolve => {
+        let settled = false;
+        let timeout = null;
+        const removeWaiter = (waiter) => {
+            const list = sessionWaiters.get(sessionId);
+            if (!list)
+                return;
+            const next = list.filter(fn => fn !== waiter);
+            if (next.length) {
+                sessionWaiters.set(sessionId, next);
+            }
+            else {
+                sessionWaiters.delete(sessionId);
+            }
+        };
+        const done = () => {
+            if (settled)
+                return;
+            settled = true;
+            if (timeout) {
+                try {
+                    clearTimeout(timeout);
+                }
+                catch { }
+                timeout = null;
+            }
+            removeWaiter(done);
+            resolve();
+        };
         const list = sessionWaiters.get(sessionId) || [];
-        list.push(resolve);
+        list.push(done);
         sessionWaiters.set(sessionId, list);
         if (SESSION_DRAIN_TIMEOUT_MS > 0 && Number.isFinite(SESSION_DRAIN_TIMEOUT_MS)) {
-            setTimeout(resolve, SESSION_DRAIN_TIMEOUT_MS);
+            timeout = setTimeout(done, SESSION_DRAIN_TIMEOUT_MS);
+            try {
+                timeout.unref?.();
+            }
+            catch { }
         }
     });
 }
@@ -3999,6 +4028,22 @@ function reproMiddleware(cfg) {
     const masking = normalizeMaskingConfig(cfg.masking);
     patchMongoDb(cfg);
     return async function (req, res, next) {
+        const reqAny = req;
+        if (reqAny?.[REPRO_REQUEST_MIDDLEWARE_RAN]) {
+            return next();
+        }
+        try {
+            Object.defineProperty(reqAny, REPRO_REQUEST_MIDDLEWARE_RAN, {
+                value: true,
+                configurable: true,
+            });
+        }
+        catch {
+            try {
+                reqAny[REPRO_REQUEST_MIDDLEWARE_RAN] = true;
+            }
+            catch { }
+        }
         const querySid = readRequestQueryParam(req, REPRO_QUERY_SESSION_PARAM);
         const queryAid = readRequestQueryParam(req, REPRO_QUERY_ACTION_PARAM);
         const queryTs = readRequestQueryNumber(req, REPRO_QUERY_START_PARAM);
@@ -4035,30 +4080,50 @@ function reproMiddleware(cfg) {
         const maskReq = { method: String(req.method || 'GET').toUpperCase(), path: urlPathOnly, key };
         const requestHeaders = sanitizeHeaders(req.headers, cfg.captureHeaders);
         beginSessionRequest(sid);
-        // ---- response body capture (unchanged) ----
+        // ---- response body capture ----
         let capturedBody = undefined;
-        const origJson = res.json.bind(res);
-        res.json = (body) => { capturedBody = body; return origJson(body); };
-        const origSend = res.send.bind(res);
-        res.send = (body) => {
-            if (capturedBody === undefined) {
-                capturedBody = coerceBodyToStorable(body, res.getHeader?.('content-type'));
-            }
-            return origSend(body);
-        };
-        const origWrite = res.write.bind(res);
-        const origEnd = res.end.bind(res);
+        const originalJson = res.json;
+        if (typeof originalJson === 'function') {
+            const origJson = originalJson.bind(res);
+            res.json = (body) => { capturedBody = body; return origJson(body); };
+        }
+        const originalSend = res.send;
+        if (typeof originalSend === 'function') {
+            const origSend = originalSend.bind(res);
+            res.send = (body) => {
+                if (capturedBody === undefined) {
+                    capturedBody = coerceBodyToStorable(body, res.getHeader?.('content-type'));
+                }
+                return origSend(body);
+            };
+        }
+        const origWrite = typeof res.write === 'function'
+            ? res.write.bind(res)
+            : null;
+        const origEnd = typeof res.end === 'function'
+            ? res.end.bind(res)
+            : null;
         const chunks = [];
-        res.write = (chunk, ...args) => { try {
-            if (chunk != null)
-                chunks.push(chunk);
+        if (origWrite) {
+            res.write = (chunk, ...args) => {
+                try {
+                    if (chunk != null)
+                        chunks.push(chunk);
+                }
+                catch { }
+                return origWrite(chunk, ...args);
+            };
         }
-        catch { } return origWrite(chunk, ...args); };
-        res.end = (chunk, ...args) => { try {
-            if (chunk != null)
-                chunks.push(chunk);
+        if (origEnd) {
+            res.end = (chunk, ...args) => {
+                try {
+                    if (chunk != null)
+                        chunks.push(chunk);
+                }
+                catch { }
+                return origEnd(chunk, ...args);
+            };
         }
-        catch { } return origEnd(chunk, ...args); };
         // ---- our ALS (unchanged) ----
         const tracerApiWithTrace = __TRACER__;
         const runInTrace = (fn) => {
@@ -4546,6 +4611,7 @@ function nativeMongoOpsForMongooseOp(op) {
     return [op];
 }
 function reproMongoosePlugin(cfg) {
+    patchMongoDb(cfg);
     return function (schema) {
         // -------- pre/post save (unchanged) --------
         schema.pre('save', { document: true }, function (next) {
@@ -7942,6 +8008,50 @@ function patchMongoCollectionPrototype(proto, cfg) {
         patchMongoCollectionMethod(proto, op, true, cfg);
     }
 }
+function patchMongoCollectionInstance(collection, cfg) {
+    if (!collection || typeof collection !== 'object')
+        return collection;
+    patchMongoCollectionPrototype(Object.getPrototypeOf(collection), cfg);
+    patchMongoCollectionPrototype(collection, cfg);
+    return collection;
+}
+function patchMongoDbPrototype(proto, cfg) {
+    if (!proto)
+        return;
+    const marker = '__repro_mongodb_db_collection_patched';
+    if (proto[marker])
+        return;
+    const original = proto.collection;
+    if (typeof original !== 'function')
+        return;
+    proto[marker] = true;
+    proto.collection = function reproPatchedMongoDbCollection(...args) {
+        const collection = original.apply(this, args);
+        return patchMongoCollectionInstance(collection, cfg);
+    };
+}
+function patchMongoDbInstance(db, cfg) {
+    if (!db || typeof db !== 'object')
+        return db;
+    patchMongoDbPrototype(Object.getPrototypeOf(db), cfg);
+    patchMongoDbPrototype(db, cfg);
+    return db;
+}
+function patchMongoClientPrototype(proto, cfg) {
+    if (!proto)
+        return;
+    const marker = '__repro_mongodb_client_db_patched';
+    if (proto[marker])
+        return;
+    const original = proto.db;
+    if (typeof original !== 'function')
+        return;
+    proto[marker] = true;
+    proto.db = function reproPatchedMongoClientDb(...args) {
+        const db = original.apply(this, args);
+        return patchMongoDbInstance(db, cfg);
+    };
+}
 function patchMongoDbFromModuleExports(mongoModule, cfg) {
     if (!mongoModule)
         return;
@@ -7956,6 +8066,28 @@ function patchMongoDbFromModuleExports(mongoModule, cfg) {
     for (const ctor of constructors) {
         patchMongoCollectionPrototype(ctor.prototype, cfg);
     }
+    const dbConstructors = new Set();
+    const clientConstructors = new Set();
+    const addDbCtor = (candidate) => {
+        if (typeof candidate === 'function' && candidate.prototype) {
+            dbConstructors.add(candidate);
+        }
+    };
+    const addClientCtor = (candidate) => {
+        if (typeof candidate === 'function' && candidate.prototype) {
+            clientConstructors.add(candidate);
+        }
+    };
+    addDbCtor(mongoModule?.Db);
+    addDbCtor(mongoModule?.default?.Db);
+    addClientCtor(mongoModule?.MongoClient);
+    addClientCtor(mongoModule?.default?.MongoClient);
+    for (const ctor of dbConstructors) {
+        patchMongoDbPrototype(ctor.prototype, cfg);
+    }
+    for (const ctor of clientConstructors) {
+        patchMongoClientPrototype(ctor.prototype, cfg);
+    }
 }
 function installMongoDbRequireHook() {
     if (mongoDbRequireHookInstalled)
@@ -7997,9 +8129,13 @@ const REPRO_AUTO_MIDDLEWARE_INSTALLED = Symbol.for('__repro_auto_middleware_inst
 const REPRO_AUTO_NEST_FACTORY_PATCHED = Symbol.for('__repro_auto_nest_factory_patched');
 const REPRO_AUTO_EXPRESS_LISTEN_PATCHED = Symbol.for('__repro_auto_express_listen_patched');
 const REPRO_AUTO_NEST_LISTEN_PATCHED = Symbol.for('__repro_auto_nest_listen_patched');
+const REPRO_AUTO_HTTP_CREATE_SERVER_PATCHED = Symbol.for('__repro_auto_http_create_server_patched');
+const REPRO_AUTO_HTTP_SERVER_PATCHED = Symbol.for('__repro_auto_http_server_patched');
+const REPRO_AUTO_HTTP_LISTENER_WRAPPED = Symbol.for('__repro_auto_http_listener_wrapped');
 const REPRO_AUTO_MONGOOSE_PLUGIN_PATCHED = Symbol.for('__repro_auto_mongoose_plugin_patched');
 let reproAutoMiddlewareConfig = null;
 const mongoosePluginConfigs = [];
+const httpRequestListenerWrappers = new WeakMap();
 let frameworkRequireHookInstalled = false;
 let mongooseRequireHookInstalled = false;
 function hasMarker(target, marker) {
@@ -8104,6 +8240,81 @@ function patchExpressFromModuleExports(expressModule) {
     setMarker(wrappedListen, REPRO_AUTO_EXPRESS_LISTEN_PATCHED);
     application.listen = wrappedListen;
 }
+function wrapHttpRequestListener(listener, cfg) {
+    if (typeof listener !== 'function')
+        return listener;
+    if (hasMarker(listener, REPRO_AUTO_HTTP_LISTENER_WRAPPED))
+        return listener;
+    const existing = httpRequestListenerWrappers.get(listener);
+    if (existing)
+        return existing;
+    const middleware = reproMiddleware(cfg);
+    const wrapped = function reproPatchedHttpRequestListener(req, res) {
+        return middleware(req, res, (err) => {
+            if (err) {
+                try {
+                    if (!res.headersSent) {
+                        res.statusCode = 500;
+                        res.end('Internal Server Error');
+                        return undefined;
+                    }
+                }
+                catch { }
+                throw err;
+            }
+            return listener.call(this, req, res);
+        });
+    };
+    setMarker(wrapped, REPRO_AUTO_HTTP_LISTENER_WRAPPED);
+    httpRequestListenerWrappers.set(listener, wrapped);
+    return wrapped;
+}
+function patchHttpServerRequestListenerMethods(server) {
+    const cfg = reproAutoMiddlewareConfig;
+    if (!cfg || !server || hasMarker(server, REPRO_AUTO_HTTP_SERVER_PATCHED))
+        return;
+    for (const methodName of ['on', 'addListener', 'prependListener', 'once']) {
+        const original = server[methodName];
+        if (typeof original !== 'function')
+            continue;
+        server[methodName] = function reproPatchedHttpServerListenerMethod(...args) {
+            if (args[0] === 'request' && typeof args[1] === 'function' && reproAutoMiddlewareConfig) {
+                args[1] = wrapHttpRequestListener(args[1], reproAutoMiddlewareConfig);
+            }
+            return original.apply(this, args);
+        };
+    }
+    setMarker(server, REPRO_AUTO_HTTP_SERVER_PATCHED);
+}
+function patchHttpCreateServerFromModuleExports(httpModule) {
+    const original = httpModule?.createServer;
+    if (!httpModule || typeof original !== 'function')
+        return;
+    if (hasMarker(original, REPRO_AUTO_HTTP_CREATE_SERVER_PATCHED))
+        return;
+    const wrappedCreateServer = function reproPatchedHttpCreateServer(...args) {
+        const cfg = reproAutoMiddlewareConfig;
+        if (cfg) {
+            const listenerIndex = typeof args[0] === 'function'
+                ? 0
+                : typeof args[1] === 'function'
+                    ? 1
+                    : -1;
+            if (listenerIndex >= 0) {
+                args[listenerIndex] = wrapHttpRequestListener(args[listenerIndex], cfg);
+            }
+        }
+        const server = original.apply(this, args);
+        patchHttpServerRequestListenerMethods(server);
+        return server;
+    };
+    setMarker(wrappedCreateServer, REPRO_AUTO_HTTP_CREATE_SERVER_PATCHED);
+    httpModule.createServer = wrappedCreateServer;
+}
+function patchHttpServerAutoAttach() {
+    patchHttpCreateServerFromModuleExports(requireOptionalHostModule('node:http') ?? requireOptionalHostModule('http'));
+    patchHttpCreateServerFromModuleExports(requireOptionalHostModule('node:https') ?? requireOptionalHostModule('https'));
+}
 function patchKnownFrameworkModules() {
     const nestModules = new Set();
     const resolvedNest = requireOptionalHostModule('@nestjs/core');
@@ -8407,6 +8618,7 @@ async function initRepro(cfg) {
     };
     shareRuntimePrivacyState(reproAutoMiddlewareConfig, integrationCfg);
     installFrameworkRequireHook();
+    patchHttpServerAutoAttach();
     patchKnownFrameworkModules();
     patchHttpClients({
         resolveContext: cfg.resolveContext,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reproapp/node-sdk",
-  "version": "0.0.28",
+  "version": "0.0.30",
   "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/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",
     "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

@@ -313,6 +313,7 @@ const REQUEST_START_HEADER = 'x-bug-request-start';
 const REPRO_QUERY_SESSION_PARAM = '__repro_sid';
 const REPRO_QUERY_ACTION_PARAM = '__repro_aid';
 const REPRO_QUERY_START_PARAM = '__repro_start';
+const REPRO_REQUEST_MIDDLEWARE_RAN = Symbol.for('__repro_request_middleware_ran');
 const KAFKA_HEADER_TRACE_ID = 'x-repro-trace-id';
 const KAFKA_HEADER_SPAN_ID = 'x-repro-span-id';
 const KAFKA_HEADER_PARENT_SPAN_ID = 'x-repro-parent-span-id';
@@ -1429,7 +1430,6 @@ function scheduleTraceMaterializerIdleShutdown(): void {
             try { void worker.terminate?.(); } catch {}
         }
     }, TRACE_MATERIALIZE_WORKER_IDLE_SHUTDOWN_MS);
-    try { (traceMaterializerIdleTimer as any).unref?.(); } catch {}
 }
 
 function rejectTraceMaterializerPending(error: Error): void {
@@ -1745,11 +1745,34 @@ function waitForSessionDrain(sessionId: string): Promise<void> {
     if (remaining <= 0) return Promise.resolve();
 
     return new Promise(resolve => {
+        let settled = false;
+        let timeout: NodeJS.Timeout | null = null;
+        const removeWaiter = (waiter: () => void) => {
+            const list = sessionWaiters.get(sessionId);
+            if (!list) return;
+            const next = list.filter(fn => fn !== waiter);
+            if (next.length) {
+                sessionWaiters.set(sessionId, next);
+            } else {
+                sessionWaiters.delete(sessionId);
+            }
+        };
+        const done = () => {
+            if (settled) return;
+            settled = true;
+            if (timeout) {
+                try { clearTimeout(timeout); } catch {}
+                timeout = null;
+            }
+            removeWaiter(done);
+            resolve();
+        };
         const list = sessionWaiters.get(sessionId) || [];
-        list.push(resolve);
+        list.push(done);
         sessionWaiters.set(sessionId, list);
         if (SESSION_DRAIN_TIMEOUT_MS > 0 && Number.isFinite(SESSION_DRAIN_TIMEOUT_MS)) {
-            setTimeout(resolve, SESSION_DRAIN_TIMEOUT_MS);
+            timeout = setTimeout(done, SESSION_DRAIN_TIMEOUT_MS);
+            try { (timeout as any).unref?.(); } catch {}
         }
     });
 }
@@ -4894,6 +4917,19 @@ export function reproMiddleware(cfg: ReproMiddlewareConfig) {
     const masking = normalizeMaskingConfig(cfg.masking);
     patchMongoDb(cfg);
     return async function (req: Request, res: Response, next: NextFunction) {
+        const reqAny = req as any;
+        if (reqAny?.[REPRO_REQUEST_MIDDLEWARE_RAN]) {
+            return next();
+        }
+        try {
+            Object.defineProperty(reqAny, REPRO_REQUEST_MIDDLEWARE_RAN, {
+                value: true,
+                configurable: true,
+            });
+        } catch {
+            try { reqAny[REPRO_REQUEST_MIDDLEWARE_RAN] = true; } catch {}
+        }
+
         const querySid = readRequestQueryParam(req, REPRO_QUERY_SESSION_PARAM);
         const queryAid = readRequestQueryParam(req, REPRO_QUERY_ACTION_PARAM);
         const queryTs = readRequestQueryNumber(req, REPRO_QUERY_START_PARAM);
@@ -4933,24 +4969,44 @@ export function reproMiddleware(cfg: ReproMiddlewareConfig) {
         const requestHeaders = sanitizeHeaders(req.headers, cfg.captureHeaders);
         beginSessionRequest(sid);
 
-        // ---- response body capture (unchanged) ----
+        // ---- response body capture ----
         let capturedBody: any = undefined;
-        const origJson = res.json.bind(res as any);
-        (res as any).json = (body: any) => { capturedBody = body; return origJson(body); };
-
-        const origSend = res.send.bind(res as any);
-        (res as any).send = (body: any) => {
-            if (capturedBody === undefined) {
-                capturedBody = coerceBodyToStorable(body, res.getHeader?.('content-type'));
-            }
-            return origSend(body);
-        };
+        const originalJson = (res as any).json;
+        if (typeof originalJson === 'function') {
+            const origJson = originalJson.bind(res as any);
+            (res as any).json = (body: any) => { capturedBody = body; return origJson(body); };
+        }
+
+        const originalSend = (res as any).send;
+        if (typeof originalSend === 'function') {
+            const origSend = originalSend.bind(res as any);
+            (res as any).send = (body: any) => {
+                if (capturedBody === undefined) {
+                    capturedBody = coerceBodyToStorable(body, res.getHeader?.('content-type'));
+                }
+                return origSend(body);
+            };
+        }
 
-        const origWrite = (res as any).write.bind(res as any);
-        const origEnd = (res as any).end.bind(res as any);
+        const origWrite = typeof (res as any).write === 'function'
+            ? (res as any).write.bind(res as any)
+            : null;
+        const origEnd = typeof (res as any).end === 'function'
+            ? (res as any).end.bind(res as any)
+            : null;
         const chunks: Array<Buffer | string> = [];
-        (res as any).write = (chunk: any, ...args: any[]) => { try { if (chunk != null) chunks.push(chunk); } catch {} return origWrite(chunk, ...args); };
-        (res as any).end = (chunk?: any, ...args: any[]) => { try { if (chunk != null) chunks.push(chunk); } catch {} return origEnd(chunk, ...args); };
+        if (origWrite) {
+            (res as any).write = (chunk: any, ...args: any[]) => {
+                try { if (chunk != null) chunks.push(chunk); } catch {}
+                return origWrite(chunk, ...args);
+            };
+        }
+        if (origEnd) {
+            (res as any).end = (chunk?: any, ...args: any[]) => {
+                try { if (chunk != null) chunks.push(chunk); } catch {}
+                return origEnd(chunk, ...args);
+            };
+        }
 
         // ---- our ALS (unchanged) ----
         const tracerApiWithTrace = __TRACER__ as (TracerApi & { withTrace?: (id: string, fn: () => void) => void }) | null;
@@ -5490,6 +5546,7 @@ function nativeMongoOpsForMongooseOp(op: string): string[] {
 }
 
 export function reproMongoosePlugin(cfg: ReproMiddlewareConfig) {
+    patchMongoDb(cfg);
     return function (schema: Schema) {
         // -------- pre/post save (unchanged) --------
         schema.pre('save', { document: true }, function (next) {
@@ -9185,6 +9242,46 @@ function patchMongoCollectionPrototype(proto: any, cfg: MongoDbPatchConfig): voi
     }
 }
 
+function patchMongoCollectionInstance(collection: any, cfg: MongoDbPatchConfig): any {
+    if (!collection || typeof collection !== 'object') return collection;
+    patchMongoCollectionPrototype(Object.getPrototypeOf(collection), cfg);
+    patchMongoCollectionPrototype(collection, cfg);
+    return collection;
+}
+
+function patchMongoDbPrototype(proto: any, cfg: MongoDbPatchConfig): void {
+    if (!proto) return;
+    const marker = '__repro_mongodb_db_collection_patched';
+    if (proto[marker]) return;
+    const original = proto.collection;
+    if (typeof original !== 'function') return;
+    proto[marker] = true;
+    proto.collection = function reproPatchedMongoDbCollection(this: any, ...args: any[]) {
+        const collection = original.apply(this, args);
+        return patchMongoCollectionInstance(collection, cfg);
+    };
+}
+
+function patchMongoDbInstance(db: any, cfg: MongoDbPatchConfig): any {
+    if (!db || typeof db !== 'object') return db;
+    patchMongoDbPrototype(Object.getPrototypeOf(db), cfg);
+    patchMongoDbPrototype(db, cfg);
+    return db;
+}
+
+function patchMongoClientPrototype(proto: any, cfg: MongoDbPatchConfig): void {
+    if (!proto) return;
+    const marker = '__repro_mongodb_client_db_patched';
+    if (proto[marker]) return;
+    const original = proto.db;
+    if (typeof original !== 'function') return;
+    proto[marker] = true;
+    proto.db = function reproPatchedMongoClientDb(this: any, ...args: any[]) {
+        const db = original.apply(this, args);
+        return patchMongoDbInstance(db, cfg);
+    };
+}
+
 function patchMongoDbFromModuleExports(mongoModule: any, cfg: MongoDbPatchConfig): void {
     if (!mongoModule) return;
     const constructors = new Set<any>();
@@ -9200,6 +9297,31 @@ function patchMongoDbFromModuleExports(mongoModule: any, cfg: MongoDbPatchConfig
     for (const ctor of constructors) {
         patchMongoCollectionPrototype(ctor.prototype, cfg);
     }
+
+    const dbConstructors = new Set<any>();
+    const clientConstructors = new Set<any>();
+    const addDbCtor = (candidate: any) => {
+        if (typeof candidate === 'function' && candidate.prototype) {
+            dbConstructors.add(candidate);
+        }
+    };
+    const addClientCtor = (candidate: any) => {
+        if (typeof candidate === 'function' && candidate.prototype) {
+            clientConstructors.add(candidate);
+        }
+    };
+
+    addDbCtor(mongoModule?.Db);
+    addDbCtor(mongoModule?.default?.Db);
+    addClientCtor(mongoModule?.MongoClient);
+    addClientCtor(mongoModule?.default?.MongoClient);
+
+    for (const ctor of dbConstructors) {
+        patchMongoDbPrototype(ctor.prototype, cfg);
+    }
+    for (const ctor of clientConstructors) {
+        patchMongoClientPrototype(ctor.prototype, cfg);
+    }
 }
 
 function installMongoDbRequireHook(): void {
@@ -9243,10 +9365,14 @@ const REPRO_AUTO_MIDDLEWARE_INSTALLED = Symbol.for('__repro_auto_middleware_inst
 const REPRO_AUTO_NEST_FACTORY_PATCHED = Symbol.for('__repro_auto_nest_factory_patched');
 const REPRO_AUTO_EXPRESS_LISTEN_PATCHED = Symbol.for('__repro_auto_express_listen_patched');
 const REPRO_AUTO_NEST_LISTEN_PATCHED = Symbol.for('__repro_auto_nest_listen_patched');
+const REPRO_AUTO_HTTP_CREATE_SERVER_PATCHED = Symbol.for('__repro_auto_http_create_server_patched');
+const REPRO_AUTO_HTTP_SERVER_PATCHED = Symbol.for('__repro_auto_http_server_patched');
+const REPRO_AUTO_HTTP_LISTENER_WRAPPED = Symbol.for('__repro_auto_http_listener_wrapped');
 const REPRO_AUTO_MONGOOSE_PLUGIN_PATCHED = Symbol.for('__repro_auto_mongoose_plugin_patched');
 
 let reproAutoMiddlewareConfig: ReproMiddlewareConfig | null = null;
 const mongoosePluginConfigs: ReproMiddlewareConfig[] = [];
+const httpRequestListenerWrappers = new WeakMap<Function, Function>();
 let frameworkRequireHookInstalled = false;
 let mongooseRequireHookInstalled = false;
 
@@ -9347,6 +9473,82 @@ function patchExpressFromModuleExports(expressModule: any): void {
     application.listen = wrappedListen;
 }
 
+function wrapHttpRequestListener(listener: any, cfg: ReproMiddlewareConfig): any {
+    if (typeof listener !== 'function') return listener;
+    if (hasMarker(listener, REPRO_AUTO_HTTP_LISTENER_WRAPPED)) return listener;
+
+    const existing = httpRequestListenerWrappers.get(listener);
+    if (existing) return existing;
+
+    const middleware = reproMiddleware(cfg);
+    const wrapped = function reproPatchedHttpRequestListener(this: any, req: any, res: any) {
+        return middleware(req, res, (err?: any) => {
+            if (err) {
+                try {
+                    if (!res.headersSent) {
+                        res.statusCode = 500;
+                        res.end('Internal Server Error');
+                        return undefined;
+                    }
+                } catch {}
+                throw err;
+            }
+            return listener.call(this, req, res);
+        });
+    };
+    setMarker(wrapped, REPRO_AUTO_HTTP_LISTENER_WRAPPED);
+    httpRequestListenerWrappers.set(listener, wrapped);
+    return wrapped;
+}
+
+function patchHttpServerRequestListenerMethods(server: any): void {
+    const cfg = reproAutoMiddlewareConfig;
+    if (!cfg || !server || hasMarker(server, REPRO_AUTO_HTTP_SERVER_PATCHED)) return;
+
+    for (const methodName of ['on', 'addListener', 'prependListener', 'once']) {
+        const original = server[methodName];
+        if (typeof original !== 'function') continue;
+        server[methodName] = function reproPatchedHttpServerListenerMethod(this: any, ...args: any[]) {
+            if (args[0] === 'request' && typeof args[1] === 'function' && reproAutoMiddlewareConfig) {
+                args[1] = wrapHttpRequestListener(args[1], reproAutoMiddlewareConfig);
+            }
+            return original.apply(this, args);
+        };
+    }
+    setMarker(server, REPRO_AUTO_HTTP_SERVER_PATCHED);
+}
+
+function patchHttpCreateServerFromModuleExports(httpModule: any): void {
+    const original = httpModule?.createServer;
+    if (!httpModule || typeof original !== 'function') return;
+    if (hasMarker(original, REPRO_AUTO_HTTP_CREATE_SERVER_PATCHED)) return;
+
+    const wrappedCreateServer = function reproPatchedHttpCreateServer(this: any, ...args: any[]) {
+        const cfg = reproAutoMiddlewareConfig;
+        if (cfg) {
+            const listenerIndex =
+                typeof args[0] === 'function'
+                    ? 0
+                    : typeof args[1] === 'function'
+                        ? 1
+                        : -1;
+            if (listenerIndex >= 0) {
+                args[listenerIndex] = wrapHttpRequestListener(args[listenerIndex], cfg);
+            }
+        }
+        const server = original.apply(this, args);
+        patchHttpServerRequestListenerMethods(server);
+        return server;
+    };
+    setMarker(wrappedCreateServer, REPRO_AUTO_HTTP_CREATE_SERVER_PATCHED);
+    httpModule.createServer = wrappedCreateServer;
+}
+
+function patchHttpServerAutoAttach(): void {
+    patchHttpCreateServerFromModuleExports(requireOptionalHostModule('node:http') ?? requireOptionalHostModule('http'));
+    patchHttpCreateServerFromModuleExports(requireOptionalHostModule('node:https') ?? requireOptionalHostModule('https'));
+}
+
 function patchKnownFrameworkModules(): void {
     const nestModules = new Set<any>();
     const resolvedNest = requireOptionalHostModule('@nestjs/core');
@@ -9654,6 +9856,7 @@ export async function initRepro(cfg: ReproInitConfig): Promise<void> {
     shareRuntimePrivacyState(reproAutoMiddlewareConfig, integrationCfg);
 
     installFrameworkRequireHook();
+    patchHttpServerAutoAttach();
     patchKnownFrameworkModules();
 
     patchHttpClients({

package/test/init-auto-http-server.test.js

@@ -0,0 +1,170 @@
+process.env.REPRO_SDK_BACKGROUND_MAX_DEFER_MS = '50';
+process.env.REPRO_SDK_BACKGROUND_IDLE_DELAY_MS = '25';
+process.env.TRACE_IDLE_FLUSH_MS = '10';
+process.env.TRACE_LINGER_AFTER_FINISH_MS = '0';
+
+const assert = require('node:assert/strict');
+const http = require('node:http');
+const mongodb = require('mongodb');
+
+const originalNativeMongoInsertOne = mongodb.Collection.prototype.insertOne;
+mongodb.Collection.prototype.insertOne = async function reproFakeNativeMongoInsertOne(doc, options) {
+    assert.equal(options.session.id, 'auto-http-native-session-1');
+    doc._id = doc._id ?? 'subject-auto-http-1';
+    return { acknowledged: true, insertedId: doc._id };
+};
+
+const { initRepro, reproMiddleware } = require('../dist');
+const { flushIngestQueue } = require('../dist/ingest/client');
+
+const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+
+async function waitFor(predicate, timeoutMs = 3000) {
+    const deadline = Date.now() + timeoutMs;
+    while (Date.now() < deadline) {
+        if (predicate()) return;
+        await sleep(25);
+    }
+    assert(predicate(), 'timed out waiting for expected condition');
+}
+
+function flattenEvents(posts) {
+    return posts.flatMap((body) => Array.isArray(body?.events) ? body.events : []);
+}
+
+function backendRequests(posts, sessionId) {
+    return flattenEvents(posts).filter((event) => {
+        return event?.event_type === 'backend_request' && event?.session_id === sessionId;
+    });
+}
+
+function dbChanges(posts, sessionId) {
+    return flattenEvents(posts)
+        .filter((event) => event?.event_type === 'db_change' && event?.session_id === sessionId)
+        .flatMap((event) => Array.isArray(event?.payload?.db) ? event.payload.db : []);
+}
+
+async function listen(server) {
+    await new Promise((resolve) => server.listen(0, '127.0.0.1', resolve));
+    return server.address().port;
+}
+
+async function close(server) {
+    await new Promise((resolve, reject) => {
+        server.close((err) => (err ? reject(err) : resolve()));
+    });
+}
+
+async function request(port, sessionId, actionId) {
+    await new Promise((resolve, reject) => {
+        const req = http.request({
+            host: '127.0.0.1',
+            port,
+            path: '/subjects',
+            method: 'POST',
+            headers: {
+                'content-type': 'application/json',
+                'x-bug-session-id': sessionId,
+                'x-bug-action-id': actionId,
+                'x-bug-request-start': String(Date.now()),
+            },
+        }, (res) => {
+            res.resume();
+            res.on('end', resolve);
+        });
+        req.on('error', reject);
+        req.end(JSON.stringify({ ok: true }));
+    });
+}
+
+async function main() {
+    const capturedBodies = [];
+    const originalFetch = global.fetch;
+    global.fetch = async (_url, init = {}) => {
+        capturedBodies.push(JSON.parse(String(init.body || '{}')));
+        return {
+            ok: true,
+            status: 202,
+            json: async () => ({ ok: true }),
+            text: async () => '{"ok":true}',
+        };
+    };
+
+    const cfg = {
+        tenantId: 'TENANT_test',
+        appId: 'APP_test',
+        appSecret: 'secret',
+        appName: 'auto-http-test',
+        serviceName: 'Auto HTTP Test',
+        captureHeaders: false,
+        privacy: { environment: 'dev', captureRaw: true },
+        ingestBase: 'http://127.0.0.1:65535',
+    };
+
+    let autoServer;
+    let doubleServer;
+    try {
+        await initRepro(cfg);
+
+        const subjectsCollection = new mongodb.MongoClient('mongodb://127.0.0.1:27017/clinical')
+            .db('clinical')
+            .collection('subjects');
+
+        autoServer = http.createServer(async (_req, res) => {
+            await subjectsCollection.insertOne(
+                { subjectNumber: '10013038', status: 'screened' },
+                {
+                    session: {
+                        id: 'auto-http-native-session-1',
+                        hasEnded: false,
+                        inTransaction: () => true,
+                    },
+                },
+            );
+            res.setHeader('content-type', 'application/json');
+            res.end(JSON.stringify({ ok: true, mode: 'auto' }));
+        });
+        const autoPort = await listen(autoServer);
+        await request(autoPort, 'S_auto_http_only', 'A_auto_http_only');
+        await waitFor(() => backendRequests(capturedBodies, 'S_auto_http_only').length === 1);
+        await waitFor(() => dbChanges(capturedBodies, 'S_auto_http_only').length === 1);
+        assert.equal(backendRequests(capturedBodies, 'S_auto_http_only').length, 1);
+        const [nativeMongoChange] = dbChanges(capturedBodies, 'S_auto_http_only');
+        assert.equal(nativeMongoChange.type, 'mongo');
+        assert.equal(nativeMongoChange.collection, 'subjects');
+        assert.equal(nativeMongoChange.op, 'insertOne');
+        assert.equal(nativeMongoChange.pk._id, 'subject-auto-http-1');
+        assert.equal(nativeMongoChange.query.driver, 'mongodb');
+        assert.equal(nativeMongoChange.query.db, 'clinical');
+        assert.equal(nativeMongoChange.query.doc.subjectNumber, '10013038');
+        assert.equal(nativeMongoChange.query.options.session.__kind, 'mongodb_client_session');
+        assert.equal(nativeMongoChange.query.options.session.id, 'auto-http-native-session-1');
+        assert.equal(nativeMongoChange.after.subjectNumber, '10013038');
+        assert.equal(nativeMongoChange.resultMeta.insertedId, 'subject-auto-http-1');
+
+        const manual = reproMiddleware(cfg);
+        doubleServer = http.createServer((req, res) => {
+            manual(req, res, () => {
+                res.setHeader('content-type', 'application/json');
+                res.end(JSON.stringify({ ok: true, mode: 'manual-too' }));
+            });
+        });
+        const doublePort = await listen(doubleServer);
+        await request(doublePort, 'S_auto_http_no_duplicate', 'A_auto_http_no_duplicate');
+        await waitFor(() => backendRequests(capturedBodies, 'S_auto_http_no_duplicate').length === 1);
+        assert.equal(backendRequests(capturedBodies, 'S_auto_http_no_duplicate').length, 1);
+
+        console.log('initRepro auto HTTP server middleware OK');
+    } finally {
+        if (autoServer?.listening) await close(autoServer);
+        if (doubleServer?.listening) await close(doubleServer);
+        await flushIngestQueue();
+        mongodb.Collection.prototype.insertOne = originalNativeMongoInsertOne;
+        global.fetch = originalFetch;
+    }
+}
+
+main().catch((error) => {
+    console.error(error);
+    process.exit(1);
+});

package/test/mongodb-integration.test.js

@@ -176,6 +176,83 @@ async function testReproMiddlewarePatchesPreloadedNativeMongoClientInsertOne() {
     });
 }
 
+async function testMongoClientDbCollectionChainIsPatchedWithoutCollectionExport() {
+    await withCapturedFetch(async (capturedBodies) => {
+        __reproTestHooks.resetSdkBackgroundQueuesForTest();
+        const sessionId = 'S_mongo_client_chain';
+        const actionId = 'A_mongo_client_chain';
+        const cfg = makeConfig(sessionId, actionId);
+
+        class FakeCollection {
+            constructor(dbName, collectionName) {
+                this.dbName = dbName;
+                this.collectionName = collectionName;
+            }
+
+            async insertOne(doc, options) {
+                assert.strictEqual(options.session.id, 'native-chain-session-1');
+                doc._id = doc._id ?? 'subject-chain-1';
+                return { acknowledged: true, insertedId: doc._id };
+            }
+        }
+
+        class FakeDb {
+            constructor(dbName) {
+                this.databaseName = dbName;
+            }
+
+            collection(collectionName) {
+                return new FakeCollection(this.databaseName, collectionName);
+            }
+        }
+
+        class FakeMongoClient {
+            db(dbName) {
+                return new FakeDb(dbName);
+            }
+        }
+
+        __reproTestHooks.patchMongoDbModuleForTest({
+            MongoClient: FakeMongoClient,
+            Db: FakeDb,
+        }, cfg);
+
+        const dbClient = new FakeMongoClient();
+        const subjectsCollection = dbClient.db('clinical').collection('subjects');
+
+        await runCapturedRequest(cfg, sessionId, actionId, async () => {
+            const result = await subjectsCollection.insertOne(
+                { subjectNumber: '10013038', status: 'screened' },
+                {
+                    session: {
+                        id: 'native-chain-session-1',
+                        hasEnded: false,
+                        inTransaction: () => true,
+                    },
+                },
+            );
+            assert.deepStrictEqual(result, { acknowledged: true, insertedId: 'subject-chain-1' });
+        });
+
+        await waitFor(() => dbChanges(flattenEvents(capturedBodies)).length === 1);
+        await drainSdk();
+
+        const [change] = dbChanges(flattenEvents(capturedBodies));
+        assert.strictEqual(change.type, 'mongo');
+        assert.strictEqual(change.collection, 'subjects');
+        assert.strictEqual(change.op, 'insertOne');
+        assert.strictEqual(change.pk._id, 'subject-chain-1');
+        assert.strictEqual(change.query.driver, 'mongodb');
+        assert.strictEqual(change.query.db, 'clinical');
+        assert.strictEqual(change.query.doc.subjectNumber, '10013038');
+        assert.strictEqual(change.query.options.session.__kind, 'mongodb_client_session');
+        assert.strictEqual(change.query.options.session.id, 'native-chain-session-1');
+        assert.strictEqual(change.after.subjectNumber, '10013038');
+        assert.strictEqual(change.resultMeta.insertedId, 'subject-chain-1');
+        assert.strictEqual(change.resultMeta.acknowledged, true);
+    });
+}
+
 async function testNativeInsertOneIsCaptured() {
     await withCapturedFetch(async (capturedBodies) => {
         __reproTestHooks.resetSdkBackgroundQueuesForTest();
@@ -376,6 +453,7 @@ async function testNativeOperationFailureIsCapturedAndRethrown() {
 
 (async () => {
     await testReproMiddlewarePatchesPreloadedNativeMongoClientInsertOne();
+    await testMongoClientDbCollectionChainIsPatchedWithoutCollectionExport();
     await testNativeInsertOneIsCaptured();
     await testNativeUpdateOneUpsertIsCaptured();
     await testNativeFindCursorToArrayIsCaptured();