mirror of
https://github.com/UrloMythus/UnHided.git
synced 2026-04-11 03:40:54 +00:00
27 lines
981 B
Python
27 lines
981 B
Python
from fastapi import Request, Response
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
from mediaflow_proxy.configs import settings
|
|
|
|
|
|
class UIAccessControlMiddleware(BaseHTTPMiddleware):
|
|
"""Middleware that controls access to UI components based on settings."""
|
|
|
|
async def dispatch(self, request: Request, call_next):
|
|
path = request.url.path
|
|
|
|
# Block access to home page
|
|
if settings.disable_home_page and (path == "/" or path == "/index.html"):
|
|
return Response(status_code=403, content="Forbidden")
|
|
|
|
# Block access to API docs
|
|
if settings.disable_docs and (path == "/docs" or path == "/redoc" or path.startswith("/openapi")):
|
|
return Response(status_code=403, content="Forbidden")
|
|
|
|
# Block access to speedtest UI
|
|
if settings.disable_speedtest and path.startswith("/speedtest"):
|
|
return Response(status_code=403, content="Forbidden")
|
|
|
|
return await call_next(request)
|
|
|