@ricsam/isolate-fetch 0.1.13 → 0.1.15

package/dist/mjs/index.mjs

@@ -80,25 +80,32 @@ var headersCode = `
     }
 
     forEach(callback, thisArg) {
-      for (const [key, [originalName, values]] of this.#headers) {
-        callback.call(thisArg, values.join(', '), originalName, this);
+      const sortedKeys = [...this.#headers.keys()].sort();
+      for (const key of sortedKeys) {
+        const [, values] = this.#headers.get(key);
+        callback.call(thisArg, values.join(', '), key, this);
       }
     }
 
     *entries() {
-      for (const [key, [name, values]] of this.#headers) {
-        yield [name, values.join(', ')];
+      const sortedKeys = [...this.#headers.keys()].sort();
+      for (const key of sortedKeys) {
+        const [, values] = this.#headers.get(key);
+        yield [key, values.join(', ')];
       }
     }
 
     *keys() {
-      for (const [key, [name]] of this.#headers) {
-        yield name;
+      const sortedKeys = [...this.#headers.keys()].sort();
+      for (const key of sortedKeys) {
+        yield key;
       }
     }
 
     *values() {
-      for (const [key, [name, values]] of this.#headers) {
+      const sortedKeys = [...this.#headers.keys()].sort();
+      for (const key of sortedKeys) {
+        const [, values] = this.#headers.get(key);
         yield values.join(', ');
       }
     }
@@ -1445,14 +1452,20 @@ function setupFetchFunction(context, stateMap, streamRegistry, options) {
     fetchAbortControllers.set(fetchId, hostController);
     const headers = JSON.parse(headersJson);
     const bodyBytes = bodyJson ? JSON.parse(bodyJson) : null;
-    const body = bodyBytes ? new Uint8Array(bodyBytes) : null;
-    const nativeRequest = new Request(url, {
+    const rawBody = bodyBytes ? new Uint8Array(bodyBytes) : null;
+    const init = {
       method,
       headers,
-      body,
+      rawBody,
+      body: rawBody,
       signal: hostController.signal
-    });
-    const onFetch = options?.onFetch ?? fetch;
+    };
+    const onFetch = options?.onFetch ?? ((url2, init2) => fetch(url2, {
+      method: init2.method,
+      headers: init2.headers,
+      body: init2.body,
+      signal: init2.signal
+    }));
     try {
       let cleanupAbort;
       const abortPromise = new Promise((_, reject) => {
@@ -1467,7 +1480,7 @@ function setupFetchFunction(context, stateMap, streamRegistry, options) {
         cleanupAbort = () => hostController.signal.removeEventListener("abort", onAbort);
       });
       abortPromise.catch(() => {});
-      const nativeResponse = await Promise.race([onFetch(nativeRequest), abortPromise]);
+      const nativeResponse = await Promise.race([onFetch(url, init), abortPromise]);
       cleanupAbort?.();
       const status = nativeResponse.status;
       const isNullBody = status === 204 || status === 304 || method.toUpperCase() === "HEAD";
@@ -1687,6 +1700,373 @@ function setupServerWebSocket(context, wsCommandCallbacks) {
 })();
   `);
 }
+function setupClientWebSocket(context, clientWsCommandCallbacks) {
+  const global = context.global;
+  global.setSync("__WebSocket_connect", new ivm.Callback((socketId, url, protocols) => {
+    const cmd = {
+      type: "connect",
+      socketId,
+      url,
+      protocols
+    };
+    for (const cb of clientWsCommandCallbacks)
+      cb(cmd);
+  }));
+  global.setSync("__WebSocket_send", new ivm.Callback((socketId, data) => {
+    const cmd = { type: "send", socketId, data };
+    for (const cb of clientWsCommandCallbacks)
+      cb(cmd);
+  }));
+  global.setSync("__WebSocket_close", new ivm.Callback((socketId, code, reason) => {
+    const cmd = { type: "close", socketId, code, reason };
+    for (const cb of clientWsCommandCallbacks)
+      cb(cmd);
+  }));
+  context.evalSync(`
+(function() {
+  // Socket ID counter
+  let __nextSocketId = 1;
+
+  // Active sockets registry
+  const __clientWebSockets = new Map();
+
+  // Simple Event class (if not defined globally)
+  const _Event = globalThis.Event || class Event {
+    constructor(type, options = {}) {
+      this.type = type;
+      this.bubbles = options.bubbles || false;
+      this.cancelable = options.cancelable || false;
+      this.defaultPrevented = false;
+      this.timeStamp = Date.now();
+      this.target = null;
+      this.currentTarget = null;
+    }
+    preventDefault() {
+      if (this.cancelable) this.defaultPrevented = true;
+    }
+    stopPropagation() {}
+    stopImmediatePropagation() {}
+  };
+
+  // MessageEvent class for WebSocket messages
+  const _MessageEvent = globalThis.MessageEvent || class MessageEvent extends _Event {
+    constructor(type, options = {}) {
+      super(type, options);
+      this.data = options.data !== undefined ? options.data : null;
+      this.origin = options.origin || '';
+      this.lastEventId = options.lastEventId || '';
+      this.source = options.source || null;
+      this.ports = options.ports || [];
+    }
+  };
+
+  // CloseEvent class for WebSocket close
+  const _CloseEvent = globalThis.CloseEvent || class CloseEvent extends _Event {
+    constructor(type, options = {}) {
+      super(type, options);
+      this.code = options.code !== undefined ? options.code : 0;
+      this.reason = options.reason !== undefined ? options.reason : '';
+      this.wasClean = options.wasClean !== undefined ? options.wasClean : false;
+    }
+  };
+
+  // Helper to dispatch events
+  function dispatchEvent(ws, event) {
+    const listeners = ws._listeners.get(event.type) || [];
+    for (const listener of listeners) {
+      try {
+        listener.call(ws, event);
+      } catch (e) {
+        console.error('WebSocket event listener error:', e);
+      }
+    }
+    // Also call on* handler if set
+    const handler = ws['on' + event.type];
+    if (typeof handler === 'function') {
+      try {
+        handler.call(ws, event);
+      } catch (e) {
+        console.error('WebSocket handler error:', e);
+      }
+    }
+  }
+
+  class WebSocket {
+    static CONNECTING = 0;
+    static OPEN = 1;
+    static CLOSING = 2;
+    static CLOSED = 3;
+
+    #socketId;
+    #url;
+    #readyState = WebSocket.CONNECTING;
+    #bufferedAmount = 0;
+    #extensions = '';
+    #protocol = '';
+    #binaryType = 'blob';
+    _listeners = new Map();
+
+    // Event handlers
+    onopen = null;
+    onmessage = null;
+    onerror = null;
+    onclose = null;
+
+    constructor(url, protocols) {
+      // Validate URL
+      let parsedUrl;
+      try {
+        parsedUrl = new URL(url);
+      } catch (e) {
+        throw new DOMException("Failed to construct 'WebSocket': The URL '" + url + "' is invalid.", 'SyntaxError');
+      }
+
+      if (parsedUrl.protocol !== 'ws:' && parsedUrl.protocol !== 'wss:') {
+        throw new DOMException("Failed to construct 'WebSocket': The URL's scheme must be either 'ws' or 'wss'.", 'SyntaxError');
+      }
+
+      // Per WHATWG spec, fragments must be stripped from WebSocket URLs
+      parsedUrl.hash = '';
+      this.#url = parsedUrl.href;
+      this.#socketId = String(__nextSocketId++);
+
+      // Normalize protocols to array
+      let protocolArray = [];
+      if (protocols !== undefined) {
+        if (typeof protocols === 'string') {
+          protocolArray = [protocols];
+        } else if (Array.isArray(protocols)) {
+          protocolArray = protocols;
+        } else {
+          protocolArray = [String(protocols)];
+        }
+      }
+
+      // Check for duplicate protocols
+      const seen = new Set();
+      for (const p of protocolArray) {
+        if (seen.has(p)) {
+          throw new DOMException("Failed to construct 'WebSocket': The subprotocol '" + p + "' is duplicated.", 'SyntaxError');
+        }
+        seen.add(p);
+      }
+
+      // Register socket
+      __clientWebSockets.set(this.#socketId, this);
+
+      // Call host to create connection
+      __WebSocket_connect(this.#socketId, this.#url, protocolArray);
+    }
+
+    get url() {
+      return this.#url;
+    }
+
+    get readyState() {
+      return this.#readyState;
+    }
+
+    get bufferedAmount() {
+      return this.#bufferedAmount;
+    }
+
+    get extensions() {
+      return this.#extensions;
+    }
+
+    get protocol() {
+      return this.#protocol;
+    }
+
+    get binaryType() {
+      return this.#binaryType;
+    }
+
+    set binaryType(value) {
+      if (value !== 'blob' && value !== 'arraybuffer') {
+        throw new DOMException("Failed to set the 'binaryType' property: '" + value + "' is not a valid value.", 'SyntaxError');
+      }
+      this.#binaryType = value;
+    }
+
+    // ReadyState constants
+    get CONNECTING() { return WebSocket.CONNECTING; }
+    get OPEN() { return WebSocket.OPEN; }
+    get CLOSING() { return WebSocket.CLOSING; }
+    get CLOSED() { return WebSocket.CLOSED; }
+
+    send(data) {
+      if (this.#readyState === WebSocket.CONNECTING) {
+        throw new DOMException("Failed to execute 'send' on 'WebSocket': Still in CONNECTING state.", 'InvalidStateError');
+      }
+
+      if (this.#readyState !== WebSocket.OPEN) {
+        // Silently discard if not open (per spec)
+        return;
+      }
+
+      // Convert data to string for transfer
+      let dataStr;
+      if (typeof data === 'string') {
+        dataStr = data;
+      } else if (data instanceof ArrayBuffer) {
+        // Convert ArrayBuffer to base64 for transfer
+        const bytes = new Uint8Array(data);
+        let binary = '';
+        for (let i = 0; i < bytes.byteLength; i++) {
+          binary += String.fromCharCode(bytes[i]);
+        }
+        dataStr = '__BINARY__' + btoa(binary);
+      } else if (ArrayBuffer.isView(data)) {
+        const bytes = new Uint8Array(data.buffer, data.byteOffset, data.byteLength);
+        let binary = '';
+        for (let i = 0; i < bytes.byteLength; i++) {
+          binary += String.fromCharCode(bytes[i]);
+        }
+        dataStr = '__BINARY__' + btoa(binary);
+      } else if (data instanceof Blob) {
+        // Blob.arrayBuffer() is async, but send() is sync
+        // For now, throw - this is a limitation
+        throw new DOMException("Failed to execute 'send' on 'WebSocket': Blob data is not supported in this environment.", 'NotSupportedError');
+      } else {
+        dataStr = String(data);
+      }
+
+      __WebSocket_send(this.#socketId, dataStr);
+    }
+
+    close(code, reason) {
+      if (code !== undefined) {
+        if (typeof code !== 'number' || code !== Math.floor(code)) {
+          throw new DOMException("Failed to execute 'close' on 'WebSocket': The code must be an integer.", 'InvalidAccessError');
+        }
+        if (code !== 1000 && (code < 3000 || code > 4999)) {
+          throw new DOMException("Failed to execute 'close' on 'WebSocket': The code must be either 1000, or between 3000 and 4999.", 'InvalidAccessError');
+        }
+      }
+
+      if (reason !== undefined) {
+        const encoder = new TextEncoder();
+        if (encoder.encode(reason).byteLength > 123) {
+          throw new DOMException("Failed to execute 'close' on 'WebSocket': The message must not be greater than 123 bytes.", 'SyntaxError');
+        }
+      }
+
+      if (this.#readyState === WebSocket.CLOSING || this.#readyState === WebSocket.CLOSED) {
+        return;
+      }
+
+      this.#readyState = WebSocket.CLOSING;
+      __WebSocket_close(this.#socketId, code ?? 1000, reason ?? '');
+    }
+
+    // EventTarget interface
+    addEventListener(type, listener, options) {
+      if (typeof listener !== 'function') return;
+      let listeners = this._listeners.get(type);
+      if (!listeners) {
+        listeners = [];
+        this._listeners.set(type, listeners);
+      }
+      if (!listeners.includes(listener)) {
+        listeners.push(listener);
+      }
+    }
+
+    removeEventListener(type, listener, options) {
+      const listeners = this._listeners.get(type);
+      if (!listeners) return;
+      const index = listeners.indexOf(listener);
+      if (index !== -1) {
+        listeners.splice(index, 1);
+      }
+    }
+
+    dispatchEvent(event) {
+      dispatchEvent(this, event);
+      return !event.defaultPrevented;
+    }
+
+    // Internal methods called from host
+    _setProtocol(protocol) {
+      this.#protocol = protocol;
+    }
+
+    _setExtensions(extensions) {
+      this.#extensions = extensions;
+    }
+
+    _setReadyState(state) {
+      this.#readyState = state;
+    }
+
+    _dispatchOpen() {
+      this.#readyState = WebSocket.OPEN;
+      const event = new _Event('open');
+      dispatchEvent(this, event);
+    }
+
+    _dispatchMessage(data) {
+      // Handle binary data
+      let messageData = data;
+      if (typeof data === 'string' && data.startsWith('__BINARY__')) {
+        const base64 = data.slice(10);
+        const binary = atob(base64);
+        const bytes = new Uint8Array(binary.length);
+        for (let i = 0; i < binary.length; i++) {
+          bytes[i] = binary.charCodeAt(i);
+        }
+        if (this.#binaryType === 'arraybuffer') {
+          messageData = bytes.buffer;
+        } else {
+          messageData = new Blob([bytes]);
+        }
+      }
+
+      const event = new _MessageEvent('message', { data: messageData });
+      dispatchEvent(this, event);
+    }
+
+    _dispatchError() {
+      const event = new _Event('error');
+      dispatchEvent(this, event);
+    }
+
+    _dispatchClose(code, reason, wasClean) {
+      this.#readyState = WebSocket.CLOSED;
+      const event = new _CloseEvent('close', { code, reason, wasClean });
+      dispatchEvent(this, event);
+      __clientWebSockets.delete(this.#socketId);
+    }
+  }
+
+  // Helper to dispatch events from host to a socket by ID
+  globalThis.__dispatchClientWebSocketEvent = function(socketId, eventType, data) {
+    const ws = __clientWebSockets.get(socketId);
+    if (!ws) return;
+
+    switch (eventType) {
+      case 'open':
+        ws._setProtocol(data.protocol || '');
+        ws._setExtensions(data.extensions || '');
+        ws._dispatchOpen();
+        break;
+      case 'message':
+        ws._dispatchMessage(data.data);
+        break;
+      case 'error':
+        ws._dispatchError();
+        break;
+      case 'close':
+        ws._dispatchClose(data.code, data.reason, data.wasClean);
+        break;
+    }
+  };
+
+  globalThis.WebSocket = WebSocket;
+})();
+  `);
+}
 function setupServe(context) {
   context.evalSync(`
 (function() {
@@ -1717,9 +2097,46 @@ async function setupFetch(context, options) {
     activeConnections: new Map
   };
   const wsCommandCallbacks = new Set;
+  const clientWsCommandCallbacks = new Set;
   setupServer(context, serveState);
   setupServerWebSocket(context, wsCommandCallbacks);
   setupServe(context);
+  setupClientWebSocket(context, clientWsCommandCallbacks);
+  const eventCallbacks = new Set;
+  context.global.setSync("__emit", new ivm.Callback((eventName, payloadJson) => {
+    const payload = JSON.parse(payloadJson);
+    for (const cb of eventCallbacks)
+      cb(eventName, payload);
+  }));
+  context.evalSync(`
+(function() {
+  const __eventListeners = new Map();
+
+  globalThis.__on = function(event, callback) {
+    let listeners = __eventListeners.get(event);
+    if (!listeners) {
+      listeners = new Set();
+      __eventListeners.set(event, listeners);
+    }
+    listeners.add(callback);
+    return function() {
+      listeners.delete(callback);
+      if (listeners.size === 0) {
+        __eventListeners.delete(event);
+      }
+    };
+  };
+
+  globalThis.__dispatchExternalEvent = function(event, payloadJson) {
+    const listeners = __eventListeners.get(event);
+    if (!listeners) return;
+    const payload = JSON.parse(payloadJson);
+    for (const cb of listeners) {
+      try { cb(payload); } catch (e) { console.error('Event listener error:', e); }
+    }
+  };
+})();
+  `);
   return {
     dispose() {
       stateMap.clear();
@@ -1977,6 +2394,63 @@ async function setupFetch(context, options) {
     },
     hasActiveConnections() {
       return serveState.activeConnections.size > 0;
+    },
+    dispatchClientWebSocketOpen(socketId, protocol, extensions) {
+      const safeProtocol = protocol.replace(/\\/g, "\\\\").replace(/"/g, "\\\"");
+      const safeExtensions = extensions.replace(/\\/g, "\\\\").replace(/"/g, "\\\"");
+      context.evalSync(`
+        __dispatchClientWebSocketEvent("${socketId}", "open", {
+          protocol: "${safeProtocol}",
+          extensions: "${safeExtensions}"
+        });
+      `);
+    },
+    dispatchClientWebSocketMessage(socketId, data) {
+      if (typeof data === "string") {
+        const safeData = data.replace(/\\/g, "\\\\").replace(/"/g, "\\\"").replace(/\n/g, "\\n").replace(/\r/g, "\\r");
+        context.evalSync(`
+          __dispatchClientWebSocketEvent("${socketId}", "message", { data: "${safeData}" });
+        `);
+      } else {
+        const bytes = new Uint8Array(data);
+        let binary = "";
+        for (let i = 0;i < bytes.byteLength; i++) {
+          binary += String.fromCharCode(bytes[i]);
+        }
+        const base64 = Buffer.from(binary, "binary").toString("base64");
+        context.evalSync(`
+          __dispatchClientWebSocketEvent("${socketId}", "message", { data: "__BINARY__${base64}" });
+        `);
+      }
+    },
+    dispatchClientWebSocketClose(socketId, code, reason, wasClean) {
+      const safeReason = reason.replace(/\\/g, "\\\\").replace(/"/g, "\\\"").replace(/\n/g, "\\n");
+      context.evalIgnored(`
+        __dispatchClientWebSocketEvent("${socketId}", "close", {
+          code: ${code},
+          reason: "${safeReason}",
+          wasClean: ${wasClean}
+        });
+      `);
+    },
+    dispatchClientWebSocketError(socketId) {
+      context.evalIgnored(`
+        __dispatchClientWebSocketEvent("${socketId}", "error", {});
+      `);
+    },
+    onClientWebSocketCommand(callback) {
+      clientWsCommandCallbacks.add(callback);
+      return () => clientWsCommandCallbacks.delete(callback);
+    },
+    onEvent(callback) {
+      eventCallbacks.add(callback);
+      return () => eventCallbacks.delete(callback);
+    },
+    dispatchEvent(event, payload) {
+      const json = JSON.stringify(payload);
+      const safeEvent = JSON.stringify(event);
+      const safeJson = JSON.stringify(json);
+      context.evalSync(`__dispatchExternalEvent(${safeEvent}, ${safeJson});`);
     }
   };
 }
@@ -1985,4 +2459,4 @@ export {
   clearAllInstanceState
 };
 
-//# debugId=D6C7C0BA2C0C13D164756E2164756E21
+//# debugId=98B5825CD3CA576464756E2164756E21