tools.go

package main

import (
	"last9-mcp/internal/models"
	"last9-mcp/internal/telemetry/logs"
	"last9-mcp/internal/telemetry/traces"
	"net/http"
	"time"

	"github.com/acrmp/mcp"
	"golang.org/x/time/rate"
)

// createTools creates the MCP tool definitions with appropriate rate limits
func createTools(cfg models.Config) ([]mcp.ToolDefinition, error) {
	client := &http.Client{
		Timeout: 30 * time.Second,
	}

	return []mcp.ToolDefinition{
		{
			Metadata: mcp.Tool{
				Name:        "get_exceptions",
				Description: ptr(traces.GetExceptionsDescription),
				InputSchema: mcp.ToolInputSchema{
					Type: "object",
					Properties: mcp.ToolInputSchemaProperties{
						"limit": map[string]any{
							"type":        "integer",
							"description": "Maximum number of exceptions to return",
							"default":     20,
						},
						"start_time_iso": map[string]any{
							"type":        "string",
							"description": "Start time in ISO format (YYYY-MM-DD HH:MM:SS)",
						},
						"end_time_iso": map[string]any{
							"type":        "string",
							"description": "End time in ISO format (YYYY-MM-DD HH:MM:SS)",
						},
						"span_name": map[string]any{
							"type":        "string",
							"description": "Name of the span to filter by",
						},
					},
				},
			},
			Execute:   traces.NewGetExceptionsHandler(client, cfg),
			RateLimit: rate.NewLimiter(rate.Limit(cfg.RequestRateLimit), cfg.RequestRateBurst),
		},
		{
			Metadata: mcp.Tool{
				Name:        "get_service_graph",
				Description: ptr(traces.GetServiceGraphDescription),
				InputSchema: mcp.ToolInputSchema{
					Type: "object",
					Properties: mcp.ToolInputSchemaProperties{
						"span_name": map[string]any{
							"type":        "string",
							"description": "Name of the span to get dependencies for",
						},
						"lookback_minutes": map[string]any{
							"type":        "integer",
							"description": "Number of minutes to look back",
							"default":     60,
						},
						"start_time_iso": map[string]any{
							"type":        "string",
							"description": "Start time in ISO format (YYYY-MM-DD HH:MM:SS)",
						},
					},
				},
			},
			Execute:   traces.NewGetServiceGraphHandler(client, cfg),
			RateLimit: rate.NewLimiter(rate.Limit(cfg.RequestRateLimit), cfg.RequestRateBurst),
		},
		{
			Metadata: mcp.Tool{
				Name:        "get_logs",
				Description: ptr(logs.GetLogsDescription),
				InputSchema: mcp.ToolInputSchema{
					Type: "object",
					Properties: mcp.ToolInputSchemaProperties{
						"service": map[string]any{
							"type":        "string",
							"description": "Name of the service to get logs for",
						},
						"severity": map[string]any{
							"type":        "string",
							"description": "Severity of the logs to get",
						},
						"start_time_iso": map[string]any{
							"type":        "string",
							"description": "Start time in ISO format (YYYY-MM-DD HH:MM:SS)",
						},
						"end_time_iso": map[string]any{
							"type":        "string",
							"description": "End time in ISO format (YYYY-MM-DD HH:MM:SS)",
						},
						"limit": map[string]any{
							"type":        "integer",
							"description": "Maximum number of exceptions to return",
							"default":     20,
						},
					},
				},
			},
			Execute:   logs.NewGetLogsHandler(client, cfg),
			RateLimit: rate.NewLimiter(rate.Limit(cfg.RequestRateLimit), cfg.RequestRateBurst),
		},
		{
			Metadata: mcp.Tool{
				Name:        "get_drop_rules",
				Description: ptr(logs.GetDropRulesDescription),
				InputSchema: mcp.ToolInputSchema{
					Type:       "object",
					Properties: mcp.ToolInputSchemaProperties{},
				},
			},
			Execute:   logs.NewGetDropRulesHandler(client, cfg),
			RateLimit: rate.NewLimiter(rate.Limit(cfg.RequestRateLimit), cfg.RequestRateBurst),
		},
		{
			Metadata: mcp.Tool{
				Name:        "add_drop_rule",
				Description: ptr(logs.AddDropRuleDescription),
				InputSchema: mcp.ToolInputSchema{
					Type: "object",
					Properties: mcp.ToolInputSchemaProperties{
						"name": map[string]any{
							"type":        "string",
							"description": "Name of the drop rule",
						},
						"filters": map[string]any{
							"type": "array of map[string]string",
							"description": `List of filter conditions.
								e.g.
									"filters": [
										{
											"key": "attributes[\"logtag\"]",
											"value": "P",
											"operator": "equals",
											"conjunction": "and"
										},
										{
											"key": "attributes[\"logtag\"]",
											"value": "F",
											"operator": "not_equals",
											"conjunction": "and"
										}
									]
								regex pattern and filtering on body or message is not supported as of now, will be added in a future version`,
							"items": map[string]any{
								"type": "map[string]string",
								"properties": map[string]any{
									"key": map[string]any{
										"type": "string",
										"description": `The key to filter on. Must be properly escaped with double quotes.
														For resource attributes, use format: "resource.attributes[\"key_name\"]" 
														Example: "resource.attributes[\"service.name\"]"
														For log attributes, use format: "attributes[\"key_name\"]"
														Example: "attributes[\"logtag\"]"`,
									},
									"value": map[string]any{
										"type":        "string",
										"description": "Filter value to match",
									},
									"operator": map[string]any{
										"type":        "string",
										"description": "The comparison operator for the filter condition. Must be one of: [equals, not_equals]. The request will be rejected if any other operator is specified.",
									},
									"conjunction": map[string]any{
										"type":        "string",
										"description": "The logical operator used to combine multiple filter conditions. Must be one of: [and]. The request will be rejected if any other conjunction is specified.",
									},
								},
							},
						},
					},
				},
			},
			Execute:   logs.NewAddDropRuleHandler(client, cfg),
			RateLimit: rate.NewLimiter(rate.Limit(cfg.RequestRateLimit), cfg.RequestRateBurst),
		},
	}, nil
}

// ptr returns a pointer to the provided string
func ptr(s string) *string {
	return &s
}