@nxtedition/nxt-undici 1.1.0 → 1.2.0

package/lib/index.js

@@ -126,7 +126,8 @@ class Readable extends stream.Readable {
 }
 
 const dispatchers = {
-  abort: require('./interceptor/abort.js'),
+  requestBody: require('./interceptor/request-body.js'),
+  dump: require('./interceptor/dump.js'),
   catch: require('./interceptor/catch.js'),
   responseContent: require('./interceptor/response-content.js'),
   requestContent: require('./interceptor/request-content.js'),
@@ -202,7 +203,8 @@ async function request(url, opts) {
   if (dispatch == null) {
     dispatch = (opts, handler) => dispatcher.dispatch(opts, handler)
     dispatch = dispatchers.catch(dispatch)
-    dispatch = dispatchers.abort(dispatch)
+    dispatch = dispatchers.requestBody(dispatch)
+    dispatch = dispatchers.dump(dispatch)
     dispatch = dispatchers.requestId(dispatch)
     dispatch = dispatchers.log(dispatch)
     dispatch = dispatchers.responseRetry(dispatch)
@@ -217,12 +219,6 @@ async function request(url, opts) {
     dispatcherCache.set(dispatcher, dispatch)
   }
 
-  // TODO (fix): retry support
-  let body = opts.body
-  if (typeof body === 'function') {
-    body = await opts.body({ signal: opts.signal })
-  }
-
   return new Promise((resolve, reject) =>
     dispatch(
       {
@@ -239,6 +235,7 @@ async function request(url, opts) {
         headersTimeout: opts.headersTimeout,
         bodyTimeout: opts.bodyTimeout,
         idempotent,
+        dump: opts.dump ?? 'abort',
         signal: opts.signal,
         retry: opts.retry ?? 8,
         proxy: opts.proxy,

package/lib/interceptor/{abort.js → dump.js}

@@ -1,16 +1,25 @@
 const { AbortError } = require('../utils')
 
+// TODO (fix): Configurable timeout
+// TODO (fix): Dump?
+
 class Handler {
   constructor(opts, { handler }) {
+    this.opts = opts
     this.handler = handler
     this.pos = 0
     this.reason = null
+    this.timeout = null
   }
 
   onConnect(abort) {
     this.abort = abort
     this.handler.onConnect((reason) => {
       this.reason = reason ?? new AbortError()
+      this.timeout = setTimeout(() => {
+        this.timeout = null
+        this.abort(this.reason)
+      }, this.opts.dump?.timeout ?? 10e3)
     })
   }
 
@@ -56,17 +65,22 @@ class Handler {
   }
 
   onComplete(rawTrailers) {
-    return this.reason == null
-      ? this.handler.onComplete(rawTrailers)
-      : this.handler.onError(this.reason)
+    return this.onFinally(this.reason, rawTrailers)
   }
 
   onError(err) {
-    return this.handler.onError(err)
+    return this.onFinally(err)
+  }
+
+  onFinally(err, rawTrailers) {
+    if (this.timeout) {
+      clearTimeout(this.timeout)
+      this.timeout = null
+    }
+
+    return err == null ? this.handler.onComplete(rawTrailers) : this.handler.onError(err)
   }
 }
 
 module.exports = (dispatch) => (opts, handler) =>
-  opts.method === 'GET' || opts.method === 'HEAD'
-    ? dispatch(opts, new Handler(opts, { handler }))
-    : dispatch(opts, handler)
+  opts.dump ? dispatch(opts, new Handler(opts, { handler })) : dispatch(opts, handler)

package/lib/interceptor/request-body.js

@@ -0,0 +1,7 @@
+module.exports = (dispatch) => (opts, handler) =>
+  typeof opts.body !== 'function'
+    ? dispatch(opts, handler)
+    : Promise.resolve(opts.body({ signal: opts.signal })).then(
+        (body) => dispatch({ ...opts, body }, handler),
+        (err) => handler.onError(err),
+      )

package/lib/utils.js

@@ -1,7 +1,12 @@
 const tp = require('node:timers/promises')
 
 function isDisturbed(body) {
-  if (body == null || typeof body === 'string' || Buffer.isBuffer(body)) {
+  if (
+    body == null ||
+    typeof body === 'string' ||
+    Buffer.isBuffer(body) ||
+    typeof body === 'function'
+  ) {
     return false
   }
 

package/package.json

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