@wxn0brp/falcon-frame 0.5.8 → 0.5.9

package/dist/compress.d.ts

@@ -1,4 +1,4 @@
 import { FFResponse } from "./res.js";
-import { FFRequest } from "./types.js";
+import { FFRequest, RouteHandler } from "./types.js";
 export declare function compression(req: FFRequest, res: FFResponse): void;
-export declare function compressionMiddleware(): (req: FFRequest, res: FFResponse, next: () => void) => void;
+export declare const compressionMiddleware: RouteHandler;

package/dist/compress.js

@@ -27,29 +27,54 @@ export function compression(req, res) {
     compressionStream.on("end", () => {
         originalEnd.call(res);
     });
+    compressionStream.on("error", (err) => {
+        if (!res.headersSent) {
+            res.statusCode = 500;
+            originalEnd.call(res);
+        }
+        else {
+            res.destroy(err);
+        }
+    });
+    res.on("close", () => {
+        compressionStream.destroy();
+    });
     // @ts-ignore
-    res.write = (chunk, encoding) => {
+    res.write = (...args) => {
+        const [chunk, encoding, cb] = args;
+        const callback = typeof encoding === "function" ? encoding : cb;
+        const enc = typeof encoding === "function" ? undefined : encoding;
         if (!res.headersSent) {
             res.writeHead(res.statusCode);
         }
-        return compressionStream.write(chunk, encoding);
+        return compressionStream.write(chunk, enc, callback);
     };
     // @ts-ignore
-    res.end = (chunk, encoding) => {
-        if (res.writableEnded) {
+    res.end = (...args) => {
+        if (res.writableEnded)
             return res;
+        const [chunk, encoding, cb] = args;
+        let finalChunk = chunk;
+        let finalCb = cb;
+        let finalEncoding = encoding;
+        if (typeof chunk === "function") {
+            finalCb = chunk;
+            finalChunk = undefined;
+        }
+        else if (typeof encoding === "function") {
+            finalCb = encoding;
+            finalEncoding = undefined;
         }
-        if (chunk) {
-            // @ts-ignore
-            res.write(chunk, encoding);
+        if (finalCb)
+            res.once("finish", finalCb);
+        if (finalChunk) {
+            res.write(finalChunk, finalEncoding);
         }
         compressionStream.end();
         return res;
     };
 }
-export function compressionMiddleware() {
-    return (req, res, next) => {
-        compression(req, res);
-        next();
-    };
-}
+export const compressionMiddleware = (req, res, next) => {
+    compression(req, res);
+    next();
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wxn0brp/falcon-frame",
-  "version": "0.5.8",
+  "version": "0.5.9",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "author": "wxn0brP",