HTML error pages for FastAPI

Signed-off-by: Lukas Fleischer <lfleischer@archlinux.org>

1 files changed, 15 insertions, 1 deletions

diff --git a/aurweb/asgi.py b/aurweb/asgi.py
index 60c7ade7..9293ed77 100644
--- a/aurweb/asgi.py
+++ b/aurweb/asgi.py
@@ -1,4 +1,7 @@
-from fastapi import FastAPI
+import http
+
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import HTMLResponse
 from starlette.middleware.sessions import SessionMiddleware
 
 import aurweb.config
@@ -14,3 +17,14 @@ if not session_secret:
 app.add_middleware(SessionMiddleware, secret_key=session_secret)
 
 app.include_router(sso.router)
+
+
+@app.exception_handler(HTTPException)
+async def http_exception_handler(request, exc):
+    """
+    Dirty HTML error page to replace the default JSON error responses.
+    In the future this should use a proper Arch-themed HTML template.
+    """
+    phrase = http.HTTPStatus(exc.status_code).phrase
+    return HTMLResponse(f"<h1>{exc.status_code} {phrase}</h1><p>{exc.detail}</p>",
+                        status_code=exc.status_code)