types

type CallToolRequest struct {
	Name      string      `json:"name"`
	Arguments interface{} `json:"arguments,omitempty"`
}

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

type CallToolsBatchRequest struct {
	Tools []ToolCall `json:"tools"`
}

type CallToolsResponse struct {
	Results []CallToolResponse `json:"results"`
}

type CancelRequest struct {
	RequestID interface{} `json:"requestId"`
}

type ClientCapabilities struct {
	Sampling     *SamplingCapability    `json:"sampling,omitempty"`
	Roots        *RootsCapability       `json:"roots,omitempty"`
	Elicitation  *ElicitationCapability `json:"elicitation,omitempty"`
	Experimental map[string]interface{} `json:"experimental,omitempty"`
}

type ClientDSL struct {
	ID        string        `json:"id,omitempty"`      // The ID of the MCP Client (required)
	Name      string        `json:"name"`              // The name of the MCP Client (required)
	Version   string        `json:"version,omitempty"` // The version of the MCP Client
	Type      string        `json:"type,omitempty"`    // The type of the MCP Client (e.g., "standard", "agent", "system")
	Transport TransportType `json:"transport"`         // One of the TransportType (required)

	types.MetaInfo

	Dependencies map[string]string `json:"dependencies,omitempty"` // Dependencies on other MCP Clients (name -> version constraint)

	// Client capabilities configuration
	EnableSampling    bool `json:"enable_sampling,omitempty"`    // Enable sampling capability
	EnableRoots       bool `json:"enable_roots,omitempty"`       // Enable roots capability
	RootsListChanged  bool `json:"roots_list_changed,omitempty"` // Whether to be notified of root changes
	EnableElicitation bool `json:"enable_elicitation,omitempty"` // Enable elicitation capability

	// For HTTP, SSE transport
	URL                string `json:"url,omitempty"`                 // For HTTP、SSE transport (optional)
	Endpoint           string `json:"endpoint,omitempty"`            // API endpoint path (e.g., "/api/mcp")
	AuthorizationToken string `json:"authorization_token,omitempty"` // For HTTP、SSE transport

	// For stdio transport
	Command   string            `json:"command,omitempty"`   // for stdio transport
	Arguments []string          `json:"arguments,omitempty"` // for stdio transport
	Env       map[string]string `json:"env,omitempty"`       // for stdio transport

	Timeout string `json:"timeout,omitempty"` // for HTTP、SSE transport (1s, 1m, 1h, 1d)

	// For process transport - mapping data (tools, prompts, resources)
	Tools     map[string]string `json:"tools,omitempty"`     // tool_name -> process_name mapping
	Resources map[string]string `json:"resources,omitempty"` // resource_name -> process_name mapping
	Prompts   map[string]string `json:"prompts,omitempty"`   // prompt_name -> process_name mapping
}

type ClientInfo struct {
	ID           string            `json:"id,omitempty"`           // Client ID
	Name         string            `json:"name"`                   // Client name
	Version      string            `json:"version,omitempty"`      // Client version
	Type         string            `json:"type,omitempty"`         // Client type (standard, agent, system)
	Transport    TransportType     `json:"transport,omitempty"`    // Transport type
	Label        string            `json:"label,omitempty"`        // Display label
	Description  string            `json:"description,omitempty"`  // Description
	Tags         []string          `json:"tags,omitempty"`         // Tags for search and filtering
	Dependencies map[string]string `json:"dependencies,omitempty"` // Dependencies (name -> version constraint)
}

type Config struct {
	ServerCommand []string               `json:"serverCommand,omitempty"`
	ServerEnv     map[string]string      `json:"serverEnv,omitempty"`
	InitOptions   map[string]interface{} `json:"initOptions,omitempty"`
	Timeout       time.Duration          `json:"timeout,omitempty"`
}

type ConnectionOptions struct {
	// Headers for HTTP/SSE transports (e.g., Mcp-Session-Id, Custom-Auth, etc.)
	Headers map[string]string `json:"headers,omitempty"`
	// Timeout for connection establishment
	Timeout time.Duration `json:"timeout,omitempty"`
	// Retry configuration
	MaxRetries int           `json:"max_retries,omitempty"`
	RetryDelay time.Duration `json:"retry_delay,omitempty"`
}

type ConnectionState string

type ElicitationCapability struct{}

type EmbeddedResource struct {
	URI      string `json:"uri"`
	MimeType string `json:"mimeType"`
	Text     string `json:"text,omitempty"`
	Blob     string `json:"blob,omitempty"` // base64-encoded binary data
}

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

type ErrorHandler func(ctx context.Context, err error) error

type Event struct {
	Type string      `json:"type"`
	Data interface{} `json:"data"`
}

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

type GetPromptResponse struct {
	Description string          `json:"description,omitempty"`
	Messages    []PromptMessage `json:"messages"`
}

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

type InitializeRequest struct {
	ProtocolVersion string                 `json:"protocolVersion"`
	Capabilities    ClientCapabilities     `json:"capabilities"`
	ClientInfo      ClientInfo             `json:"clientInfo"`
	Meta            map[string]interface{} `json:"meta,omitempty"`
}

type InitializeResponse struct {
	ProtocolVersion string                 `json:"protocolVersion"`
	Capabilities    ServerCapabilities     `json:"capabilities"`
	ServerInfo      ServerInfo             `json:"serverInfo"`
	Meta            map[string]interface{} `json:"meta,omitempty"`
}

type ListPromptsRequest struct {
	Cursor string `json:"cursor,omitempty"`
}

type ListPromptsResponse struct {
	Prompts    []Prompt `json:"prompts"`
	NextCursor string   `json:"nextCursor,omitempty"`
}

type ListResourcesRequest struct {
	Cursor string `json:"cursor,omitempty"`
}

type ListResourcesResponse struct {
	Resources  []Resource `json:"resources"`
	NextCursor string     `json:"nextCursor,omitempty"`
}

type ListSamplesResponse struct {
	Samples []SampleData `json:"samples"`
	Total   int          `json:"total"`
}

type ListToolsRequest struct {
	Cursor string `json:"cursor,omitempty"`
}

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

type LogLevel string

type LogMessage struct {
	Level  LogLevel               `json:"level"`
	Data   interface{}            `json:"data"`
	Logger string                 `json:"logger,omitempty"`
	Meta   map[string]interface{} `json:"meta,omitempty"`
}

type LoggingCapability struct{}

type MappingData struct {
	Tools     map[string]*ToolSchema     `json:"tools"`     // tool_name -> ToolSchema
	Resources map[string]*ResourceSchema `json:"resources"` // resource_name -> ResourceSchema
	Prompts   map[string]*PromptSchema   `json:"prompts"`   // prompt_name -> PromptSchema
}

type Message struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      interface{} `json:"id,omitempty"`
	Method  string      `json:"method,omitempty"`
	Params  interface{} `json:"params,omitempty"`
	Result  interface{} `json:"result,omitempty"`
	Error   *Error      `json:"error,omitempty"`
}

type NotificationHandler func(ctx context.Context, notification Message) error

type Progress struct {
	Token uint64 `json:"token"`
	Total uint64 `json:"total,omitempty"`
}

type ProgressNotification struct {
	Token    uint64 `json:"token"`
	Progress uint64 `json:"progress"`
	Total    uint64 `json:"total,omitempty"`
}

type Prompt struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	Arguments   []PromptArgument       `json:"arguments,omitempty"`
	Meta        map[string]interface{} `json:"meta,omitempty"`
}

type PromptArgument struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	Required    bool   `json:"required,omitempty"`
}

type PromptContent struct {
	Type PromptContentType `json:"type"`

	// Text content
	Text string `json:"text,omitempty"`

	// Image content
	Data     string `json:"data,omitempty"`     // base64-encoded data for image/audio
	MimeType string `json:"mimeType,omitempty"` // MIME type for image/audio

	// Resource content
	Resource *EmbeddedResource `json:"resource,omitempty"`
}

type PromptContentType string

type PromptMessage struct {
	Role    PromptRole    `json:"role"`
	Content PromptContent `json:"content"`
}

type PromptRole string

type PromptSchema struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	Template    string                 `json:"template"` // Prompt template content
	Arguments   []PromptArgument       `json:"arguments,omitempty"`
	Meta        map[string]interface{} `json:"meta,omitempty"`
}

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

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

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

type RequestHandler func(ctx context.Context, request Message) (interface{}, error)

type Resource struct {
	URI         string                 `json:"uri"`
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	MimeType    string                 `json:"mimeType,omitempty"`
	Meta        map[string]interface{} `json:"meta,omitempty"`
}

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

type ResourceParameter struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	Required    bool   `json:"required,omitempty"`
}

type ResourceSchema struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	Process     string                 `json:"process"` // Yao process name
	URI         string                 `json:"uri"`     // URI template (e.g., "customer://{id}")
	MimeType    string                 `json:"mimeType,omitempty"`
	Parameters  []ResourceParameter    `json:"parameters,omitempty"` // URI parameters
	Meta        map[string]interface{} `json:"meta,omitempty"`
	ProcessArgs []string               `json:"x-process-args,omitempty"` // Mapping from URI/parameters to Process positional args
}

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

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

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

type SampleData struct {
	Index    int    `json:"index"`               // Auto-generated index (set by loader)
	Name     string `json:"name,omitempty"`      // Optional sample name
	ItemName string `json:"item_name,omitempty"` // Tool/Resource name (set by loader)

	// Multi-purpose fields (semantics depend on item type)
	Input  map[string]interface{} `json:"input,omitempty"`  // For tools: input args; For resources: parsed from URI
	Output interface{}            `json:"output,omitempty"` // For tools: expected output; Not used for resources
	URI    string                 `json:"uri,omitempty"`    // For resources: full URI with params
	Data   interface{}            `json:"data,omitempty"`   // For resources: response data

	Metadata  map[string]interface{} `json:"metadata,omitempty"`  // Optional metadata (e.g., description)
	Timestamp string                 `json:"timestamp,omitempty"` // Optional timestamp
}

type SampleItemType string

type SamplingCapability struct{}

type SamplingContent struct {
	Type     string `json:"type"`
	Text     string `json:"text,omitempty"`
	ImageURL string `json:"imageUrl,omitempty"`
}

type SamplingMessage struct {
	Role    string          `json:"role"`
	Content SamplingContent `json:"content"`
}

type SamplingRequest struct {
	Model          string                 `json:"model"`
	Messages       []SamplingMessage      `json:"messages"`
	SystemPrompt   string                 `json:"systemPrompt,omitempty"`
	IncludeContext string                 `json:"includeContext,omitempty"`
	Temperature    float64                `json:"temperature,omitempty"`
	MaxTokens      int                    `json:"maxTokens,omitempty"`
	StopSequences  []string               `json:"stopSequences,omitempty"`
	Meta           map[string]interface{} `json:"meta,omitempty"`
}

type SamplingResponse struct {
	Model      string          `json:"model"`
	Role       string          `json:"role"`
	Content    SamplingContent `json:"content"`
	StopReason string          `json:"stopReason,omitempty"`
}

type ServerCapabilities struct {
	Resources    *ResourcesCapability   `json:"resources,omitempty"`
	Tools        *ToolsCapability       `json:"tools,omitempty"`
	Prompts      *PromptsCapability     `json:"prompts,omitempty"`
	Logging      *LoggingCapability     `json:"logging,omitempty"`
	Experimental map[string]interface{} `json:"experimental,omitempty"`
}

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

type SetLogLevelRequest struct {
	Level LogLevel `json:"level"`
}

type Tool struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	InputSchema json.RawMessage        `json:"inputSchema"`
	Meta        map[string]interface{} `json:"meta,omitempty"`
}

type ToolCall struct {
	Name      string      `json:"name"`
	Arguments interface{} `json:"arguments,omitempty"`
}

type ToolContent struct {
	Type ToolContentType `json:"type"`

	// Text content
	Text string `json:"text,omitempty"`

	// Image content (base64 encoded)
	Data     string `json:"data,omitempty"`     // base64-encoded image data
	MimeType string `json:"mimeType,omitempty"` // MIME type for image

	// Resource content
	Resource *EmbeddedResource `json:"resource,omitempty"`
}

type ToolContentType string

type ToolSchema struct {
	Name         string          `json:"name"`
	Description  string          `json:"description,omitempty"`
	Process      string          `json:"process"`                  // Yao process name
	InputSchema  json.RawMessage `json:"inputSchema"`              // JSON Schema for input
	OutputSchema json.RawMessage `json:"outputSchema,omitempty"`   // Optional JSON Schema for output
	ProcessArgs  []string        `json:"x-process-args,omitempty"` // Mapping from MCP arguments to Process positional args
}

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

type Transport interface {
	Start(ctx context.Context) error
	Stop() error
	Send(message Message) error
	Receive() (<-chan Message, error)
	Close() error
}

type TransportType string