BIG BIG BIG BIG BIG UPDATES

Co-authored-by: Copilot <[email protected]>

backend/app/core/config.py

@@ -0,0 +1,23 @@
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Settings(BaseSettings):
+    model_config = SettingsConfigDict(
+        env_file=(".env", "backend/.env"),
+        env_file_encoding="utf-8",
+        case_sensitive=False,
+        extra="ignore",
+    )
+
+    MONGODB_URI: str
+    MONGODB_DB_NAME: str
+    REDIS_URL: str
+    JWT_SECRET_KEY: str
+    JWT_ALGORITHM: str = "HS256"
+    JWT_EXPIRE_MINUTES: int = Field(default=30, gt=0)
+    AUTH_USERNAME: str
+    AUTH_PASSWORD: str
+
+
+settings = Settings()

backend/app/core/mongodb.py

@@ -0,0 +1,37 @@
+from core.config import settings
+from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase
+
+
+class _MongoDB:
+    client: AsyncIOMotorClient | None = None
+
+    def get_database(self) -> AsyncIOMotorDatabase:
+        if self.client is None:
+            raise RuntimeError("MongoDB client has not been initialized")
+        return self.client[settings.MONGODB_DB_NAME]
+
+
+mongodb = _MongoDB()
+
+
+async def init_mongodb() -> None:
+    mongodb.client = AsyncIOMotorClient(settings.MONGODB_URI)
+    await ensure_mongodb_available()
+
+
+async def close_mongodb() -> None:
+    client = mongodb.client
+    mongodb.client = None
+    if client is not None:
+        client.close()
+
+
+async def get_database() -> AsyncIOMotorDatabase:
+    return mongodb.get_database()
+
+
+async def ensure_mongodb_available() -> None:
+    client = mongodb.client
+    if client is None:
+        raise RuntimeError("MongoDB client has not been initialized")
+    await client.admin.command("ping")

backend/app/core/redis.py

@@ -0,0 +1,22 @@
+import redis.asyncio as aioredis
+from core.config import settings
+
+_redis_pool: aioredis.Redis | None = None
+
+
+async def init_redis() -> None:
+    global _redis_pool
+    _redis_pool = aioredis.from_url(
+        settings.REDIS_URL, encoding="utf-8", decode_responses=True
+    )
+
+
+async def close_redis() -> None:
+    if _redis_pool:
+        await _redis_pool.aclose()
+
+
+async def get_redis() -> aioredis.Redis:
+    if _redis_pool is None:
+        raise RuntimeError("Redis pool has not been initialized")
+    return _redis_pool

backend/app/core/security.py

@@ -0,0 +1,116 @@
+from dataclasses import dataclass
+from datetime import UTC, datetime, timedelta
+from typing import Annotated
+from uuid import uuid4
+
+from authlib.jose import jwt  # type: ignore
+from authlib.jose.errors import JoseError  # type: ignore
+from core.config import settings
+from fastapi import Depends, HTTPException, status
+from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
+from models.auth import AccessTokenPayload, AuthUserDocument
+from pwdlib import PasswordHash
+from pwdlib.hashers.argon2 import Argon2Hasher
+from repositories.auth import get_user_by_username, is_token_revoked
+
+pwd_context = PasswordHash((Argon2Hasher(),))
+bearer_scheme = HTTPBearer(auto_error=False)
+
+
+@dataclass(slots=True)
+class AuthContext:
+    token: str
+    user: AuthUserDocument
+    payload: AccessTokenPayload
+
+
+class InvalidTokenError(Exception):
+    pass
+
+
+def token_auth_error() -> HTTPException:
+    return HTTPException(
+        status_code=status.HTTP_401_UNAUTHORIZED,
+        detail="Could not validate credentials",
+        headers={"WWW-Authenticate": "Bearer"},
+    )
+
+
+def hash_password(password: str) -> str:
+    return pwd_context.hash(password)
+
+
+def verify_password(plain_password: str, password_hash: str) -> bool:
+    if hasattr(pwd_context, "verify"):
+        return pwd_context.verify(plain_password, password_hash)
+    return pwd_context.check(plain_password, password_hash)
+
+
+def create_access_token(*, subject: str) -> str:
+    now = datetime.now(UTC)
+    expires_delta = timedelta(minutes=settings.JWT_EXPIRE_MINUTES)
+    expires_at = int((now + expires_delta).timestamp())
+
+    header = {
+        "alg": settings.JWT_ALGORITHM,
+        "typ": "JWT",
+    }
+
+    payload = {
+        "sub": subject,
+        "jti": str(uuid4()),
+        "exp": expires_at,
+    }
+
+    token = jwt.encode(header, payload, settings.JWT_SECRET_KEY)
+
+    if isinstance(token, bytes):
+        return token.decode("utf-8")
+
+    return token
+
+
+def decode_access_token(token: str) -> AccessTokenPayload:
+    try:
+        claims = jwt.decode(token, settings.JWT_SECRET_KEY)
+        claims.validate()
+        return AccessTokenPayload.model_validate(dict(claims))
+    except JoseError, ValueError, TypeError:
+        raise InvalidTokenError from None
+
+
+async def authenticate_token(token: str) -> AuthContext:
+    try:
+        payload = decode_access_token(token)
+    except InvalidTokenError:
+        raise InvalidTokenError from None
+
+    if await is_token_revoked(payload.jti):
+        raise InvalidTokenError
+
+    user = await get_user_by_username(payload.sub)
+    if user is None:
+        raise InvalidTokenError
+
+    return AuthContext(token=token, user=user, payload=payload)
+
+
+async def get_auth_context(
+    credentials: Annotated[
+        HTTPAuthorizationCredentials | None,
+        Depends(bearer_scheme),
+    ],
+) -> AuthContext:
+    if credentials is None or credentials.scheme.lower() != "bearer":
+        raise token_auth_error()
+
+    try:
+        return await authenticate_token(credentials.credentials)
+    except InvalidTokenError:
+        raise token_auth_error() from None
+
+
+async def require_authenticated_user(
+    auth_context: Annotated[AuthContext, Depends(get_auth_context)],
+) -> AuthUserDocument:
+    return auth_context.user