@travetto/web-connect 6.0.0-rc.3 → 6.0.0-rc.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@travetto/web-connect",
-  "version": "6.0.0-rc.3",
+  "version": "6.0.0-rc.4",
   "description": "Web integration for Connect-Like Resources",
   "keywords": [
     "web",
@@ -24,7 +24,7 @@
     "directory": "module/web-connect"
   },
   "dependencies": {
-    "@travetto/web": "^6.0.0-rc.3"
+    "@travetto/web": "^6.0.0-rc.4"
   },
   "peerDependencies": {
     "@travetto/cli": "^6.0.0-rc.3",

package/src/connect.ts

@@ -50,6 +50,7 @@ export class ConnectResponse implements Pick<ServerResponse,
   #headersSent = false;
   #finished = false;
   #written: Buffer[] = [];
+  #onEndHandlers: (() => void)[] = [];
 
   constructor(response?: WebResponse) {
     this.#response = response ?? new WebResponse();
@@ -143,11 +144,19 @@ export class ConnectResponse implements Pick<ServerResponse,
     }
     this.#finished = true;
     cb?.();
+    for (const item of this.#onEndHandlers) {
+      item();
+    }
+    return this;
+  }
+
+  on(type: 'end', handler: () => void): this {
+    this.#onEndHandlers.push(handler);
     return this;
   }
 
   throwIfSent(): void {
-    if (!this.#headersSent) {
+    if (this.#headersSent) {
       this.#response.body = Buffer.concat(this.#written);
       throw this.#response;
     }

package/src/util.ts

@@ -27,6 +27,8 @@ export class WebConnectUtil {
     const connectRes = ConnectResponse.get();
 
     const p = Promise.withResolvers<T | undefined>();
+    connectRes.on('end', () => p.resolve(null!));
+
     handler(connectReq, connectRes, (err, value) => err ? p.reject(err) : p.resolve(value!));
     const result = await p.promise;
     connectRes.throwIfSent();