"""
Django management command to fetch Steam Workshop collections
"""

from django.core.management.base import BaseCommand, CommandError
from submissions.utils import create_or_update_collection
from submissions.models import SteamCollection


class Command(BaseCommand):
    help = 'Fetch Steam Workshop collection data and save to database'
    
    def add_arguments(self, parser):
        parser.add_argument(
            'url',
            type=str,
            help='Steam Workshop collection URL'
        )
        parser.add_argument(
            '--api-key',
            type=str,
            help='Steam API key (optional, can also be set via STEAM_API_KEY environment variable)'
        )
        parser.add_argument(
            '--force',
            action='store_true',
            help='Force refetch even if collection already exists'
        )
    
    def handle(self, *args, **options):
        url = options['url']
        api_key = options.get('api_key')
        force = options['force']
        
        self.stdout.write(f"Fetching Steam collection from: {url}")
        
        try:
            # Check if collection already exists
            from submissions.utils import SteamCollectionFetcher
            fetcher = SteamCollectionFetcher(api_key)
            collection_id = fetcher.extract_collection_id(url)
            
            if collection_id and not force:
                existing = SteamCollection.objects.filter(steam_id=collection_id).first()
                if existing:
                    self.stdout.write(
                        self.style.WARNING(
                            f"Collection {collection_id} already exists (ID: {existing.id}). "
                            "Use --force to refetch."
                        )
                    )
                    return
            
            # Fetch and create/update collection
            collection, created = create_or_update_collection(url)
            
            if created:
                self.stdout.write(
                    self.style.SUCCESS(
                        f"Successfully created collection: {collection.title} (ID: {collection.id})"
                    )
                )
            else:
                self.stdout.write(
                    self.style.SUCCESS(
                        f"Successfully updated collection: {collection.title} (ID: {collection.id})"
                    )
                )
            
            # Display collection info
            self.stdout.write("\nCollection Details:")
            self.stdout.write(f"  Steam ID: {collection.steam_id}")
            self.stdout.write(f"  Title: {collection.title}")
            self.stdout.write(f"  Author: {collection.author_name or 'Unknown'}")
            self.stdout.write(f"  Description: {collection.description[:100]}{'...' if len(collection.description) > 100 else ''}")
            self.stdout.write(f"  Total Items: {collection.total_items}")
            self.stdout.write(f"  Unique Visitors: {collection.unique_visitors}")
            self.stdout.write(f"  Current Favorites: {collection.current_favorites}")
            self.stdout.write(f"  Total Favorites: {collection.total_favorites}")
            
            if collection.items.exists():
                self.stdout.write(f"\nCollection Items ({collection.items.count()}):")
                for item in collection.items.all()[:10]:  # Show first 10 items
                    self.stdout.write(f"  - {item.title} (Steam ID: {item.steam_item_id})")
                
                if collection.items.count() > 10:
                    self.stdout.write(f"  ... and {collection.items.count() - 10} more items")
            else:
                self.stdout.write("\nNo items found in collection.")
            
        except Exception as e:
            raise CommandError(f"Failed to fetch collection: {e}")