@nxtedition/nxt-undici 6.1.4 → 6.1.6

package/lib/interceptor/cache.js

@@ -190,22 +190,26 @@ export default () => (dispatch) => (opts, handler) => {
     }
   }
 
-  handler.onConnect(abort)
-  if (aborted) {
-    return
-  }
-
-  handler.onHeaders(statusCode, headers, NOOP)
-  if (aborted) {
-    return
-  }
+  try {
+    handler.onConnect(abort)
+    if (aborted) {
+      return
+    }
 
-  if (body?.byteLength) {
-    handler.onData(body)
+    handler.onHeaders(statusCode, headers, NOOP)
     if (aborted) {
       return
     }
-  }
 
-  handler.onComplete({})
+    if (body?.byteLength) {
+      handler.onData(body)
+      if (aborted) {
+        return
+      }
+    }
+
+    handler.onComplete({})
+  } catch (err) {
+    abort(err)
+  }
 }

package/lib/utils.js

@@ -1,5 +1,6 @@
 import cacheControlParser from 'cache-control-parser'
 import stream from 'node:stream'
+import assert from 'node:assert'
 import { util } from '@nxtedition/undici'
 
 let fastNow = Date.now()
@@ -200,6 +201,7 @@ export class DecoratorHandler {
   #handler
   #aborted = false
   #errored = false
+  #completed = false
 
   constructor(handler) {
     if (typeof handler !== 'object' || handler === null) {
@@ -211,10 +213,13 @@ export class DecoratorHandler {
   onConnect(abort) {
     this.#aborted = false
     this.#errored = false
+    this.#completed = false
 
     return this.#handler.onConnect?.((reason) => {
-      this.#aborted = true
-      abort(reason)
+      if (!this.#aborted && !this.#completed && !this.#errored) {
+        this.#aborted = true
+        abort(reason)
+      }
     })
   }
 
@@ -232,19 +237,23 @@ export class DecoratorHandler {
 
   onData(data) {
     if (!this.#aborted && !this.#errored && data != null) {
+      assert(!this.#completed)
       return this.#handler.onData?.(data)
     }
   }
 
   onComplete(trailers) {
-    if (!this.#aborted && !this.#errored) {
+    if (!this.#aborted && !this.#completed) {
+      this.#completed = true
+      assert(!this.#errored)
       return this.#handler.onComplete?.(trailers)
     }
   }
 
   onError(err) {
-    if (!this.#errored) {
+    if (!this.#errored && !this.#completed) {
       this.#errored = true
+      assert(!this.#completed)
       return this.#handler.onError?.(err)
     }
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/nxt-undici",
-  "version": "6.1.4",
+  "version": "6.1.6",
   "license": "MIT",
   "author": "Robert Nagy <robert.nagy@boffins.se>",
   "main": "lib/index.js",