Add support for pulling favicon and logo from a URL

This commit is contained in:
Solirs 2023-03-22 21:41:32 +01:00
parent 0dcf64cf8e
commit 54993ad2fa

View file

@ -307,14 +307,64 @@ def get_favicon_or_logo(imgtype: str, request):
path = os.path.expanduser(settings.get("brand").get(imgtype))
relpath = os.path.join(app.root_path, path)
mimetype = 'image/vnd.microsoft.icon' if imgtype == "favicon" else None
if not os.path.isfile(path) and not os.path.isfile(relpath):
# If path doesn't exist neither relatively nor absolutely, fallback to whatever is in the theme
return send_from_directory(
fallback = ( send_from_directory(
os.path.join(app.root_path, settings['ui']['static_path'], 'themes', theme, 'img'), # pyright: ignore
'favicon.png' if imgtype == "favicon" else "searxng.png",
mimetype=mimetype,
)
mimetype=mimetype
) )
# If path is a URL
if "://" in path:
resp_ok = False
resp = None
# TODO: Make a a function for this used by both this function and the image proxy
try:
# Pull image from it
request_headers = {
'User-Agent': gen_useragent(),
'Accept': 'image/webp,*/*',
'Accept-Encoding': 'gzip, deflate',
'Sec-GPC': '1',
'DNT': '1',
}
resp, stream = http_stream(method='GET', url=path, headers=request_headers, allow_redirects=True)
if resp.status_code != 200:
logger.debug(f"{imgtype}: Bad status code %i", resp.status_code)
if resp.status_code >= 400:
return fallback
resp_ok = True
except httpx.HTTPError:
logger.debug(f"{imgtype}: HTTP error")
return fallback
finally:
if resp and not resp_ok:
try:
resp.close()
except httpx.HTTPError:
logger.exception(f'{imgtype}: HTTP error on closing')
def close_stream():
nonlocal resp, stream
try:
if resp:
resp.close()
del resp
del stream
except httpx.HTTPError as e:
logger.debug(f'{imgtype}Exception while closing response', e)
try:
headers = dict_subset(resp.headers, {'Content-Type', 'Content-Encoding', 'Content-Length', 'Length'})
response = Response(stream, mimetype=resp.headers['Content-Type'], headers=headers, direct_passthrough=True)
response.call_on_close(close_stream)
return response
except httpx.HTTPError:
close_stream()
return fallback
elif not os.path.isfile(path) and not os.path.isfile(relpath):
# If path doesn't exist neither relatively nor absolutely, fallback to whatever is in the theme
return fallback
else:
# Otherwise we're good to go, send whatever is specified.
# Works with both relative and absolute.