import random from urllib.parse import quote as url_escape import requests from django.core.handlers.wsgi import WSGIRequest from django.http import HttpResponse from django.template import loader from tmessage.settings import config COLORS = ["rosewater", "flamingo", "pink", "mauve", "red", "maroon", "peach", "yellow", "green", "teal", "sky", "sapphire", "blue", "lavender"] def render_template( request: WSGIRequest, template: str, /, *, status: int=200, headers: dict[str, str]={}, content_type: str="text/html", **context ) -> HttpResponse: c = { "accent": random.choice(COLORS), "config": config } for key, val in context.items(): c[key] = val resp = HttpResponse( loader.get_template(template).render(c, request), status=status, content_type=content_type ) for key, val in headers.items(): resp[key] = val return resp def is_logged_in(request: WSGIRequest) -> dict | None: resp = requests.get(config["services"]["auth"]["url"]["int"] + f"/api/authenticated/?token={url_escape(config['services']['message']['token'])}&service=message", cookies={**request.COOKIES}).json() return resp["success"] and resp["auth"]