republisher/repub/web.py

from __future__ import annotations

import hashlib

import htpy as h
from quart import Quart, Response, request, url_for

from repub.pages import admin_component, shim_page


def _render_shim_page(*, stylesheet_href: str, datastar_src: str) -> tuple[str, str]:
    head = (
        h.title["Republisher Admin UI"],
        h.link(rel="stylesheet", href=stylesheet_href),
    )
    body = str(shim_page(datastar_src=datastar_src, head=head))
    etag = hashlib.sha256(body.encode("utf-8")).hexdigest()
    return body, etag


def create_app() -> Quart:
    app = Quart(__name__)

    @app.get("/")
    async def index() -> Response:
        body, etag = _render_shim_page(
            stylesheet_href=url_for("static", filename="app.css"),
            datastar_src=url_for("static", filename="datastar@1.0.0-RC.8.js"),
        )
        if request.if_none_match.contains(etag):
            response = Response(status=304)
            response.set_etag(etag)
            return response

        response = Response(body, mimetype="text/html")
        response.set_etag(etag)
        return response

    @app.post("/")
    async def index_patch() -> Response:
        return Response(str(admin_component()), mimetype="text/html")

    return app