Function getOrCreateWs

web/js/chat.js:251–316

(options = {})

}

async function getOrCreateWs(options = {}) {
  const { background = false } = options;
  if (ws && ws.readyState === WebSocket.OPEN) {
    return ws;
  }

  if (wsConnectPromise) {
    return wsConnectPromise;
  }

  clearReconnectTimer();

  wsConnectPromise = new Promise((resolve, reject) => {
    const protocol = window.location.protocol === "https:" ? "wss:" : "ws:";
    const provider = state.config && state.config.provider ? state.config.provider : "openai";
    const params = new URLSearchParams({
      provider,
      channelId: currentChannelId || DEFAULT_WEB_CHANNEL_ID,
    });
    const wsUrl = `${protocol}//${window.location.host}${state.API_BASE}/api/chat?${params.toString()}`;

    const socket = new WebSocket(wsUrl);
    ws = socket;

    socket.onopen = () => {
      shouldMaintainWs = true;
      wsConnectPromise = null;
      resolve(socket);
    };
    socket.onerror = (err) => {
      console.error(err);
      wsConnectPromise = null;
      reject(new Error("WebSocket connection failed"));
    };

    socket.onmessage = (event) => {
      try {
        const parsed = JSON.parse(event.data);
        if (parsed.error) {
          if (background && !currentAssistantMsg) {
            console.warn("Background WebSocket message error:", parsed.error);
          } else {
            handleError(parsed.error);
          }
        } else if (parsed.done) {
          handleDone();
        } else if (parsed.type) {
          handleAgentEvent(parsed);
        }
      } catch (e) {
        console.error("Failed to parse message", e);
      }
    };

    socket.onclose = () => {
      if (ws === socket) {
        ws = null;
      }