mcp

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

type ClientMetadata struct {
	Status            ClientStatus `json:"status"`
	Initialized       bool         `json:"initialized"`
	ToolNameMode      ToolNameMode `json:"tool_name_mode"`
	LastError         string       `json:"last_error,omitempty"`
	LastInitializedAt *time.Time   `json:"last_initialized_at,omitempty"`
	ServerInfo        *ServerInfo  `json:"server_info,omitempty"`
	// ProtocolVersion is the MCP version agreed during initialize handshake.
	ProtocolVersion string `json:"protocol_version,omitempty"`
}

type ClientStatus string

type ConfigLoadResult struct {
	Servers map[string]ScopedMcpServerConfig
	Errors  []ValidationError
}

type ConfigScope string

type ConnectServerConfig struct {
	// Name is the server name
	Name string `json:"name"`

	// Command is the command to start the server (for stdio)
	Command string `json:"command,omitempty"`

	// Args are the command arguments (for stdio)
	Args []string `json:"args,omitempty"`

	// URL is the server URL (for http)
	URL string `json:"url,omitempty"`

	// Transport is the transport type ("stdio" or "http")
	Transport string `json:"transport"`

	// Env are environment variables
	Env map[string]string `json:"env,omitempty"`

	// Timeout is the timeout in seconds
	Timeout int `json:"timeout"`

	// Headers are HTTP headers (for http transport)
	Headers map[string]string `json:"headers,omitempty"`
}

type Discovery struct {
	Tools     []Tool     `json:"tools,omitempty"`
	Resources []Resource `json:"resources,omitempty"`
	Prompts   []Prompt   `json:"prompts,omitempty"`
}

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

type HTTPTransportConfig struct {
	// URL is the server URL
	URL string

	// Headers are HTTP headers
	Headers map[string]string

	// Timeout is the request timeout
	Timeout time.Duration
}

type InitializeResult struct {
	// ProtocolVersion is the MCP protocol version
	ProtocolVersion string `json:"protocolVersion"`

	// ServerInfo contains server information
	ServerInfo ServerInfo `json:"serverInfo"`

	// Capabilities are the server capabilities
	Capabilities ServerCapabilities `json:"capabilities"`
}

type IntegrationOptions struct {
	ToolNameMode ToolNameMode `json:"tool_name_mode,omitempty"`
}

type IntegrationResult struct {
	// MCPTools are the wrapped MCP tools (across all servers that succeeded).
	MCPTools []tool.Tool

	// ServerResults holds per-server outcome — one entry per configured server.
	// Inspect this to diagnose which servers loaded and which failed.
	ServerResults []ServerResult

	// Error is set only when zero tools were registered across all configured
	// servers. For partial failures, inspect ServerResults instead.
	Error error
	// contains filtered or unexported fields
}

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

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

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

type JSONRPCResponse struct {
	JSONRPC string         `json:"jsonrpc"`
	ID      int64          `json:"id"`
	Result  map[string]any `json:"result,omitempty"`
	Error   *JSONRPCError  `json:"error,omitempty"`
}

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

type MCPOAuthConfig struct {
	ClientID     string
	ClientSecret string
	AuthURL      string
	TokenURL     string
	Scopes       []string
}

type MCPServerStatus struct {
	Name        string       `json:"name"`
	Status      ClientStatus `json:"status"`
	Scope       ConfigScope  `json:"scope"`
	Type        string       `json:"type"`
	LastError   string       `json:"last_error,omitempty"`
	ConnectedAt time.Time    `json:"connected_at,omitempty"`
	ToolCount   int          `json:"tool_count"`
}

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

type McpJsonConfig struct {
	MCPServers map[string]McpServerConfig `json:"mcpServers,omitempty"`
}

type McpServerConfig struct {
	Type    McpServerType     `json:"type,omitempty"`
	Command string            `json:"command,omitempty"`
	Args    []string          `json:"args,omitempty"`
	Env     map[string]string `json:"env,omitempty"`
	URL     string            `json:"url,omitempty"`
	Headers map[string]string `json:"headers,omitempty"`
	Timeout int               `json:"timeout,omitempty"`
	Scope   ConfigScope       `json:"scope,omitempty"`
}

type McpServerType string

type MethodHandler func(ctx context.Context, request *JSONRPCRequest) (*JSONRPCResponse, error)

type NotificationHandler func(notification *JSONRPCNotification)

type PluginMcpServer struct {
	Name   string
	Source string
	Config McpServerConfig
}

type ProgressCallback func(progress float64, total *float64, message string)

type Prompt struct {
	// Name is the prompt name
	Name string `json:"name"`

	// Description explains the prompt
	Description string `json:"description,omitempty"`

	// Arguments are the prompt arguments
	Arguments []PromptArgument `json:"arguments,omitempty"`
}

type PromptArgument struct {
	// Name is the argument name
	Name string `json:"name"`

	// Description explains the argument
	Description string `json:"description,omitempty"`

	// Required indicates if the argument is required
	Required bool `json:"required"`
}

type PromptMessage struct {
	// Role is the message role
	Role string `json:"role"`

	// Content is the message content
	Content any `json:"content"`
}

type PromptResult struct {
	Messages []PromptMessage `json:"messages,omitempty"`
}

type PromptsCapability struct {
	// ListChanged indicates if the prompts list can change
	ListChanged bool `json:"listChanged,omitempty"`
}

type Resource struct {
	// URI is the resource URI
	URI string `json:"uri"`

	// Name is the resource name
	Name string `json:"name"`

	// Description explains the resource
	Description string `json:"description,omitempty"`

	// MimeType is the resource MIME type
	MimeType string `json:"mimeType,omitempty"`
}

type ResourceContent struct {
	// URI is the resource URI
	URI string `json:"uri"`

	// MimeType is the content type
	MimeType string `json:"mimeType,omitempty"`

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

	// Blob is the binary content (base64)
	Blob string `json:"blob,omitempty"`
}

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

type ResourcesCapability struct {
	// Subscribe indicates if resources can be subscribed to
	Subscribe bool `json:"subscribe,omitempty"`

	// ListChanged indicates if the resources list can change
	ListChanged bool `json:"listChanged,omitempty"`
}

type ScopedMcpServerConfig struct {
	McpServerConfig
	Scope ConfigScope
}

type ServerCapabilities struct {
	// Tools are the tool capabilities
	Tools *ToolsCapability `json:"tools,omitempty"`

	// Resources are the resource capabilities
	Resources *ResourcesCapability `json:"resources,omitempty"`

	// Prompts are the prompt capabilities
	Prompts *PromptsCapability `json:"prompts,omitempty"`
}

type ServerConfig struct {
	// Name is the server name
	Name string `json:"name"`

	// Command is the command to start the server (for stdio)
	Command string `json:"command,omitempty"`

	// Args are the command arguments (for stdio)
	Args []string `json:"args,omitempty"`

	// URL is the server URL (for http/sse/websocket)
	URL string `json:"url,omitempty"`

	// Transport is the transport type
	Transport TransportType `json:"transport"`

	// Env are environment variables for the server process
	Env map[string]string `json:"env,omitempty"`

	// Timeout is the timeout for operations
	Timeout time.Duration `json:"timeout"`

	// Headers are HTTP headers (for http/sse)
	Headers map[string]string `json:"headers,omitempty"`
}

type ServerInfo struct {
	// Name is the server name
	Name string `json:"name"`

	// Version is the server version
	Version string `json:"version"`

	// ProtocolVersion is the MCP protocol version
	ProtocolVersion string `json:"protocolVersion"`
}

type ServerResult struct {
	// Name is the server name from ServerConfig.
	Name string
	// ToolsRegistered is the number of tools successfully registered.
	ToolsRegistered int
	// Error is non-nil if the server could not be connected, initialized, or wrapped.
	Error error
}

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

type StdioTransportConfig struct {
	// Command is the command to run
	Command string

	// Args are the command arguments
	Args []string

	// Env are environment variables
	Env map[string]string

	// Stdin is the stdin to use (for testing)
	Stdin io.Reader

	// Stdout is the stdout to use (for testing)
	Stdout io.Writer

	// Stderr is the stderr to use (for testing)
	Stderr io.Writer
}

type Tool struct {
	// Name is the tool name
	Name string `json:"name"`

	// Description explains what the tool does
	Description string `json:"description"`

	// InputSchema is the JSON schema for the tool input
	InputSchema map[string]any `json:"inputSchema"`
}

type ToolCallResult struct {
	Raw        map[string]any `json:"raw,omitempty"`
	Text       string         `json:"text,omitempty"`
	IsError    bool           `json:"is_error,omitempty"`
	Structured map[string]any `json:"structured,omitempty"`
	Meta       map[string]any `json:"meta,omitempty"`
}

type ToolNameMode string

type ToolServerConfig struct {
	// Name is the server name
	Name string

	// Command is the command to start the server (for stdio)
	Command string

	// Args are the command arguments (for stdio)
	Args []string

	// URL is the server URL (for http)
	URL string

	// Transport is the transport type (stdio or http)
	Transport string

	// Env are environment variables
	Env map[string]string

	// Headers are HTTP headers (for http transport)
	Headers map[string]string
}

type ToolWrapperMetadata struct {
	ServerName string       `json:"server_name"`
	ToolName   string       `json:"tool_name"`
	NameMode   ToolNameMode `json:"name_mode"`
}

type ToolsCapability struct {
	// ListChanged indicates if the tools list can change
	ListChanged bool `json:"listChanged,omitempty"`
}

type Transport interface {
	// Start starts the transport
	Start(ctx context.Context) error

	// Send sends a request
	Send(ctx context.Context, req *JSONRPCRequest) (*JSONRPCResponse, error)

	// SendNotification sends a notification
	SendNotification(ctx context.Context, notification *JSONRPCNotification) error

	// Close closes the transport
	Close() error

	// SetNotificationsHandler sets the handler for incoming notifications
	SetNotificationsHandler(handler NotificationHandler)

	// SetMethodsHandler sets the handler for incoming method calls
	SetMethodsHandler(handler MethodHandler)
}

type TransportType string

type ValidationError struct {
	File       string `json:"file,omitempty"`
	Path       string `json:"path,omitempty"`
	Message    string `json:"message"`
	Suggestion string `json:"suggestion,omitempty"`
	Severity   string `json:"severity"`
}

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