94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
from aiohttp import web
|
|
|
|
from custom_components.hacs.helpers.functions.logger import getLogger
|
|
from custom_components.hacs.helpers.functions.path_exsist import async_path_exsist
|
|
from custom_components.hacs.helpers.functions.file_etag import async_get_etag
|
|
|
|
from custom_components.hacs.share import get_hacs
|
|
|
|
_LOGGER = getLogger()
|
|
|
|
|
|
async def async_serve_category_file(request, requested_file):
|
|
hacs = get_hacs()
|
|
response = None
|
|
|
|
try:
|
|
if requested_file.startswith("themes/"):
|
|
servefile = f"{hacs.core.config_path}/{requested_file}"
|
|
response = await async_serve_static_file(request, servefile, requested_file)
|
|
else:
|
|
servefile = f"{hacs.core.config_path}/www/community/{requested_file}"
|
|
response = await async_serve_static_file_with_etag(
|
|
request, servefile, requested_file
|
|
)
|
|
except (Exception, BaseException):
|
|
_LOGGER.exception("Error trying to serve %s", requested_file)
|
|
|
|
if response is not None:
|
|
return response
|
|
|
|
return web.Response(status=404)
|
|
|
|
|
|
async def async_serve_static_file(request, servefile, requested_file):
|
|
"""Serve a static file without an etag."""
|
|
if await async_path_exsist(servefile):
|
|
_LOGGER.debug("Serving %s from %s", requested_file, servefile)
|
|
response = web.FileResponse(servefile)
|
|
response.headers["Cache-Control"] = "public, max-age=2678400"
|
|
return response
|
|
|
|
_LOGGER.error(
|
|
"%s tried to request '%s' but the file does not exist",
|
|
request.remote,
|
|
servefile,
|
|
)
|
|
return None
|
|
|
|
|
|
async def async_serve_static_file_with_etag(request, servefile, requested_file):
|
|
"""Serve a static file with an etag."""
|
|
etag = await async_get_etag(servefile)
|
|
if_none_match_header = request.headers.get("if-none-match")
|
|
|
|
if (
|
|
etag is not None
|
|
and if_none_match_header is not None
|
|
and _match_etag(etag, if_none_match_header)
|
|
):
|
|
_LOGGER.debug(
|
|
"Serving %s from %s with etag %s (not-modified)",
|
|
requested_file,
|
|
servefile,
|
|
etag,
|
|
)
|
|
return web.Response(status=304)
|
|
|
|
if etag is not None:
|
|
_LOGGER.debug(
|
|
"Serving %s from %s with etag %s (not cached)",
|
|
requested_file,
|
|
servefile,
|
|
etag,
|
|
)
|
|
response = web.FileResponse(servefile)
|
|
response.headers["Cache-Control"] = "must-revalidate, max-age=0"
|
|
response.headers["Etag"] = etag
|
|
return response
|
|
|
|
_LOGGER.error(
|
|
"%s tried to request '%s' but the file does not exist",
|
|
request.remote,
|
|
servefile,
|
|
)
|
|
return None
|
|
|
|
|
|
def _match_etag(etag, if_none_match_header):
|
|
"""Check to see if an etag matches."""
|
|
for if_none_match_ele in if_none_match_header.split(","):
|
|
if if_none_match_ele.strip() == etag:
|
|
return True
|
|
return False
|