integrate static assets into webui command

2023-08-25 17:08:39 +02:00
parent 79b02dffab
commit 9ae14a4f7d
11 changed files with 149 additions and 28 deletions
--- a/pkgs/clan-cli/clan_cli/webui/routers/root.py
+++ b/pkgs/clan-cli/clan_cli/webui/routers/root.py
@@ -1,9 +1,28 @@
+import os
+from mimetypes import guess_type
+from pathlib import Path
+
 from fastapi import APIRouter, Response

+from ..assets import asset_path
+
 router = APIRouter()


-@router.get("/")
-async def root() -> Response:
-    body = "<html><body><h1>Welcome</h1></body></html>"
-    return Response(content=body, media_type="text/html")
+@router.get("/{path_name:path}")
+async def root(path_name: str) -> Response:
+    if path_name == "":
+        path_name = "index.html"
+    filename = Path(os.path.normpath((asset_path() / path_name)))
+
+    if not filename.is_relative_to(asset_path()):
+        # prevent directory traversal
+        return Response(status_code=403)
+
+    if not filename.is_file():
+        print(filename)
+        print(asset_path())
+        return Response(status_code=404)
+
+    content_type, _ = guess_type(filename)
+    return Response(filename.read_bytes(), media_type=content_type)