#!/usr/bin/env python3
"""
Integration test to verify basic functionality after security hardening
"""

import tempfile
from pathlib import Path
from reportlab.pdfgen import canvas
from src.mcp_pdf_tools.server import create_server, validate_pdf_path, validate_page_count
import fitz


def create_test_pdf():
    """Create a simple test PDF file"""
    with tempfile.NamedTemporaryFile(suffix='.pdf', delete=False) as tmp_file:
        c = canvas.Canvas(tmp_file.name)
        c.drawString(100, 750, "This is a test PDF document.")
        c.drawString(100, 700, "It has some sample text for testing.")
        c.save()
        return Path(tmp_file.name)


def test_basic_functionality():
    """Test basic functionality after security hardening"""
    
    print("🧪 Testing MCP PDF Tools Integration")
    print("=" * 50)
    
    # 1. Test server creation
    print("1. Testing server creation...")
    try:
        server = create_server()
        print("   ✅ Server created successfully")
    except Exception as e:
        print(f"   ❌ Server creation failed: {e}")
        return False
    
    # 2. Test PDF file validation
    print("2. Testing PDF validation...")
    test_pdf = create_test_pdf()
    try:
        validated_path = validate_pdf_path(str(test_pdf))
        print(f"   ✅ PDF validation successful: {validated_path}")
    except Exception as e:
        print(f"   ❌ PDF validation failed: {e}")
        test_pdf.unlink()
        return False
    
    # 3. Test page count validation
    print("3. Testing page count validation...")
    try:
        doc = fitz.open(str(test_pdf))
        validate_page_count(doc, "integration test")
        doc.close()
        print("   ✅ Page count validation successful")
    except Exception as e:
        print(f"   ❌ Page count validation failed: {e}")
        test_pdf.unlink()
        return False
    
    # 4. Test file size limits
    print("4. Testing file size checking...")
    file_size = test_pdf.stat().st_size
    print(f"   📏 Test PDF size: {file_size} bytes")
    print(f"   📏 Max allowed: 100MB ({100 * 1024 * 1024} bytes)")
    if file_size < 100 * 1024 * 1024:
        print("   ✅ File size within limits")
    else:
        print("   ❌ File size exceeds limits")
        test_pdf.unlink()
        return False
    
    # 5. Clean up
    test_pdf.unlink()
    print("   🧹 Test file cleaned up")
    
    print("\n🎉 All integration tests passed!")
    print("🔒 Security features are working correctly")
    print("⚡ Core functionality is intact")
    
    return True


if __name__ == "__main__":
    success = test_basic_functionality()
    exit(0 if success else 1)