zulip/zerver/views/camo.py

24 lines
965 B
Python

from django.conf import settings
from django.shortcuts import redirect
from django.utils.translation import ugettext as _
from django.http import (
HttpRequest, HttpResponse, HttpResponseForbidden, HttpResponseNotFound
)
from zerver.lib.camo import is_camo_url_valid
from zerver.lib.thumbnail import generate_thumbnail_url
import codecs
def handle_camo_url(request: HttpRequest, digest: str,
received_url: str) -> HttpResponse:
if not settings.THUMBOR_SERVES_CAMO:
return HttpResponseNotFound()
hex_encoded_url = received_url.encode('utf-8')
hex_decoded_url = codecs.decode(hex_encoded_url, 'hex')
original_url = hex_decoded_url.decode('utf-8') # type: ignore # https://github.com/python/typeshed/issues/300
if is_camo_url_valid(digest, original_url):
return redirect(generate_thumbnail_url(original_url, is_camo_url=True))
else:
return HttpResponseForbidden(_("<p>Not a valid URL.</p>"))