package tool

import (
	"context"

	"github.com/mark3labs/mcp-go/mcp"
	"github.com/mark3labs/mcp-go/server"
)

// Tool 是所有 MCP 工具必须实现的接口。
type Tool interface {
	Name() string
	Description() string
	Register(mcpServer *server.MCPServer) error
	Initialize(ctx context.Context) error
	Shutdown(ctx context.Context) error
	HealthCheck(ctx context.Context) error
}

// getArgs 把 req.Params.Arguments 转换为 map[string]any。
func getArgs(req mcp.CallToolRequest) map[string]any {
	if args, ok := req.Params.Arguments.(map[string]any); ok {
		return args
	}
	return map[string]any{}
}