@ai-ecoverse/slicc-kernel 1.26.0 → 1.26.1

package/dist/node.js

@@ -2454,6 +2454,7 @@ Sec-WebSocket-Accept: ${accept2}\r
 // src/cdp/facade.ts
 var MAX_HEAD = 64 * 1024;
 var LIST_TIMEOUT = 1e4;
+var CLOSE_WAIT_MS = 1e3;
 function respond(status, body, type = "text/plain; charset=utf-8") {
   const bytes = new TextEncoder().encode(body);
   const head = latin1Bytes(
@@ -2599,7 +2600,11 @@ var CdpFacade = class {
   async upgrade(conn, incoming, req, runtime) {
     const accept2 = await acceptKey(req.headers);
     const host = await this.host(runtime);
-    await new Relay(conn, incoming, host, this.options.maxMessage ?? MAX_MESSAGE, this.stop.signal).start(switchingHead(accept2)).finally(() => host.close());
+    const limits = {
+      max: this.options.maxMessage ?? MAX_MESSAGE,
+      closeWaitMs: this.options.closeWaitMs ?? CLOSE_WAIT_MS
+    };
+    await new Relay(conn, incoming, host, limits, this.stop.signal).start(switchingHead(accept2)).finally(() => host.close());
   }
 };
 var Relay = class {
@@ -2609,12 +2614,14 @@ var Relay = class {
   incoming;
   host;
   max;
+  closeWaitMs;
   stop;
-  constructor(conn, incoming, host, max, stop) {
+  constructor(conn, incoming, host, limits, stop) {
     this.conn = conn;
     this.incoming = incoming;
     this.host = host;
-    this.max = max;
+    this.max = limits.max;
+    this.closeWaitMs = limits.closeWaitMs;
     this.stop = stop;
   }
   send(bytes) {
@@ -2640,8 +2647,15 @@ var Relay = class {
         void this.end(CLOSE.tooBig, `a message is over ${this.max} bytes`);
       else void this.send(frame(OP_TEXT, bytes));
     };
-    await Promise.race([this.pump(), hostGone]);
+    const pumping = this.pump();
+    await Promise.race([pumping, hostGone]);
     await this.writes;
+    let timer;
+    const waited = new Promise((resolve) => {
+      timer = setTimeout(resolve, this.closeWaitMs);
+    });
+    await Promise.race([pumping, waited]);
+    clearTimeout(timer);
   }
   async pump() {
     const reader = new WsReader(this.incoming, this.max);