mcp-nomad

// File: tools/deployments.go package tools import ( "context" "encoding/json" "log" "github.com/kocierik/mcp-nomad/utils" "github.com/mark3labs/mcp-go/mcp" "github.com/mark3labs/mcp-go/server" ) // RegisterDeploymentTools registers all deployment-related tools func RegisterDeploymentTools(s *server.MCPServer, nomadClient *utils.NomadClient, logger *log.Logger) { // List deployments tool listDeploymentsTool := mcp.NewTool("list_deployments", mcp.WithDescription("List all deployments"), mcp.WithString("namespace", mcp.Description("The namespace to list deployments from (default: default)"), ), ) s.AddTool(listDeploymentsTool, ListDeploymentsHandler(nomadClient, logger)) // Get deployment tool getDeploymentTool := mcp.NewTool("get_deployment", mcp.WithDescription("Get deployment details by ID"), mcp.WithString("deployment_id", mcp.Required(), mcp.Description("The ID of the deployment to retrieve"), ), ) s.AddTool(getDeploymentTool, GetDeploymentHandler(nomadClient, logger)) } // ListDeploymentsHandler returns a handler for listing deployments func ListDeploymentsHandler(client *utils.NomadClient, logger *log.Logger) func(context.Context, mcp.CallToolRequest) (*mcp.CallToolResult, error) { return func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { arguments, ok := request.Params.Arguments.(map[string]interface{}) if !ok { return mcp.NewToolResultError("Invalid arguments"), nil } namespace := "default" if ns, ok := arguments["namespace"].(string); ok && ns != "" { namespace = ns } deployments, err := client.ListDeployments(namespace) if err != nil { logger.Printf("Error listing deployments: %v", err) return mcp.NewToolResultErrorFromErr("Failed to list deployments", err), nil } deploymentsJSON, err := json.MarshalIndent(deployments, "", " ") if err != nil { return mcp.NewToolResultErrorFromErr("Failed to format deployments", err), nil } return mcp.NewToolResultText(string(deploymentsJSON)), nil } } // GetDeploymentHandler returns a handler for getting deployment details func GetDeploymentHandler(client *utils.NomadClient, logger *log.Logger) func(context.Context, mcp.CallToolRequest) (*mcp.CallToolResult, error) { return func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { arguments, ok := request.Params.Arguments.(map[string]interface{}) if !ok { return mcp.NewToolResultError("Invalid arguments"), nil } deploymentID, ok := arguments["deployment_id"].(string) if !ok || deploymentID == "" { return mcp.NewToolResultError("deployment_id is required"), nil } deployment, err := client.GetDeployment(deploymentID) if err != nil { logger.Printf("Error getting deployment: %v", err) return mcp.NewToolResultErrorFromErr("Failed to get deployment", err), nil } deploymentJSON, err := json.MarshalIndent(deployment, "", " ") if err != nil { return mcp.NewToolResultErrorFromErr("Failed to format deployment", err), nil } return mcp.NewToolResultText(string(deploymentJSON)), nil } }