fixed

videoarchiver/processor/__init__.py

@@ -1,45 +1,212 @@
 """Video processing module for VideoArchiver"""
 
+from typing import Dict, Any, Optional, Union, List, Tuple
+import discord
+
 from .core import VideoProcessor
-from .constants import REACTIONS
-from .progress_tracker import ProgressTracker
+from .constants import (
+    REACTIONS,
+    ReactionType,
+    ReactionEmojis,
+    ProgressEmojis,
+    get_reaction,
+    get_progress_emoji
+)
+from .url_extractor import (
+    URLExtractor,
+    URLMetadata,
+    URLPattern,
+    URLType,
+    URLPatternManager,
+    URLValidator,
+    URLMetadataExtractor
+)
+from .message_validator import (
+    MessageValidator,
+    ValidationContext,
+    ValidationRule,
+    ValidationResult,
+    ValidationRuleManager,
+    ValidationCache,
+    ValidationStats,
+    ValidationCacheEntry,
+    ValidationError
+)
 from .message_handler import MessageHandler
 from .queue_handler import QueueHandler
+from .reactions import (
+    handle_archived_reaction,
+    update_queue_position_reaction,
+    update_progress_reaction,
+    update_download_progress_reaction
+)
 
 # Export public classes and constants
 __all__ = [
+    # Core components
     "VideoProcessor",
-    "REACTIONS",
-    "ProgressTracker",
     "MessageHandler",
     "QueueHandler",
+    
+    # URL Extraction
+    "URLExtractor",
+    "URLMetadata",
+    "URLPattern",
+    "URLType",
+    "URLPatternManager",
+    "URLValidator",
+    "URLMetadataExtractor",
+    
+    # Message Validation
+    "MessageValidator",
+    "ValidationContext",
+    "ValidationRule",
+    "ValidationResult",
+    "ValidationRuleManager",
+    "ValidationCache",
+    "ValidationStats",
+    "ValidationCacheEntry",
+    "ValidationError",
+    
+    # Constants and enums
+    "REACTIONS",
+    "ReactionType",
+    "ReactionEmojis",
+    "ProgressEmojis",
+    
+    # Helper functions
+    "get_reaction",
+    "get_progress_emoji",
+    "extract_urls",
+    "validate_message",
+    "update_download_progress",
+    "complete_download",
+    "increment_download_retries",
+    "get_download_progress",
+    "get_active_operations",
+    "get_validation_stats",
+    "clear_caches",
+    
+    # Reaction handlers
+    "handle_archived_reaction",
+    "update_queue_position_reaction",
+    "update_progress_reaction",
+    "update_download_progress_reaction",
 ]
 
-# Create a shared progress tracker instance for module-level access
-progress_tracker = ProgressTracker()
+# Version information
+__version__ = "1.0.0"
+__author__ = "VideoArchiver Team"
+__description__ = "Video processing module for archiving Discord videos"
 
+# Create shared instances for module-level access
+url_extractor = URLExtractor()
+message_validator = MessageValidator()
 
-# Export progress tracking functions that wrap the instance methods
-def update_download_progress(url, progress_data):
-    """Update download progress for a specific URL"""
+# URL extraction helper functions
+async def extract_urls(
+    message: discord.Message,
+    enabled_sites: Optional[List[str]] = None
+) -> List[URLMetadata]:
+    """
+    Extract video URLs from a Discord message.
+    
+    Args:
+        message: Discord message to extract URLs from
+        enabled_sites: Optional list of enabled site identifiers
+        
+    Returns:
+        List of URLMetadata objects for extracted URLs
+    """
+    return await url_extractor.extract_urls(message, enabled_sites)
+
+async def validate_message(
+    message: discord.Message,
+    settings: Dict[str, Any]
+) -> Tuple[bool, Optional[str]]:
+    """
+    Validate a Discord message.
+    
+    Args:
+        message: Discord message to validate
+        settings: Guild settings dictionary
+        
+    Returns:
+        Tuple of (is_valid, reason)
+        
+    Raises:
+        ValidationError: If validation fails unexpectedly
+    """
+    return await message_validator.validate_message(message, settings)
+
+# Progress tracking helper functions
+def update_download_progress(url: str, progress_data: Dict[str, Any]) -> None:
+    """
+    Update download progress for a specific URL.
+    
+    Args:
+        url: The URL being downloaded
+        progress_data: Dictionary containing progress information
+    """
     progress_tracker.update_download_progress(url, progress_data)
 
-
-def complete_download(url):
-    """Mark a download as complete"""
+def complete_download(url: str) -> None:
+    """
+    Mark a download as complete.
+    
+    Args:
+        url: The URL that completed downloading
+    """
     progress_tracker.complete_download(url)
 
-
-def increment_download_retries(url):
-    """Increment retry count for a download"""
+def increment_download_retries(url: str) -> None:
+    """
+    Increment retry count for a download.
+    
+    Args:
+        url: The URL being retried
+    """
     progress_tracker.increment_download_retries(url)
 
-
-def get_download_progress(url=None):
-    """Get download progress for a specific URL or all downloads"""
+def get_download_progress(url: Optional[str] = None) -> Union[Dict[str, Any], Dict[str, Dict[str, Any]]]:
+    """
+    Get download progress for a specific URL or all downloads.
+    
+    Args:
+        url: Optional URL to get progress for. If None, returns all download progress.
+    
+    Returns:
+        Dictionary containing progress information for one or all downloads
+    """
     return progress_tracker.get_download_progress(url)
 
-
-def get_active_operations():
-    """Get all active operations"""
+def get_active_operations() -> Dict[str, Dict[str, Any]]:
+    """
+    Get all active operations.
+    
+    Returns:
+        Dictionary containing information about all active operations
+    """
     return progress_tracker.get_active_operations()
+
+def get_validation_stats() -> ValidationStats:
+    """
+    Get message validation statistics.
+    
+    Returns:
+        Dictionary containing validation statistics and rule information
+    """
+    return message_validator.get_stats()
+
+def clear_caches(message_id: Optional[int] = None) -> None:
+    """
+    Clear URL and validation caches.
+    
+    Args:
+        message_id: Optional message ID to clear caches for. If None, clears all caches.
+    """
+    url_extractor.clear_cache(message_id)
+    message_validator.clear_cache(message_id)
+
+# Initialize shared progress tracker instance
+progress_tracker = ProgressTracker()