#!/usr/bin/env python3
"""
Quick demonstration of using extracted StarCraft assets

This script shows basic examples of loading and using different
asset types from the extracted StarCraft MPQ.
"""

import sys
from pathlib import Path


def demo_list_assets():
    """List all extracted assets by category"""
    assets_dir = Path("assets")
    
    if not assets_dir.exists():
        print("❌ Assets directory not found!")
        print("Please run: python extract_starcraft_assets.py")
        return False
    
    print("=" * 80)
    print("StarCraft Extracted Assets Overview")
    print("=" * 80)
    print()
    
    total_files = 0
    total_size = 0
    
    for category_dir in sorted(assets_dir.iterdir()):
        if not category_dir.is_dir():
            continue
        
        files = list(category_dir.rglob("*"))
        files = [f for f in files if f.is_file()]
        
        if not files:
            continue
        
        category_size = sum(f.stat().st_size for f in files)
        total_files += len(files)
        total_size += category_size
        
        print(f"📁 {category_dir.name}/")
        print(f"   Files: {len(files)}")
        print(f"   Size:  {format_size(category_size)}")
        
        # Show some example files
        examples = sorted(files)[:3]
        if examples:
            print(f"   Examples:")
            for f in examples:
                print(f"     - {f.name}")
        print()
    
    print("-" * 80)
    print(f"Total: {total_files} files, {format_size(total_size)}")
    print()
    
    return True


def demo_audio_info():
    """Show information about audio files"""
    audio_dir = Path("assets/audio")
    
    if not audio_dir.exists():
        print("❌ Audio directory not found!")
        return False
    
    print("=" * 80)
    print("Audio Assets Analysis")
    print("=" * 80)
    print()
    
    wav_files = list(audio_dir.glob("*.wav"))
    
    if not wav_files:
        print("No WAV files found!")
        return False
    
    print(f"Found {len(wav_files)} WAV files")
    print()
    
    # Try to read WAV headers (if wave module available)
    try:
        import wave
        
        print("Sample WAV file information:")
        print("-" * 80)
        
        for wav_file in sorted(wav_files)[:5]:
            try:
                with wave.open(str(wav_file), 'rb') as wf:
                    channels = wf.getnchannels()
                    sample_width = wf.getsampwidth()
                    framerate = wf.getframerate()
                    n_frames = wf.getnframes()
                    duration = n_frames / framerate
                    
                    print(f"📢 {wav_file.name}")
                    print(f"   Channels:    {channels} ({'Mono' if channels == 1 else 'Stereo'})")
                    print(f"   Sample Rate: {framerate} Hz")
                    print(f"   Bit Depth:   {sample_width * 8} bit")
                    print(f"   Duration:    {duration:.2f} seconds")
                    print(f"   Size:        {format_size(wav_file.stat().st_size)}")
                    print()
            except Exception as e:
                print(f"   ⚠️  Could not read: {e}")
    
    except ImportError:
        print("⚠️  wave module not available for detailed analysis")
        print("Showing file sizes instead:")
        print("-" * 80)
        
        for wav_file in sorted(wav_files)[:10]:
            size = wav_file.stat().st_size
            print(f"  {wav_file.name:.<50} {format_size(size):>10}")
    
    print()
    return True


def demo_graphics_info():
    """Show information about graphics files"""
    graphics_dir = Path("assets/graphics")
    
    if not graphics_dir.exists():
        print("❌ Graphics directory not found!")
        return False
    
    print("=" * 80)
    print("Graphics Assets Analysis")
    print("=" * 80)
    print()
    
    # Count by extension
    from collections import defaultdict
    by_ext = defaultdict(list)
    
    for gfx_file in graphics_dir.rglob("*"):
        if gfx_file.is_file():
            ext = gfx_file.suffix.lower()
            by_ext[ext].append(gfx_file)
    
    print("File types:")
    print("-" * 80)
    for ext in sorted(by_ext.keys()):
        files = by_ext[ext]
        total_size = sum(f.stat().st_size for f in files)
        print(f"  {ext or '(no ext)':.<15} {len(files):>4} files  {format_size(total_size):>10}")
    print()
    
    # Try to analyze PCX files if PIL available
    try:
        from PIL import Image
        
        pcx_files = list(graphics_dir.glob("*.pcx"))
        if pcx_files:
            print("Sample PCX images:")
            print("-" * 80)
            
            for pcx_file in sorted(pcx_files)[:5]:
                try:
                    with Image.open(pcx_file) as img:
                        print(f"🖼️  {pcx_file.name}")
                        print(f"   Size:   {img.width}x{img.height}")
                        print(f"   Mode:   {img.mode}")
                        print(f"   Format: {img.format}")
                        print()
                except Exception as e:
                    print(f"   ⚠️  Could not read: {e}")
        
    except ImportError:
        print("⚠️  PIL/Pillow not available for image analysis")
        print("Install with: pip install Pillow")
    
    print()
    return True


def demo_search_files(query: str):
    """Search for files matching a pattern"""
    assets_dir = Path("assets")
    
    if not assets_dir.exists():
        print("❌ Assets directory not found!")
        return False
    
    query_lower = query.lower()
    matches = []
    
    for file_path in assets_dir.rglob("*"):
        if file_path.is_file() and query_lower in file_path.name.lower():
            matches.append(file_path)
    
    print(f"\nSearch results for '{query}':")
    print("=" * 80)
    
    if not matches:
        print("No files found!")
        return False
    
    print(f"Found {len(matches)} matching files:\n")
    
    for match in sorted(matches)[:20]:
        rel_path = match.relative_to(assets_dir)
        size = match.stat().st_size
        print(f"  {str(rel_path):.<60} {format_size(size):>10}")
    
    if len(matches) > 20:
        print(f"\n  ... and {len(matches) - 20} more files")
    
    print()
    return True


def format_size(size_bytes: int) -> str:
    """Format size in bytes to human-readable format"""
    if size_bytes < 1024:
        return f"{size_bytes} B"
    elif size_bytes < 1024 * 1024:
        return f"{size_bytes / 1024:.1f} KB"
    else:
        return f"{size_bytes / 1024 / 1024:.1f} MB"


def main():
    """Main entry point"""
    print("\n" + "=" * 80)
    print("StarCraft Asset Demo")
    print("=" * 80)
    print()
    
    # Check if assets exist
    if not Path("assets").exists():
        print("❌ Assets not extracted yet!")
        print("\nPlease run the extraction script first:")
        print("  python extract_starcraft_assets.py")
        print()
        return 1
    
    # Show main menu if no arguments
    if len(sys.argv) == 1:
        print("Available demos:")
        print("  1. List all assets")
        print("  2. Analyze audio files")
        print("  3. Analyze graphics files")
        print("  4. Search files")
        print()
        
        choice = input("Select demo (1-4) or 'q' to quit: ").strip()
        
        if choice == '1':
            demo_list_assets()
        elif choice == '2':
            demo_audio_info()
        elif choice == '3':
            demo_graphics_info()
        elif choice == '4':
            query = input("Enter search query: ").strip()
            demo_search_files(query)
        elif choice.lower() == 'q':
            return 0
        else:
            print("Invalid choice!")
            return 1
    
    # Command line usage
    elif sys.argv[1] == 'list':
        demo_list_assets()
    elif sys.argv[1] == 'audio':
        demo_audio_info()
    elif sys.argv[1] == 'graphics':
        demo_graphics_info()
    elif sys.argv[1] == 'search':
        if len(sys.argv) < 3:
            print("Usage: python demo_assets.py search <query>")
            return 1
        demo_search_files(sys.argv[2])
    else:
        print("Usage:")
        print("  python demo_assets.py [list|audio|graphics|search <query>]")
        print("\nOr run without arguments for interactive menu")
        return 1
    
    return 0


if __name__ == "__main__":
    sys.exit(main())