mcp

type AudioContent struct {
	// Must be "audio"
	Type string `json:"type"`
	// The audio data in base64 format
	Data string `json:"data"`
	// The MIME type of the audio. e.g., "audio/wav", "audio/mpeg"
	MimeType string `json:"mimeType"`
}

type BaseRequestParams struct {
	// Meta information for the request
	Meta RequestMeta `json:"_meta"`
}

type CallToolParams struct {
	BaseRequestParams
	// The name of the tool to call
	Name string `json:"name"`
	// The arguments to pass to the tool
	Arguments json.RawMessage `json:"arguments"`
}

type CallToolRequest struct {
	JSONRPCRequest
	Params CallToolParams `json:"params"`
}

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

type CapabilitiesInfo struct {
	Tools             []MCPTool             `json:"tools"`
	Prompts           []MCPPrompt           `json:"prompts"`
	Resources         []MCPResource         `json:"resources"`
	ResourceTemplates []MCPResourceTemplate `json:"resourceTemplates"`
	LastSynced        time.Time             `json:"lastSynced"`
	ServerInfo        ImplementationSchema  `json:"serverInfo"`
}

type ClientCapabilitiesSchema struct {
	Experimental map[string]any        `json:"experimental"`
	Sampling     map[string]any        `json:"sampling"`
	Roots        RootsCapabilitySchema `json:"roots"`
}

type Content interface {
	// GetType returns the type of the content
	GetType() string
}

type ExperimentalCapabilitySchema struct {
}

type ImageContent struct {
	// Must be "image"
	Type string `json:"type"`
	// The image data in base64 format
	Data string `json:"data"`
	// The MIME type of the image. e.g., "image/png", "image/jpeg"
	MimeType string `json:"mimeType"`
}

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

type InitializeRequestParams struct {
	BaseRequestParams
	// The latest version of the Model Context Protocol that the client supports
	ProtocolVersion string `json:"protocolVersion"`
	// Client capabilities
	Capabilities ClientCapabilitiesSchema `json:"capabilities"`
	// Client implementation information
	ClientInfo ImplementationSchema `json:"clientInfo"`
}

type InitializeRequestSchema struct {
	JSONRPCRequest
}

type InitializeResult struct {
	JSONRPCBaseResult
	Result InitializedResult `json:"result"`
}

type InitializedNotification struct {
	JSONRPCRequest
}

type InitializedResult struct {
	// The version of the Model Context Protocol that the server wants to use
	ProtocolVersion string `json:"protocolVersion"`
	// Server capabilities
	Capabilities ServerCapabilitiesSchema `json:"capabilities"`
	// Server implementation information
	ServerInfo ImplementationSchema `json:"serverInfo"`
	// Instructions describing how to use the server and its features
	Instructions string `json:"instructions"`
}

type JSONRPCBaseResult struct {
	JSONRPC string `json:"jsonrpc"`
	ID      any    `json:"id"`
}

type JSONRPCError struct {
	// The error type that occurred
	Code int `json:"code"`
	// A short description of the error
	Message string `json:"message"`
	// Additional information about the error
	Data any `json:"data,omitempty"`
}

type JSONRPCErrorSchema struct {
	JSONRPCBaseResult
	Error JSONRPCError `json:"error"`
}

type JSONRPCNotification struct {
	JSONRPCBaseResult
	Method string          `json:"method"`
	Params json.RawMessage `json:"params"`
}

type JSONRPCRequest struct {
	// JSONRPC version, must be "2.0"
	JSONRPC string `json:"jsonrpc"`
	// A uniquely identifying ID for a request in JSON-RPC
	Id any `json:"id"`
	// The method to be invoked
	Method string `json:"method"`
	// The parameters to be passed to the method
	Params json.RawMessage `json:"params"`
}

type JSONRPCResponse struct {
	JSONRPCBaseResult
	Result any `json:"result"`
}

type ListToolsResult struct {
	Tools []ToolSchema `json:"tools"`
}

type LoggingCapabilitySchema struct {
}

type MCPPrompt struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Arguments   []PromptArgumentSchema `json:"arguments"`
	LastSynced  time.Time              `json:"lastSynced"`
}

type MCPResource struct {
	Uri         string    `json:"uri"`
	Name        string    `json:"name"`
	Description string    `json:"description"`
	MimeType    string    `json:"mimeType"`
	LastSynced  time.Time `json:"lastSynced"`
}

type MCPResourceTemplate struct {
	UriTemplate string    `json:"uriTemplate"`
	Name        string    `json:"name"`
	Description string    `json:"description"`
	MimeType    string    `json:"mimeType"`
	LastSynced  time.Time `json:"lastSynced"`
}

type MCPTool struct {
	Name        string           `json:"name"`
	Description string           `json:"description"`
	InputSchema ToolInputSchema  `json:"inputSchema"`
	Annotations *ToolAnnotations `json:"annotations,omitempty"`
	Enabled     bool             `json:"enabled"`
	LastSynced  time.Time        `json:"lastSynced"`
}

type PingRequest struct {
	JSONRPCRequest
}

type PromptArgumentSchema struct {
	Name        string `json:"name" yaml:"name"`
	Description string `json:"description" yaml:"description"`
	Required    bool   `json:"required" yaml:"required"`
}

type PromptResponseContentSchema struct {
	Type string `json:"type" yaml:"type"`
	Text string `json:"text" yaml:"text"`
}

type PromptResponseSchema struct {
	Role    string                      `json:"role" yaml:"role"`
	Content PromptResponseContentSchema `json:"content" yaml:"content"`
}

type PromptSchema struct {
	Name           string                 `json:"name" yaml:"name"`
	Description    string                 `json:"description" yaml:"description"`
	Arguments      []PromptArgumentSchema `json:"arguments" yaml:"arguments"`
	PromptResponse []PromptResponseSchema `json:"promptResponse,omitempty" yaml:"promptResponse,omitempty"`
}

type PromptsCapabilitySchema struct {
	ListChanged bool `json:"listChanged"`
}

type RequestMeta struct {
	// Progress token for tracking request progress
	// Can be string or number
	ProgressToken any `json:"progressToken"`
}

type ResourcesCapabilitySchema struct {
	Subscribe   bool `json:"subscribe"`
	ListChanged bool `json:"listChanged"`
}

type RootsCapabilitySchema struct {
	ListChanged bool `json:"listChanged"`
}

type ServerCapabilitiesSchema struct {
	Experimental ExperimentalCapabilitySchema `json:"experimental"`
	Logging      LoggingCapabilitySchema      `json:"logging"`
	Prompts      PromptsCapabilitySchema      `json:"prompts"`
	Resources    ResourcesCapabilitySchema    `json:"resources"`
	Tools        ToolsCapabilitySchema        `json:"tools"`
}

type TextContent struct {
	// Must be "text"
	Type string `json:"type"`
	// The text content
	Text string `json:"text"`
}

type ToolAnnotations struct {
	DestructiveHint bool `json:"destructiveHint,omitempty"`
	IdempotentHint  bool `json:"idempotentHint,omitempty"`
	OpenWorldHint   bool `json:"openWorldHint,omitempty"`
	ReadOnlyHint    bool `json:"readOnlyHint,omitempty"`
	// A human-readable title for the tool.
	Title string `json:"title,omitempty"`
}

type ToolInputSchema struct {
	Type       string         `json:"type"`
	Properties map[string]any `json:"properties"`
	Required   []string       `json:"required,omitempty"`
	Title      string         `json:"title"`
	Enum       []any          `json:"enum,omitempty"`
}

type ToolSchema struct {
	// The name of the tool
	Name string `json:"name"`
	// A human-readable description of the tool
	Description string `json:"description"`
	// A JSON Schema object defining the expected parameters for the tool
	InputSchema ToolInputSchema `json:"inputSchema"`
	// Annotations for the tool
	Annotations *ToolAnnotations `json:"annotations,omitempty"`
	// Meta information for the tool
	// https://github.com/modelcontextprotocol/python-sdk/blob/main/src/mcp/types.py
	Meta map[string]any `json:"_meta,omitempty"`
}

type ToolsCapabilitySchema struct {
	ListChanged bool `json:"listChanged"`
}