feat: home-gateway 초기 구성 — Mac mini에서 GPU 서버로 전면 이전
OrbStack 라이선스 만료로 Mac mini Docker 서비스를 GPU 서버로 통합. nginx → Caddy 전환, 12개 서브도메인 자동 HTTPS, fail2ban Caddy JSON 연동. 주요 변경: - home-caddy: Caddy 리버스 프록시 (Let's Encrypt 자동 HTTPS) - home-fail2ban: Caddy JSON 로그 기반 보안 모니터링 - home-ddns: Cloudflare DDNS (API 키 .env 분리) - gpu-hub-api/web: AI 백엔드 라우터 + 웹 UI (gpu-services에서 이전) - AI 런타임(Ollama) 내부망 전용, 외부는 gpu-hub 인증 게이트웨이 경유 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
96
hub-api/middleware/auth.py
Normal file
96
hub-api/middleware/auth.py
Normal file
@@ -0,0 +1,96 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import time
|
||||
|
||||
from jose import JWTError, jwt
|
||||
from starlette.middleware.base import BaseHTTPMiddleware
|
||||
from starlette.requests import Request
|
||||
|
||||
from config import settings
|
||||
|
||||
# Paths that don't require authentication
|
||||
PUBLIC_PATHS = {"/", "/health", "/auth/login", "/docs", "/openapi.json"}
|
||||
PUBLIC_PREFIXES = ("/health/",)
|
||||
|
||||
|
||||
class AuthMiddleware(BaseHTTPMiddleware):
|
||||
async def dispatch(self, request: Request, call_next):
|
||||
path = request.url.path
|
||||
|
||||
# Skip auth for public paths
|
||||
if path in PUBLIC_PATHS or any(path.startswith(p) for p in PUBLIC_PREFIXES):
|
||||
request.state.role = "anonymous"
|
||||
return await call_next(request)
|
||||
|
||||
# Skip auth for OPTIONS (CORS preflight)
|
||||
if request.method == "OPTIONS":
|
||||
return await call_next(request)
|
||||
|
||||
# Try Bearer token first, then cookie
|
||||
token = _extract_token(request)
|
||||
if not token:
|
||||
request.state.role = "anonymous"
|
||||
return await call_next(request)
|
||||
|
||||
# Verify JWT
|
||||
payload = _verify_token(token)
|
||||
if payload:
|
||||
request.state.role = payload.get("role", "guest")
|
||||
else:
|
||||
request.state.role = "anonymous"
|
||||
|
||||
return await call_next(request)
|
||||
|
||||
|
||||
def create_token(role: str) -> str:
|
||||
payload = {
|
||||
"role": role,
|
||||
"exp": time.time() + settings.jwt_expire_hours * 3600,
|
||||
"iat": time.time(),
|
||||
}
|
||||
return jwt.encode(payload, settings.jwt_secret, algorithm=settings.jwt_algorithm)
|
||||
|
||||
|
||||
def _extract_token(request: Request) -> str | None:
|
||||
# 1. Authorization: Bearer header
|
||||
auth_header = request.headers.get("authorization", "")
|
||||
if auth_header.startswith("Bearer "):
|
||||
return auth_header[7:]
|
||||
|
||||
# 2. httpOnly cookie
|
||||
return request.cookies.get("token")
|
||||
|
||||
|
||||
def _verify_token(token: str) -> dict | None:
|
||||
try:
|
||||
payload = jwt.decode(
|
||||
token, settings.jwt_secret, algorithms=[settings.jwt_algorithm]
|
||||
)
|
||||
if payload.get("exp", 0) < time.time():
|
||||
return None
|
||||
return payload
|
||||
except JWTError:
|
||||
return None
|
||||
|
||||
|
||||
# Login rate limiting (IP-based)
|
||||
_login_attempts: dict[str, list[float]] = {}
|
||||
MAX_ATTEMPTS = 5
|
||||
LOCKOUT_SECONDS = 60
|
||||
|
||||
|
||||
def check_login_rate_limit(ip: str) -> bool:
|
||||
"""Returns True if login is allowed for this IP."""
|
||||
now = time.time()
|
||||
attempts = _login_attempts.get(ip, [])
|
||||
# Clean old attempts
|
||||
attempts = [t for t in attempts if now - t < LOCKOUT_SECONDS]
|
||||
_login_attempts[ip] = attempts
|
||||
return len(attempts) < MAX_ATTEMPTS
|
||||
|
||||
|
||||
def record_login_attempt(ip: str):
|
||||
now = time.time()
|
||||
if ip not in _login_attempts:
|
||||
_login_attempts[ip] = []
|
||||
_login_attempts[ip].append(now)
|
||||
Reference in New Issue
Block a user