mcp

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

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

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

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"` // Base64 for binary
}

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

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

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

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

type JSONRPCNotification struct {
	JSONRPC string `json:"jsonrpc"`
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
}

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

type JSONRPCResponse struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      int64           `json:"id"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *JSONRPCError   `json:"error,omitempty"`
}

type MCPResource struct {
	URI         string `json:"uri"`
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	MimeType    string `json:"mimeType,omitempty"`
}

type MCPTool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	InputSchema map[string]any `json:"inputSchema"`
	ServerName  string         `json:"-"` // Set by client
}

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

type RegistryEntry struct {
	// Name is the short name used in DSL (e.g. "filesystem").
	Name string

	// Description briefly explains what the server provides.
	Description string

	// Command is the executable to run.
	Command string

	// Args are default command-line arguments.
	Args []string

	// RequiredEnv lists environment variables that must be set.
	RequiredEnv []string

	// OptionalEnv lists environment variables that are useful but not required.
	OptionalEnv []string

	// BuiltinGo indicates this server has a native Go implementation
	// that runs in-process without requiring Node.js or any external binary.
	BuiltinGo bool
}

type ResourceContent struct {
	URI      string `json:"uri"`
	MimeType string `json:"mimeType,omitempty"`
	Text     string `json:"text,omitempty"`
	Blob     string `json:"blob,omitempty"` // Base64
}

type ResourceReadParams struct {
	URI string `json:"uri"`
}

type ResourceReadResult struct {
	Contents []ResourceContent `json:"contents"`
}

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

type ResourcesListResult struct {
	Resources []MCPResource `json:"resources"`
}

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

type SamplingCapability struct{}

type ServerConfig struct {
	// Name is a human-readable identifier for the server.
	Name string

	// Transport specifies the transport type.
	Transport TransportType

	// Command is the executable for stdio transport.
	Command string

	// Args are command-line arguments for stdio transport.
	Args []string

	// Env are environment variables for stdio transport.
	Env map[string]string

	// URL is the endpoint for HTTP/SSE transport.
	URL string

	// Headers are HTTP headers for HTTP/SSE transport.
	Headers map[string]string

	// Timeout for operations (default: 30s).
	Timeout time.Duration
}

type ServerInfo struct {
	Name            string       `json:"name"`
	Version         string       `json:"version"`
	ProtocolVersion string       `json:"protocolVersion"`
	Capabilities    Capabilities `json:"capabilities"`
}

type StdioTransport 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 ToolsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

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

type Transport interface {
	// Connect establishes the connection.
	Connect(ctx context.Context) error

	// Send sends a JSON-RPC request and returns the result.
	Send(ctx context.Context, method string, params any) (json.RawMessage, error)

	// Close closes the connection.
	Close() error

	// OnNotification registers a handler for server notifications.
	OnNotification(handler func(method string, params json.RawMessage))
}

type TransportType string