#!/usr/bin/env python3
"""生成 LiveKit Token - 简化版"""
import asyncio
import sys
from dotenv import load_dotenv

load_dotenv('/Users/leo/.hermes/workspace/livekit-agents/.env.dev')

from livekit import api

def generate_token(room: str = "voice-ai-meeting", identity: str = "user1") -> str:
    """生成 LiveKit Token"""
    token = (
        api.AccessToken(
            "devkey", 
            "secret"
        )
        .with_identity(identity)
        .with_name(identity.capitalize())
        .with_grants(api.VideoGrants(
            room_join=True,
            room=room,
            can_publish=True,
            can_subscribe=True,
        ))
        .to_jwt()
    )
    return token


async def main():
    import sys
    if len(sys.argv) > 2:
        room = sys.argv[1]
        identity = sys.argv[2]
    else:
        room = "voice-ai-meeting"
        identity = "user1"
    
    token = generate_token(room, identity)
    
    print(f"Room: {room}")
    print(f"Identity: {identity}")
    print(f"Token: {token}")
    print()
    print(f"Server URL: ws://192.168.31.229:7880")


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