Pac-cogs/videoarchiver/__init__.py

"""VideoArchiver cog for Red-DiscordBot"""

import sys
import asyncio
import logging
import importlib
from typing import Optional
from redbot.core.bot import Red  # type: ignore

# Force reload of all modules
modules_to_reload = [
    "utils.exceptions",
    "utils",
    "processor",
    "processor.core",
    "processor.queue_processor",
    "queue",
    "queue.types",  # Added types module
    "queue.models",
    "queue.manager",
    "queue.cleaners",
    "queue.cleaners.guild_cleaner",
    "queue.cleaners.history_cleaner",
    "queue.cleaners.tracking_cleaner",
    "queue.monitoring",
    "queue.recovery_manager",
    "queue.state_manager",
    "ffmpeg",
    "ffmpeg.binary_manager",
    "ffmpeg.encoder_params",
    "ffmpeg.exceptions",
    "ffmpeg.ffmpeg_downloader",
    "ffmpeg.ffmpeg_manager",
    "ffmpeg.gpu_detector",
    "ffmpeg.process_manager",
    "ffmpeg.verification_manager",
    "ffmpeg.video_analyzer",
    "database",
    "database.connection_manager",
    "database.query_manager",
    "database.schema_manager",
    "database.video_archive_db",
    "config",
    "config.channel_manager",
    "config.exceptions",
    "config.role_manager",
    "config.settings_formatter",
    "config.validation_manager",
    "core",
    "core.base",
    "core.cleanup",
    "core.commands",
    "core.commands.archiver_commands",
    "core.commands.database_commands",
    "core.commands.settings_commands",
    "core.component_manager",
    "core.error_handler",
    "core.events",
    "core.guild",
    "core.initialization",
    "core.lifecycle",
    "core.response_handler",
    "core.settings",
]

# Remove modules to force fresh import
for module in modules_to_reload:
    if module in sys.modules:
        del sys.modules[module]

    # try:
    # Try relative imports first
    from . import utils
    from . import processor
    from . import queue
    from . import ffmpeg
    from . import database
    from . import config
    from . import core
    from .core.base import VideoArchiver
    from .core.initialization import initialize_cog, init_callback
    from .core.cleanup import cleanup_resources
    from .utils.exceptions import (
        VideoArchiverError,
        CommandError,
        EventError,
        CogError,
        ErrorContext,
        ErrorSeverity,
        ProcessingError,
    )

    # except ImportError:
    # Fall back to absolute imports if relative imports fail
    #  from videoarchiver import utils
    #  from videoarchiver import processor
    #  from videoarchiver import queue
    #  from videoarchiver import ffmpeg
    #  from videoarchiver import database
    #  from videoarchiver import config
    #  from videoarchiver import core
    # from videoarchiver.core.base import VideoArchiver
    # from videoarchiver.core.initialization import initialize_cog, init_callback
    # from videoarchiver.core.cleanup import cleanup_resources
    # from videoarchiver.utils.exceptions import (
    #    VideoArchiverError,
    #    CommandError,
    #    EventError,
    #    CogError,
    #    ErrorContext,
    #    ErrorSeverity,
    #    ProcessingError,
    # )

    # Reload all modules
    importlib.reload(utils)
    importlib.reload(processor)
    importlib.reload(queue)
    importlib.reload(ffmpeg)
    importlib.reload(database)
    importlib.reload(config)
    importlib.reload(core)

# Import all submodules
from database import *
from ffmpeg import *
from queue import *
from processor import *
from config_manager import *
from update_checker import *
from queue.cleaners import *
from database import *
from utils import *
from core import *
from config import *

logger = logging.getLogger("VideoArchiver")

# Track initialization task
_init_task: Optional[asyncio.Task] = None

# Version information
__version__ = "1.0.0"
__author__ = "VideoArchiver Team"
__description__ = "Video archiving cog for Red-DiscordBot"


async def setup(bot: Red) -> None:
    """Load VideoArchiver with proper initialization."""
    try:
        # Create cog instance
        cog = VideoArchiver(bot)

        # Start initialization in background
        global _init_task
        _init_task = asyncio.create_task(initialize_cog(cog))
        _init_task.add_done_callback(lambda t: init_callback(cog, t))

        # Add cog to bot
        await bot.add_cog(cog)

        logger.info("VideoArchiver cog loaded successfully")

    except Exception as e:
        logger.error(f"Failed to load VideoArchiver: {str(e)}")
        if _init_task and not _init_task.done():
            _init_task.cancel()
        raise


async def teardown(bot: Red) -> None:
    """Clean up when unloading."""
    try:
        # Cancel initialization if still running
        if _init_task and not _init_task.done():
            _init_task.cancel()

        # Remove cog and clean up resources
        if "VideoArchiver" in bot.cogs:
            cog = bot.get_cog("VideoArchiver")
            if cog:
                await cleanup_resources(cog)
            await bot.remove_cog("VideoArchiver")

        logger.info("VideoArchiver cog unloaded successfully")

    except Exception as e:
        logger.error(f"Error during teardown: {str(e)}")
        raise


__all__ = [
    # Core classes
    "VideoArchiver",
    "VideoArchiveDB",
    "FFmpegManager",
    "EnhancedVideoQueueManager",
    "VideoProcessor",
    "ConfigManager",
    "UpdateChecker",
    "QueueCleaner",
    "QueueStateManager",
    "QueueMetricsManager",
    "QueueProcessor",
    "BinaryManager",
    "EncoderParams",
    "GPUDetector",
    "ProcessManager",
    "DatabaseConnectionManager",
    "DatabaseQueryManager",
    "DatabaseSchemaManager",
    "CleanupManager",
    "MessageHandler",
    "CompressionManager",
    "DirectoryManager",
    "DownloadManager",
    "FileOperations",
    # Base exceptions
    "VideoArchiverError",
    "CommandError",
    "EventError",
    "CogError",
    "ErrorContext",
    "ErrorSeverity",
    "ProcessingError",
    # Setup functions
    "setup",
    "teardown",
]