Initial commit

app/core/security.py

@@ -0,0 +1,50 @@
+import uuid
+from typing import Dict, Optional
+from fastapi import Depends, HTTPException, status, Request
+from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+
+# Простое хранилище активных сессий (в реальном приложении лучше использовать Redis или другое хранилище)
+ACTIVE_SESSIONS: Dict[str, str] = {}
+
+# Схема безопасности для проверки токена
+security = HTTPBearer(auto_error=False)
+
+def create_session_token(username: str) -> str:
+    """Создание простого сессионного токена"""
+    token = str(uuid.uuid4())
+    ACTIVE_SESSIONS[token] = username
+    return token
+
+def remove_session_token(token: str) -> None:
+    """Удаление сессионного токена"""
+    if token in ACTIVE_SESSIONS:
+        del ACTIVE_SESSIONS[token]
+
+async def validate_session_token(
+    request: Request,
+    credentials: Optional[HTTPAuthorizationCredentials] = Depends(security)
+) -> Optional[str]:
+    """Проверка сессионного токена"""
+    token_value = None
+    
+    # Проверяем заголовок Authorization
+    if credentials:
+        token_value = credentials.credentials
+    # Или query-параметр token (для обратной совместимости)
+    elif 'token' in request.query_params:
+        token_value = request.query_params['token']
+    
+    if not token_value or token_value not in ACTIVE_SESSIONS:
+        return None
+    
+    return ACTIVE_SESSIONS[token_value]
+
+async def require_auth(token: Optional[str] = Depends(validate_session_token)) -> str:
+    """Требование обязательной аутентификации"""
+    if not token:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Не авторизован",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+    return token