repro-nest 0.0.205 → 0.0.208

package/dist/index.d.ts

@@ -29,6 +29,11 @@ type TracerApi = {
     init?: (opts: any) => void;
     tracer?: {
         on: (fn: (ev: any) => void) => () => void;
+        getCurrentTraceContext?: () => {
+            traceId: any;
+            spanId: any;
+            depth?: number;
+        };
     };
     getCurrentTraceId?: () => string | null;
     patchHttp?: () => void;

package/dist/index.js

@@ -439,6 +439,13 @@ function initReproTracing(opts) {
         const initOpts = { ...defaultTracerInitOpts(), ...rest };
         tracerPkg.init?.(initOpts);
         tracerPkg.patchHttp?.();
+        // Ensure tracer exposes current trace context on the on-event API when available.
+        try {
+            if (!tracerPkg.tracer?.getCurrentTraceContext && tracerPkg.getCurrentTraceContext) {
+                tracerPkg.tracer.getCurrentTraceContext = tracerPkg.getCurrentTraceContext;
+            }
+        }
+        catch { }
         applyTraceLogPreference(tracerPkg);
         __TRACER_READY = true;
         patchAllKnownMongooseInstances();
@@ -800,7 +807,8 @@ const SESSION_DRAIN_TIMEOUT_MS = (() => {
     if (Number.isFinite(env) && env >= 0)
         return env;
     // Bound wait for draining sessions to avoid lost flushes when a request hangs.
-    return 10000;
+    // Default to 1 minute; caller can override via env.
+    return 60000;
 })();
 function isThenable(value) {
     return value != null && typeof value === 'object' && typeof value.then === 'function';
@@ -1327,6 +1335,9 @@ function reproMiddleware(cfg) {
                             if (ev.args !== undefined) {
                                 evt.args = sanitizeTraceArgs(ev.args);
                             }
+                            if (ev.receiver !== undefined) {
+                                evt.receiver = sanitizeTraceValue(ev.receiver);
+                            }
                             if (ev.returnValue !== undefined) {
                                 evt.returnValue = sanitizeTraceValue(ev.returnValue);
                             }
@@ -1888,6 +1899,12 @@ function dehydrateComplexValue(value) {
 function emitDbQuery(cfg, sid, aid, payload) {
     if (!sid)
         return;
+    let traceCtx = null;
+    try {
+        const getCtx = __TRACER__?.tracer?.getCurrentTraceContext;
+        traceCtx = typeof getCtx === 'function' ? getCtx() : null;
+    }
+    catch { }
     post(cfg.apiBase, cfg.tenantId, cfg.appId, cfg.appSecret, sid, {
         entries: [{
                 actionId: aid ?? null,
@@ -1897,6 +1914,8 @@ function emitDbQuery(cfg, sid, aid, payload) {
                         query: payload.query ?? undefined,
                         resultMeta: payload.resultMeta ?? undefined,
                         durMs: payload.durMs ?? undefined,
+                        traceId: traceCtx?.traceId ?? null,
+                        spanId: traceCtx?.spanId ?? null,
                         pk: null, before: null, after: null,
                         error: payload.error ?? undefined,
                     }],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "repro-nest",
-  "version": "0.0.205",
+  "version": "0.0.208",
   "description": "Repro Nest SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -174,7 +174,10 @@ function patchAllKnownMongooseInstances() {
 // ====== tiny, safe tracer auto-init (no node_modules patches) ======
 type TracerApi = {
     init?: (opts: any) => void;
-    tracer?: { on: (fn: (ev: any) => void) => () => void };
+    tracer?: {
+        on: (fn: (ev: any) => void) => () => void;
+        getCurrentTraceContext?: () => { traceId: any; spanId: any; depth?: number };
+    };
     getCurrentTraceId?: () => string | null;
     patchHttp?: () => void; // optional in your tracer
     setFunctionLogsEnabled?: (enabled: boolean) => void;
@@ -210,6 +213,7 @@ type TraceEventRecord = {
     spanId?: string | number | null;
     parentSpanId?: string | number | null;
     args?: any;
+    receiver?: any;
     returnValue?: any;
     threw?: boolean;
     error?: any;
@@ -622,6 +626,12 @@ export function initReproTracing(opts?: ReproTracingInitOptions) {
         const initOpts = { ...defaultTracerInitOpts(), ...(rest as TracerInitOpts) };
         tracerPkg.init?.(initOpts);
         tracerPkg.patchHttp?.();
+        // Ensure tracer exposes current trace context on the on-event API when available.
+        try {
+            if (!tracerPkg.tracer?.getCurrentTraceContext && (tracerPkg as any).getCurrentTraceContext) {
+                (tracerPkg.tracer as any).getCurrentTraceContext = (tracerPkg as any).getCurrentTraceContext;
+            }
+        } catch {}
         applyTraceLogPreference(tracerPkg);
         __TRACER_READY = true;
         patchAllKnownMongooseInstances();
@@ -1023,7 +1033,8 @@ const SESSION_DRAIN_TIMEOUT_MS = (() => {
     const env = Number(process.env.SESSION_DRAIN_TIMEOUT_MS);
     if (Number.isFinite(env) && env >= 0) return env;
     // Bound wait for draining sessions to avoid lost flushes when a request hangs.
-    return 10000;
+    // Default to 1 minute; caller can override via env.
+    return 60000;
 })();
 
 function isThenable(value: any): value is PromiseLike<any> {
@@ -1532,6 +1543,9 @@ export function reproMiddleware(cfg: ReproMiddlewareConfig) {
                             if (ev.args !== undefined) {
                                 evt.args = sanitizeTraceArgs(ev.args);
                             }
+                            if (ev.receiver !== undefined) {
+                                evt.receiver = sanitizeTraceValue(ev.receiver);
+                            }
                             if (ev.returnValue !== undefined) {
                                 evt.returnValue = sanitizeTraceValue(ev.returnValue);
                             }
@@ -2100,6 +2114,12 @@ function dehydrateComplexValue(value: any) {
 
 function emitDbQuery(cfg: any, sid?: string, aid?: string, payload?: any) {
     if (!sid) return;
+    let traceCtx: { traceId: any; spanId: any } | null = null;
+    try {
+        const getCtx = __TRACER__?.tracer?.getCurrentTraceContext;
+        traceCtx = typeof getCtx === 'function' ? getCtx() : null;
+    } catch {}
+
     post(cfg.apiBase, cfg.tenantId, cfg.appId, cfg.appSecret, sid, {
         entries: [{
             actionId: aid ?? null,
@@ -2109,6 +2129,8 @@ function emitDbQuery(cfg: any, sid?: string, aid?: string, payload?: any) {
                 query: payload.query ?? undefined,
                 resultMeta: payload.resultMeta ?? undefined,
                 durMs: payload.durMs ?? undefined,
+                traceId: traceCtx?.traceId ?? null,
+                spanId: traceCtx?.spanId ?? null,
                 pk: null, before: null, after: null,
                 error: payload.error ?? undefined,
             }],

package/tracer/runtime.js

@@ -100,6 +100,17 @@ function popSpan(ctx) {
 const trace = {
     on(fn){ listeners.add(fn); return () => listeners.delete(fn); },
     withTrace(id, fn, depth = 0){ return als.run({ traceId: id, depth }, fn); },
+    getCurrentTraceContext(){
+        const store = als.getStore();
+        if (!store) return { traceId: null, spanId: null, depth: 0 };
+        const spanStack = Array.isArray(store.__repro_span_stack) ? store.__repro_span_stack : [];
+        const top = spanStack.length ? spanStack[spanStack.length - 1] : null;
+        return {
+            traceId: store.traceId || null,
+            spanId: top ? top.id ?? null : null,
+            depth: typeof store.depth === 'number' ? store.depth : (top?.depth ?? 0),
+        };
+    },
     enter(fn, meta, detail){
         const parentSpanIdOverride = meta && Object.prototype.hasOwnProperty.call(meta, 'parentSpanId')
             ? meta.parentSpanId
@@ -132,6 +143,7 @@ const trace = {
             traceId: ctx.traceId,
             depth: ctx.depth,
             args: detail?.args,
+            receiver: detail?.receiver,
             spanId: span.id,
             parentSpanId: span.parentId
         });
@@ -159,7 +171,8 @@ const trace = {
             returnValue: detail?.returnValue,
             error: detail?.error,
             threw: detail?.threw === true,
-            unawaited: detail?.unawaited === true || frameUnawaited
+            unawaited: detail?.unawaited === true || frameUnawaited,
+            receiver: detail?.receiver
         };
 
         const promiseTaggedUnawaited = !!(baseDetail.returnValue && baseDetail.returnValue[SYM_UNAWAITED]);
@@ -179,6 +192,9 @@ const trace = {
                 unawaited: overrides.hasOwnProperty('unawaited')
                     ? overrides.unawaited
                     : forceUnawaited,
+                receiver: overrides.hasOwnProperty('receiver')
+                    ? overrides.receiver
+                    : baseDetail.receiver,
                 args: overrides.hasOwnProperty('args')
                     ? overrides.args
                     : baseDetail.args
@@ -199,6 +215,7 @@ const trace = {
                 threw: finalDetail.threw === true,
                 error: finalDetail.error,
                 args: finalDetail.args,
+                receiver: finalDetail.receiver,
                 unawaited: finalDetail.unawaited === true
             });
         };
@@ -475,6 +492,24 @@ function forkAlsStoreForUnawaited(baseStore) {
     return cloned;
 }
 
+// Only capture receiver/collection snapshots for common array iteration helpers to keep noise low.
+const RECEIVER_METHOD_NAMES = new Set([
+    // Array iterators
+    'map', 'forEach', 'filter', 'find', 'findIndex', 'findLast', 'findLastIndex',
+    'some', 'every', 'reduce', 'reduceRight', 'flatMap', 'flat',
+    // Set / Map mutators
+    'add', 'delete', 'clear', 'set'
+]);
+
+function shouldCaptureReceiver(label, receiver) {
+    if (!label || receiver === null || receiver === undefined) return false;
+    const method = String(label).split('.').pop() || '';
+    if (!RECEIVER_METHOD_NAMES.has(method)) return false;
+    if (Array.isArray(receiver)) return true;
+    if (typeof receiver === 'object' && typeof receiver[Symbol.iterator] === 'function') return true;
+    return false;
+}
+
 // ========= Generic call-site shim (used by Babel transform) =========
 // Decides whether to emit a top-level event based on callee origin tags.
 // No hardcoded library names or file paths.
@@ -596,13 +631,15 @@ if (!global.__repro_call) {
                         ? (label && label.length ? label : fn.name)
                         : '(anonymous)';
                     const sourceFile = fn[SYM_SRC_FILE];
+                    const receiverForTrace = shouldCaptureReceiver(label, thisArg) ? thisArg : undefined;
                     const meta = {
-                        file: sourceFile || callFile || null,
-                        line: sourceFile ? null : (callLine || null),
+                        // Prefer callsite file/line so traces point to where the function was invoked.
+                        file: callFile || sourceFile || null,
+                        line: callLine ?? null,
                         parentSpanId: forcedParentSpanId
                     };
 
-                    trace.enter(name, meta, { args });
+                    trace.enter(name, meta, { args, receiver: receiverForTrace });
                     // After entering, snapshot a clean store that captures the parent + this call’s span,
                     // so callbacks invoked during the callee run are isolated per invocation.
                     const snapshotForArgs = cloneStore(als.getStore());
@@ -617,7 +654,8 @@ if (!global.__repro_call) {
                         const exitDetailBase = {
                             returnValue: out,
                             args,
-                            unawaited: shouldForceExit
+                            unawaited: shouldForceExit,
+                            receiver: receiverForTrace
                         };
 
                         if (shouldForceExit) markPromiseUnawaited(out);
@@ -659,7 +697,8 @@ if (!global.__repro_call) {
                                     args,
                                     threw,
                                     error,
-                                    unawaited: shouldForceExit
+                                    unawaited: shouldForceExit,
+                                    receiver: receiverForTrace
                                 };
                                 runExit(detail);
                                 return value;
@@ -679,7 +718,7 @@ if (!global.__repro_call) {
                         runExit(exitDetailBase);
                         return out;
                     } catch (e) {
-                        trace.exit({ fn: name, file: meta.file, line: meta.line }, { threw: true, error: e, args });
+                        trace.exit({ fn: name, file: meta.file, line: meta.line }, { threw: true, error: e, args, receiver: receiverForTrace });
                         throw e;
                     } finally {
                         if (pendingMarker) {