orchestrator-client 5.7.1 → 5.7.2

package/dist/index.cjs

@@ -148,6 +148,7 @@ var OrchestratorAsync = class {
     this._getToken = opts.getToken;
     this._timeoutMs = opts.timeoutMs ?? DEFAULT_TIMEOUT_MS;
     this._maxRetries = opts.maxRetries ?? DEFAULT_MAX_RETRIES;
+    this._locale = opts.locale;
     this._fetch = opts.fetch ?? globalThis.fetch;
     if (opts.insecure && !opts.fetch && typeof process !== "undefined" && process.versions?.node) {
       this._insecure = true;
@@ -172,6 +173,9 @@ var OrchestratorAsync = class {
         headers.Authorization = `Bearer ${token}`;
       }
     }
+    if (this._locale) {
+      headers["X-Locale"] = this._locale;
+    }
     return headers;
   }
   async _request(method, path, opts) {
@@ -402,10 +406,9 @@ var OrchestratorAsync = class {
     );
   }
   async getTaskCompactions(taskId) {
-    const data = await this._get("/task/compactions", {
+    return this._get("/task/compactions", {
       task_id: taskId
     });
-    return data.compactions ?? [];
   }
   async getTaskJournal(taskId) {
     const data = await this._get("/task/journal", {
@@ -982,8 +985,10 @@ var OrchestratorAsync = class {
   async getErrorDetail(errorId) {
     return this._get(`/errors/${errorId}`);
   }
-  async getErrorStats(since) {
-    const params = {};
+  async getErrorStats(since, topN = 10) {
+    const params = {
+      top_n: topN
+    };
     if (since) params.since = since;
     return this._get("/errors/stats", params);
   }
@@ -1693,9 +1698,15 @@ var RealtimeClient = class {
     this._socket = null;
     this._handlers = /* @__PURE__ */ new Map();
     this._connected = false;
+    // Multi-subscriber room-diffing state (mirrors WebSocketProvider)
+    this._subscriptions = /* @__PURE__ */ new Map();
+    this._currentRooms = /* @__PURE__ */ new Set();
+    this._subIdCounter = 0;
     this._baseUrl = baseUrl.replace(/\/+$/, "");
     this._socketOptions = opts.socketOptions ?? {};
     this._getToken = opts.getToken;
+    this._clientId = opts.clientId ?? "orchestrator-client";
+    this._locale = opts.locale;
     if (opts.autoConnect) {
       this.connect();
     }
@@ -1708,18 +1719,26 @@ var RealtimeClient = class {
     if (this._getToken) {
       auth.token = typeof this._getToken === "function" ? await this._getToken() : this._getToken;
     }
+    const query = { client_id: this._clientId };
+    if (this._locale) {
+      query.locale = this._locale;
+    }
     const { io } = await import("socket.io-client");
     this._socket = io(this._baseUrl, {
       path: "/socket.io",
       auth,
+      query,
       transports: ["websocket", "polling"],
       ...this._socketOptions
     });
     this._socket.on("connect", () => {
       this._connected = true;
+      this._currentRooms = /* @__PURE__ */ new Set();
+      this._syncRooms();
     });
     this._socket.on("disconnect", () => {
       this._connected = false;
+      this._currentRooms = /* @__PURE__ */ new Set();
     });
     this._socket.on("message", (payload) => this._dispatch(payload));
     return new Promise((resolve, reject) => {
@@ -1742,7 +1761,7 @@ var RealtimeClient = class {
     this._connected = false;
   }
   /**
-   * Dispatch a message envelope to registered handlers.
+   * Dispatch a message envelope to registered handlers and subscribers.
    * The server sends: socket.emit("message", {type: "message", event: {..., event_type: "...", ...}})
    */
   _dispatch(payload) {
@@ -1756,6 +1775,9 @@ var RealtimeClient = class {
         h(event);
       }
     }
+    for (const sub of this._subscriptions.values()) {
+      sub.handler(event);
+    }
   }
   /**
    * Subscribe to realtime events for a specific task.
@@ -1825,6 +1847,64 @@ var RealtimeClient = class {
     if (!this._socket) throw new Error("RealtimeClient not connected");
     this._socket.emit("leave", { rooms });
   }
+  // ------------------------------------------------------------------
+  // Multi-subscriber API (mirrors WebSocketProvider room-diffing)
+  // ------------------------------------------------------------------
+  /**
+   * Register a subscriber that receives all events from the given rooms.
+   *
+   * Room membership is managed automatically: the client joins the union
+   * of all active subscribers' rooms and leaves rooms that are no longer
+   * needed when the last subscriber referencing them is removed.
+   *
+   * The `handler` receives the unwrapped event dict (same object that
+   * `on()` handlers receive).
+   *
+   * Returns an opaque subscription id that must be passed to
+   * `unsubscribe()` to remove the subscription.
+   *
+   * Example — mirror the webui's per-component subscription pattern:
+   *
+   *   const id = rt.subscribe((event) => {
+   *     if (event.event_type === "task_status_changed") { ... }
+   *   }, [`task:${taskId}`]);
+   *
+   *   // later, on cleanup:
+   *   rt.unsubscribe(id);
+   */
+  subscribe(handler, rooms) {
+    const id = `sub_${++this._subIdCounter}`;
+    this._subscriptions.set(id, { handler, rooms });
+    this._syncRooms();
+    return id;
+  }
+  /**
+   * Remove a subscription registered via `subscribe()`.
+   *
+   * Rooms that are no longer referenced by any remaining subscriber are
+   * left automatically.
+   */
+  unsubscribe(id) {
+    this._subscriptions.delete(id);
+    this._syncRooms();
+  }
+  /**
+   * Diff the union of all subscribers' rooms against the currently joined
+   * rooms and emit `join`/`leave` for the delta.  No-ops when not connected.
+   */
+  _syncRooms() {
+    const socket = this._socket;
+    if (!socket?.connected) return;
+    const needed = /* @__PURE__ */ new Set();
+    for (const sub of this._subscriptions.values()) {
+      for (const r of sub.rooms) needed.add(r);
+    }
+    const toJoin = [...needed].filter((r) => !this._currentRooms.has(r));
+    const toLeave = [...this._currentRooms].filter((r) => !needed.has(r));
+    if (toJoin.length) socket.emit("join", { rooms: toJoin });
+    if (toLeave.length) socket.emit("leave", { rooms: toLeave });
+    this._currentRooms = needed;
+  }
   /**
    * Send a ping to the server.
    */