From bec02795b8ed6ceede02e64034208b7047906633 Mon Sep 17 00:00:00 2001
From: Jakob Ketterl <dd5jfk@darc.de>
Date: Fri, 4 Sep 2020 15:44:25 +0200
Subject: [PATCH] implement gzip compression for assets

---
 owrx/controllers/assets.py | 36 +++++++++++++++++++++++++++++++++---
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/owrx/controllers/assets.py b/owrx/controllers/assets.py
index b15ff64..d9e5b73 100644
--- a/owrx/controllers/assets.py
+++ b/owrx/controllers/assets.py
@@ -5,9 +5,39 @@ import mimetypes
 import os
 import pkg_resources
 from abc import ABCMeta, abstractmethod
+import gzip
+
+import logging
+
+logger = logging.getLogger(__name__)
 
 
-class ModificationAwaraController(Controller, metaclass=ABCMeta):
+class GzipMixin(object):
+    def send_response(self, content, headers=None, content_type="text/html",  *args, **kwargs):
+        if self.zipable(content_type) and "accept-encoding" in self.request.headers:
+            accepted = [s.strip().lower() for s in self.request.headers['accept-encoding'].split(",")]
+            if "gzip" in accepted:
+                if type(content) == str:
+                    content = content.encode()
+                content = self.gzip(content)
+                if headers is None:
+                    headers = {}
+                headers["Content-Encoding"] = "gzip"
+        super().send_response(content, headers=headers, content_type=content_type, *args, **kwargs)
+
+    def zipable(self, content_type):
+        types = [
+            "application/javascript",
+            "text/css",
+            "text/html"
+        ]
+        return content_type in types
+
+    def gzip(self, content):
+        return gzip.compress(content)
+
+
+class ModificationAwareController(Controller, metaclass=ABCMeta):
     @abstractmethod
     def getModified(self, file):
         pass
@@ -28,7 +58,7 @@ class ModificationAwaraController(Controller, metaclass=ABCMeta):
         return True
 
 
-class AssetsController(ModificationAwaraController, metaclass=ABCMeta):
+class AssetsController(GzipMixin, ModificationAwareController, metaclass=ABCMeta):
     def getModified(self, file):
         return datetime.fromtimestamp(os.path.getmtime(self.getFilePath(file)), timezone.utc)
 
@@ -80,7 +110,7 @@ class AprsSymbolsController(AssetsController):
         return self.path + file
 
 
-class CompiledAssetsController(ModificationAwaraController):
+class CompiledAssetsController(GzipMixin, ModificationAwareController):
     profiles = {
         "receiver.js": [
             "openwebrx.js",