mcp

type AgentClient interface {
	Name() string
	Initialize(ctx context.Context) error
	RefreshTools(ctx context.Context) error
	Tools() []Tool
	CallTool(ctx context.Context, name string, arguments map[string]any) (*ToolCallResult, error)
	IsInitialized() bool
	ServerInfo() ServerInfo
}

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

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

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

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

type Content struct {
	Type string `json:"type"`
	Text string `json:"text,omitempty"`
}

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

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

type HealthStatus struct {
	Healthy     bool      // Whether the server is responding to pings
	LastCheck   time.Time // When the last health check ran
	LastHealthy time.Time // When the server was last seen healthy
	Error       string    // Error message if unhealthy (empty when healthy)
}

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

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

type InputSchemaObject struct {
	Type       string              `json:"type"`
	Properties map[string]Property `json:"properties,omitempty"`
	Required   []string            `json:"required,omitempty"`
}

type MCPPrompt struct {
	Name        string           `json:"name"`
	Description string           `json:"description,omitempty"`
	Arguments   []PromptArgument `json:"arguments,omitempty"`
}

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

type MCPServerConfig struct {
	Name            string
	Transport       Transport
	Endpoint        string               // For HTTP/SSE transport
	ContainerID     string               // For Docker Stdio transport
	External        bool                 // True for external URL servers (no container)
	LocalProcess    bool                 // True for local process servers (no container)
	SSH             bool                 // True for SSH servers (remote process over SSH)
	OpenAPI         bool                 // True for OpenAPI-based servers
	Command         []string             // For local process or SSH transport
	WorkDir         string               // For local process transport
	Env             map[string]string    // For local process or SSH transport
	SSHHost         string               // SSH hostname (for SSH servers)
	SSHUser         string               // SSH username (for SSH servers)
	SSHPort         int                  // SSH port (for SSH servers, 0 = default 22)
	SSHIdentityFile string               // SSH identity file path (for SSH servers)
	OpenAPIConfig   *OpenAPIClientConfig // OpenAPI configuration (for OpenAPI servers)
	Tools           []string             // Tool whitelist (empty = all tools)
}

type MCPServerStatus struct {
	Name         string     `json:"name"`
	Transport    Transport  `json:"transport"`
	Endpoint     string     `json:"endpoint,omitempty"`
	ContainerID  string     `json:"containerId,omitempty"`
	Initialized  bool       `json:"initialized"`
	ToolCount    int        `json:"toolCount"`
	Tools        []string   `json:"tools"`
	External     bool       `json:"external"`              // True for external URL servers
	LocalProcess bool       `json:"localProcess"`          // True for local process servers
	SSH          bool       `json:"ssh"`                   // True for SSH servers
	SSHHost      string     `json:"sshHost,omitempty"`     // SSH hostname
	OpenAPI      bool       `json:"openapi"`               // True for OpenAPI servers
	OpenAPISpec  string     `json:"openapiSpec,omitempty"` // OpenAPI spec location
	Healthy      *bool      `json:"healthy,omitempty"`     // Health check result (nil if not yet checked)
	LastCheck    *time.Time `json:"lastCheck,omitempty"`   // When last health check ran
	HealthError  string     `json:"healthError,omitempty"` // Error message if unhealthy
}

type OpenAPIClient struct {
	ClientBase
	// contains filtered or unexported fields
}

type OpenAPIClientConfig struct {
	Spec       string   // URL or local file path to OpenAPI spec
	BaseURL    string   // Override server URL from spec
	AuthType   string   // "bearer" or "header"
	AuthToken  string   // Resolved bearer token (from env)
	AuthHeader string   // Header name for header-based auth
	AuthValue  string   // Resolved header value (from env)
	Include    []string // Operation IDs to include
	Exclude    []string // Operation IDs to exclude
	NoExpand   bool     // If true, skip environment variable expansion in spec file
}

type OpenAPIOperation struct {
	Method       string
	Path         string
	PathParams   []string // Parameter names in path order (always required)
	QueryParams  map[string]*openapi3.Parameter
	HeaderParams map[string]*openapi3.Parameter
	RequestBody  *openapi3.RequestBodyRef
}

type Pingable interface {
	Ping(ctx context.Context) error
}

type ProcessClient struct {
	RPCClient
	// contains filtered or unexported fields
}

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

type PromptArgumentData struct {
	Name        string
	Description string
	Required    bool
	Default     string
}

type PromptData struct {
	Name        string
	Description string
	Content     string
	Arguments   []PromptArgumentData
}

type PromptMessage struct {
	Role    string  `json:"role"` // "user" or "assistant"
	Content Content `json:"content"`
}

type PromptProvider interface {
	ListPromptData() []PromptData
	GetPromptData(name string) (*PromptData, error)
}

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

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

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

type PromptsListResult struct {
	Prompts []MCPPrompt `json:"prompts"`
}

type Property struct {
	Type        string   `json:"type"`
	Description string   `json:"description,omitempty"`
	Enum        []string `json:"enum,omitempty"`
	Default     any      `json:"default,omitempty"`
}

type RPCClient struct {
	ClientBase
	// contains filtered or unexported fields
}

type Reconnectable interface {
	Reconnect(ctx context.Context) error
}

type ResourceContents struct {
	URI      string `json:"uri"`
	MimeType string `json:"mimeType,omitempty"`
	Text     string `json:"text,omitempty"`
}

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

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

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

type ResourcesReadResult struct {
	Contents []ResourceContents `json:"contents"`
}

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

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

type SSESession struct {
	ID        string
	AgentName string // Agent identity for tool access control
	Writer    http.ResponseWriter
	Flusher   http.Flusher
	Done      chan struct{}
	MessageID atomic.Int64
}

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

type Session struct {
	ID          string
	ClientInfo  ClientInfo
	Initialized bool
	CreatedAt   time.Time
	LastSeen    time.Time
}

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

type StdioClient struct {
	RPCClient
	// contains filtered or unexported fields
}

type Tool struct {
	Name        string          `json:"name"`
	Title       string          `json:"title,omitempty"`
	Description string          `json:"description,omitempty"`
	InputSchema json.RawMessage `json:"inputSchema"`
}

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

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

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

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

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

type Transport string