30 lines
876 B
Python
30 lines
876 B
Python
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Middleware for checking if client IP is banned.
|
|
"""
|
|
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
from starlette.requests import Request
|
|
from starlette.responses import Response
|
|
|
|
from dependencies import get_client_ip
|
|
|
|
|
|
class BanCheckMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
# Skip ban check for dashboard routes
|
|
config = request.app.state.config
|
|
dashboard_prefix = "/" + config.dashboard_secret_path.lstrip("/")
|
|
if request.url.path.startswith(dashboard_prefix):
|
|
return await call_next(request)
|
|
|
|
client_ip = get_client_ip(request)
|
|
tracker = request.app.state.tracker
|
|
|
|
if tracker.is_banned_ip(client_ip):
|
|
return Response(status_code=500)
|
|
|
|
response = await call_next(request)
|
|
return response
|