bitfab 0.36.5 → 0.36.6

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.36.5";
+    __version__ = "0.36.6";
   }
 });
 
@@ -2393,12 +2393,13 @@ function sleep(ms) {
     unrefTimer(timer);
   });
 }
-async function mapWithConcurrency2(tasks, maxConcurrency, onSettled) {
+async function mapWithConcurrency2(tasks, maxConcurrency, onSettled, onStarted) {
   const results = new Array(tasks.length);
   let nextIndex = 0;
   async function worker() {
     while (nextIndex < tasks.length) {
       const index = nextIndex++;
+      onStarted?.(index);
       const result = await tasks[index]();
       results[index] = result;
       onSettled?.(result, index);
@@ -2485,13 +2486,15 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     )
   );
   const total = tasks.length;
+  const onItemFinish = options?.onItemFinish ?? options?.onProgress;
   let completed = 0;
+  let started = 0;
   let succeeded = 0;
   let errored = 0;
   const resultItems = await mapWithConcurrency2(
     tasks,
     maxConcurrency,
-    options?.onProgress ? (item) => {
+    (item) => {
       completed += 1;
       if (item.error === null) {
         succeeded += 1;
@@ -2499,18 +2502,17 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
         errored += 1;
       }
       try {
-        options?.onProgress?.({
+        onItemFinish?.({
           testRunId,
           completed,
           total,
           succeeded,
           errored,
           item: {
-            // The server replay trace id isn't known until completeReplay
-            // runs (below), so it can't be reported mid-run and we never
-            // emit the client-side placeholder. originalTraceId (the
-            // historical trace) is known now and is what a UI keys on to
-            // identify what just settled.
+            // The server replay trace id isn't known until completeReplay runs
+            // (below), so it can't be reported mid-run and we never emit the
+            // client-side placeholder. originalTraceId (the historical trace)
+            // is known now and is what a UI keys on to identify what settled.
             traceId: null,
             originalTraceId: item.originalTraceId ?? null,
             originalSpanId: item.originalSpanId ?? null,
@@ -2531,6 +2533,30 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
         });
       } catch {
       }
+    },
+    options?.onItemStart ? (index) => {
+      started += 1;
+      const serverItem = serverItems[index];
+      const originalTraceId = serverItem.originalTraceId ?? serverItem.sourceTraceId;
+      const originalSpanId = serverItem.originalSpanId ?? serverItem.sourceSpanId;
+      try {
+        options.onItemStart?.({
+          type: "started",
+          testRunId,
+          started,
+          completed,
+          total,
+          succeeded,
+          errored,
+          item: {
+            originalTraceId,
+            originalSpanId,
+            sourceTraceId: originalTraceId,
+            sourceSpanId: originalSpanId
+          }
+        });
+      } catch {
+      }
     } : void 0
   );
   await preserveReplayFailure(
@@ -2593,17 +2619,19 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     testRunUrl: fullTestRunUrl
   };
   await writeReplayResultFile(result);
-  try {
-    options?.onProgress?.({
-      type: "complete",
-      testRunId,
-      completed: total,
-      total,
-      succeeded,
-      errored,
-      result
-    });
-  } catch {
+  if (!options?.onItemFinish) {
+    try {
+      options?.onProgress?.({
+        type: "complete",
+        testRunId,
+        completed: total,
+        total,
+        succeeded,
+        errored,
+        result
+      });
+    } catch {
+    }
   }
   return result;
 }