from contextlib import AbstractContextManager
from .cliente_mcp import ClienteMCP


class SessaoMCP(AbstractContextManager["SessaoMCP"]):
    """Garante o ciclo de vida de uma conexão MCP durante uma leitura."""

    def __init__(self, cliente: ClienteMCP) -> None:
        self.cliente = cliente

    def __enter__(self) -> "SessaoMCP":
        self.cliente.conectar()
        return self

    def __exit__(self, tipo, valor, traceback) -> None:
        self.cliente.desconectar()