decodingtrust-agent-sdk 0.1.0__py3-none-any.whl

dt_arena/mcp_server/calendar/example_multi_user.py

@@ -0,0 +1,162 @@
+#!/usr/bin/env python3
+"""
+Example: Multi-User Calendar MCP Server Usage
+
+This example demonstrates how to use the Calendar MCP Server with multiple users
+by passing access_token as a parameter to each tool call.
+"""
+import asyncio
+import json
+import os
+import sys
+
+# Add current directory to path
+sys.path.insert(0, os.path.dirname(__file__))
+
+from main import list_calendars, list_events, create_event
+
+
+async def get_access_token(email: str, password: str) -> str:
+    """Get access token for a user"""
+    import httpx
+    async with httpx.AsyncClient() as client:
+        resp = await client.post(
+            "http://localhost:8032/api/v1/auth/login",
+            json={"email": email, "password": password}
+        )
+        if resp.status_code == 200:
+            return resp.json()["access_token"]
+        else:
+            raise Exception(f"Login failed: {resp.text}")
+
+
+async def demo_multi_user():
+    """Demonstrate multi-user usage"""
+    
+    print("=" * 70)
+    print("Calendar MCP Server - Multi-User Demo")
+    print("=" * 70)
+    
+    # Get tokens for different users
+    print("\n1. Authenticating users...")
+    alice_token = await get_access_token("alice@example.com", "password123")
+    bob_token = await get_access_token("bob@example.com", "password123")
+    print(f"   ✅ Alice token: {alice_token[:30]}...")
+    print(f"   ✅ Bob token: {bob_token[:30]}...")
+    
+    # Alice lists her calendars
+    print("\n2. Alice lists her calendars...")
+    result = await list_calendars(access_token=alice_token)
+    data = json.loads(result)
+    if "items" in data:
+        print(f"   ✅ Alice has {len(data['items'])} calendar(s)")
+        for cal in data["items"]:
+            print(f"      - {cal['summary']}")
+    
+    # Bob lists his calendars
+    print("\n3. Bob lists his calendars...")
+    result = await list_calendars(access_token=bob_token)
+    data = json.loads(result)
+    if "items" in data:
+        print(f"   ✅ Bob has {len(data['items'])} calendar(s)")
+        for cal in data["items"]:
+            print(f"      - {cal['summary']}")
+    
+    # Alice creates an event and invites Bob
+    print("\n4. Alice creates an event and invites Bob...")
+    result = await create_event(
+        summary="Team Meeting",
+        start_datetime="2025-10-26T14:00:00",
+        end_datetime="2025-10-26T15:00:00",
+        location="Conference Room",
+        description="Discuss Q4 goals",
+        attendees=["bob@example.com"],
+        send_updates="all",  # Send email invitation to Bob
+        access_token=alice_token
+    )
+    data = json.loads(result)
+    if "id" in data:
+        event_id = data["id"]
+        print(f"   ✅ Event created: {data['summary']} (ID: {event_id})")
+        print(f"      Attendees: {[a['email'] for a in data.get('attendees', [])]}")
+    
+    # Bob lists his events (should see the invitation)
+    print("\n5. Bob lists his events...")
+    result = await list_events(
+        max_results=5,
+        access_token=bob_token
+    )
+    data = json.loads(result)
+    if "items" in data:
+        print(f"   ✅ Bob has {len(data['items'])} event(s)")
+        for event in data["items"][:3]:
+            summary = event.get("summary", "Untitled")
+            start = event.get("start", {}).get("dateTime", event.get("start", {}).get("date"))
+            print(f"      - {summary} ({start})")
+    
+    # Bob creates his own event
+    print("\n6. Bob creates his own event...")
+    result = await create_event(
+        summary="Bob's Personal Task",
+        start_datetime="2025-10-27T10:00:00",
+        end_datetime="2025-10-27T11:00:00",
+        description="Work on project",
+        access_token=bob_token
+    )
+    data = json.loads(result)
+    if "id" in data:
+        print(f"   ✅ Event created: {data['summary']}")
+    
+    print("\n" + "=" * 70)
+    print("Multi-User Demo Completed!")
+    print("=" * 70)
+    print("\nKey Points:")
+    print("  • Each user has their own access_token")
+    print("  • Tokens are passed as parameters to each tool call")
+    print("  • No need to set USER_ACCESS_TOKEN environment variable")
+    print("  • Perfect for multi-user AI agent scenarios")
+    print("=" * 70)
+
+
+async def demo_single_user_with_env():
+    """Demonstrate single-user usage with environment variable"""
+    
+    print("\n" + "=" * 70)
+    print("Calendar MCP Server - Single-User Demo (Environment Variable)")
+    print("=" * 70)
+    
+    # Check if USER_ACCESS_TOKEN is set
+    if not os.getenv("USER_ACCESS_TOKEN"):
+        print("\n⚠️  USER_ACCESS_TOKEN not set!")
+        print("   Set it with: export USER_ACCESS_TOKEN='your_token'")
+        print("   Skipping single-user demo...")
+        return
+    
+    print("\n1. Listing calendars (using env token)...")
+    result = await list_calendars()  # No access_token parameter
+    data = json.loads(result)
+    if "items" in data:
+        print(f"   ✅ Found {len(data['items'])} calendar(s)")
+    
+    print("\n2. Listing events (using env token)...")
+    result = await list_events(max_results=5)  # No access_token parameter
+    data = json.loads(result)
+    if "items" in data:
+        print(f"   ✅ Found {len(data['items'])} event(s)")
+    
+    print("\n" + "=" * 70)
+    print("Single-User Demo Completed!")
+    print("=" * 70)
+
+
+if __name__ == "__main__":
+    print("\n" + "=" * 70)
+    print("CALENDAR MCP SERVER - AUTHENTICATION MODES")
+    print("=" * 70)
+    
+    # Run multi-user demo (recommended)
+    asyncio.run(demo_multi_user())
+    
+    # Run single-user demo if env var is set
+    asyncio.run(demo_single_user_with_env())
+