@ricsam/isolate-client 0.1.14 → 0.1.16

package/dist/cjs/connection.cjs

@@ -42,6 +42,7 @@ var __export = (target, all) => {
 // packages/isolate-client/src/connection.ts
 var exports_connection = {};
 __export(exports_connection, {
+  isBenignDisposeError: () => isBenignDisposeError,
   connect: () => connect
 });
 module.exports = __toCommonJS(exports_connection);
@@ -49,8 +50,10 @@ var import_node_net = require("node:net");
 var import_node_path = __toESM(require("node:path"));
 var import_isolate_protocol = require("@ricsam/isolate-protocol");
 var import_client = require("@ricsam/isolate-playwright/client");
-var DEFAULT_TIMEOUT = 30000;
 var isolateWsCallbacks = new Map;
+var isolateClientWebSockets = new Map;
+var isolateWebSocketCallbacks = new Map;
+var isolateEventListeners = new Map;
 async function connect(options = {}) {
   const socket = await createSocket(options);
   const state = {
@@ -80,9 +83,6 @@ async function connect(options = {}) {
   socket.on("close", () => {
     state.connected = false;
     for (const [, pending] of state.pendingRequests) {
-      if (pending.timeoutId) {
-        clearTimeout(pending.timeoutId);
-      }
       pending.reject(new Error("Connection closed"));
     }
     state.pendingRequests.clear();
@@ -121,7 +121,7 @@ async function connect(options = {}) {
 }
 function createSocket(options) {
   return new Promise((resolve, reject) => {
-    const timeout = options.timeout ?? DEFAULT_TIMEOUT;
+    const timeout = options.timeout;
     let socket;
     const onError = (err) => {
       reject(err);
@@ -136,13 +136,15 @@ function createSocket(options) {
       socket = import_node_net.connect(options.port ?? 47891, options.host ?? "127.0.0.1", onConnect);
     }
     socket.on("error", onError);
-    const timeoutId = setTimeout(() => {
-      socket.destroy();
-      reject(new Error("Connection timeout"));
-    }, timeout);
-    socket.once("connect", () => {
-      clearTimeout(timeoutId);
-    });
+    if (timeout && timeout > 0) {
+      const timeoutId = setTimeout(() => {
+        socket.destroy();
+        reject(new Error("Connection timeout"));
+      }, timeout);
+      socket.once("connect", () => {
+        clearTimeout(timeoutId);
+      });
+    }
   });
 }
 function handleMessage(message, state) {
@@ -152,8 +154,6 @@ function handleMessage(message, state) {
       const pending = state.pendingRequests.get(response.requestId);
       if (pending) {
         state.pendingRequests.delete(response.requestId);
-        if (pending.timeoutId)
-          clearTimeout(pending.timeoutId);
         pending.resolve(response.data);
       }
       break;
@@ -163,8 +163,6 @@ function handleMessage(message, state) {
       const pending = state.pendingRequests.get(response.requestId);
       if (pending) {
         state.pendingRequests.delete(response.requestId);
-        if (pending.timeoutId)
-          clearTimeout(pending.timeoutId);
         const error = new Error(response.message);
         if (response.details) {
           error.name = response.details.name;
@@ -191,27 +189,9 @@ function handleMessage(message, state) {
       }
       break;
     }
-    case import_isolate_protocol.MessageType.WS_COMMAND: {
+    case import_isolate_protocol.MessageType.ISOLATE_EVENT: {
       const msg = message;
-      const callbacks = isolateWsCallbacks.get(msg.isolateId);
-      if (callbacks) {
-        let data;
-        if (msg.command.data instanceof Uint8Array) {
-          data = msg.command.data.buffer.slice(msg.command.data.byteOffset, msg.command.data.byteOffset + msg.command.data.byteLength);
-        } else {
-          data = msg.command.data;
-        }
-        const cmd = {
-          type: msg.command.type,
-          connectionId: msg.command.connectionId,
-          data,
-          code: msg.command.code,
-          reason: msg.command.reason
-        };
-        for (const cb of callbacks) {
-          cb(cmd);
-        }
-      }
+      handleIsolateEvent(msg, state);
       break;
     }
     case import_isolate_protocol.MessageType.RESPONSE_STREAM_START: {
@@ -281,8 +261,6 @@ function handleMessage(message, state) {
       const pending = state.pendingRequests.get(msg.requestId);
       if (pending) {
         state.pendingRequests.delete(msg.requestId);
-        if (pending.timeoutId)
-          clearTimeout(pending.timeoutId);
         const response = new Response(readableStream, {
           status: msg.metadata?.status ?? 200,
           statusText: msg.metadata?.statusText ?? "OK",
@@ -407,25 +385,24 @@ function sendMessage(socket, message) {
   const frame = import_isolate_protocol.buildFrame(message);
   socket.write(frame);
 }
-function sendRequest(state, message, timeout = DEFAULT_TIMEOUT) {
+function sendRequest(state, message) {
   return new Promise((resolve, reject) => {
     if (!state.connected) {
       reject(new Error("Not connected"));
       return;
     }
     const requestId = message.requestId;
-    const timeoutId = setTimeout(() => {
-      state.pendingRequests.delete(requestId);
-      reject(new Error("Request timeout"));
-    }, timeout);
     state.pendingRequests.set(requestId, {
       resolve,
-      reject,
-      timeoutId
+      reject
     });
     sendMessage(state.socket, message);
   });
 }
+function isBenignDisposeError(error) {
+  const message = error instanceof Error ? error.message : String(error ?? "");
+  return /isolate not owned by this connection|isolate not found|not connected|connection closed/i.test(message);
+}
 async function createRuntime(state, options = {}, namespaceId) {
   const callbacks = {};
   if (options.console) {
@@ -449,73 +426,76 @@ async function createRuntime(state, options = {}, namespaceId) {
   const networkResponses = [];
   const pageListenerCleanups = [];
   if (options.playwright) {
-    playwrightHandler = import_client.createPlaywrightHandler(options.playwright.page, {
-      timeout: options.playwright.timeout
-    });
+    playwrightHandler = options.playwright.handler;
+    if (!playwrightHandler) {
+      throw new Error("playwright.handler is required when using playwright options");
+    }
+    const page = import_client.getDefaultPlaywrightHandlerMetadata(playwrightHandler)?.page;
     const handlerCallbackId = state.nextCallbackId++;
     state.callbacks.set(handlerCallbackId, async (opJson) => {
       const op = JSON.parse(opJson);
       const result2 = await playwrightHandler(op);
       return JSON.stringify(result2);
     });
-    const page = options.playwright.page;
-    const onConsole = (msg) => {
-      const entry = {
-        level: msg.type(),
-        stdout: msg.text(),
-        timestamp: Date.now()
+    if (page) {
+      const onConsole = (msg) => {
+        const entry = {
+          level: msg.type(),
+          stdout: msg.text(),
+          timestamp: Date.now()
+        };
+        browserConsoleLogs.push(entry);
+        if (options.playwright.onEvent) {
+          options.playwright.onEvent({
+            type: "browserConsoleLog",
+            ...entry
+          });
+        }
+        if (options.playwright.console && options.console?.onEntry) {
+          options.console.onEntry({
+            type: "browserOutput",
+            ...entry
+          });
+        } else if (options.playwright.console) {
+          const prefix = entry.level === "error" ? "[browser:error]" : "[browser]";
+          console.log(prefix, entry.stdout);
+        }
       };
-      browserConsoleLogs.push(entry);
-      if (options.playwright.onEvent) {
-        options.playwright.onEvent({
-          type: "browserConsoleLog",
-          ...entry
-        });
-      }
-      if (options.playwright.console && options.console?.onEntry) {
-        options.console.onEntry({
-          type: "browserOutput",
-          ...entry
-        });
-      } else if (options.playwright.console) {
-        const prefix = entry.level === "error" ? "[browser:error]" : "[browser]";
-        console.log(prefix, entry.stdout);
-      }
-    };
-    const onRequest = (request2) => {
-      const info = {
-        url: request2.url(),
-        method: request2.method(),
-        headers: request2.headers(),
-        timestamp: Date.now()
+      const onRequest = (request2) => {
+        const info = {
+          url: request2.url(),
+          method: request2.method(),
+          headers: request2.headers(),
+          timestamp: Date.now()
+        };
+        networkRequests.push(info);
+        if (options.playwright.onEvent) {
+          options.playwright.onEvent({
+            type: "networkRequest",
+            ...info
+          });
+        }
       };
-      networkRequests.push(info);
-      if (options.playwright.onEvent) {
-        options.playwright.onEvent({
-          type: "networkRequest",
-          ...info
-        });
-      }
-    };
-    const onResponse = (response) => {
-      const info = {
-        url: response.url(),
-        status: response.status(),
-        headers: response.headers(),
-        timestamp: Date.now()
+      const onResponse = (response) => {
+        const info = {
+          url: response.url(),
+          status: response.status(),
+          headers: response.headers(),
+          timestamp: Date.now()
+        };
+        networkResponses.push(info);
+        if (options.playwright.onEvent) {
+          options.playwright.onEvent({
+            type: "networkResponse",
+            ...info
+          });
+        }
       };
-      networkResponses.push(info);
-      if (options.playwright.onEvent) {
-        options.playwright.onEvent({
-          type: "networkResponse",
-          ...info
-        });
-      }
-    };
-    page.on("console", onConsole);
-    page.on("request", onRequest);
-    page.on("response", onResponse);
-    pageListenerCleanups.push(() => page.removeListener("console", onConsole), () => page.removeListener("request", onRequest), () => page.removeListener("response", onResponse));
+      page.on("console", onConsole);
+      page.on("request", onRequest);
+      page.on("response", onResponse);
+      pageListenerCleanups.push(() => page.removeListener("console", onConsole), () => page.removeListener("request", onRequest), () => page.removeListener("response", onResponse));
+    }
     callbacks.playwright = {
       handlerCallbackId,
       console: options.playwright.console && !options.console?.onEntry
@@ -563,8 +543,18 @@ async function createRuntime(state, options = {}, namespaceId) {
   const reused = result.reused ?? false;
   const wsCommandCallbacks = new Set;
   isolateWsCallbacks.set(isolateId, wsCommandCallbacks);
+  if (options.onWebSocketCommand) {
+    wsCommandCallbacks.add(options.onWebSocketCommand);
+  }
+  if (options.webSocket) {
+    isolateWebSocketCallbacks.set(isolateId, options.webSocket);
+  }
   const fetchHandle = {
     async dispatchRequest(req, opts) {
+      const signal = opts?.signal;
+      if (signal?.aborted) {
+        throw new DOMException("The operation was aborted", "AbortError");
+      }
       const reqId = state.nextRequestId++;
       const serialized = await serializeRequestWithStreaming(state, req);
       const { bodyStream, ...serializableRequest } = serialized;
@@ -572,24 +562,40 @@ async function createRuntime(state, options = {}, namespaceId) {
         type: import_isolate_protocol.MessageType.DISPATCH_REQUEST,
         requestId: reqId,
         isolateId,
-        request: serializableRequest,
-        options: opts
+        request: serializableRequest
       };
       const handleResponse = (res) => {
         if (res.__streaming && res.response instanceof Response) {
           return res.response;
         }
-        return deserializeResponse(res.response);
+        return import_isolate_protocol.deserializeResponse(res.response);
       };
-      if (serialized.bodyStreamId !== undefined && bodyStream) {
-        const streamId = serialized.bodyStreamId;
-        const responsePromise = sendRequest(state, request2, opts?.timeout ?? DEFAULT_TIMEOUT);
-        await sendBodyStream(state, streamId, bodyStream);
-        const res = await responsePromise;
-        return handleResponse(res);
-      } else {
-        const res = await sendRequest(state, request2, opts?.timeout ?? DEFAULT_TIMEOUT);
-        return handleResponse(res);
+      let onAbort;
+      if (signal) {
+        onAbort = () => {
+          const pending = state.pendingRequests.get(reqId);
+          if (pending) {
+            state.pendingRequests.delete(reqId);
+            pending.reject(new DOMException("The operation was aborted", "AbortError"));
+          }
+        };
+        signal.addEventListener("abort", onAbort, { once: true });
+      }
+      try {
+        if (serialized.bodyStreamId !== undefined && bodyStream) {
+          const streamId = serialized.bodyStreamId;
+          const responsePromise = sendRequest(state, request2);
+          await sendBodyStream(state, streamId, bodyStream);
+          const res = await responsePromise;
+          return handleResponse(res);
+        } else {
+          const res = await sendRequest(state, request2);
+          return handleResponse(res);
+        }
+      } finally {
+        if (signal && onAbort) {
+          signal.removeEventListener("abort", onAbort);
+        }
       }
     },
     async getUpgradeRequest() {
@@ -736,7 +742,7 @@ async function createRuntime(state, options = {}, namespaceId) {
         isolateId,
         timeout
       };
-      return sendRequest(state, req, timeout ?? DEFAULT_TIMEOUT);
+      return sendRequest(state, req);
     },
     async hasTests() {
       if (!testEnvironmentEnabled) {
@@ -778,7 +784,7 @@ async function createRuntime(state, options = {}, namespaceId) {
   const playwrightHandle = {
     getCollectedData() {
       if (!playwrightEnabled) {
-        throw new Error("Playwright not configured. Provide playwright.page in createRuntime options.");
+        throw new Error("Playwright not configured. Provide playwright.handler in createRuntime options.");
       }
       return {
         browserConsoleLogs: [...browserConsoleLogs],
@@ -788,7 +794,7 @@ async function createRuntime(state, options = {}, namespaceId) {
     },
     clearCollectedData() {
       if (!playwrightEnabled) {
-        throw new Error("Playwright not configured. Provide playwright.page in createRuntime options.");
+        throw new Error("Playwright not configured. Provide playwright.handler in createRuntime options.");
       }
       browserConsoleLogs.length = 0;
       networkRequests.length = 0;
@@ -797,7 +803,6 @@ async function createRuntime(state, options = {}, namespaceId) {
   };
   return {
     id: isolateId,
-    isolateId,
     reused,
     fetch: fetchHandle,
     timers: timersHandle,
@@ -812,24 +817,69 @@ async function createRuntime(state, options = {}, namespaceId) {
         requestId: reqId,
         isolateId,
         code,
-        filename: options2?.filename,
-        maxExecutionMs: options2?.maxExecutionMs,
-        module: true
+        filename: options2?.filename
       };
       await sendRequest(state, req);
     },
+    on(event, callback) {
+      let listeners = isolateEventListeners.get(isolateId);
+      if (!listeners) {
+        listeners = new Map;
+        isolateEventListeners.set(isolateId, listeners);
+      }
+      let eventListeners = listeners.get(event);
+      if (!eventListeners) {
+        eventListeners = new Set;
+        listeners.set(event, eventListeners);
+      }
+      eventListeners.add(callback);
+      return () => {
+        eventListeners.delete(callback);
+        if (eventListeners.size === 0) {
+          listeners.delete(event);
+          if (listeners.size === 0) {
+            isolateEventListeners.delete(isolateId);
+          }
+        }
+      };
+    },
+    emit(event, payload) {
+      sendMessage(state.socket, {
+        type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+        isolateId,
+        event,
+        payload
+      });
+    },
     dispose: async () => {
       for (const cleanup of pageListenerCleanups) {
         cleanup();
       }
       isolateWsCallbacks.delete(isolateId);
+      isolateWebSocketCallbacks.delete(isolateId);
+      isolateEventListeners.delete(isolateId);
+      const clientSockets = isolateClientWebSockets.get(isolateId);
+      if (clientSockets) {
+        for (const ws of clientSockets.values()) {
+          if (ws.readyState === WebSocket.OPEN || ws.readyState === WebSocket.CONNECTING) {
+            ws.close(1000, "Isolate disposed");
+          }
+        }
+        isolateClientWebSockets.delete(isolateId);
+      }
       const reqId = state.nextRequestId++;
       const req = {
         type: import_isolate_protocol.MessageType.DISPOSE_RUNTIME,
         requestId: reqId,
         isolateId
       };
-      await sendRequest(state, req);
+      try {
+        await sendRequest(state, req);
+      } catch (error) {
+        if (!isBenignDisposeError(error)) {
+          throw error;
+        }
+      }
     }
   };
 }
@@ -857,8 +907,15 @@ function registerFetchCallback(state, callback) {
   const callbackId = state.nextCallbackId++;
   state.callbacksNeedingRequestId.add(callbackId);
   state.callbacks.set(callbackId, async (serialized, requestId) => {
-    const request = deserializeRequest(serialized);
-    const response = await callback(request);
+    const data = serialized;
+    const init = {
+      method: data.method,
+      headers: data.headers,
+      rawBody: data.body ?? null,
+      body: data.body ?? null,
+      signal: new AbortController().signal
+    };
+    const response = await callback(data.url, init);
     const contentLength = response.headers.get("content-length");
     const knownSize = contentLength ? parseInt(contentLength, 10) : null;
     const isNetworkResponse = response.url && (response.url.startsWith("http://") || response.url.startsWith("https://"));
@@ -883,7 +940,7 @@ function registerFetchCallback(state, callback) {
       streamCallbackResponseBody(state, streamId, requestId, response.body);
       return { __callbackStreaming: true, streamId };
     }
-    return serializeResponse(response);
+    return import_isolate_protocol.serializeResponse(response);
   });
   return { callbackId, name: "fetch", type: "async" };
 }
@@ -1011,12 +1068,6 @@ var clientIteratorSessions = new Map;
 var nextClientIteratorId = 1;
 var returnedPromiseRegistry = new Map;
 var returnedIteratorRegistry = new Map;
-function isPromiseRef(value) {
-  return typeof value === "object" && value !== null && value.__type === "PromiseRef";
-}
-function isAsyncIteratorRef(value) {
-  return typeof value === "object" && value !== null && value.__type === "AsyncIteratorRef";
-}
 function registerCustomFunctions(state, customFunctions) {
   const registrations = {};
   for (const [name, def] of Object.entries(customFunctions)) {
@@ -1099,7 +1150,7 @@ function registerCustomFunctions(state, customFunctions) {
           if (value === null || typeof value !== "object") {
             return value;
           }
-          if (isPromiseRef(value)) {
+          if (import_isolate_protocol.isPromiseRef(value)) {
             const resolveCallbackId = state.nextCallbackId++;
             state.callbacks.set(resolveCallbackId, async (...args2) => {
               const promiseId = args2[0];
@@ -1117,7 +1168,7 @@ function registerCustomFunctions(state, customFunctions) {
               __resolveCallbackId: resolveCallbackId
             };
           }
-          if (isAsyncIteratorRef(value)) {
+          if (import_isolate_protocol.isAsyncIteratorRef(value)) {
             const nextCallbackId = state.nextCallbackId++;
             state.callbacks.set(nextCallbackId, async (...args2) => {
               const iteratorId = args2[0];
@@ -1200,36 +1251,6 @@ function registerCustomFunctions(state, customFunctions) {
   }
   return registrations;
 }
-async function serializeResponse(response) {
-  const headers = [];
-  response.headers.forEach((value, key) => {
-    headers.push([key, value]);
-  });
-  let body = null;
-  if (response.body) {
-    body = new Uint8Array(await response.arrayBuffer());
-  }
-  return {
-    status: response.status,
-    statusText: response.statusText,
-    headers,
-    body
-  };
-}
-function deserializeRequest(data) {
-  return new Request(data.url, {
-    method: data.method,
-    headers: data.headers,
-    body: data.body
-  });
-}
-function deserializeResponse(data) {
-  return new Response(data.body, {
-    status: data.status,
-    statusText: data.statusText,
-    headers: data.headers
-  });
-}
 async function serializeRequestWithStreaming(state, request) {
   const headers = [];
   request.headers.forEach((value, key) => {
@@ -1328,5 +1349,196 @@ async function sendBodyStream(state, streamId, body) {
     state.uploadStreams.delete(streamId);
   }
 }
+function handleIsolateEvent(message, state) {
+  switch (message.event) {
+    case import_isolate_protocol.IsolateEvents.WS_COMMAND: {
+      const payload = message.payload;
+      const callbacks = isolateWsCallbacks.get(message.isolateId);
+      if (callbacks) {
+        let data;
+        if (payload.data instanceof Uint8Array) {
+          data = payload.data.buffer.slice(payload.data.byteOffset, payload.data.byteOffset + payload.data.byteLength);
+        } else {
+          data = payload.data;
+        }
+        const cmd = {
+          type: payload.type,
+          connectionId: payload.connectionId,
+          data,
+          code: payload.code,
+          reason: payload.reason
+        };
+        for (const cb of callbacks) {
+          cb(cmd);
+        }
+      }
+      break;
+    }
+    case import_isolate_protocol.IsolateEvents.WS_CLIENT_CONNECT: {
+      const payload = message.payload;
+      handleClientWsConnect(message.isolateId, payload, state);
+      break;
+    }
+    case import_isolate_protocol.IsolateEvents.WS_CLIENT_SEND: {
+      const payload = message.payload;
+      handleClientWsSend(message.isolateId, payload, state);
+      break;
+    }
+    case import_isolate_protocol.IsolateEvents.WS_CLIENT_CLOSE: {
+      const payload = message.payload;
+      handleClientWsClose(message.isolateId, payload, state);
+      break;
+    }
+    default: {
+      const listeners = isolateEventListeners.get(message.isolateId);
+      if (listeners) {
+        const eventListeners = listeners.get(message.event);
+        if (eventListeners) {
+          for (const cb of eventListeners) {
+            cb(message.payload);
+          }
+        }
+      }
+      break;
+    }
+  }
+}
+function handleClientWsConnect(isolateId, payload, state) {
+  const { socketId, url, protocols } = payload;
+  let sockets = isolateClientWebSockets.get(isolateId);
+  if (!sockets) {
+    sockets = new Map;
+    isolateClientWebSockets.set(isolateId, sockets);
+  }
+  const setupWebSocket = (ws) => {
+    sockets.set(socketId, ws);
+    ws.onopen = () => {
+      sendMessage(state.socket, {
+        type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+        isolateId,
+        event: import_isolate_protocol.ClientEvents.WS_CLIENT_OPENED,
+        payload: { socketId, protocol: ws.protocol || "", extensions: ws.extensions || "" }
+      });
+    };
+    ws.onmessage = (event) => {
+      let data;
+      if (typeof event.data === "string") {
+        data = event.data;
+      } else if (event.data instanceof ArrayBuffer) {
+        data = new Uint8Array(event.data);
+      } else if (event.data instanceof Blob) {
+        event.data.arrayBuffer().then((buffer) => {
+          sendMessage(state.socket, {
+            type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+            isolateId,
+            event: import_isolate_protocol.ClientEvents.WS_CLIENT_MESSAGE,
+            payload: { socketId, data: new Uint8Array(buffer) }
+          });
+        });
+        return;
+      } else {
+        data = String(event.data);
+      }
+      sendMessage(state.socket, {
+        type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+        isolateId,
+        event: import_isolate_protocol.ClientEvents.WS_CLIENT_MESSAGE,
+        payload: { socketId, data }
+      });
+    };
+    ws.onerror = () => {
+      sendMessage(state.socket, {
+        type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+        isolateId,
+        event: import_isolate_protocol.ClientEvents.WS_CLIENT_ERROR,
+        payload: { socketId }
+      });
+    };
+    ws.onclose = (event) => {
+      sendMessage(state.socket, {
+        type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+        isolateId,
+        event: import_isolate_protocol.ClientEvents.WS_CLIENT_CLOSED,
+        payload: { socketId, code: event.code, reason: event.reason, wasClean: event.wasClean }
+      });
+      sockets?.delete(socketId);
+      if (sockets?.size === 0) {
+        isolateClientWebSockets.delete(isolateId);
+      }
+    };
+  };
+  const sendConnectionFailed = (reason) => {
+    sendMessage(state.socket, {
+      type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+      isolateId,
+      event: import_isolate_protocol.ClientEvents.WS_CLIENT_ERROR,
+      payload: { socketId }
+    });
+    sendMessage(state.socket, {
+      type: import_isolate_protocol.MessageType.CLIENT_EVENT,
+      isolateId,
+      event: import_isolate_protocol.ClientEvents.WS_CLIENT_CLOSED,
+      payload: { socketId, code: 1006, reason, wasClean: false }
+    });
+  };
+  const callback = isolateWebSocketCallbacks.get(isolateId);
+  if (callback) {
+    try {
+      const result = callback(url, protocols || []);
+      if (result instanceof Promise) {
+        result.then((ws) => {
+          if (ws === null) {
+            sendConnectionFailed("Connection blocked");
+          } else {
+            setupWebSocket(ws);
+          }
+        }).catch(() => {
+          sendConnectionFailed("Callback error");
+        });
+      } else if (result === null) {
+        sendConnectionFailed("Connection blocked");
+      } else {
+        setupWebSocket(result);
+      }
+    } catch {
+      sendConnectionFailed("Callback error");
+    }
+  } else {
+    try {
+      const ws = protocols && protocols.length > 0 ? new WebSocket(url, protocols) : new WebSocket(url);
+      setupWebSocket(ws);
+    } catch {
+      sendConnectionFailed("Connection failed");
+    }
+  }
+}
+function handleClientWsSend(isolateId, payload, state) {
+  const { socketId, data } = payload;
+  const sockets = isolateClientWebSockets.get(isolateId);
+  const ws = sockets?.get(socketId);
+  if (!ws || ws.readyState !== WebSocket.OPEN) {
+    return;
+  }
+  if (typeof data === "string" && data.startsWith("__BINARY__")) {
+    const base64 = data.slice(10);
+    const binary = Buffer.from(base64, "base64");
+    ws.send(binary);
+  } else if (data instanceof Uint8Array) {
+    ws.send(data);
+  } else {
+    ws.send(data);
+  }
+}
+function handleClientWsClose(isolateId, payload, state) {
+  const { socketId, code, reason } = payload;
+  const sockets = isolateClientWebSockets.get(isolateId);
+  const ws = sockets?.get(socketId);
+  if (!ws) {
+    return;
+  }
+  if (ws.readyState === WebSocket.OPEN || ws.readyState === WebSocket.CONNECTING) {
+    ws.close(code ?? 1000, reason ?? "");
+  }
+}
 
-//# debugId=1C20082A75E8BCB364756E2164756E21
+//# debugId=1DAE4D8C13C4D6B364756E2164756E21