@ricsam/isolate-daemon 0.1.20 → 0.1.21

package/dist/cjs/connection.cjs

@@ -47,6 +47,7 @@ var import_isolate_protocol = require("@ricsam/isolate-protocol");
 var import_callback_fs_handler = require("./callback-fs-handler.cjs");
 var import_isolate_runtime = require("@ricsam/isolate-runtime");
 var LINKER_CONFLICT_ERROR = "Module is currently being linked by another linker";
+var NULL_BODY_STATUSES = new Set([101, 103, 204, 205, 304]);
 function getErrorText(error) {
   if (error instanceof Error) {
     const cause = error.cause;
@@ -1480,16 +1481,36 @@ function handleCallbackStreamStart(message, connection) {
       return Promise.resolve();
     }
   });
+  const cancelStream = () => {
+    connection.callbackStreamReceivers.delete(message.streamId);
+    sendMessage(connection.socket, {
+      type: import_isolate_protocol.MessageType.CALLBACK_STREAM_CANCEL,
+      streamId: message.streamId
+    });
+  };
   connection.callbackStreamReceivers.set(message.streamId, receiver);
   const pending = connection.pendingCallbacks.get(message.requestId);
-  if (pending) {
-    connection.pendingCallbacks.delete(message.requestId);
-    const response = new Response(readableStream, {
+  if (!pending) {
+    cancelStream();
+    return;
+  }
+  try {
+    const body = NULL_BODY_STATUSES.has(message.metadata.status) ? null : readableStream;
+    const response = new Response(body, {
       status: message.metadata.status,
       statusText: message.metadata.statusText,
       headers: message.metadata.headers
     });
+    connection.pendingCallbacks.delete(message.requestId);
     pending.resolve({ __streamingResponse: true, response });
+    if (body === null) {
+      cancelStream();
+    }
+  } catch (err) {
+    connection.pendingCallbacks.delete(message.requestId);
+    cancelStream();
+    const error = err instanceof Error ? err : new Error(String(err));
+    pending.reject(error);
   }
 }
 function handleCallbackStreamChunk(message, connection) {
@@ -1752,4 +1773,4 @@ async function handleClearCollectedData(message, connection, state) {
   }
 }
 
-//# debugId=3783342607D7276564756E2164756E21
+//# debugId=CCDD5E4DA24900F364756E2164756E21