Function connectWebsocket

extension/background.js:87–149

(request, senderTabId, sendResponse)

}

function connectWebsocket(request, senderTabId, sendResponse) {
  const wsId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
  let ws;
  try {
    const headers = request.request.headers?.reduce((acc, el) => {
      acc[el.key] = el.value;
      return acc;
    }, {}) ?? {};
    ws = new WebSocket(request.request.data.uri, [], {
      headers
    });
  } catch (err) {
    console.error("error connecting to websocket", err);
    sendResponse({ status: "error", err: err.message });
    return;
  }

  const connectionTimeout = setTimeout(() => {
    if (ws.readyState !== WebSocket.OPEN) {
      ws.close();
      websockets.delete(wsId);
      console.error("websocket connection timeout", wsId);
      sendResponse({ status: "error", err: "Connection timeout" });
    }
  }, 3000);

  ws.onclose = () => {
    if (websockets.has(wsId)) {
      websockets.delete(wsId);
      clearTimeout(connectionTimeout);
      chrome.tabs.sendMessage(senderTabId, {
        type: "ws-close",
        result: {
          status: "error",
          err: "WebSocket closed unexpectedly",
          wsId: wsId,
        }
      });
    }
  };
  ws.onerror = (e) => {
    ws.close();
    clearTimeout(connectionTimeout);
    sendResponse({ status: "error", err: "WebSocket error", wsId, metaData: request.metaData });
  };
  ws.onopen = () => {
    websockets.set(wsId, ws);
    sendResponse({ status: "success", wsId, metaData: request.metaData });
  };
  ws.onmessage = (event) => {
    let parsedData = event.data;
    if (typeof event.data === 'object') {
      parsedData = JSON.stringify(event.data);
    }
    chrome.tabs.sendMessage(senderTabId, {
      type: "ws-read",
      result: {
        wsId: wsId,