mcp

type CallToolParams struct {
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments,omitempty"`
}

type CallToolResult struct {
	Content []ContentItem `json:"content"`
	IsError bool          `json:"isError,omitempty"`
}

type Client struct {
	// contains filtered or unexported fields
}

type ClientCapabilities struct {
	Roots    *RootsCapability    `json:"roots,omitempty"`
	Sampling *SamplingCapability `json:"sampling,omitempty"`
}

type ClientConfig struct {
	Name    string // 客户端名称
	Version string // 客户端版本
}

type ClientInfo struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type ConfigFile struct {
	MCPServers []ServerConfig `json:"mcpServers"`
}

type ContentItem struct {
	Type     string `json:"type"`               // "text", "image", "resource"
	Text     string `json:"text,omitempty"`     // type=text 时
	MimeType string `json:"mimeType,omitempty"` // type=image/resource 时
	Data     string `json:"data,omitempty"`     // base64 编码的数据
}

type InitializeParams struct {
	ProtocolVersion string             `json:"protocolVersion"`
	Capabilities    ClientCapabilities `json:"capabilities"`
	ClientInfo      ClientInfo         `json:"clientInfo"`
}

type InitializeResult struct {
	ProtocolVersion string             `json:"protocolVersion"`
	Capabilities    ServerCapabilities `json:"capabilities"`
	ServerInfo      ServerInfo         `json:"serverInfo"`
}

type ListToolsResult struct {
	Tools      []MCPTool `json:"tools"`
	NextCursor string    `json:"nextCursor,omitempty"`
}

type LoggingCapability struct{}

type MCPTool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	InputSchema map[string]any `json:"inputSchema"`
}

type Manager struct {
	// contains filtered or unexported fields
}

type PromptsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type RPCError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Data    any    `json:"data,omitempty"`
}

type Request struct {
	JSONRPC string `json:"jsonrpc"`
	ID      int    `json:"id"`
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
}

type ResourcesCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type Response struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      int             `json:"id"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *RPCError       `json:"error,omitempty"`
}

type RootsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type SSEConfig struct {
	// URL SSE 端点地址，如 http://localhost:8080/sse
	URL string
	// Headers 自定义 HTTP 请求头（如认证头）
	Headers map[string]string
	// HTTPClient 自定义 HTTP 客户端（可选）
	HTTPClient *http.Client
}

type SSETransport struct {
	// contains filtered or unexported fields
}

type SamplingCapability struct{}

type ServerCapabilities struct {
	Tools     *ToolsCapability     `json:"tools,omitempty"`
	Resources *ResourcesCapability `json:"resources,omitempty"`
	Prompts   *PromptsCapability   `json:"prompts,omitempty"`
	Logging   *LoggingCapability   `json:"logging,omitempty"`
}

type ServerConfig struct {
	Name      string          `json:"name"`      // 服务器名称（用于日志和工具名前缀）
	Transport TransportConfig `json:"transport"` // 传输配置
	Enabled   bool            `json:"enabled"`   // 是否启用（默认 true）
	Prefix    string          `json:"prefix"`    // 工具名前缀（默认用 Name）
	Tags      []string        `json:"tags"`      // 额外标签
}

type ServerInfo struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type StdioConfig struct {
	Command string   // 可执行文件路径
	Args    []string // 命令参数
	Env     []string // 额外环境变量（格式 KEY=VALUE）
	WorkDir string   // 工作目录
}

type StdioTransport struct {
	// contains filtered or unexported fields
}

type ToolsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type Transport interface {
	// Connect 建立连接
	Connect(ctx context.Context) error
	// Send 发送数据（一帧）
	Send(data []byte) error
	// Recv 接收数据（一帧）
	Recv() ([]byte, error)
	// Close 关闭连接
	Close() error
}

type TransportConfig struct {
	Type    string   `json:"type"`    // "stdio" 或 "sse"
	Command string   `json:"command"` // stdio: 命令路径
	Args    []string `json:"args"`    // stdio: 命令参数
	Env     []string `json:"env"`     // stdio: 环境变量
	WorkDir string   `json:"work_dir"`

	// SSE 传输配置
	URL     string            `json:"url"`     // sse: SSE 端点地址
	Headers map[string]string `json:"headers"` // sse: 自定义请求头（如认证）
}