@srfnstack/spliffy 1.1.4 → 1.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@srfnstack/spliffy",
-  "version": "1.1.4",
+  "version": "1.1.5",
   "author": "snowbldr",
   "private": false,
   "homepage": "https://github.com/narcolepticsnowman/spliffy",

package/src/decorator.mjs

@@ -97,19 +97,21 @@ export function decorateResponse (res, req, finalizeResponse, errorTransformer,
     if (!res.statusCode) res.statusCode = httpStatusCodes.OK
     if (!res.statusMessage) res.statusMessage = defaultStatusMessages[res.statusCode]
     res.headersSent = true
-    res.writeStatus(`${res.statusCode} ${res.statusMessage}`)
-    if (typeof res.onFlushHeaders === 'function') {
-      res.onFlushHeaders(res)
-    }
-    for (const header of Object.keys(res.headers)) {
-      if (Array.isArray(res.headers[header])) {
-        for (const multiple of res.headers[header]) {
-          res.writeHeader(header, multiple.toString())
+    res.cork(() => {
+      res.writeStatus(`${res.statusCode} ${res.statusMessage}`)
+      if (typeof res.onFlushHeaders === 'function') {
+        res.onFlushHeaders(res)
+      }
+      for (const header of Object.keys(res.headers)) {
+        if (Array.isArray(res.headers[header])) {
+          for (const multiple of res.headers[header]) {
+            res.writeHeader(header, multiple.toString())
+          }
+        } else {
+          res.writeHeader(header, res.headers[header].toString())
         }
-      } else {
-        res.writeHeader(header, res.headers[header].toString())
       }
-    }
+    })
   }
   res.writeHead = (status, headers) => {
     res.statusCode = status
@@ -128,22 +130,26 @@ export function decorateResponse (res, req, finalizeResponse, errorTransformer,
     return this
   }
 
-  res.writeArrayBuffer = res.write
+  res.uwsWrite = res.write
   res.write = (chunk, encoding, cb) => {
     try {
-      res.streaming = true
-      res.flushHeaders()
       let result
-      if (chunk instanceof Buffer) {
-        result = res.writeArrayBuffer(toArrayBuffer(chunk))
-      } else if (typeof chunk === 'string') {
-        result = res.writeArrayBuffer(toArrayBuffer(Buffer.from(chunk, encoding || 'utf8')))
-      } else {
-        result = res.writeArrayBuffer(toArrayBuffer(Buffer.from(JSON.stringify(chunk), encoding || 'utf8')))
-      }
-      if (typeof cb === 'function') {
-        cb()
-      }
+      res.cork(() => {
+        res.streaming = true
+        res.flushHeaders()
+        let data
+        if (chunk instanceof Buffer) {
+          data = toArrayBuffer(chunk)
+        } else if (typeof chunk === 'string') {
+          data = toArrayBuffer(Buffer.from(chunk, encoding || 'utf8'))
+        } else {
+          data = toArrayBuffer(Buffer.from(JSON.stringify(chunk), encoding || 'utf8'))
+        }
+        result = res.uwsWrite(data)
+        if (typeof cb === 'function') {
+          cb()
+        }
+      })
       return result
     } catch (e) {
       if (typeof cb === 'function') {
@@ -181,10 +187,12 @@ export function decorateResponse (res, req, finalizeResponse, errorTransformer,
     }
     // provide writableEnded like node does, with slightly different behavior
     if (!res.writableEnded) {
-      res.flushHeaders()
-      uwsEnd.call(res, body)
-      res.writableEnded = true
-      res.ended = true
+      res.cork(() => {
+        res.flushHeaders()
+        uwsEnd.call(res, body)
+        res.writableEnded = true
+        res.ended = true
+      })
     }
     if (typeof res.onEnd === 'function') {
       res.onEnd()

package/src/handler.mjs

@@ -56,7 +56,6 @@ const end = (res, defaultStatusCode, statusCodeOverride, body) => {
   if (body instanceof Readable || res.streaming) {
     res.streaming = true
     if (body instanceof Readable) {
-      res.flushHeaders()
       pipeResponse(res, body)
     }
     // handler is responsible for ending the response if they are streaming
@@ -65,10 +64,14 @@ const end = (res, defaultStatusCode, statusCodeOverride, body) => {
   }
 }
 
+const ipv6CompressRegex = /\b:?(?:0+:?){2,}/g
+
+const compressIpv6 = ip => ip && ip.includes(':') ? ip.replaceAll(ipv6CompressRegex, '::') : ip
+
 const writeAccess = function (req, res) {
   const start = new Date().getTime()
   return () => {
-    log.access(req.remoteAddress, res.proxiedRemoteAddress || '', res.statusCode, req.method, req.url, new Date().getTime() - start + 'ms')
+    log.access(compressIpv6(req.remoteAddress), compressIpv6(res.proxiedRemoteAddress) || '', res.statusCode, req.method, req.url, new Date().getTime() - start + 'ms')
   }
 }
 

package/src/middleware.mjs

@@ -50,6 +50,7 @@ export const validateMiddleware = (middleware) => {
 }
 
 export const validateMiddlewareArray = (arr) => {
+  if (!arr) return
   if (!Array.isArray(arr)) {
     throw new Error('middleware must be an array of functions')
   }

package/src/staticHandler.mjs

@@ -71,4 +71,4 @@ export function createStaticHandler (fullPath, contentType, cacheStatic, staticC
       }
     }
   }
-}
+}