@pear-protocol/symmio-client 0.3.12 → 0.3.14

package/dist/react/provider.js

@@ -57,6 +57,9 @@ function resolveBinanceSymbol(symmSymbol) {
 // src/utils/binance-ws.ts
 var BINANCE_WS_URL = "wss://fstream.binance.com/market/ws";
 var RECONNECT_DELAYS = [1e3, 2e3, 4e3, 8e3, 16e3, 3e4];
+var IDLE_CLOSE_DELAY_MS = 3e4;
+var STALE_CONNECTION_MS = 3e4;
+var STALE_CHECK_INTERVAL_MS = 1e4;
 var STABLE_QUOTES = ["USDT0", "USDT", "USDC", "USDE", "USDH", "USD"];
 function normalizeBaseSymbol(symbol) {
   const normalized = symbol.toUpperCase().trim();
@@ -106,8 +109,11 @@ var BinanceWsManager = class {
   streams = /* @__PURE__ */ new Map();
   reconnectAttempt = 0;
   reconnectTimer = null;
+  idleCloseTimer = null;
+  staleCheckTimer = null;
   intentionalClose = false;
-  pendingSubscribes = [];
+  pendingSubscribes = /* @__PURE__ */ new Set();
+  lastMessageAt = 0;
   idCounter = 0;
   /**
    * Subscribe to a kline stream. Returns an unsubscribe function.
@@ -180,10 +186,10 @@ var BinanceWsManager = class {
    */
   destroy() {
     this.intentionalClose = true;
-    if (this.reconnectTimer) {
-      clearTimeout(this.reconnectTimer);
-      this.reconnectTimer = null;
-    }
+    this.clearReconnectTimer();
+    this.clearIdleCloseTimer();
+    this.clearStaleCheckTimer();
+    this.pendingSubscribes.clear();
     if (this.ws) {
       this.ws.close();
       this.ws = null;
@@ -203,6 +209,7 @@ var BinanceWsManager = class {
     }
     sub.callbacks.set(id, cb);
     if (isNew) {
+      this.clearIdleCloseTimer();
       this.ensureConnected();
       this.sendSubscribe([streamName]);
     }
@@ -214,18 +221,14 @@ var BinanceWsManager = class {
     if (sub.callbacks.size === 0) {
       this.streams.delete(streamName);
       this.sendUnsubscribe([streamName]);
-      if (this.streams.size === 0 && this.ws) {
-        this.intentionalClose = true;
-        this.ws.close();
-        this.ws = null;
-        this.intentionalClose = false;
-      }
+      this.scheduleIdleClose();
     }
   }
   ensureConnected() {
     if (this.ws && (this.ws.readyState === WebSocket.OPEN || this.ws.readyState === WebSocket.CONNECTING)) {
       return;
     }
+    this.clearReconnectTimer();
     this.connect();
   }
   connect() {
@@ -234,23 +237,27 @@ var BinanceWsManager = class {
     this.ws = new WebSocket(BINANCE_WS_URL);
     this.ws.onopen = () => {
       this.reconnectAttempt = 0;
-      const activeStreams = Array.from(this.streams.keys());
+      this.lastMessageAt = Date.now();
+      this.startStaleCheck();
+      const activeStreams = Array.from(
+        /* @__PURE__ */ new Set([...this.streams.keys(), ...this.pendingSubscribes])
+      );
+      this.pendingSubscribes.clear();
       if (activeStreams.length > 0) {
         this.sendSubscribe(activeStreams);
       }
-      if (this.pendingSubscribes.length > 0) {
-        this.sendSubscribe(this.pendingSubscribes);
-        this.pendingSubscribes = [];
-      }
     };
     this.ws.onmessage = (event) => {
       try {
         const data = JSON.parse(event.data);
+        this.lastMessageAt = Date.now();
         this.handleMessage(data);
       } catch {
       }
     };
     this.ws.onclose = () => {
+      this.ws = null;
+      this.clearStaleCheckTimer();
       if (this.intentionalClose) return;
       this.scheduleReconnect();
     };
@@ -284,25 +291,27 @@ var BinanceWsManager = class {
   }
   sendSubscribe(streams) {
     if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      this.pendingSubscribes.push(...streams);
+      streams.forEach((stream) => this.pendingSubscribes.add(stream));
       return;
     }
+    const params = Array.from(new Set(streams));
+    if (params.length === 0) return;
     this.ws.send(JSON.stringify({
       method: "SUBSCRIBE",
-      params: streams,
+      params,
       id: Date.now()
     }));
   }
   sendUnsubscribe(streams) {
     if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      this.pendingSubscribes = this.pendingSubscribes.filter(
-        (s) => !streams.includes(s)
-      );
+      streams.forEach((stream) => this.pendingSubscribes.delete(stream));
       return;
     }
+    const params = Array.from(new Set(streams));
+    if (params.length === 0) return;
     this.ws.send(JSON.stringify({
       method: "UNSUBSCRIBE",
-      params: streams,
+      params,
       id: Date.now()
     }));
   }
@@ -316,6 +325,44 @@ var BinanceWsManager = class {
       this.connect();
     }, delay);
   }
+  scheduleIdleClose() {
+    if (this.streams.size > 0 || this.idleCloseTimer) return;
+    this.clearReconnectTimer();
+    if (!this.ws) return;
+    this.idleCloseTimer = setTimeout(() => {
+      this.idleCloseTimer = null;
+      if (this.streams.size > 0 || !this.ws) return;
+      this.intentionalClose = true;
+      this.ws.close();
+      this.ws = null;
+      this.intentionalClose = false;
+      this.clearStaleCheckTimer();
+    }, IDLE_CLOSE_DELAY_MS);
+  }
+  startStaleCheck() {
+    this.clearStaleCheckTimer();
+    this.staleCheckTimer = setInterval(() => {
+      if (!this.ws || this.ws.readyState !== WebSocket.OPEN) return;
+      if (this.streams.size === 0) return;
+      if (Date.now() - this.lastMessageAt < STALE_CONNECTION_MS) return;
+      this.ws.close();
+    }, STALE_CHECK_INTERVAL_MS);
+  }
+  clearReconnectTimer() {
+    if (!this.reconnectTimer) return;
+    clearTimeout(this.reconnectTimer);
+    this.reconnectTimer = null;
+  }
+  clearIdleCloseTimer() {
+    if (!this.idleCloseTimer) return;
+    clearTimeout(this.idleCloseTimer);
+    this.idleCloseTimer = null;
+  }
+  clearStaleCheckTimer() {
+    if (!this.staleCheckTimer) return;
+    clearInterval(this.staleCheckTimer);
+    this.staleCheckTimer = null;
+  }
 };
 var _instance = null;
 function getBinanceWsManager() {
@@ -346,6 +393,28 @@ function getNextRefCount(binanceSymbol) {
 function getPrevRefCount(binanceSymbol) {
   return Math.max(0, (refCounts.get(binanceSymbol) ?? 0) - 1);
 }
+function addMappedSymbol(binanceSymbol, symmSymbol) {
+  const symbols = streamSymbols.get(binanceSymbol) ?? /* @__PURE__ */ new Map();
+  symbols.set(symmSymbol, (symbols.get(symmSymbol) ?? 0) + 1);
+  streamSymbols.set(binanceSymbol, symbols);
+}
+function removeMappedSymbol(binanceSymbol, symmSymbol) {
+  const symbols = streamSymbols.get(binanceSymbol);
+  if (!symbols) return false;
+  const currentCount = symbols.get(symmSymbol) ?? 0;
+  if (currentCount <= 0) return false;
+  if (currentCount === 1) {
+    symbols.delete(symmSymbol);
+  } else {
+    symbols.set(symmSymbol, currentCount - 1);
+  }
+  if (symbols.size === 0) {
+    streamSymbols.delete(binanceSymbol);
+  } else {
+    streamSymbols.set(binanceSymbol, symbols);
+  }
+  return true;
+}
 var useBinanceMarkPriceStore = zustand.create((set) => ({
   markPrices: {},
   fundingRates: {},
@@ -354,9 +423,7 @@ var useBinanceMarkPriceStore = zustand.create((set) => ({
     const binanceSymbol = normalizeBinanceSymbol(rawBinanceSymbol);
     const nextRefCount = getNextRefCount(binanceSymbol);
     refCounts.set(binanceSymbol, nextRefCount);
-    const symbols = streamSymbols.get(binanceSymbol) ?? /* @__PURE__ */ new Set();
-    symbols.add(symmSymbol);
-    streamSymbols.set(binanceSymbol, symbols);
+    addMappedSymbol(binanceSymbol, symmSymbol);
     if (allMarkPricesRefCount === 0) {
       const wsManager = getBinanceWsManager();
       allMarkPricesUnsubscribe = wsManager.subscribeAllMarkPrices((entries) => {
@@ -371,7 +438,7 @@ var useBinanceMarkPriceStore = zustand.create((set) => ({
             nextMarkPrices ??= { ...state.markPrices };
             nextFundingRates ??= { ...state.fundingRates };
             nextFundingTimes ??= { ...state.nextFundingTimes };
-            mappedSymbols.forEach((mappedSymbol) => {
+            mappedSymbols.forEach((_count, mappedSymbol) => {
               nextMarkPrices[mappedSymbol] = entry.markPrice;
               nextFundingRates[mappedSymbol] = entry.fundingRate;
               nextFundingTimes[mappedSymbol] = entry.nextFundingTime;
@@ -392,14 +459,9 @@ var useBinanceMarkPriceStore = zustand.create((set) => ({
   },
   unsubscribeSymbol: (symmSymbol, rawBinanceSymbol) => {
     const binanceSymbol = normalizeBinanceSymbol(rawBinanceSymbol);
-    const symbols = streamSymbols.get(binanceSymbol);
-    if (symbols) {
-      symbols.delete(symmSymbol);
-      if (symbols.size === 0) {
-        streamSymbols.delete(binanceSymbol);
-      } else {
-        streamSymbols.set(binanceSymbol, symbols);
-      }
+    const removedSubscription = removeMappedSymbol(binanceSymbol, symmSymbol);
+    if (!removedSubscription) {
+      return;
     }
     const nextRefCount = getPrevRefCount(binanceSymbol);
     if (nextRefCount === 0) {
@@ -413,6 +475,10 @@ var useBinanceMarkPriceStore = zustand.create((set) => ({
       allMarkPricesUnsubscribe = null;
     }
     set((state) => {
+      const hasMappedSymbol = Boolean(streamSymbols.get(binanceSymbol)?.has(symmSymbol));
+      if (hasMappedSymbol) {
+        return state;
+      }
       if (state.markPrices[symmSymbol] == null && state.fundingRates[symmSymbol] == null && state.nextFundingTimes[symmSymbol] == null) {
         return state;
       }
@@ -497,7 +563,7 @@ var symmKeys = {
   signature: (address, chainId) => ["symm", "signature", address, chainId],
   auth: (accountAddress, chainId, signerAddress) => ["symm", "auth", accountAddress, chainId, signerAddress],
   approval: (owner, spender, chainId, token) => ["symm", "approval", owner, spender, chainId, token],
-  balances: (address, chainId) => ["symm", "balances", address, chainId],
+  balances: (address, chainId, multiAccountAddress) => multiAccountAddress === void 0 ? ["symm", "balances", address, chainId] : ["symm", "balances", address, chainId, multiAccountAddress],
   positions: (params) => ["symm", "positions", params],
   openOrders: (params) => ["symm", "openOrders", params],
   tradeHistory: (params) => ["symm", "tradeHistory", params],
@@ -530,8 +596,33 @@ var useSymmWsStore = zustand.create((set) => ({
 }));
 
 // src/react/hooks/use-symm-ws.ts
-function asUnsubscribeFn(value) {
-  return typeof value === "function" ? value : null;
+var wsOwnerCounts = /* @__PURE__ */ new WeakMap();
+var wsConnectPromises = /* @__PURE__ */ new WeakMap();
+function addWsOwner(ws) {
+  wsOwnerCounts.set(ws, (wsOwnerCounts.get(ws) ?? 0) + 1);
+}
+function removeWsOwner(ws) {
+  const nextCount = Math.max(0, (wsOwnerCounts.get(ws) ?? 0) - 1);
+  if (nextCount === 0) {
+    wsOwnerCounts.delete(ws);
+  } else {
+    wsOwnerCounts.set(ws, nextCount);
+  }
+  return nextCount;
+}
+function connectShared(ws) {
+  if (ws.isConnected()) {
+    return Promise.resolve();
+  }
+  const existing = wsConnectPromises.get(ws);
+  if (existing) {
+    return existing;
+  }
+  const promise = ws.connect().finally(() => {
+    wsConnectPromises.delete(ws);
+  });
+  wsConnectPromises.set(ws, promise);
+  return promise;
 }
 function useSymmWs(params = {}) {
   const ctx = react.useContext(SymmContext);
@@ -548,114 +639,76 @@ function useSymmWs(params = {}) {
     }
     const ws = symmCoreClient.ws;
     const addr = accountAddress;
-    const unsubscribers = [];
     let cancelled = false;
-    const removeOnConnect = ws.onConnect(() => {
-      setConnected(true);
+    addWsOwner(ws);
+    const removeOnConnect = ws.onConnect(() => setConnected(true));
+    const removeOnDisconnect = ws.onDisconnect(() => setConnected(false));
+    const subscriptions = [];
+    const addSubscription = (channel, handler) => {
+      ws.subscribe(channel, addr, chainId, handler);
+      subscriptions.push({ channel, handler });
+    };
+    addSubscription("positions", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.positionsRoot });
     });
-    const removeOnDisconnect = ws.onDisconnect(() => {
-      setConnected(false);
+    addSubscription("open-orders", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
     });
-    const removeOnError = asUnsubscribeFn(ws.onError?.(() => {
-    }));
-    const removeOnWelcome = asUnsubscribeFn(ws.onWelcome?.(() => {
-    }));
-    unsubscribers.push(removeOnConnect, removeOnDisconnect);
-    if (removeOnError) unsubscribers.push(removeOnError);
-    if (removeOnWelcome) unsubscribers.push(removeOnWelcome);
-    const positionsUnsub = asUnsubscribeFn(
-      ws.subscribeToPositions(addr, chainId, () => {
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.positionsRoot
-        });
-      })
-    );
-    if (positionsUnsub) unsubscribers.push(positionsUnsub);
-    const openOrdersUnsub = asUnsubscribeFn(
-      ws.subscribeToOpenOrders(addr, chainId, () => {
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.openOrdersRoot
-        });
-      })
-    );
-    if (openOrdersUnsub) unsubscribers.push(openOrdersUnsub);
-    const tradesUnsub = asUnsubscribeFn(
-      ws.subscribeToTrades(addr, chainId, () => {
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.tradeHistoryRoot
-        });
-      })
-    );
-    if (tradesUnsub) unsubscribers.push(tradesUnsub);
-    const accountSummaryUnsub = asUnsubscribeFn(
-      ws.subscribeToAccountSummary(addr, chainId, () => {
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.balances(accountAddress, chainId)
-        });
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.accountSummary(accountAddress, chainId)
-        });
-      })
-    );
-    if (accountSummaryUnsub) unsubscribers.push(accountSummaryUnsub);
-    const notificationsUnsub = asUnsubscribeFn(
-      ws.subscribeToNotifications(addr, chainId, () => {
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.notifications({ accountAddress, chainId })
-        });
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.unreadCount({ accountAddress, chainId })
-        });
-      })
-    );
-    if (notificationsUnsub) unsubscribers.push(notificationsUnsub);
-    const tpslUnsub = asUnsubscribeFn(
-      ws.subscribeToTpsl(addr, chainId, () => {
-        queryClient.invalidateQueries({ queryKey: symmKeys.tpslOrdersRoot });
-        queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
-      })
-    );
-    if (tpslUnsub) unsubscribers.push(tpslUnsub);
-    const twapUnsub = asUnsubscribeFn(
-      ws.subscribeToTwapOrders(addr, chainId, () => {
-        queryClient.invalidateQueries({ queryKey: symmKeys.twapOrdersRoot });
-        queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
-      })
-    );
-    if (twapUnsub) unsubscribers.push(twapUnsub);
-    const triggerOrdersUnsub = asUnsubscribeFn(
-      ws.subscribeToTriggerOrders(addr, chainId, () => {
-        queryClient.invalidateQueries({ queryKey: symmKeys.triggerOrdersRoot });
-        queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
-      })
-    );
-    if (triggerOrdersUnsub) unsubscribers.push(triggerOrdersUnsub);
-    const executionsUnsub = asUnsubscribeFn(
-      ws.subscribeToExecutions(addr, chainId, () => {
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.positionsRoot
-        });
-        queryClient.invalidateQueries({
-          queryKey: symmKeys.portfolioRoot
-        });
-      })
-    );
-    if (executionsUnsub) unsubscribers.push(executionsUnsub);
-    void ws.connect().then(() => {
-      if (cancelled) {
-        return;
+    addSubscription("trades", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.tradeHistoryRoot });
+    });
+    addSubscription("account-summary", () => {
+      queryClient.invalidateQueries({
+        queryKey: symmKeys.balances(accountAddress, chainId)
+      });
+      queryClient.invalidateQueries({
+        queryKey: symmKeys.accountSummary(accountAddress, chainId)
+      });
+    });
+    addSubscription("notifications", () => {
+      queryClient.invalidateQueries({
+        queryKey: symmKeys.notifications({ accountAddress, chainId })
+      });
+      queryClient.invalidateQueries({
+        queryKey: symmKeys.unreadCount({ accountAddress, chainId })
+      });
+    });
+    addSubscription("tpsl", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.tpslOrdersRoot });
+      queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
+    });
+    addSubscription("twap-orders", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.twapOrdersRoot });
+      queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
+    });
+    addSubscription("trigger-orders", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.triggerOrdersRoot });
+      queryClient.invalidateQueries({ queryKey: symmKeys.openOrdersRoot });
+    });
+    addSubscription("executions", () => {
+      queryClient.invalidateQueries({ queryKey: symmKeys.positionsRoot });
+      queryClient.invalidateQueries({ queryKey: symmKeys.portfolioRoot });
+    });
+    void connectShared(ws).then(() => {
+      if (cancelled) return;
+      if (ws.isConnected()) {
+        setConnected(true);
       }
     }).catch(() => {
-      if (cancelled) {
-        return;
-      }
+      if (cancelled) return;
       setConnected(false);
     });
     return () => {
       cancelled = true;
-      unsubscribers.forEach((unsubscribe) => unsubscribe());
-      ws.disconnect();
-      setConnected(false);
+      removeOnConnect();
+      removeOnDisconnect();
+      subscriptions.forEach(({ channel, handler }) => {
+        ws.unsubscribe(channel, addr, chainId, handler);
+      });
+      if (removeWsOwner(ws) === 0) {
+        ws.disconnect();
+        setConnected(false);
+      }
     };
   }, [symmCoreClient, accountAddress, chainId, queryClient, setConnected]);
   return { isConnected };