main.py

from __future__ import annotations

from mcp.server.fastmcp import FastMCP

from controllers.prometheus_controller import PrometheusController


# Controller de domínio (Prometheus)
controller = PrometheusController()

# Instância do servidor FastMCP
mcp = FastMCP(
    "prometheus-agent-mcp",
    json_response=True
)


@mcp.tool()
def promql_execute(
    promql: str,
    range: bool = False,
    start: str | None = None,
    end: str | None = None,
    step: str | None = None,
) -> dict:
    """
    Handler para a ferramenta promql_execute do MCP
    
    Args:
        promql: Query PromQL
        range: Se True, executa query_range
        start: Data/hora de início (ISO format)
        end: Data/hora de fim (ISO format)
        step: Intervalo em segundos
        
    Returns:
        Resposta formatada no padrão MCP
    """
    return controller.execute_query(
        promql=promql,
        range_query=range,
        start=start,
        end=end,
        step=step
    )


if __name__ == "__main__":
    # Executa via STDIO para integrar com clientes MCP (ex.: Cursor)
    mcp.run(transport="stdio")