Use custom ASGI app to serve SPA properly

Author: scootermon

server/app.py

@@ -4,13 +4,13 @@
 from fastapi import FastAPI
 from fastapi.routing import APIRoute
 from starlette.middleware.cors import CORSMiddleware
-from starlette.staticfiles import StaticFiles
 
 from events.router import events_router
 from lifespan import lifespan
 from search.router import search_router
 from server_info.router import settings_router
 from settings import Settings
+from spa import SpaFiles
 from tasks.router import tasks_router
 from workers.router import workers_router
 from ws.router import ws_router
@@ -58,4 +58,4 @@ async def health_check():
 app.include_router(settings_router)
 
 if Path("static").exists():
-    app.mount("/", StaticFiles(directory="static", html=True), name="static")
+    app.mount("/", SpaFiles("static", ignore_prefixes=["/api", "/docs", "/redoc", "/ws"]), name="static")

server/spa.py

@@ -0,0 +1,36 @@
+import os
+
+from fastapi import HTTPException
+from starlette.exceptions import HTTPException as StarletteHTTPException
+from starlette.staticfiles import StaticFiles
+
+
+class SpaFiles(StaticFiles):
+    def __init__(
+        self,
+        directory: os.PathLike[str],
+        *,
+        ignore_prefixes: list[str] | None = None,
+    ):
+        super().__init__(
+            directory=directory,
+            packages=None,
+            html=True,
+            check_dir=True,
+            follow_symlink=False,
+        )
+        self._ignore_prefixes = ignore_prefixes or []
+
+    def _is_ignored(self, path: str) -> bool:
+        return any(path.startswith(prefix) for prefix in self._ignore_prefixes)
+
+    async def get_response(self, path: str, scope):
+        if self._is_ignored(path):
+            raise HTTPException(status_code=404)
+        try:
+            return await super().get_response(path, scope)
+        except (HTTPException, StarletteHTTPException) as ex:
+            if ex.status_code == 404:
+                return await super().get_response("index.html", scope)
+            else:
+                raise ex