openwebrx-clone/owrx/controllers/assets.py

189 lines
6.3 KiB
Python
Raw Normal View History

from . import Controller
2021-02-11 18:31:44 +00:00
from owrx.config.core import CoreConfig
2020-06-21 20:35:40 +00:00
from datetime import datetime, timezone
import mimetypes
import os
import pkg_resources
from abc import ABCMeta, abstractmethod
2020-09-04 13:44:25 +00:00
import gzip
import logging
logger = logging.getLogger(__name__)
class GzipMixin(object):
2021-02-08 22:29:24 +00:00
def send_response(self, content, code=200, headers=None, content_type="text/html", *args, **kwargs):
2020-09-04 13:44:25 +00:00
if self.zipable(content_type) and "accept-encoding" in self.request.headers:
2021-01-20 16:01:46 +00:00
accepted = [s.strip().lower() for s in self.request.headers["accept-encoding"].split(",")]
2020-09-04 13:44:25 +00:00
if "gzip" in accepted:
if type(content) == str:
content = content.encode()
content = self.gzip(content)
if headers is None:
headers = {}
headers["Content-Encoding"] = "gzip"
2021-02-08 22:29:24 +00:00
super().send_response(content, code, headers=headers, content_type=content_type, *args, **kwargs)
2020-09-04 13:44:25 +00:00
def zipable(self, content_type):
2021-01-20 16:01:46 +00:00
types = ["application/javascript", "text/css", "text/html"]
2020-09-04 13:44:25 +00:00
return content_type in types
def gzip(self, content):
return gzip.compress(content)
2020-09-04 13:44:25 +00:00
class ModificationAwareController(Controller, metaclass=ABCMeta):
2020-06-21 20:35:40 +00:00
@abstractmethod
def getModified(self, file):
pass
2021-01-20 16:01:46 +00:00
2020-06-21 20:35:40 +00:00
def wasModified(self, file):
try:
modified = self.getModified(file).replace(microsecond=0)
2021-01-20 16:01:46 +00:00
2020-06-21 20:35:40 +00:00
if modified is not None and "If-Modified-Since" in self.handler.headers:
client_modified = datetime.strptime(
self.handler.headers["If-Modified-Since"], "%a, %d %b %Y %H:%M:%S %Z"
).replace(tzinfo=timezone.utc)
if modified <= client_modified:
return False
except FileNotFoundError:
pass
2021-01-20 16:01:46 +00:00
2020-06-21 20:35:40 +00:00
return True
2020-09-04 13:44:25 +00:00
class AssetsController(GzipMixin, ModificationAwareController, metaclass=ABCMeta):
def getModified(self, file):
return datetime.fromtimestamp(os.path.getmtime(self.getFilePath(file)), timezone.utc)
def openFile(self, file):
return open(self.getFilePath(file), "rb")
@abstractmethod
def getFilePath(self, file):
pass
def serve_file(self, file, content_type=None):
try:
modified = self.getModified(file)
2020-06-21 20:35:40 +00:00
if not self.wasModified(file):
self.send_response("", code=304)
return
f = self.openFile(file)
data = f.read()
f.close()
if content_type is None:
(content_type, encoding) = mimetypes.guess_type(self.getFilePath(file))
self.send_response(data, content_type=content_type, last_modified=modified, max_age=3600)
except FileNotFoundError:
self.send_response("file not found", code=404)
def indexAction(self):
filename = self.request.matches.group(1)
self.serve_file(filename)
class OwrxAssetsController(AssetsController):
def getFilePath(self, file):
mappedFiles = {
"gfx/openwebrx-avatar.png": "receiver_avatar",
"gfx/openwebrx-top-photo.jpg": "receiver_top_photo",
}
2021-02-10 20:29:46 +00:00
if file in mappedFiles and ("mapped" not in self.request.query or self.request.query["mapped"][0] != "false"):
config = CoreConfig()
for ext in ["png", "jpg"]:
user_file = "{}/{}.{}".format(config.get_data_directory(), mappedFiles[file], ext)
if os.path.exists(user_file) and os.path.isfile(user_file):
return user_file
return pkg_resources.resource_filename("htdocs", file)
class AprsSymbolsController(AssetsController):
2020-02-27 17:43:44 +00:00
def __init__(self, handler, request, options):
2021-02-06 23:21:57 +00:00
path = CoreConfig().get_aprs_symbols_path()
if not path.endswith("/"):
path += "/"
self.path = path
2020-02-27 17:43:44 +00:00
super().__init__(handler, request, options)
def getFilePath(self, file):
return self.path + file
2020-09-04 13:44:25 +00:00
class CompiledAssetsController(GzipMixin, ModificationAwareController):
profiles = {
"receiver.js": [
2020-09-17 19:33:11 +00:00
"lib/chroma.min.js",
"openwebrx.js",
"lib/jquery-3.2.1.min.js",
"lib/jquery.nanoscroller.min.js",
2020-05-08 22:11:20 +00:00
"lib/Header.js",
"lib/Demodulator.js",
"lib/DemodulatorPanel.js",
"lib/BookmarkBar.js",
"lib/BookmarkDialog.js",
"lib/AudioEngine.js",
"lib/ProgressBar.js",
"lib/Measurement.js",
"lib/FrequencyDisplay.js",
"lib/MessagePanel.js",
"lib/Js8Threads.js",
"lib/Modes.js",
"lib/MetaPanel.js",
2020-05-02 11:57:19 +00:00
],
"map.js": [
"lib/jquery-3.2.1.min.js",
"lib/chroma.min.js",
"lib/Header.js",
"map.js",
],
2020-05-31 18:25:41 +00:00
"settings.js": [
"lib/jquery-3.2.1.min.js",
"lib/bootstrap.bundle.min.js",
2020-05-31 18:25:41 +00:00
"lib/Header.js",
2021-02-15 19:19:43 +00:00
"lib/settings/MapInput.js",
2021-02-08 22:29:24 +00:00
"lib/settings/ImageUpload.js",
2021-02-13 22:53:16 +00:00
"lib/settings/BookmarkTable.js",
2021-02-15 19:19:43 +00:00
"lib/settings/WsjtDecodingDepthsInput.js",
"lib/settings/WaterfallDropdown.js",
2021-02-19 20:07:13 +00:00
"lib/settings/GainInput.js",
2021-02-22 22:49:28 +00:00
"lib/settings/OptionalSection.js",
"lib/settings/SchedulerInput.js",
"lib/settings/ExponentialInput.js",
2020-05-31 18:25:41 +00:00
"settings.js",
2021-01-20 16:01:46 +00:00
],
}
def indexAction(self):
profileName = self.request.matches.group(1)
if profileName not in CompiledAssetsController.profiles:
self.send_response("profile not found", code=404)
2020-05-02 14:59:27 +00:00
return
files = CompiledAssetsController.profiles[profileName]
files = [pkg_resources.resource_filename("htdocs", f) for f in files]
modified = self.getModified(files)
2020-06-21 20:35:40 +00:00
if not self.wasModified(files):
self.send_response("", code=304)
return
contents = [self.getContents(f) for f in files]
(content_type, encoding) = mimetypes.guess_type(profileName)
self.send_response("\n".join(contents), content_type=content_type, last_modified=modified, max_age=3600)
def getContents(self, file):
with open(file) as f:
return f.read()
def getModified(self, files):
modified = [os.path.getmtime(f) for f in files]
return datetime.fromtimestamp(max(*modified), timezone.utc)