Pac-cogs/videoarchiver/utils/__init__.py

"""Utility functions and classes for VideoArchiver"""

from typing import Dict, Optional, Any, Union, List

from .file_ops import (
    cleanup_downloads,
    ensure_directory,
    get_file_size,
    is_valid_path,
    safe_delete
)
from .file_deletion import FileDeleter
from .directory_manager import DirectoryManager
from .permission_manager import PermissionManager
from .download_manager import DownloadManager
from .compression_manager import CompressionManager
from .progress_tracker import (
    ProgressTracker,
    ProgressStatus,
    DownloadProgress,
    CompressionProgress,
    CompressionParams
)
from .path_manager import PathManager
from .exceptions import (
    # Base exception
    VideoArchiverError,
    ErrorSeverity,
    ErrorContext,
    
    # File operations
    FileOperationError,
    DirectoryError,
    PermissionError,
    FileCleanupError,
    
    # Video operations
    VideoDownloadError,
    VideoProcessingError,
    VideoVerificationError,
    VideoUploadError,
    VideoCleanupError,
    
    # Resource management
    ResourceError,
    ResourceExhaustedError,
    
    # Network and API
    NetworkError,
    DiscordAPIError,
    
    # Component operations
    ComponentError,
    ConfigurationError,
    DatabaseError,
    FFmpegError,
    
    # Queue operations
    QueueError,
    QueueHandlerError,
    QueueProcessorError,
    
    # Processing operations
    ProcessingError,
    ProcessorError,
    ValidationError,
    DisplayError,
    URLExtractionError,
    MessageHandlerError,
    
    # Cleanup operations
    CleanupError,
    
    # Health monitoring
    HealthCheckError,
    
    # Command and Event operations
    CommandError,
    EventError,
    
    # Cog operations
    CogError,
    
    # Progress tracking
    TrackingError
)

__all__ = [
    # File Operations
    'cleanup_downloads',
    'ensure_directory',
    'get_file_size',
    'is_valid_path',
    'safe_delete',
    
    # Managers
    'FileDeleter',
    'DirectoryManager',
    'PermissionManager',
    'DownloadManager',
    'CompressionManager',
    'ProgressTracker',
    'PathManager',
    
    # Progress Tracking Types
    'ProgressStatus',
    'DownloadProgress',
    'CompressionProgress',
    'CompressionParams',
    
    # Base Exceptions
    'VideoArchiverError',
    'ErrorSeverity',
    'ErrorContext',
    
    # File Operation Exceptions
    'FileOperationError',
    'DirectoryError',
    'PermissionError',
    'FileCleanupError',
    
    # Video Operation Exceptions
    'VideoDownloadError',
    'VideoProcessingError',
    'VideoVerificationError',
    'VideoUploadError',
    'VideoCleanupError',
    
    # Resource Exceptions
    'ResourceError',
    'ResourceExhaustedError',
    
    # Network and API Exceptions
    'NetworkError',
    'DiscordAPIError',
    
    # Component Exceptions
    'ComponentError',
    'ConfigurationError',
    'DatabaseError',
    'FFmpegError',
    
    # Queue Exceptions
    'QueueError',
    'QueueHandlerError',
    'QueueProcessorError',
    
    # Processing Exceptions
    'ProcessingError',
    'ProcessorError',
    'ValidationError',
    'DisplayError',
    'URLExtractionError',
    'MessageHandlerError',
    
    # Cleanup Exceptions
    'CleanupError',
    
    # Health Monitoring Exceptions
    'HealthCheckError',
    
    # Command and Event Exceptions
    'CommandError',
    'EventError',
    
    # Cog Exceptions
    'CogError',
    
    # Progress Tracking Exceptions
    'TrackingError',
    
    # Helper Functions
    'get_download_progress',
    'get_compression_progress',
    'get_active_downloads',
    'get_active_compressions'
]

# Version information
__version__ = "1.0.0"
__author__ = "VideoArchiver Team"
__description__ = "Utility functions and classes for VideoArchiver"

# Initialize shared instances for module-level access
directory_manager = DirectoryManager()
permission_manager = PermissionManager()
download_manager = DownloadManager()
compression_manager = CompressionManager()
progress_tracker = ProgressTracker()
path_manager = PathManager()

# Progress tracking helper functions
def get_download_progress(url: Optional[str] = None) -> Union[Dict[str, DownloadProgress], Optional[DownloadProgress]]:
    """
    Get progress information for a download.
    
    Args:
        url: Optional URL to get progress for. If None, returns all progress.
        
    Returns:
        If url is provided, returns progress for that URL or None if not found.
        If url is None, returns dictionary of all download progress.
        
    Raises:
        TrackingError: If there's an error getting progress information
    """
    try:
        return progress_tracker.get_download_progress(url)
    except Exception as e:
        raise TrackingError(f"Failed to get download progress: {str(e)}")

def get_compression_progress(input_file: Optional[str] = None) -> Union[Dict[str, CompressionProgress], Optional[CompressionProgress]]:
    """
    Get progress information for a compression operation.
    
    Args:
        input_file: Optional file to get progress for. If None, returns all progress.
        
    Returns:
        If input_file is provided, returns progress for that file or None if not found.
        If input_file is None, returns dictionary of all compression progress.
        
    Raises:
        TrackingError: If there's an error getting progress information
    """
    try:
        return progress_tracker.get_compression_progress(input_file)
    except Exception as e:
        raise TrackingError(f"Failed to get compression progress: {str(e)}")

def get_active_downloads() -> Dict[str, DownloadProgress]:
    """
    Get all active downloads.
    
    Returns:
        Dictionary mapping URLs to their download progress information
        
    Raises:
        TrackingError: If there's an error getting active downloads
    """
    try:
        return progress_tracker.get_active_downloads()
    except Exception as e:
        raise TrackingError(f"Failed to get active downloads: {str(e)}")

def get_active_compressions() -> Dict[str, CompressionProgress]:
    """
    Get all active compression operations.
    
    Returns:
        Dictionary mapping file paths to their compression progress information
        
    Raises:
        TrackingError: If there's an error getting active compressions
    """
    try:
        return progress_tracker.get_active_compressions()
    except Exception as e:
        raise TrackingError(f"Failed to get active compressions: {str(e)}")

# Error handling helper functions
def create_error_context(
    component: str,
    operation: str,
    details: Optional[Dict[str, Any]] = None,
    severity: ErrorSeverity = ErrorSeverity.MEDIUM
) -> ErrorContext:
    """
    Create an error context object.
    
    Args:
        component: Component where error occurred
        operation: Operation that failed
        details: Optional error details
        severity: Error severity level
        
    Returns:
        ErrorContext object
    """
    return ErrorContext(component, operation, details, severity)