30 lines
813 B
Python
30 lines
813 B
Python
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("/{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():
|
|
if filename.suffix == "":
|
|
filename = filename.with_suffix(".html")
|
|
if not filename.is_file():
|
|
return Response(status_code=404)
|
|
|
|
content_type, _ = guess_type(filename)
|
|
return Response(filename.read_bytes(), media_type=content_type)
|