test_direct_applescript.py

#!/usr/bin/env python3
"""
Direct AppleScript Test - Bypass AI entirely

Test the new AppleScript capabilities directly without going through the AI service.
"""

import sys
import asyncio
from pathlib import Path

# Add automation framework to path
sys.path.insert(0, str(Path(__file__).parent / "automation_framework"))

from rich.console import Console
from rich.panel import Panel
from rich.json import JSON

console = Console()


async def test_applescript_directly():
    """Test AppleScript capabilities directly"""
    
    console.print("[bold cyan]🍎 Direct AppleScript Test[/bold cyan]")
    console.print("=" * 50)
    
    try:
        # Import the generic actor directly
        from actors.generic.mouse_keyboard import GenericActorStack
        
        console.print("1. 🚀 Initializing Generic Actor...")
        actor = GenericActorStack()
        
        console.print("2. 📋 Testing capabilities...")
        capabilities = actor.get_capabilities()
        
        applescript_actions = {k: v for k, v in capabilities.items() 
                             if 'script' in k.lower() or k in ['get_current_window', 'screenshot']}
        
        console.print("Available AppleScript actions:")
        for action, details in applescript_actions.items():
            console.print(f"  • [green]{action}[/green]: {details['description']}")
        
        console.print("\n3. 🪟 Getting current window info...")
        try:
            window_info = await actor.execute_action('get_current_window')
            console.print(Panel(
                JSON.from_data(window_info),
                title="Current Window Info"
            ))
        except Exception as e:
            console.print(f"[red]❌ get_current_window failed: {e}[/red]")
        
        console.print("\n4. 📸 Testing screenshot...")
        try:
            screenshot_result = await actor.execute_action('screenshot', filename='test_screenshot.png')
            if screenshot_result:
                console.print("[green]✅ Screenshot taken successfully![/green]")
            else:
                console.print("[yellow]⚠️ Screenshot may have failed[/yellow]")
        except Exception as e:
            console.print(f"[red]❌ Screenshot failed: {e}[/red]")
        
        console.print("\n5. 🍎 Testing custom AppleScript...")
        custom_script = '''
        tell application "System Events"
            try
                set appList to name of every application process whose visible is true
                set appCount to count of appList
                return "Running apps: " & appCount & " - " & (item 1 of appList) & ", " & (item 2 of appList)
            on error
                return "Error getting app list"
            end try
        end tell
        '''
        
        try:
            script_result = await actor.execute_action('run_applescript', script=custom_script)
            console.print(f"[cyan]AppleScript result:[/cyan] {script_result}")
        except Exception as e:
            console.print(f"[red]❌ Custom AppleScript failed: {e}[/red]")
        
        console.print("\n6. 🧹 Cleaning up...")
        await actor.cleanup()
        
        console.print("\n[green]✅ Direct AppleScript test complete![/green]")
        
    except Exception as e:
        console.print(f"[red]❌ Test failed: {e}[/red]")
        import traceback
        traceback.print_exc()


async def test_simple_applescript():
    """Test even simpler AppleScript execution"""
    
    console.print("\n[bold yellow]🔧 Simple AppleScript Test[/bold yellow]")
    console.print("=" * 30)
    
    try:
        # Test direct osascript execution
        console.print("Testing direct osascript...")
        
        process = await asyncio.create_subprocess_exec(
            'osascript', '-e', 'tell application "System Events" to get name of first application process whose frontmost is true',
            stdout=asyncio.subprocess.PIPE,
            stderr=asyncio.subprocess.PIPE
        )
        stdout, stderr = await process.communicate()
        
        if stderr:
            console.print(f"[red]Error: {stderr.decode()}[/red]")
        else:
            console.print(f"[green]Front app: {stdout.decode().strip()}[/green]")
            
    except Exception as e:
        console.print(f"[red]❌ Simple test failed: {e}[/red]")


async def main():
    """Run all tests"""
    
    # Test direct actor capabilities
    await test_applescript_directly()
    
    # Test simple AppleScript
    await test_simple_applescript()
    
    console.print("\n🎯 Summary")
    console.print("This test bypassed the AI entirely and called AppleScript actions directly.")
    console.print("If this works, we know the AppleScript integration is functional.")


if __name__ == "__main__":
    asyncio.run(main())