Merge pull request #70 from tjhop/feat/prom-management-endpoints

feat/prom management endpoints

Author: tjhop

README.md

@@ -60,14 +60,18 @@ Please see [Flags](#command-line-flags) for more information on the available fl
 | `docs_search` | Search the markdown files containing official Prometheus documentation from the prometheus/docs repo |
 | `exemplars_query` | Performs a query for exemplars by the given query and time range |
 | `flags` | Get runtime flags |
+| `healthy` | Management API endpoint that can be used to check Prometheus health |
 | `label_names` | Returns the unique label names present in the block in sorted order by given time range and matchers |
 | `label_values` | Performs a query for the values of the given label, time range and matchers |
 | `list_alerts` | List all active alerts |
 | `list_rules` | List all alerting and recording rules that are loaded |
 | `list_targets` | Get overview of Prometheus target discovery |
 | `metric_metadata` | Returns metadata about metrics currently scraped by the metric name | 
 | `query` | Execute an instant query against the Prometheus datasource |
+| `quit` | Management API endpoint that can be used to trigger a graceful shutdown of Prometheus |
 | `range_query` | Execute a range query against the Prometheus datasource |
+| `ready` | Management API endpoint that can be used to check Prometheus is ready to serve traffic (i.e. respond to queries |
+| `reload` | Management API endpoint that can be used to trigger a reload of the Prometheus configuration and rule files |
 | `runtime_info` | Get Prometheus runtime information |
 | `series` | Finds series by label matchers |
 | `targets_metadata` | Returns metadata about metrics currently scraped by the target |
@@ -144,6 +148,8 @@ Qualifications and support criteria are still under consideration, please open a
 | [`thanos`](https://thanos.io/) | `config` | remove | Thanos does not use a centralized config, so it doesn't implement the endpoint and the tool returns a `404`. |
 | [`thanos`](https://thanos.io/) | `wal_replay_status` | remove | Thanos does not implement the endpoint and the tool returns a `404`. |
 | [`thanos`](https://thanos.io/) | `list_stores` | add | Thanos provides an additional endpoint to list store API servers. |
+| [`thanos`](https://thanos.io/) | `reload` | remove | Thanos does not implement the endpoint and the tool returns a `404`. |
+| [`thanos`](https://thanos.io/) | `quit` | remove | Thanos does not implement the endpoint and the tool returns a `404`. |
 
 ### Resources
 

pkg/mcp/promapi.go

@@ -2,13 +2,11 @@ package mcp
 
 import (
 	"context"
-	"encoding/json"
 	"fmt"
 	"net/http"
 	"strings"
 	"time"
 
-	"github.com/alpkeskin/gotoon"
 	promv1 "github.com/prometheus/client_golang/api/prometheus/v1"
 	"github.com/prometheus/client_golang/prometheus"
 
@@ -57,25 +55,6 @@ func NewAPIClient(prometheusUrl string, rt http.RoundTripper) (promv1.API, error
 	return client, nil
 }
 
-func toonOrJsonOutput(ctx context.Context, data any) (string, error) {
-	toonEnabled := getToonOutputFromContext(ctx)
-	if toonEnabled {
-		toonEncodedData, err := gotoon.Encode(data)
-		if err != nil {
-			return "", fmt.Errorf("error TOON encoding data: %w", err)
-		}
-
-		return toonEncodedData, nil
-	}
-
-	jsonEncodedData, err := json.Marshal(data)
-	if err != nil {
-		return "", fmt.Errorf("error marshaling to JSON: %w", err)
-	}
-
-	return string(jsonEncodedData), nil
-}
-
 func alertmanagersApiCall(ctx context.Context) (string, error) {
 	client, err := getApiClientFromContext(ctx)
 	if err != nil {

pkg/mcp/prometheus_tools.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"math"
+	"net/http"
 	"path"
 	"strings"
 	"time"
@@ -206,6 +207,23 @@ var (
 		),
 	)
 
+	// Tools for Prometheus Management API.
+	prometheusHealthyTool = mcp.NewTool("healthy",
+		mcp.WithDescription("Management API endpoint that can be used to check Prometheus health."),
+	)
+
+	prometheusReadyTool = mcp.NewTool("ready",
+		mcp.WithDescription("Management API endpoint that can be used to check Prometheus is ready to serve traffic (i.e. respond to queries.)"),
+	)
+
+	prometheusReloadTool = mcp.NewTool("reload",
+		mcp.WithDescription("Management API endpoint that can be used to trigger a reload of the Prometheus configuration and rule files."),
+	)
+
+	prometheusQuitTool = mcp.NewTool("quit",
+		mcp.WithDescription("Management API endpoint that can be used to trigger a graceful shutdown of Prometheus."),
+	)
+
 	// Tools for Prometheus documentation.
 	prometheusDocsListTool = mcp.NewTool("docs_list",
 		mcp.WithDescription("List of Official Prometheus Documentation Files."),
@@ -700,3 +718,43 @@ func prometheusDocsSearchToolHandler(ctx context.Context, request mcp.CallToolRe
 
 	return toolRes, nil
 }
+
+func doPrometheusManagementApiCall(ctx context.Context, method, path string) (*mcp.CallToolResult, error) {
+	client, err := getApiClientFromContext(ctx)
+	if err != nil {
+		return mcp.NewToolResultError("error getting prometheus api client from context: " + err.Error()), nil
+	}
+	ctx, cancel := context.WithTimeout(ctx, apiTimeout)
+	defer cancel()
+
+	data, err := doHttpRequest(ctx, method, client.roundtripper, client.url, path, false)
+	if err != nil {
+		return mcp.NewToolResultError(fmt.Sprintf("error making Prometheus Management API call to %s: %s", path, err.Error())), nil
+	}
+
+	return mcp.NewToolResultText(strings.Trim(data, "\\n\"")), nil
+}
+
+const (
+	mgmtApiEndpointPrefix  = "/-/"
+	mgmtApiHealthyEndpoint = mgmtApiEndpointPrefix + "healthy"
+	mgmtApiReadyEndpoint   = mgmtApiEndpointPrefix + "ready"
+	mgmtApiReloadEndpoint  = mgmtApiEndpointPrefix + "reload"
+	mgmtApiQuitEndpoint    = mgmtApiEndpointPrefix + "quit"
+)
+
+func prometheusHealthyToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	return doPrometheusManagementApiCall(ctx, http.MethodGet, mgmtApiHealthyEndpoint)
+}
+
+func prometheusReadyToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	return doPrometheusManagementApiCall(ctx, http.MethodGet, mgmtApiReadyEndpoint)
+}
+
+func prometheusReloadToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	return doPrometheusManagementApiCall(ctx, http.MethodPost, mgmtApiReloadEndpoint)
+}
+
+func prometheusQuitToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	return doPrometheusManagementApiCall(ctx, http.MethodPost, mgmtApiQuitEndpoint)
+}