mcp

type Auth struct {
	Type     string            `yaml:"type" json:"type"`         // bearer, basic, api-key
	Token    string            `yaml:"token" json:"token"`       // Bearer token
	Username string            `yaml:"username" json:"username"` // Basic auth username
	Password string            `yaml:"password" json:"password"` // Basic auth password
	APIKey   string            `yaml:"api_key" json:"api_key"`   // API key
	Headers  map[string]string `yaml:"headers" json:"headers"`   // Custom auth headers
}

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

type Config struct {
	Name        string            `yaml:"name" json:"name"`                                   // Server name
	ServerURL   string            `yaml:"url" json:"url"`                                     // HTTP URL or stdio command
	Transport   Transport         `yaml:"transport" json:"transport"`                         // http or stdio
	Auth        *Auth             `yaml:"auth,omitempty" json:"auth,omitempty"`               // Authentication config
	Timeout     time.Duration     `yaml:"timeout" json:"timeout"`                             // Request timeout
	MaxRetries  int               `yaml:"max_retries" json:"max_retries"`                     // Retry attempts
	Headers     map[string]string `yaml:"headers,omitempty" json:"headers,omitempty"`         // Custom headers (HTTP only)
	Enabled     bool              `yaml:"enabled" json:"enabled"`                             // Whether server is enabled
	Tools       []string          `yaml:"tools,omitempty" json:"tools,omitempty"`             // Available tools (optional)
	Description string            `yaml:"description,omitempty" json:"description,omitempty"` // Server description
}

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

type MCPClient interface {
	// Connection management
	Connect(ctx context.Context) error
	Close() error
	Ping(ctx context.Context) error

	// Tool discovery
	ListTools(ctx context.Context) ([]Tool, error)

	// Tool invocation
	CallTool(ctx context.Context, toolName string, args map[string]interface{}) (interface{}, error)

	// Configuration
	Config() *Config
}

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

type MCPRequest struct {
	JSONRPC string                 `json:"jsonrpc"`
	ID      *int                   `json:"id,omitempty"` // Optional for notifications
	Method  string                 `json:"method"`
	Params  map[string]interface{} `json:"params,omitempty"`
}

type MCPResponse struct {
	JSONRPC  string                 `json:"jsonrpc"`
	ID       int                    `json:"id"`
	Result   map[string]interface{} `json:"result,omitempty"`
	Error    *MCPError              `json:"-"` // Handled via UnmarshalJSON
	RawError json.RawMessage        `json:"error,omitempty"`
}

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

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

type Transport string