mcp

type AgentTool struct {
	Type     string `json:"type"`
	Function struct {
		Name        string      `json:"name"`
		Description string      `json:"description"`
		Parameters  interface{} `json:"parameters"`
	} `json:"function"`
}

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

type EnvVarTemplate struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Required    bool   `json:"required"`
	Secret      bool   `json:"secret"`  // Should be masked in display
	Default     string `json:"default"` // Default value if any
}

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

type MCPConfig struct {
	Enabled      bool                       `json:"enabled"`
	Servers      map[string]MCPServerConfig `json:"servers"`
	AutoStart    bool                       `json:"auto_start"`
	AutoDiscover bool                       `json:"auto_discover"`
	Timeout      time.Duration              `json:"timeout"`
}

type MCPContent struct {
	Type        string                 `json:"type"`
	Text        string                 `json:"text,omitempty"`
	Data        string                 `json:"data,omitempty"`
	MimeType    string                 `json:"mimeType,omitempty"`
	Annotations map[string]interface{} `json:"annotations,omitempty"`
}

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

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

type MCPManager interface {
	// AddServer adds a new MCP server
	AddServer(config MCPServerConfig) error

	// RemoveServer removes an MCP server
	RemoveServer(name string) error

	// GetServer gets an MCP server by name
	GetServer(name string) (MCPServer, bool)

	// ListServers lists all registered servers
	ListServers() []MCPServer

	// StartAll starts all configured servers
	StartAll(ctx context.Context) error

	// StopAll stops all running servers
	StopAll(ctx context.Context) error

	// GetAllTools gets all tools from all running servers
	GetAllTools(ctx context.Context) ([]MCPTool, error)

	// CallTool calls a tool on the appropriate server
	CallTool(ctx context.Context, serverName, toolName string, args map[string]interface{}) (*MCPToolCallResult, error)
}

type MCPMessage 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   *MCPError   `json:"error,omitempty"`
}

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

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

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

type MCPServer interface {
	// Start starts the MCP server process
	Start(ctx context.Context) error

	// Stop stops the MCP server process
	Stop(ctx context.Context) error

	// IsRunning checks if the server is running
	IsRunning() bool

	// GetName returns the server name
	GetName() string

	// GetConfig returns the server configuration
	GetConfig() MCPServerConfig

	// Initialize sends initialize request to server
	Initialize(ctx context.Context) error

	// ListTools lists available tools from the server
	ListTools(ctx context.Context) ([]MCPTool, error)

	// CallTool calls a tool on the server
	CallTool(ctx context.Context, request MCPToolCallRequest) (*MCPToolCallResult, error)

	// ListResources lists available resources from the server
	ListResources(ctx context.Context) ([]MCPResource, error)

	// ReadResource reads a resource from the server
	ReadResource(ctx context.Context, uri string) (*MCPContent, error)

	// ListPrompts lists available prompts from the server
	ListPrompts(ctx context.Context) ([]MCPPrompt, error)

	// GetPrompt gets a prompt from the server
	GetPrompt(ctx context.Context, name string, args map[string]interface{}) (*MCPContent, error)
}

type MCPServerConfig struct {
	Name        string            `json:"name"`
	Type        string            `json:"type,omitempty"`    // "stdio" or "http"
	Command     string            `json:"command,omitempty"` // For stdio servers
	Args        []string          `json:"args,omitempty"`    // For stdio servers
	URL         string            `json:"url,omitempty"`     // For HTTP servers
	Env         map[string]string `json:"env,omitempty"`
	WorkingDir  string            `json:"working_dir,omitempty"` // For stdio servers
	Timeout     time.Duration     `json:"timeout,omitempty"`
	AutoStart   bool              `json:"auto_start"`
	MaxRestarts int               `json:"max_restarts"`
}

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

type MCPServerTemplate struct {
	ID          string           `json:"id"`
	Name        string           `json:"name"`
	Description string           `json:"description"`
	Type        string           `json:"type"`     // "stdio", "http"
	URL         string           `json:"url"`      // For HTTP servers
	Command     string           `json:"command"`  // For stdio servers
	Args        []string         `json:"args"`     // For stdio servers
	EnvVars     []EnvVarTemplate `json:"env_vars"` // Required environment variables
	Timeout     time.Duration    `json:"timeout"`
	Features    []string         `json:"features"`  // List of capabilities
	AuthType    string           `json:"auth_type"` // "bearer", "basic", "none"
	Docs        string           `json:"docs"`      // Documentation URL
}

type MCPTool struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	InputSchema map[string]interface{} `json:"inputSchema"`
	ServerName  string                 `json:"server_name"`
}

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

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

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

type Parameters struct {
	Args   []string
	Kwargs map[string]interface{}
}

type Result struct {
	Success       bool                   `json:"success"`
	Output        interface{}            `json:"output"`
	Errors        []string               `json:"errors"`
	Metadata      map[string]interface{} `json:"metadata"`
	ExecutionTime time.Duration          `json:"execution_time"`
}