From eedb57f893986091887d1a889d6752ea4b06ba2a Mon Sep 17 00:00:00 2001
From: UltraQbik <no1skill@yandex.ru>
Date: Thu, 22 Aug 2024 03:00:37 +0300
Subject: Add simple HTML minimizer for gooder compression

---
 main.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'main.py')

diff --git a/main.py b/main.py
index e85e536..f22b957 100644
--- a/main.py
+++ b/main.py
@@ -11,6 +11,7 @@ import signal
 import asyncio
 import aiofiles
 from src.request import Request
+from src.minimizer import minimize_html
 
 
 # path mapping
@@ -168,6 +169,10 @@ class HTTPServer:
             async with aiofiles.open(PATH_MAP[request.path]["path"], "rb") as f:
                 data = await f.read()
 
+            # pre-compress data for HTML files
+            if PATH_MAP[request.path]["path"][-4:] == "html":
+                data = minimize_html(data)
+
             # add gzip compression header (if supported)
             headers = {}
             if "gzip" in compressions:
-- 
cgit 1.4.1