mirror of
https://github.com/zulip/zulip.git
synced 2025-10-31 12:03:46 +00:00
7dc1b591cb
This will help us eliminate camo from our production installs. Basically it helps us de duplicate some code from upcoming code which will help us check validity of a camo url.
20 lines
765 B
Python
20 lines
765 B
Python
from django.conf import settings
|
|
import codecs
|
|
import hashlib
|
|
import hmac
|
|
|
|
def generate_camo_url(url: str) -> str:
|
|
encoded_url = url.encode("utf-8")
|
|
encoded_camo_key = settings.CAMO_KEY.encode("utf-8")
|
|
digest = hmac.new(encoded_camo_key, encoded_url, hashlib.sha1).hexdigest()
|
|
hex_encoded_url = codecs.encode(encoded_url, "hex") # type: ignore # https://github.com/python/typeshed/issues/300
|
|
return "%s/%s" % (digest, hex_encoded_url.decode("utf-8"))
|
|
|
|
# Encodes the provided URL using the same algorithm used by the camo
|
|
# caching https image proxy
|
|
def get_camo_url(url: str) -> str:
|
|
# Only encode the url if Camo is enabled
|
|
if settings.CAMO_URI == '':
|
|
return url
|
|
return "%s%s" % (settings.CAMO_URI, generate_camo_url(url))
|