@reactor-team/js-sdk 2.5.1 → 2.6.0

package/dist/index.mjs

@@ -129,6 +129,7 @@ var IceServersResponseSchema = z.object({
 // src/utils/webrtc.ts
 var DEFAULT_DATA_CHANNEL_LABEL = "data";
 var FORCE_RELAY_MODE = false;
+var DEFAULT_MAX_MESSAGE_BYTES = 256 * 1024;
 function createPeerConnection(config) {
   return new RTCPeerConnection({
     iceServers: config.iceServers,
@@ -285,14 +286,21 @@ function waitForIceGathering(pc, timeoutMs = 5e3) {
     }, timeoutMs);
   });
 }
-function sendMessage(channel, command, data, scope = "application") {
+function sendMessage(channel, command, data, scope = "application", maxBytes = DEFAULT_MAX_MESSAGE_BYTES) {
   if (channel.readyState !== "open") {
     throw new Error(`Data channel not open: ${channel.readyState}`);
   }
   const jsonData = typeof data === "string" ? JSON.parse(data) : data;
   const inner = { type: command, data: jsonData };
   const payload = { scope, data: inner };
-  channel.send(JSON.stringify(payload));
+  const serialized = JSON.stringify(payload);
+  const byteLength = new TextEncoder().encode(serialized).byteLength;
+  if (byteLength > maxBytes) {
+    throw new Error(
+      `Data channel message too large: ${byteLength} bytes exceeds limit of ${maxBytes} bytes (command: "${command}")`
+    );
+  }
+  channel.send(serialized);
 }
 function parseMessage(data) {
   if (typeof data === "string") {
@@ -613,7 +621,7 @@ var CoordinatorClient = class {
         if (response.status === 200) {
           const answerData = yield response.json();
           console.debug("[CoordinatorClient] Received SDP answer via polling");
-          return answerData.sdp_answer;
+          return { sdpAnswer: answerData.sdp_answer, attempts: attempt };
         }
         if (response.status === 202) {
           console.warn(
@@ -637,7 +645,7 @@ var CoordinatorClient = class {
    * @param sessionId - The session ID to connect to
    * @param sdpOffer - Optional SDP offer from the local WebRTC peer connection
    * @param maxAttempts - Optional maximum number of polling attempts before giving up
-   * @returns The SDP answer from the server
+   * @returns The SDP answer and the number of polling attempts made (0 if answered immediately via PUT)
    */
   connect(sessionId, sdpOffer, maxAttempts) {
     return __async(this, null, function* () {
@@ -645,10 +653,11 @@ var CoordinatorClient = class {
       if (sdpOffer) {
         const answer = yield this.sendSdpOffer(sessionId, sdpOffer);
         if (answer !== null) {
-          return answer;
+          return { sdpAnswer: answer, sdpPollingAttempts: 0 };
         }
       }
-      return this.pollSdpAnswer(sessionId, maxAttempts);
+      const result = yield this.pollSdpAnswer(sessionId, maxAttempts);
+      return { sdpAnswer: result.sdpAnswer, sdpPollingAttempts: result.attempts };
     });
   }
   /**
@@ -730,9 +739,10 @@ var LocalCoordinatorClient = class extends CoordinatorClient {
   }
   /**
    * Connects to the local session by posting SDP params to /sdp_params.
+   * Local connections are always immediate (no polling).
    * @param sessionId - The session ID (ignored for local)
    * @param sdpMessage - The SDP offer from the local WebRTC peer connection
-   * @returns The SDP answer from the server
+   * @returns The SDP answer and polling attempts (always 0 for local)
    */
   connect(sessionId, sdpMessage) {
     return __async(this, null, function* () {
@@ -758,7 +768,7 @@ var LocalCoordinatorClient = class extends CoordinatorClient {
       }
       const sdpAnswer = yield response.json();
       console.debug("[LocalCoordinatorClient] Received SDP answer");
-      return sdpAnswer.sdp;
+      return { sdpAnswer: sdpAnswer.sdp, sdpPollingAttempts: 0 };
     });
   }
   terminateSession() {
@@ -903,6 +913,9 @@ var GPUMachineClient = class {
         );
       }
       this.setStatus("connecting");
+      this.iceStartTime = performance.now();
+      this.iceNegotiationMs = void 0;
+      this.dataChannelMs = void 0;
       try {
         let answer = sdpAnswer;
         if (this.midMapping) {
@@ -942,6 +955,7 @@ var GPUMachineClient = class {
       this.midMapping = void 0;
       this.peerConnected = false;
       this.dataChannelOpen = false;
+      this.resetConnectionTimings();
       this.setStatus("disconnected");
       console.debug("[GPUMachineClient] Disconnected");
     });
@@ -966,6 +980,14 @@ var GPUMachineClient = class {
   // ─────────────────────────────────────────────────────────────────────────────
   // Messaging
   // ─────────────────────────────────────────────────────────────────────────────
+  /**
+   * Returns the negotiated SCTP max message size (bytes) if available,
+   * otherwise `undefined` so `sendMessage` falls back to its built-in default.
+   */
+  get maxMessageBytes() {
+    var _a, _b, _c;
+    return (_c = (_b = (_a = this.peerConnection) == null ? void 0 : _a.sctp) == null ? void 0 : _b.maxMessageSize) != null ? _c : void 0;
+  }
   /**
    * Sends a command to the GPU machine via the data channel.
    * @param command The command to send
@@ -977,7 +999,13 @@ var GPUMachineClient = class {
       throw new Error("[GPUMachineClient] Data channel not available");
     }
     try {
-      sendMessage(this.dataChannel, command, data, scope);
+      sendMessage(
+        this.dataChannel,
+        command,
+        data,
+        scope,
+        this.maxMessageBytes
+      );
     } catch (error) {
       console.warn("[GPUMachineClient] Failed to send message:", error);
     }
@@ -1105,6 +1133,24 @@ var GPUMachineClient = class {
   getStats() {
     return this.stats;
   }
+  /**
+   * Returns the ICE/data-channel durations recorded during the last connect(),
+   * or undefined if no connection has completed yet.
+   */
+  getConnectionTimings() {
+    if (this.iceNegotiationMs == null || this.dataChannelMs == null) {
+      return void 0;
+    }
+    return {
+      iceNegotiationMs: this.iceNegotiationMs,
+      dataChannelMs: this.dataChannelMs
+    };
+  }
+  resetConnectionTimings() {
+    this.iceStartTime = void 0;
+    this.iceNegotiationMs = void 0;
+    this.dataChannelMs = void 0;
+  }
   startStatsPolling() {
     this.stopStatsPolling();
     this.statsInterval = setInterval(() => __async(this, null, function* () {
@@ -1148,6 +1194,9 @@ var GPUMachineClient = class {
       if (state) {
         switch (state) {
           case "connected":
+            if (this.iceStartTime != null && this.iceNegotiationMs == null) {
+              this.iceNegotiationMs = performance.now() - this.iceStartTime;
+            }
             this.peerConnected = true;
             this.checkFullyConnected();
             break;
@@ -1197,6 +1246,9 @@ var GPUMachineClient = class {
     if (!this.dataChannel) return;
     this.dataChannel.onopen = () => {
       console.debug("[GPUMachineClient] Data channel open");
+      if (this.iceStartTime != null && this.dataChannelMs == null) {
+        this.dataChannelMs = performance.now() - this.iceStartTime;
+      }
       this.dataChannelOpen = true;
       this.startPing();
       this.checkFullyConnected();
@@ -1236,13 +1288,13 @@ var GPUMachineClient = class {
 // src/core/Reactor.ts
 import { z as z2 } from "zod";
 var LOCAL_COORDINATOR_URL = "http://localhost:8080";
-var PROD_COORDINATOR_URL = "https://api.reactor.inc";
+var DEFAULT_BASE_URL = "https://api.reactor.inc";
 var TrackConfigSchema = z2.object({
   name: z2.string(),
   kind: z2.enum(["audio", "video"])
 });
 var OptionsSchema = z2.object({
-  coordinatorUrl: z2.string().default(PROD_COORDINATOR_URL),
+  apiUrl: z2.string().default(DEFAULT_BASE_URL),
   modelName: z2.string(),
   local: z2.boolean().default(false),
   /**
@@ -1267,12 +1319,12 @@ var Reactor = class {
     // Generic event map
     this.eventListeners = /* @__PURE__ */ new Map();
     const validatedOptions = OptionsSchema.parse(options);
-    this.coordinatorUrl = validatedOptions.coordinatorUrl;
+    this.coordinatorUrl = validatedOptions.apiUrl;
     this.model = validatedOptions.modelName;
     this.local = validatedOptions.local;
     this.receive = validatedOptions.receive;
     this.send = validatedOptions.send;
-    if (this.local && options.coordinatorUrl === void 0) {
+    if (this.local && options.apiUrl === void 0) {
       this.coordinatorUrl = LOCAL_COORDINATOR_URL;
     }
   }
@@ -1393,7 +1445,7 @@ var Reactor = class {
         receive: this.receive
       });
       try {
-        const sdpAnswer = yield this.coordinatorClient.connect(
+        const { sdpAnswer } = yield this.coordinatorClient.connect(
           this.sessionId,
           sdpOffer,
           options == null ? void 0 : options.maxAttempts
@@ -1410,7 +1462,7 @@ var Reactor = class {
         this.createError(
           "RECONNECTION_FAILED",
           `Failed to reconnect: ${error}`,
-          "coordinator",
+          "api",
           true
         );
       }
@@ -1433,6 +1485,7 @@ var Reactor = class {
         throw new Error("Already connected or connecting");
       }
       this.setStatus("connecting");
+      this.connectStartTime = performance.now();
       try {
         console.debug(
           "[Reactor] Connecting to coordinator with authenticated URL"
@@ -1450,13 +1503,25 @@ var Reactor = class {
           send: this.send,
           receive: this.receive
         });
+        const tSession = performance.now();
         const sessionId = yield this.coordinatorClient.createSession(sdpOffer);
+        const sessionCreationMs = performance.now() - tSession;
         this.setSessionId(sessionId);
-        const sdpAnswer = yield this.coordinatorClient.connect(
+        const tSdp = performance.now();
+        const { sdpAnswer, sdpPollingAttempts } = yield this.coordinatorClient.connect(
           sessionId,
           void 0,
           options == null ? void 0 : options.maxAttempts
         );
+        const sdpPollingMs = performance.now() - tSdp;
+        this.connectionTimings = {
+          sessionCreationMs,
+          sdpPollingMs,
+          sdpPollingAttempts,
+          iceNegotiationMs: 0,
+          dataChannelMs: 0,
+          totalMs: 0
+        };
         yield this.machineClient.connect(sdpAnswer);
       } catch (error) {
         if (isAbortError(error)) return;
@@ -1464,7 +1529,7 @@ var Reactor = class {
         this.createError(
           "CONNECTION_FAILED",
           `Connection failed: ${error}`,
-          "coordinator",
+          "api",
           true
         );
         try {
@@ -1502,6 +1567,7 @@ var Reactor = class {
       if (this.machineClient !== client) return;
       switch (status) {
         case "connected":
+          this.finalizeConnectionTimings(client);
           this.setStatus("ready");
           break;
         case "disconnected":
@@ -1527,7 +1593,9 @@ var Reactor = class {
     );
     client.on("statsUpdate", (stats) => {
       if (this.machineClient !== client) return;
-      this.emit("statsUpdate", stats);
+      this.emit("statsUpdate", __spreadProps(__spreadValues({}, stats), {
+        connectionTimings: this.connectionTimings
+      }));
     });
   }
   /**
@@ -1561,6 +1629,7 @@ var Reactor = class {
         }
       }
       this.setStatus("disconnected");
+      this.resetConnectionTimings();
       if (!recoverable) {
         this.setSessionExpiration(void 0);
         this.setSessionId(void 0);
@@ -1623,7 +1692,23 @@ var Reactor = class {
   }
   getStats() {
     var _a;
-    return (_a = this.machineClient) == null ? void 0 : _a.getStats();
+    const stats = (_a = this.machineClient) == null ? void 0 : _a.getStats();
+    if (!stats) return void 0;
+    return __spreadProps(__spreadValues({}, stats), { connectionTimings: this.connectionTimings });
+  }
+  resetConnectionTimings() {
+    this.connectStartTime = void 0;
+    this.connectionTimings = void 0;
+  }
+  finalizeConnectionTimings(client) {
+    var _a, _b;
+    if (!this.connectionTimings || this.connectStartTime == null) return;
+    const webrtcTimings = client.getConnectionTimings();
+    this.connectionTimings.iceNegotiationMs = (_a = webrtcTimings == null ? void 0 : webrtcTimings.iceNegotiationMs) != null ? _a : 0;
+    this.connectionTimings.dataChannelMs = (_b = webrtcTimings == null ? void 0 : webrtcTimings.dataChannelMs) != null ? _b : 0;
+    this.connectionTimings.totalMs = performance.now() - this.connectStartTime;
+    this.connectStartTime = void 0;
+    console.debug("[Reactor] Connection timings:", this.connectionTimings);
   }
   /**
    * Create and store an error
@@ -1663,7 +1748,7 @@ var initReactorStore = (props) => {
 };
 var createReactorStore = (initProps, publicState = defaultInitState) => {
   console.debug("[ReactorStore] Creating store", {
-    coordinatorUrl: initProps.coordinatorUrl,
+    apiUrl: initProps.apiUrl,
     jwtToken: initProps.jwtToken,
     initialState: publicState
   });
@@ -1831,7 +1916,7 @@ function ReactorProvider(_a) {
     console.debug("[ReactorProvider] Reactor store created successfully");
   }
   const _a2 = connectOptions != null ? connectOptions : {}, { autoConnect = false } = _a2, pollingOptions = __objRest(_a2, ["autoConnect"]);
-  const { coordinatorUrl, modelName, local, receive, send } = props;
+  const { apiUrl, modelName, local, receive, send } = props;
   const maxAttempts = pollingOptions.maxAttempts;
   useEffect(() => {
     const handleBeforeUnload = () => {
@@ -1884,7 +1969,7 @@ function ReactorProvider(_a) {
     console.debug("[ReactorProvider] Updating reactor store");
     storeRef.current = createReactorStore(
       initReactorStore({
-        coordinatorUrl,
+        apiUrl,
         modelName,
         local,
         receive,
@@ -1916,7 +2001,7 @@ function ReactorProvider(_a) {
       });
     };
   }, [
-    coordinatorUrl,
+    apiUrl,
     modelName,
     autoConnect,
     local,
@@ -2745,12 +2830,12 @@ function WebcamStream({
 }
 
 // src/utils/tokens.ts
-function fetchInsecureJwtToken(_0) {
-  return __async(this, arguments, function* (apiKey, coordinatorUrl = PROD_COORDINATOR_URL) {
+function fetchInsecureToken(_0) {
+  return __async(this, arguments, function* (apiKey, apiUrl = DEFAULT_BASE_URL) {
     console.warn(
-      "[Reactor] \u26A0\uFE0F SECURITY WARNING: fetchInsecureJwtToken() exposes your API key in client-side code. This should ONLY be used for local development or testing. In production, fetch tokens from your server instead."
+      "[Reactor] \u26A0\uFE0F SECURITY WARNING: fetchInsecureToken() exposes your API key in client-side code. This should ONLY be used for local development or testing. In production, fetch tokens from your server instead."
     );
-    const response = yield fetch(`${coordinatorUrl}/tokens`, {
+    const response = yield fetch(`${apiUrl}/tokens`, {
       method: "GET",
       headers: {
         "X-API-Key": apiKey
@@ -2767,14 +2852,14 @@ function fetchInsecureJwtToken(_0) {
 export {
   AbortError,
   ConflictError,
-  PROD_COORDINATOR_URL,
+  DEFAULT_BASE_URL,
   Reactor,
   ReactorController,
   ReactorProvider,
   ReactorView,
   WebcamStream,
   audio,
-  fetchInsecureJwtToken,
+  fetchInsecureToken,
   isAbortError,
   useReactor,
   useReactorInternalMessage,