repro-nest 0.0.201 → 0.0.203

package/dist/index.js

@@ -1289,6 +1289,27 @@ function reproMiddleware(cfg) {
                                 depth: evt.depth,
                                 library: inferLibraryNameFromFile(evt.file),
                             };
+                            if (evt.type === 'exit') {
+                                // If the function returned a Mongoose Query, update the exit value when the query settles.
+                                const rv = ev.returnValue;
+                                const setReturnValue = (val) => {
+                                    try {
+                                        evt.returnValue = sanitizeTraceValue(val);
+                                    }
+                                    catch { }
+                                };
+                                const qp = rv && rv.__repro_result_promise;
+                                const hasResult = rv && Object.prototype.hasOwnProperty.call(rv, '__repro_result');
+                                if (hasResult) {
+                                    setReturnValue(rv.__repro_result);
+                                }
+                                else if (qp && typeof qp.then === 'function') {
+                                    try {
+                                        qp.then((val) => setReturnValue(val), () => { });
+                                    }
+                                    catch { }
+                                }
+                            }
                             const spanKey = normalizeSpanId(evt.spanId);
                             if (evt.type === 'enter') {
                                 lastEventAt = Date.now();

package/package.json

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

package/src/index.ts

@@ -1499,6 +1499,26 @@ export function reproMiddleware(cfg: ReproMiddlewareConfig) {
                                 library: inferLibraryNameFromFile(evt.file),
                             };
 
+                            if (evt.type === 'exit') {
+                                // If the function returned a Mongoose Query, update the exit value when the query settles.
+                                const rv: any = ev.returnValue;
+                                const setReturnValue = (val: any) => {
+                                    try { evt.returnValue = sanitizeTraceValue(val); } catch {}
+                                };
+                                const qp = rv && rv.__repro_result_promise;
+                                const hasResult = rv && Object.prototype.hasOwnProperty.call(rv, '__repro_result');
+                                if (hasResult) {
+                                    setReturnValue(rv.__repro_result);
+                                } else if (qp && typeof qp.then === 'function') {
+                                    try {
+                                        qp.then(
+                                            (val: any) => setReturnValue(val),
+                                            () => {}
+                                        );
+                                    } catch {}
+                                }
+                            }
+
                             const spanKey = normalizeSpanId(evt.spanId);
                             if (evt.type === 'enter') {
                                 lastEventAt = Date.now();

package/tracer/runtime.js

@@ -645,11 +645,6 @@ if (!global.__repro_call) {
                                 }
                             } catch {}
 
-                            if (isQuery) {
-                                runExit(exitDetailBase);
-                                return out;
-                            }
-
                             let settled = false;
                             const finalize = (value, threw, error) => {
                                 if (settled) return value;