Function WebSocketProvider

app/_providers/WebSocketProvider.tsx:35–158

({ children }: { children: ReactNode })

const REFRESH_DEBOUNCE = 500;

export const WebSocketProvider = ({ children }: { children: ReactNode }) => {
  const router = useRouter();
  const { user } = useAppMode();
  const [isConnected, setIsConnected] = useState(false);
  const wsRef = useRef<WebSocket | null>(null);
  const connectionIdRef = useRef<string | null>(null);
  const reconnectDelayRef = useRef(INITIAL_RECONNECT_DELAY);
  const reconnectTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
  const hasPendingUpdates = useRef(false);
  const refreshTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
  const mountedRef = useRef(true);
  const listenersRef = useRef<Set<(event: WsEvent) => void>>(new Set());

  const isEditorActive = useEditorActivityStore((s) => s.isActive);

  const subscribe = useCallback((handler: (event: WsEvent) => void) => {
    listenersRef.current.add(handler);
    return () => { listenersRef.current.delete(handler); };
  }, []);

  const debouncedRefresh = useCallback(() => {
    if (refreshTimerRef.current) clearTimeout(refreshTimerRef.current);
    refreshTimerRef.current = setTimeout(() => {
      router.refresh();
    }, REFRESH_DEBOUNCE);
  }, [router]);

  const handleMessage = useCallback(
    (event: MessageEvent) => {
      try {
        const raw = JSON.parse(event.data) as Record<string, unknown>;

        if (raw.type === "connected") {
          connectionIdRef.current = (raw.connectionId as string) ?? null;
          return;
        }

        const data = raw as unknown as WsEvent;
        listenersRef.current.forEach((handler) => handler(data));

        if (data.type === "notification") return;

        if (isEditorActive()) {
          hasPendingUpdates.current = true;
          return;
        }

        debouncedRefresh();
      } catch {}
    },
    [isEditorActive, debouncedRefresh],
  );

  const connect = useCallback(() => {
    if (!mountedRef.current) return;
    if (!user) return;
    if (wsRef.current?.readyState === WebSocket.OPEN) return;