mcp

type Client interface {
	// Start initializes the connection and performs MCP handshake.
	Start(ctx context.Context) error

	// ListTools retrieves available tools from the server.
	ListTools(ctx context.Context) ([]ToolInfo, error)

	// CallTool executes a tool on the server.
	CallTool(ctx context.Context, name string, args map[string]any) (*ToolCallResult, error)

	// Notifications returns a channel for server-initiated messages.
	// Returns nil for transports that don't support notifications (stdio).
	Notifications() <-chan Notification

	// Close shuts down the connection.
	Close() error
}

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

type ContentBlock struct {
	Type     string `json:"type"`
	Text     string `json:"text,omitempty"`
	MimeType string `json:"mimeType,omitempty"`
	Data     string `json:"data,omitempty"`
}

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

type Notification struct {
	Method string          `json:"method"`
	Params json.RawMessage `json:"params,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      uint64 `json:"id"`
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
}

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

type ServerConfig struct {
	Name      string            `json:"name"`
	Transport string            `json:"transport"`
	Command   string            `json:"command"`
	Args      []string          `json:"args,omitempty"`
	Env       map[string]string `json:"env,omitempty"`

	// HTTP transport fields
	URL     string            `json:"url,omitempty"`
	Headers map[string]string `json:"headers,omitempty"`
}

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

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

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

type ToolCaller interface {
	CallTool(ctx context.Context, name string, args map[string]any) (*ToolCallResult, error)
}

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

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

type ToolProvider interface {
	ToolCaller
	ListTools(ctx context.Context) ([]ToolInfo, error)
}

type ToolsListResult struct {
	Tools []ToolInfo `json:"tools"`
}