@chainstream-io/sdk 0.1.21 → 0.1.22

package/dist/stream/index.cjs

@@ -301,10 +301,24 @@ var StreamApi = class {
   constructor(context) {
     __publicField(this, "realtimeClient");
     __publicField(this, "listenersMap");
-    const realtimeEndpoint = this.buildWsUrl(context.streamUrl, context.accessToken);
+    __publicField(this, "context");
+    __publicField(this, "initPromise", null);
+    this.context = context;
+    this.listenersMap = /* @__PURE__ */ new Map();
+  }
+  /**
+   * Initialize Centrifuge client with current token
+   * This is called lazily when connect() is called
+   */
+  async initClient() {
+    if (this.realtimeClient) {
+      return;
+    }
+    const token = await this.getTokenValue();
+    const realtimeEndpoint = this.buildWsUrl(this.context.streamUrl, token);
     this.realtimeClient = new import_centrifuge.Centrifuge(realtimeEndpoint, {
       getToken: async (_ctx) => {
-        return typeof context.accessToken === "string" ? context.accessToken : await context.accessToken.getToken();
+        return this.getTokenValue();
       }
     });
     this.realtimeClient.on("connected", () => {
@@ -314,34 +328,73 @@ var StreamApi = class {
     }).on("error", (err) => {
       console.error("[streaming] error: ", err);
     });
-    this.listenersMap = /* @__PURE__ */ new Map();
+  }
+  /**
+   * Wait for initialization to complete
+   * If not initialized, auto-connect first
+   */
+  async ensureInitialized() {
+    if (!this.initPromise) {
+      this.connect();
+    }
+    await this.initPromise;
+  }
+  /**
+   * Get token value from string or TokenProvider
+   */
+  async getTokenValue() {
+    return typeof this.context.accessToken === "string" ? this.context.accessToken : await this.context.accessToken.getToken();
   }
   /**
    * Build WebSocket URL with token as query parameter
    */
-  buildWsUrl(endpoint, accessToken) {
+  buildWsUrl(endpoint, token) {
     const url = new URL(endpoint);
-    if (typeof accessToken === "string") {
-      url.searchParams.set("token", accessToken);
-    }
+    url.searchParams.set("token", token);
     return url.toString();
   }
+  /**
+   * Connect to the WebSocket server.
+   * This is automatically called when you use subscribe methods if not already connected.
+   * You can also call this method manually for explicit control.
+   */
   connect() {
-    this.realtimeClient.connect();
+    this.initPromise = this.initClient().then(() => {
+      this.realtimeClient.connect();
+    });
+  }
+  /**
+   * Disconnect from the WebSocket server.
+   * All subscriptions will be automatically removed.
+   */
+  disconnect() {
+    if (this.realtimeClient) {
+      this.realtimeClient.disconnect();
+    }
+  }
+  /**
+   * Check if the WebSocket is currently connected.
+   */
+  isConnected() {
+    return this.realtimeClient?.state === "connected";
   }
   /**
    * Start batching commands for efficient bulk operations
    * All subscription commands after this call will be batched until stopBatching is called
    */
   startBatching() {
-    this.realtimeClient.startBatching();
+    if (this.realtimeClient) {
+      this.realtimeClient.startBatching();
+    }
   }
   /**
    * Stop batching and flush all collected commands to the server
    * This will send all batched subscription commands in a single network request
    */
   stopBatching() {
-    this.realtimeClient.stopBatching();
+    if (this.realtimeClient) {
+      this.realtimeClient.stopBatching();
+    }
   }
   /**
    * Batch subscribe method that accepts a function containing subscription calls
@@ -371,26 +424,31 @@ var StreamApi = class {
     });
   }
   subscribe(channel, fn, filter, methodName) {
-    let sub = this.realtimeClient.getSubscription(channel);
     let listeners = this.listenersMap.get(channel);
-    if (!sub) {
+    if (!listeners) {
       listeners = /* @__PURE__ */ new Set();
       this.listenersMap.set(channel, listeners);
-      console.log("[xrealtime] create new sub: ", channel);
-      const processedFilter = filter && methodName ? replaceFilterFields(filter, methodName) : filter;
-      sub = this.realtimeClient.newSubscription(channel, {
-        delta: "fossil",
-        ...processedFilter && { filter: processedFilter }
-      });
-      sub.on("subscribed", () => {
-        console.log("[xrealtime] subscribed", channel);
-      }).on("unsubscribed", () => {
-        console.log("[xrealtime] unsubscribed", channel);
-      }).on("publication", (ctx) => {
-        listeners?.forEach((it) => it(ctx.data));
-      }).subscribe();
     }
-    listeners?.add(fn);
+    listeners.add(fn);
+    this.ensureInitialized().then(() => {
+      let sub = this.realtimeClient.getSubscription(channel);
+      if (!sub) {
+        console.log("[xrealtime] create new sub: ", channel);
+        const processedFilter = filter && methodName ? replaceFilterFields(filter, methodName) : filter;
+        sub = this.realtimeClient.newSubscription(channel, {
+          delta: "fossil",
+          ...processedFilter && { filter: processedFilter }
+        });
+        sub.on("subscribed", () => {
+          console.log("[xrealtime] subscribed", channel);
+        }).on("unsubscribed", () => {
+          console.log("[xrealtime] unsubscribed", channel);
+        }).on("publication", (ctx) => {
+          const currentListeners = this.listenersMap.get(channel);
+          currentListeners?.forEach((it) => it(ctx.data));
+        }).subscribe();
+      }
+    });
     return new StreamUnsubscrible(this, channel, fn);
   }
   unsubscribe(channel, fn) {
@@ -402,12 +460,14 @@ var StreamApi = class {
     console.log("unsubscribe, remain listeners: ", listeners.size);
     if (listeners.size === 0) {
       console.log("unsubscribe channel: ", channel);
-      const sub = this.realtimeClient.getSubscription(channel);
-      if (sub) {
-        sub.unsubscribe();
-        this.realtimeClient.removeSubscription(sub);
-      }
       this.listenersMap.delete(channel);
+      this.ensureInitialized().then(() => {
+        const sub = this.realtimeClient.getSubscription(channel);
+        if (sub) {
+          sub.unsubscribe();
+          this.realtimeClient.removeSubscription(sub);
+        }
+      });
     }
   }
   formatScientificNotation(value) {