AgilePlace MCP Server

#!/usr/bin/env python3
"""Verification script for AgilePlace MCP Server OKR tools."""

import sys
import os

def verify_tools():
    """Verify that OKR tools are properly configured."""
    print("🔍 Verifying AgilePlace MCP Server OKR Tools")
    print("=" * 50)
    
    # Check 1: Verify server.py exists and has OKR tools
    print("\n1. Checking main server.py...")
    if os.path.exists('server.py'):
        with open('server.py', 'r') as f:
            content = f.read()
        
        okr_functions = [
            'create_objective',
            'create_key_result',
            'update_objective',
            'update_key_result', 
            'get_objectives_by_board_id',
            'get_key_results_by_objective_id'
        ]
        
        activity_functions = [
            'connect_activities_to_key_result',
            'create_activity',
            'list_activities',
            'list_activity_containers',
            'search_activities',
            'list_activity_types',
            'search_users',
            'get_current_user'
        ]
        
        print("   ✓ OKR Functions:")
        for func in okr_functions:
            if f'async def {func}' in content:
                print(f"     ✓ {func}")
            else:
                print(f"     ✗ {func} - MISSING")
        
        print("   ✓ Activity Functions:")
        for func in activity_functions:
            if f'async def {func}' in content:
                print(f"     ✓ {func}")
            else:
                print(f"     ✗ {func} - MISSING")
        
        # Check for tool decorators
        tool_count = content.count('@mcp.tool()')
        print(f"   ✓ Tool decorators found: {tool_count}")
        
    else:
        print("   ✗ server.py not found!")
        return False
    
    # Check 2: Verify tools directory structure
    print("\n2. Checking tools directory...")
    if os.path.exists('tools'):
        tools_files = os.listdir('tools')
        print("   ✓ Tools directory exists")
        print("   ✓ Files:", [f for f in tools_files if f.endswith('.py')])
        
        # Check for OKR modules
        okr_files = [f for f in tools_files if 'okr' in f]
        if 'okr.py' in tools_files and 'okr_activities.py' in tools_files:
            print("   ✓ OKR modules found")
        else:
            print("   ✗ OKR modules missing!")
            return False
    else:
        print("   ✗ Tools directory not found!")
        return False
    
    # Check 3: Verify models
    print("\n3. Checking models...")
    if os.path.exists('models.py'):
        print("   ✓ models.py exists")
        
        # Check for key models
        with open('models.py', 'r') as f:
            models_content = f.read()
        
        key_models = ['Objective', 'KeyResult', 'WorkItemContainer', 'WorkItem']
        for model in key_models:
            if f'class {model}' in models_content:
                print(f"   ✓ {model} model")
            else:
                print(f"   ✗ {model} model - MISSING")
    else:
        print("   ✗ models.py not found!")
        return False
    
    # Check 4: Verify imports work
    print("\n4. Testing imports...")
    try:
        sys.path.insert(0, '.')
        
        # Test tool imports
        from agileplace_mcp.tools import okr, okr_activities
        print("   ✓ Can import okr module")
        print("   ✓ Can import okr_activities module")
        
        # Test model imports
        from agileplace_mcp.models import Objective, KeyResult, WorkItemContainer
        print("   ✓ Can import models")
        
    except ImportError as e:
        print(f"   ✗ Import error: {e}")
        return False
    
    print("\n🎉 All checks passed!")
    print("\nNext steps:")
    print("1. Install FastMCP: pip install fastmcp")
    print("2. Configure Claude Desktop with the updated server")
    print("3. Restart Claude Desktop")
    print("4. Try using the new OKR tools!")
    
    return True

if __name__ == "__main__":
    success = verify_tools()
    if not success:
        sys.exit(1)