@ai-ecoverse/slicc-kernel 1.103.3 → 1.106.2

package/dist/index.js

@@ -1030,6 +1030,28 @@ async function* chunks(body, idleMs) {
     await reader.cancel().catch(() => void 0);
   }
 }
+function streamOf(body) {
+  const it = body[Symbol.asyncIterator]();
+  return new ReadableStream(
+    {
+      async pull(controller) {
+        const next = await it.next();
+        if (next.done) controller.close();
+        else controller.enqueue(next.value);
+      },
+      async cancel(reason) {
+        await it.return?.(reason);
+      }
+    },
+    { highWaterMark: 0 }
+  );
+}
+function requestBody(request) {
+  if (request.bodyStream) {
+    return { body: streamOf(request.bodyStream), duplex: "half" };
+  }
+  return request.body ? { body: request.body } : {};
+}
 function fetchTransport(options = {}) {
   const send2 = options.fetch ?? globalThis.fetch.bind(globalThis);
   const Socket = options.webSocket === true ? globalThis.WebSocket : options.webSocket || void 0;
@@ -1041,7 +1063,8 @@ function fetchTransport(options = {}) {
       encodedBodies: false,
       maxRequestBody: options.maxRequestBody ?? MAX_REQUEST_BODY,
       crossOrigin: "cors",
-      ...Socket ? { websocket: true } : {}
+      ...Socket ? { websocket: true } : {},
+      ...options.streamingBodies ? { streamingBodies: true } : {}
     },
     ...Socket ? { websocket: (request) => openWebSocket(Socket, request, headers) } : {},
     async fetch(request) {
@@ -1050,7 +1073,7 @@ function fetchTransport(options = {}) {
         response = await send2(request.url, {
           method: request.method,
           headers: joinHeaders(request.headers),
-          ...request.body ? { body: request.body } : {},
+          ...requestBody(request),
           signal: request.signal,
           redirect,
           credentials: "omit",
@@ -1087,6 +1110,29 @@ function fetchTransport(options = {}) {
     }
   };
 }
+function uploadsFor(peer) {
+  const pulls = /* @__PURE__ */ new Map();
+  const of = (nid, signal) => ({
+    async *[Symbol.asyncIterator]() {
+      for (; ; ) {
+        signal.throwIfAborted();
+        const got = await new Promise((resolve, reject) => {
+          const stop = () => reject(signal.reason);
+          signal.addEventListener("abort", stop, { once: true });
+          pulls.set(nid, (call) => {
+            signal.removeEventListener("abort", stop);
+            resolve(call);
+          });
+          peer.postMessage({ net: "body-pull", nid });
+        }).finally(() => pulls.delete(nid));
+        if (got.net === "body-end") return;
+        if (got.net === "body-error") throw new Error(got.message);
+        if (got.net === "body-chunk") yield got.bytes;
+      }
+    }
+  });
+  return { of, answer: (call) => pulls.get(call.nid)?.(call) };
+}
 function serveTransport(peer, transport, { closeWaitMs = CLOSE_WAIT_MS } = {}) {
   const open = /* @__PURE__ */ new Map();
   const fail = (nid, e) => {
@@ -1100,6 +1146,7 @@ function serveTransport(peer, transport, { closeWaitMs = CLOSE_WAIT_MS } = {}) {
       ...typeof code === "string" ? { code } : {}
     });
   };
+  const uploads = uploadsFor(peer);
   const start = async (call) => {
     const abort = new AbortController();
     const entry = { abort };
@@ -1109,7 +1156,7 @@ function serveTransport(peer, transport, { closeWaitMs = CLOSE_WAIT_MS } = {}) {
         url: call.url,
         method: call.method,
         headers: call.headers,
-        ...call.body ? { body: call.body } : {},
+        ...call.streamed ? { bodyStream: uploads.of(call.nid, abort.signal) } : call.body ? { body: call.body } : {},
         signal: abort.signal
       });
       if (!open.has(call.nid)) {
@@ -1247,7 +1294,9 @@ function serveTransport(peer, transport, { closeWaitMs = CLOSE_WAIT_MS } = {}) {
   return {
     answer(call) {
       if (call.net === "fetch") void start(call);
-      else if (call.net === "read") void read(call.nid);
+      else if (call.net === "body-chunk" || call.net === "body-end" || call.net === "body-error") {
+        uploads.answer(call);
+      } else if (call.net === "read") void read(call.nid);
       else if (call.net === "ws-open") void openSocket(call);
       else if (call.net === "ws-send") void sendOn(call);
       else if (call.net === "ws-close") {
@@ -1278,12 +1327,17 @@ var RemoteTransport = class {
   waiting = /* @__PURE__ */ new Map();
   nextId = 0;
   sockets = /* @__PURE__ */ new Map();
+  uploads = /* @__PURE__ */ new Map();
   failure;
   constructor(port, traits) {
     this.port = port;
     this.traits = traits;
   }
   receive(reply) {
+    if (reply.net === "body-pull") {
+      void this.pull(reply.nid);
+      return;
+    }
     const socket = this.sockets.get(reply.nid);
     if (socket && this.toSocket(reply, socket)) return;
     const waiter = this.waiting.get(reply.nid);
@@ -1302,8 +1356,35 @@ var RemoteTransport = class {
     } else return false;
     return true;
   }
+  async pull(nid) {
+    const upload = this.uploads.get(nid);
+    if (!upload) {
+      this.port.postMessage({ net: "body-error", nid, message: "the request body is gone" });
+      return;
+    }
+    try {
+      const next = await upload.next();
+      if (next.done) {
+        this.uploads.delete(nid);
+        this.port.postMessage({ net: "body-end", nid });
+      } else this.port.postMessage({ net: "body-chunk", nid, bytes: next.value });
+    } catch (e) {
+      this.uploads.delete(nid);
+      this.port.postMessage({
+        net: "body-error",
+        nid,
+        message: e instanceof Error ? e.message : String(e)
+      });
+    }
+  }
+  dropUpload(nid) {
+    const upload = this.uploads.get(nid);
+    this.uploads.delete(nid);
+    void upload?.return?.().catch(() => void 0);
+  }
   fail(error) {
     this.failure = error;
+    for (const nid of [...this.uploads.keys()]) this.dropUpload(nid);
     for (const waiter of this.waiting.values()) waiter.reject(error);
     this.waiting.clear();
     for (const socket of this.sockets.values())
@@ -1330,14 +1411,17 @@ var RemoteTransport = class {
     let finished = false;
     const abort = () => {
       finished = true;
+      this.dropUpload(nid);
       this.cancel(nid);
     };
     signal.addEventListener("abort", abort, { once: true });
     const finish = () => {
+      this.dropUpload(nid);
       if (finished) return;
       finished = true;
       signal.removeEventListener("abort", abort);
     };
+    if (request.bodyStream) this.uploads.set(nid, request.bodyStream[Symbol.asyncIterator]());
     let head;
     try {
       head = await this.ask({
@@ -1346,7 +1430,7 @@ var RemoteTransport = class {
         url: request.url,
         method: request.method,
         headers: request.headers,
-        ...request.body ? { body: request.body } : {}
+        ...request.bodyStream ? { streamed: true } : request.body ? { body: request.body } : {}
       });
     } catch (e) {
       finish();