xiaoxia-saas/apps/api/app/dependencies.py

"""Dependency injection providers for FastAPI endpoints.

All repository and service factories are defined here as FastAPI dependencies,
ensuring proper lifecycle management and testability.
"""
from __future__ import annotations

import redis
from typing import Generator

from app.config import settings
from fastapi import Depends
from sqlalchemy.orm import Session

from packages.adapters.redis import NoopSessionStore, SessionStore
from packages.adapters.smtp import EmailConfig, NoopEmailService, get_email_service
from packages.adapters.sqlalchemy_impl.asset_library_repository import (
    SQLAlchemyAssetLibraryRepository,
)
from packages.adapters.sqlalchemy_impl.asset_repository import SQLAlchemyAssetRepository
from packages.adapters.sqlalchemy_impl.classification_job_repository import (
    SQLAlchemyClassificationJobRepository,
)
from packages.adapters.sqlalchemy_impl.generated_video_repository import (
    SQLAlchemyGeneratedVideoRepository,
)
from packages.adapters.sqlalchemy_impl.generation_task_repository import (
    SQLAlchemyGenerationTaskRepository,
)
from packages.adapters.sqlalchemy_impl.ingest_job_repository import (
    SQLAlchemyIngestJobRepository,
)
from packages.adapters.sqlalchemy_impl.project_repository import (
    SQLAlchemyProjectRepository,
)
from packages.adapters.sqlalchemy_impl.project_title_repository import (
    SQLAlchemyProjectTitleRepository,
)
from packages.adapters.sqlalchemy_impl.session import build_session_factory
from packages.adapters.sqlalchemy_impl.user_repository import SQLAlchemyUserRepository
from packages.adapters.sqlalchemy_impl.workspace_invitation_repository import SQLAlchemyWorkspaceInvitationRepository
from packages.adapters.sqlalchemy_impl.workspace_member_repository import SQLAlchemyWorkspaceMemberRepository
from packages.adapters.sqlalchemy_impl.workspace_repository import SQLAlchemyWorkspaceRepository
from packages.ports.asset_repository import AssetRepository
from packages.ports.asset_library_repository import AssetLibraryRepository
from packages.ports.user_repository import UserRepository
from packages.ports.workspace_invitation_repository import WorkspaceInvitationRepository
from packages.ports.workspace_member_repository import WorkspaceMemberRepository
from packages.ports.workspace_repository import WorkspaceRepository
from packages.ports.classification_job_repository import ClassificationJobRepository
from packages.ports.generation_task_repository import GenerationTaskRepository
from packages.ports.generated_video_repository import GeneratedVideoRepository
from packages.ports.ingest_job_repository import IngestJobRepository
from packages.ports.project_repository import ProjectRepository
from packages.ports.project_title_repository import ProjectTitleRepository

_engine, _SessionLocal = build_session_factory(settings.DATABASE_URL)


def get_db_session() -> Generator[Session, None, None]:
    """Provide a database session with automatic cleanup."""
    session: Session = _SessionLocal()
    try:
        yield session
    finally:
        session.close()


def get_asset_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyAssetRepository:
    """Provide the SQLAlchemy asset repository implementation."""
    return SQLAlchemyAssetRepository(session)


def get_asset_library_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyAssetLibraryRepository:
    """Provide the SQLAlchemy asset library repository implementation."""
    return SQLAlchemyAssetLibraryRepository(session)


def get_ingest_job_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyIngestJobRepository:
    """Provide the SQLAlchemy ingest job repository implementation."""
    return SQLAlchemyIngestJobRepository(session)


def get_classification_job_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyClassificationJobRepository:
    """Provide the SQLAlchemy classification job repository implementation."""
    return SQLAlchemyClassificationJobRepository(session)


def get_generation_task_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyGenerationTaskRepository:
    """Provide the SQLAlchemy generation task repository implementation."""
    return SQLAlchemyGenerationTaskRepository(session)


def get_generated_video_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyGeneratedVideoRepository:
    """Provide the SQLAlchemy generated video repository implementation."""
    return SQLAlchemyGeneratedVideoRepository(session)


def get_project_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyProjectRepository:
    """Provide the SQLAlchemy project repository implementation."""
    return SQLAlchemyProjectRepository(session)


def get_project_title_repository(
    session: Session = Depends(get_db_session),
) -> SQLAlchemyProjectTitleRepository:
    """Provide the SQLAlchemy project title repository implementation."""
    return SQLAlchemyProjectTitleRepository(session)


def get_user_repository(
    session: Session = Depends(get_db_session),
) -> UserRepository:
    """Provide the SQLAlchemy user repository implementation."""
    return SQLAlchemyUserRepository(session)


def get_workspace_repository(
    session: Session = Depends(get_db_session),
) -> WorkspaceRepository:
    """Provide the SQLAlchemy workspace repository implementation."""
    return SQLAlchemyWorkspaceRepository(session)


def get_workspace_member_repository(
    session: Session = Depends(get_db_session),
) -> WorkspaceMemberRepository:
    """Provide the SQLAlchemy workspace member repository implementation."""
    return SQLAlchemyWorkspaceMemberRepository(session)


def get_workspace_invitation_repository(
    session: Session = Depends(get_db_session),
) -> WorkspaceInvitationRepository:
    """Provide the SQLAlchemy workspace invitation repository implementation."""
    return SQLAlchemyWorkspaceInvitationRepository(session)


def get_auth_session_store() -> SessionStore | NoopSessionStore:
    """Provide the session store based on configuration."""
    if not settings.ENABLE_REDIS_SESSIONS:
        return NoopSessionStore()
    return SessionStore(redis_client=redis.from_url(settings.REDIS_URL, decode_responses=True))


def get_auth_email_service() -> NoopEmailService | EmailService:
    """Provide the email service based on configuration."""
    if not settings.ENABLE_EMAIL_DELIVERY:
        return NoopEmailService()
    return get_email_service(
        EmailConfig(
            smtp_host=settings.SMTP_HOST,
            smtp_port=settings.SMTP_PORT,
            smtp_user=settings.SMTP_USER,
            smtp_password=settings.SMTP_PASSWORD,
            from_email=settings.SMTP_FROM_EMAIL,
            from_name=settings.SMTP_FROM_NAME,
            use_tls=settings.SMTP_USE_TLS,
        ),
        enabled=True,
    )