import sys
import asyncio
import argparse
import requests
from mcp.client.session import ClientSession
from mcp.client.stdio import stdio_client
from mcp.client.sse import sse_client

async def main():
    parser = argparse.ArgumentParser(description="MCP Remote Proxy Client")
    parser.add_argument("--url", default="http://dietpi.tail706a7a.ts.net:8001/sse", help="URL do servidor MCP SSE")
    args = parser.parse_args()

    print(f"Conectando ao servidor MCP remoto: {args.url}", file=sys.stderr)
    
    try:
        async with sse_client(args.url) as streams:
            async with ClientSession(streams[0], streams[1]) as session:
                await session.initialize()
                print("Conectado com sucesso! Aguardando comandos da IA...", file=sys.stderr)
                # Mantém a conexão aberta e repassa STDIO para SSE
                # Nota: A biblioteca MCP lida com o repasse automaticamente dentro do session
                await asyncio.Future() # Roda para sempre
    except Exception as e:
        print(f"Erro na conexão: {e}", file=sys.stderr)
        sys.exit(1)

if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        pass