@nxtedition/lib 19.1.7 → 19.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/lib",
-  "version": "19.1.7",
+  "version": "19.3.0",
   "license": "MIT",
   "author": "Robert Nagy <robert.nagy@boffins.se>",
   "type": "module",

package/s3.js

@@ -3,10 +3,73 @@ import stream from 'node:stream'
 import tp from 'node:timers/promises'
 import AWS from '@aws-sdk/client-s3'
 import PQueue from 'p-queue'
+import { NodeHttpHandler } from '@smithy/node-http-handler'
+import undici from 'undici'
 
 const QUEUE = new PQueue({ concurrency: 8 })
 const MD5_HEX_EXPR = /^[A-F0-9]{32}$/i
 
+/**
+ * @typedef {import('undici').Dispatcher} Dispatcher
+ * @typedef {import('@aws-sdk/client-s3').S3ClientConfig & {dispatcher?: Dispatcher}} S3ClientConfig
+ */
+
+export class S3Client extends AWS.S3Client {
+  /**
+   * @param {S3ClientConfig} config
+   */
+  constructor(config) {
+    const { dispatcher, ...options } = config
+    super({
+      requestHandler: new UndiciRequestHandler(dispatcher),
+      ...options,
+    })
+  }
+}
+
+class UndiciRequestHandler extends NodeHttpHandler {
+  #dispatcher
+
+  /**
+   * @param {Dispatcher=} dispatcher
+   */
+  constructor(dispatcher) {
+    super()
+    this.#dispatcher = dispatcher
+  }
+
+  /**
+   * @param {import('@smithy/protocol-http').HttpRequest} request
+   * @param {import('@smithy/types').HttpHandlerOptions=} options
+   */
+  async handle(request, options) {
+    const abortSignal = options?.abortSignal
+
+    const { protocol, hostname, port, path, ...requestOptions } = request
+    // NOTE: Expect header is not supported by undici
+    const { Expect, ...headers } = request.headers
+    const typedMethod = /** @type {import('undici').Dispatcher.HttpMethod} */ (request.method)
+
+    const url = `${request.protocol}//${request.hostname}${request.port ? `:${request.port}` : ''}${request.path}`
+
+    const response = await undici.request(url, {
+      ...requestOptions,
+      method: typedMethod,
+      signal: abortSignal,
+      dispatcher: this.#dispatcher,
+      headers,
+      body: request.body,
+    })
+
+    return {
+      response: {
+        ...response,
+        headers: getTransformedHeaders(response.headers),
+      },
+    }
+  }
+}
+
 /**
  * Uploads a file to S3 using multipart upload.
  *
@@ -241,3 +304,25 @@ export async function upload({
     signal?.removeEventListener('abort', onAbort)
   }
 }
+
+/**
+ * @see https://github.com/smithy-lang/smithy-typescript/blob/main/packages/node-http-handler/src/get-transformed-headers.ts
+ *
+ * @param {import('http2').IncomingHttpHeaders} headers
+ * @returns {import('@smithy/types').HeaderBag}
+ */
+function getTransformedHeaders(headers) {
+  /**
+   * @type {import('@smithy/types').HeaderBag}
+   */
+  const transformedHeaders = {}
+
+  for (const name of Object.keys(headers)) {
+    const headerValues = headers[name]
+    transformedHeaders[name] = Array.isArray(headerValues)
+      ? headerValues.join(',')
+      : headerValues ?? ''
+  }
+
+  return transformedHeaders
+}