fix_ssl.py

#!/usr/bin/env python3
"""
SSL Certificate Fix for Codestral CLI
Helps resolve common SSL certificate issues
"""

import ssl
import sys
import subprocess
from urllib.request import urlopen

def test_ssl_connection():
    """Test SSL connection to Codestral API"""
    try:
        print("🔒 Testing SSL connection to Codestral API...")
        
        # Test the connection
        with urlopen("https://codestral.mistral.ai", timeout=10) as response:
            print(f"✅ SSL connection successful! Status: {response.status}")
            return True
            
    except Exception as e:
        print(f"❌ SSL connection failed: {e}")
        return False

def fix_certificates():
    """Try to fix certificate issues"""
    print("\n🔧 Attempting to fix SSL certificates...")
    
    fixes = [
        ("Upgrading certifi", ["pip3", "install", "--upgrade", "certifi"]),
        ("Updating CA certificates", ["pip3", "install", "--upgrade", "requests", "urllib3"]),
    ]
    
    for desc, cmd in fixes:
        try:
            print(f"  {desc}...")
            subprocess.run(cmd, check=True, capture_output=True)
            print(f"  ✅ {desc} completed")
        except subprocess.CalledProcessError as e:
            print(f"  ⚠️ {desc} failed: {e}")
        except FileNotFoundError:
            print(f"  ⚠️ Command not found: {' '.join(cmd)}")

def show_workarounds():
    """Show SSL workaround options"""
    print("\n🛠️  SSL Workarounds:")
    print("1. Temporary SSL bypass (not recommended for production):")
    print("   export PYTHONHTTPSVERIFY=0")
    print("   python3 main.py")
    print()
    print("2. Update Python certificates (macOS):")
    print("   /Applications/Python\\ 3.x/Install\\ Certificates.command")
    print()
    print("3. Manual certificate bundle:")
    print("   pip3 install --upgrade certifi")
    print("   python3 -c \"import certifi; print(certifi.where())\"")
    print()
    print("4. Check your firewall/proxy settings")

def main():
    print("🔒 Codestral CLI - SSL Certificate Troubleshoot")
    print("=" * 50)
    
    # Test current connection
    if test_ssl_connection():
        print("🎉 No SSL issues detected!")
        return
    
    # Try to fix
    fix_certificates()
    
    # Test again
    print("\n🔄 Testing connection after fixes...")
    if test_ssl_connection():
        print("🎉 SSL issues resolved!")
    else:
        show_workarounds()
        print("\n💡 If issues persist, try the workarounds above")

if __name__ == "__main__":
    main()