@wxn0brp/falcon-frame 0.5.4 → 0.5.5

package/dist/compress.d.ts

@@ -0,0 +1,3 @@
+import { FFResponse } from "./res.js";
+import { FFRequest } from "./types.js";
+export declare function compressionMiddleware(req: FFRequest, res: FFResponse): void;

package/dist/compress.js

@@ -0,0 +1,49 @@
+import { createGzip, createDeflate } from "zlib";
+export function compressionMiddleware(req, res) {
+    const encoding = req.headers["accept-encoding"];
+    if (!encoding || typeof encoding !== "string" || res.headersSent) {
+        return;
+    }
+    let compressionStream;
+    let encodingType;
+    if (/\bgzip\b/.test(encoding)) {
+        encodingType = "gzip";
+        compressionStream = createGzip();
+    }
+    else if (/\bdeflate\b/.test(encoding)) {
+        encodingType = "deflate";
+        compressionStream = createDeflate();
+    }
+    else {
+        return;
+    }
+    res.setHeader("Content-Encoding", encodingType);
+    res.removeHeader("Content-Length");
+    const originalWrite = res.write;
+    const originalEnd = res.end;
+    compressionStream.on("data", (chunk) => {
+        originalWrite.call(res, chunk);
+    });
+    compressionStream.on("end", () => {
+        originalEnd.call(res);
+    });
+    // @ts-ignore
+    res.write = (chunk, encoding) => {
+        if (!res.headersSent) {
+            res.writeHead(res.statusCode);
+        }
+        return compressionStream.write(chunk, encoding);
+    };
+    // @ts-ignore
+    res.end = (chunk, encoding) => {
+        if (res.writableEnded) {
+            return res;
+        }
+        if (chunk) {
+            // @ts-ignore
+            res.write(chunk, encoding);
+        }
+        compressionStream.end();
+        return res;
+    };
+}

package/dist/req.js

@@ -3,6 +3,7 @@ import { parseCookies } from "./helpers.js";
 import { FFResponse } from "./res.js";
 import { validate } from "./valid.js";
 import { getMiddlewares, matchMiddleware } from "./middleware.js";
+import { compressionMiddleware } from "./compress.js";
 export function handleRequest(req, res, FF) {
     Object.setPrototypeOf(res, FFResponse.prototype);
     res.FF = FF;
@@ -39,6 +40,8 @@ export function handleRequest(req, res, FF) {
         res.status(404).end("404: File had second thoughts");
         return;
     }
+    if (!FF.getVar("disable compression"))
+        compressionMiddleware(req, res);
     let middlewareIndex = 0;
     async function next() {
         if (middlewareIndex >= matchedMiddlewares.length) {

package/package.json

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