@torrent-tv/proxy 2.104.1 → 2.104.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@torrent-tv/proxy",
-  "version": "2.104.1",
+  "version": "2.104.2",
   "description": "Torrent proxy client that exposes webseed-like HTTP stream endpoint.",
   "license": "GPL-3.0-or-later",
   "repository": {

package/services/torrent/torrent-pool.js

@@ -1520,6 +1520,11 @@ export class TorrentPool {
       const context = this.#waitTraceByTorrent.get(torrent);
       if (context?.trace === true) context.emitWaitTrace?.(context.sourceKey, process, phase, fields);
     };
+    if (waitTrace?.trace === true) {
+      // WebTorrent assigns the store during _onStore, after client.add first
+      // returns. Revisit this hook when its ready callback calls us.
+      tracePieceStorePut(pieceStoreOf(torrent), trace);
+    }
     const pool = this;
     const wrap = (method, process) => {
       const original = torrent[method];
@@ -1564,16 +1569,28 @@ export class TorrentPool {
       this.#waitTraceHooks.add(torrent);
       wrap("_startDiscovery", "W2");
       wrap("_onStore", "W6");
-      tracePieceStorePut(pieceStoreOf(torrent), trace);
       const verify = torrent._verifyPieces;
       if (typeof verify === "function") {
         torrent._verifyPieces = function (callback, ...args) {
           const startedAt = performance.now();
           trace("W5", "begin", { startedAt });
-          return verify.call(this, (...results) => {
-            trace("W5", "end", { startedAt, endedAt: performance.now(), outcome: "verified" });
-            callback?.(...results);
-          }, ...args);
+          let finished = false;
+          const finish = (outcome, fields = {}) => {
+            if (finished) return;
+            finished = true;
+            const endedAt = performance.now();
+            trace("W5", "end", { startedAt, endedAt, durationMs: endedAt - startedAt, outcome, ...fields });
+          };
+          try {
+            return verify.call(this, (...results) => {
+              const error = results[0];
+              finish(error ? "error" : "verified", error ? { error: String(error?.message ?? error) } : {});
+              callback?.(...results);
+            }, ...args);
+          } catch (error) {
+            finish("error", { error: String(error?.message ?? error) });
+            throw error;
+          }
         };
       }
       const timing = this.#swarmTimingByTorrent.get(torrent);

package/services/torrent/worker/trace-piece-store-put.js

@@ -1,7 +1,9 @@
 /** Record the torrent worker's verified-piece storage write without changing its callback API. */
+const tracedStores = new WeakSet();
+
 export function tracePieceStorePut(store, trace, now = () => performance.now()) {
   const put = store?.put;
-  if (typeof put !== "function") return false;
+  if (!store || tracedStores.has(store) || typeof put !== "function") return false;
 
   store.put = function (index, bytes, callback) {
     const startedAt = now();
@@ -24,5 +26,6 @@ export function tracePieceStorePut(store, trace, now = () => performance.now())
       throw error;
     }
   };
+  tracedStores.add(store);
   return true;
 }

package/services/torrent/worker/worker.js

@@ -541,6 +541,16 @@ async function runCommand(command, params, id) {
     }
 
     case Command.PRIORITY_MAP: {
+      const startedAt = performance.now();
+      emitWaitTrace(params.sourceKey, "W7", "begin", { startedAt, fileIndex: params.fileIndex });
+      let traceFinished = false;
+      const finishPriorityTrace = (outcome, fields = {}) => {
+        if (traceFinished) return;
+        traceFinished = true;
+        const endedAt = performance.now();
+        emitWaitTrace(params.sourceKey, "W7", "end", { startedAt, endedAt,
+          durationMs: endedAt - startedAt, outcome, fileIndex: params.fileIndex, ...fields });
+      };
       // When this handler began, on the clock the main thread posted by: the
       // post to here holds the transfer and this thread's queue together
       // (torrent-tv/meta#177).
@@ -549,27 +559,21 @@ async function runCommand(command, params, id) {
       // only update an existing source, never revive one after its viewer left.
       const zones = Array.isArray(params.zones) ? params.zones : [];
       // Nor may one for a file held whole, which has nothing left to fetch.
-      const torrent = await knownTorrent(params.sourceKey);
-      if (!torrent) {
-        emitWaitTrace(params.sourceKey, "W7", "event", { outcome: "source-not-ready", fileIndex: params.fileIndex });
-        return { appliedMs: null, receivedAt, repliedAt: performance.timeOrigin + performance.now() };
-      }
-      // What applying a map costs this thread, told to whoever sent it: every
-      // zone becomes a window of the register and the swarm is told again
-      // (torrent-tv/meta#166).
-      const startedAt = performance.now();
-      emitWaitTrace(params.sourceKey, "W7", "begin", { startedAt, fileIndex: params.fileIndex });
       try {
+        const torrent = await knownTorrent(params.sourceKey);
+        if (!torrent) {
+          finishPriorityTrace("source-not-ready");
+          return { appliedMs: null, receivedAt, repliedAt: performance.timeOrigin + performance.now() };
+        }
+        // What applying a map costs this thread, told to whoever sent it: every
+        // zone becomes a window of the register and the swarm is told again
+        // (torrent-tv/meta#166).
         ensureArrivalsWired(params.sourceKey, torrent);
         pool.applyPriorityMap(torrent, params.fileIndex, zones, params.durationSeconds);
         torrent.emit("priority-map-changed", params.fileIndex);
-        const endedAt = performance.now();
-        emitWaitTrace(params.sourceKey, "W7", "end", { startedAt, endedAt, durationMs: endedAt - startedAt,
-          outcome: "applied", fileIndex: params.fileIndex });
+        finishPriorityTrace("applied");
       } catch (error) {
-        const endedAt = performance.now();
-        emitWaitTrace(params.sourceKey, "W7", "end", { startedAt, endedAt, durationMs: endedAt - startedAt,
-          outcome: "error", error: String(error?.message ?? error), fileIndex: params.fileIndex });
+        finishPriorityTrace("error", { error: String(error?.message ?? error) });
         throw error;
       }
       return { appliedMs: performance.now() - startedAt, receivedAt, repliedAt: performance.timeOrigin + performance.now() };