Aniworld/src/server/utils/dependencies.py

"""
Dependency injection utilities for FastAPI.

This module provides dependency injection functions for the FastAPI
application, including SeriesApp instances, AnimeService, DownloadService,
database sessions, and authentication dependencies.
"""
from typing import AsyncGenerator, Optional

from fastapi import Depends, HTTPException, status
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer

try:
    from sqlalchemy.ext.asyncio import AsyncSession
except Exception:  # pragma: no cover - optional dependency
    AsyncSession = object

from src.config.settings import settings
from src.core.SeriesApp import SeriesApp
from src.server.services.auth_service import AuthError, auth_service

# Security scheme for JWT authentication
security = HTTPBearer()


# Global SeriesApp instance
_series_app: Optional[SeriesApp] = None

# Global service instances
_anime_service: Optional[object] = None
_download_service: Optional[object] = None


def get_series_app() -> SeriesApp:
    """
    Dependency to get SeriesApp instance.
    
    Returns:
        SeriesApp: The main application instance for anime management
        
    Raises:
        HTTPException: If SeriesApp is not initialized or anime directory
                      is not configured
    """
    global _series_app
    
    if not settings.anime_directory:
        raise HTTPException(
            status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
            detail="Anime directory not configured. Please complete setup."
        )
    
    if _series_app is None:
        try:
            _series_app = SeriesApp(settings.anime_directory)
        except Exception as e:
            raise HTTPException(
                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
                detail=f"Failed to initialize SeriesApp: {str(e)}"
            )
    
    return _series_app


def reset_series_app() -> None:
    """Reset the global SeriesApp instance (for testing or config changes)."""
    global _series_app
    _series_app = None


async def get_database_session() -> AsyncGenerator[Optional[object], None]:
    """
    Dependency to get database session.
    
    Yields:
        AsyncSession: Database session for async operations
    """
    # TODO: Implement database session management
    # This is a placeholder for future database implementation
    raise HTTPException(
        status_code=status.HTTP_501_NOT_IMPLEMENTED,
        detail="Database functionality not yet implemented"
    )


def get_current_user(
    credentials: HTTPAuthorizationCredentials = Depends(security)
) -> dict:
    """
    Dependency to get current authenticated user.
    
    Args:
        credentials: JWT token from Authorization header
        
    Returns:
        dict: User information
        
    Raises:
        HTTPException: If token is invalid or user is not authenticated
    """
    if not credentials:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail="Missing authorization credentials",
        )

    token = credentials.credentials
    try:
        # Validate and decode token using the auth service
        session = auth_service.create_session_model(token)
        return session.dict()
    except AuthError as e:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail=str(e),
        )


def require_auth(
    current_user: dict = Depends(get_current_user)
) -> dict:
    """
    Dependency that requires authentication.
    
    Args:
        current_user: Current authenticated user from get_current_user
        
    Returns:
        dict: User information
    """
    return current_user


def optional_auth(
    credentials: Optional[HTTPAuthorizationCredentials] = Depends(
        HTTPBearer(auto_error=False)
    )
) -> Optional[dict]:
    """
    Dependency for optional authentication.
    
    Args:
        credentials: Optional JWT token from Authorization header
        
    Returns:
        Optional[dict]: User information if authenticated, None otherwise
    """
    if credentials is None:
        return None
    
    try:
        return get_current_user(credentials)
    except HTTPException:
        return None


def get_current_user_optional(
    credentials: Optional[HTTPAuthorizationCredentials] = Depends(
        HTTPBearer(auto_error=False)
    )
) -> Optional[str]:
    """
    Dependency to get optional current user ID.
    
    Args:
        credentials: Optional JWT token from Authorization header
        
    Returns:
        Optional[str]: User ID if authenticated, None otherwise
    """
    user_dict = optional_auth(credentials)
    if user_dict:
        return user_dict.get("user_id")
    return None


class CommonQueryParams:
    """Common query parameters for API endpoints."""
    
    def __init__(self, skip: int = 0, limit: int = 100):
        self.skip = skip
        self.limit = limit


def common_parameters(
    skip: int = 0,
    limit: int = 100
) -> CommonQueryParams:
    """
    Dependency for common query parameters.
    
    Args:
        skip: Number of items to skip (for pagination)
        limit: Maximum number of items to return
        
    Returns:
        CommonQueryParams: Common query parameters
    """
    return CommonQueryParams(skip=skip, limit=limit)


# Dependency for rate limiting (placeholder)
async def rate_limit_dependency():
    """
    Dependency for rate limiting API requests.
    
    TODO: Implement rate limiting logic
    """
    pass


# Dependency for request logging (placeholder)
async def log_request_dependency():
    """
    Dependency for logging API requests.
    
    TODO: Implement request logging logic
    """
    pass


def get_anime_service() -> object:
    """
    Dependency to get AnimeService instance.
    
    Returns:
        AnimeService: The anime service for async operations
        
    Raises:
        HTTPException: If anime directory is not configured or
                      AnimeService initialization fails
    """
    global _anime_service
    
    if not settings.anime_directory:
        raise HTTPException(
            status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
            detail="Anime directory not configured. Please complete setup.",
        )
    
    if _anime_service is None:
        try:
            from src.server.services.anime_service import AnimeService
            _anime_service = AnimeService(settings.anime_directory)
        except Exception as e:
            raise HTTPException(
                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
                detail=f"Failed to initialize AnimeService: {str(e)}",
            ) from e
    
    return _anime_service


def get_download_service() -> object:
    """
    Dependency to get DownloadService instance.
    
    Returns:
        DownloadService: The download queue service
        
    Raises:
        HTTPException: If DownloadService initialization fails
    """
    global _download_service
    
    if _download_service is None:
        try:
            from src.server.services.download_service import DownloadService
            from src.server.services.websocket_service import get_websocket_service

            # Get anime service first (required dependency)
            anime_service = get_anime_service()
            
            # Initialize download service with anime service
            _download_service = DownloadService(anime_service)
            
            # Setup WebSocket broadcast callback
            ws_service = get_websocket_service()
            
            async def broadcast_callback(update_type: str, data: dict):
                """Broadcast download updates via WebSocket."""
                if update_type == "download_progress":
                    await ws_service.broadcast_download_progress(
                        data.get("download_id", ""), data
                    )
                elif update_type == "download_complete":
                    await ws_service.broadcast_download_complete(
                        data.get("download_id", ""), data
                    )
                elif update_type == "download_failed":
                    await ws_service.broadcast_download_failed(
                        data.get("download_id", ""), data
                    )
                elif update_type == "queue_status":
                    await ws_service.broadcast_queue_status(data)
                else:
                    # Generic queue update
                    await ws_service.broadcast_queue_status(data)
            
            _download_service.set_broadcast_callback(broadcast_callback)
            
        except HTTPException:
            raise
        except Exception as e:
            raise HTTPException(
                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
                detail=f"Failed to initialize DownloadService: {str(e)}",
            ) from e
    
    return _download_service


def reset_anime_service() -> None:
    """Reset global AnimeService instance (for testing/config changes)."""
    global _anime_service
    _anime_service = None


def reset_download_service() -> None:
    """Reset global DownloadService instance (for testing/config changes)."""
    global _download_service
    _download_service = None