#!/usr/bin/env python3
"""Quick test script for SS Crawler components."""

import sys
from pathlib import Path

def test_imports():
    """Test if all required modules can be imported."""
    print("🔍 Testing imports...")
    
    try:
        import scrapy
        print(f"✅ Scrapy {scrapy.__version__}")
    except ImportError as e:
        print(f"❌ Scrapy not found: {e}")
        print("   Install: pip install scrapy")
        return False
    
    try:
        import click
        print("✅ Click")
    except ImportError as e:
        print(f"❌ Click not found: {e}")
        print("   Install: pip install click")
        return False
    
    try:
        import pydantic
        print(f"✅ Pydantic {pydantic.__version__}")
    except ImportError as e:
        print(f"❌ Pydantic not found: {e}")
        print("   Install: pip install pydantic")
        return False
    
    try:
        import pandas
        print(f"✅ Pandas {pandas.__version__}")
    except ImportError as e:
        print(f"❌ Pandas not found: {e}")
        print("   Install: pip install pandas")
        return False
    
    try:
        import yaml
        print("✅ PyYAML")
    except ImportError as e:
        print(f"❌ PyYAML not found: {e}")
        print("   Install: pip install pyyaml")
        return False
    
    return True


def test_project_structure():
    """Test if project structure is correct."""
    print("\n🔍 Testing project structure...")
    
    required_files = [
        'scrapy_project/__init__.py',
        'scrapy_project/items.py',
        'scrapy_project/spiders/__init__.py',
        'scrapy_project/spiders/generic_spider.py',
        'scrapy_project/pipelines.py',
        'scrapy_project/middlewares.py',
        'scrapy_project/settings.py',
        'scrapy_project/utils/detector.py',
        'main.py',
        'scrapy.cfg',
        'requirements.txt'
    ]
    
    all_ok = True
    for file_path in required_files:
        path = Path(file_path)
        if path.exists():
            print(f"✅ {file_path}")
        else:
            print(f"❌ {file_path} - NOT FOUND")
            all_ok = False
    
    return all_ok


def test_detector():
    """Test website detector."""
    print("\n🔍 Testing website detector...")
    
    try:
        from scrapy_project.utils.detector import WebsiteDetector
        
        detector = WebsiteDetector()
        
        # Test detection
        test_urls = [
            'https://shopee.vn',
            'https://tiki.vn',
            'https://example.com'
        ]
        
        for url in test_urls:
            config = detector.detect(url)
            if config:
                print(f"✅ {url} -> {config.name}")
            else:
                print(f"⚠️  {url} -> Not detected")
        
        return True
    except Exception as e:
        print(f"❌ Detector test failed: {e}")
        import traceback
        traceback.print_exc()
        return False


def test_items():
    """Test items and models."""
    print("\n🔍 Testing items and models...")
    
    try:
        from scrapy_project.items import ProductItem, Product
        
        # Test ProductItem
        item = ProductItem()
        item['name'] = 'Test Product'
        item['price'] = 100000
        item['currency'] = 'VND'
        print("✅ ProductItem created")
        
        # Test Product model
        product = Product.from_scrapy_item(item)
        print(f"✅ Product model created: {product.name}")
        
        # Test export methods
        wc_row = product.to_woocommerce_csv_row()
        print(f"✅ WooCommerce export: {wc_row['Name']}")
        
        shopify_row = product.to_shopify_csv_row()
        print(f"✅ Shopify export: {shopify_row['Title']}")
        
        return True
    except Exception as e:
        print(f"❌ Items test failed: {e}")
        import traceback
        traceback.print_exc()
        return False


def main():
    """Run all tests."""
    print("=" * 50)
    print("🧪 SS Crawler - Quick Test")
    print("=" * 50)
    
    results = []
    
    # Test imports
    results.append(("Imports", test_imports()))
    
    # Test structure
    results.append(("Project Structure", test_project_structure()))
    
    # Test detector (even without scrapy installed)
    results.append(("Website Detector", test_detector()))
    
    # Test items (even without scrapy installed)
    results.append(("Items & Models", test_items()))
    
    # Summary
    print("\n" + "=" * 50)
    print("📊 Test Summary")
    print("=" * 50)
    
    passed = sum(1 for _, result in results if result)
    total = len(results)
    
    for name, result in results:
        status = "✅ PASS" if result else "❌ FAIL"
        print(f"{status} - {name}")
    
    print(f"\nTotal: {passed}/{total} tests passed")
    
    if passed == total:
        print("\n🎉 All tests passed! Ready to use.")
        return 0
    else:
        print("\n⚠️  Some tests failed. Please install missing dependencies.")
        print("\nInstall dependencies:")
        print("  pip install -r requirements.txt")
        return 1


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