@arbidocs/react 0.5.28 → 0.5.30

package/dist/index.js

@@ -6283,6 +6283,8 @@ function ArbiWebSocketProvider({
   children
 }) {
   const arbi = useArbi();
+  const arbiRef = useRef(arbi);
+  arbiRef.current = arbi;
   const queryClient = useQueryClient();
   const subscribersRef = useRef(/* @__PURE__ */ new Set());
   const setStatus = useArbiWsStore((s) => s.setStatus);
@@ -6321,10 +6323,10 @@ function ArbiWebSocketProvider({
       else {
         const type = msg.type;
         if (type.startsWith("chat.") || type === "agent_dm_status")
-          void applyChatEvent(arbi, queryClient, msg);
+          void applyChatEvent(arbiRef.current, queryClient, msg);
         else if (type.startsWith("call."))
           applyCallEvent(queryClient, msg, {
-            me: arbi.isLoggedIn ? arbi.user.identity()?.externalId ?? null : null,
+            me: arbiRef.current.isLoggedIn ? arbiRef.current.user.identity()?.externalId ?? null : null,
             acceptRings: acceptRingsRef.current
           });
         else void applyLiveUpdate(queryClient, msg);
@@ -6350,8 +6352,11 @@ function ArbiWebSocketProvider({
     };
     const start = async () => {
       if (cancelled) return;
-      const info = arbi.getConnectionInfo();
-      if (!info) return;
+      const info = arbiRef.current.getConnectionInfo();
+      if (!info) {
+        scheduleRestart();
+        return;
+      }
       conn?.close();
       conn = null;
       setStatus("connecting");
@@ -6363,9 +6368,9 @@ function ArbiWebSocketProvider({
           ),
           accessToken: info.accessToken,
           // Resolve the freshest token on every (re)connect so rotation is picked up.
-          getAccessToken: () => arbi.getConnectionInfo()?.accessToken ?? null,
+          getAccessToken: () => arbiRef.current.getConnectionInfo()?.accessToken ?? null,
           // A reconnect refused for its token re-logs in silently first.
-          refreshAuth: () => arbi.refreshAccess(),
+          refreshAuth: () => arbiRef.current.refreshAccess(),
           onAuthenticated: (message) => onAuthenticatedRef.current?.(message),
           // The session's: a sign-out closes the socket before the server revokes the token.
           signal: info.signal,
@@ -6406,7 +6411,7 @@ function ArbiWebSocketProvider({
       conn?.close();
       reset();
     };
-  }, [enabled, arbi, queryClient, setStatus, setEvicted, applyTaskUpdate, applyPresence, reset]);
+  }, [enabled, queryClient, setStatus, setEvicted, applyTaskUpdate, applyPresence, reset]);
   return /* @__PURE__ */ jsx(ArbiWsContext.Provider, { value: context, children });
 }
 function useArbiWebSocket(onMessage) {