mcp

type AuthConfig struct {
	Type     string `json:"type"` // api_key, oauth2, basic
	APIKey   string `json:"api_key,omitempty"`
	Username string `json:"username,omitempty"`
	Password string `json:"password,omitempty"`
}

type Client interface {
	ID() string
	Name() string
	Status() ClientStatus
	Connect(ctx context.Context) error
	Disconnect(ctx context.Context) error
	ListTools(ctx context.Context) ([]Tool, error)
	CallTool(ctx context.Context, name string, args map[string]any) (*ToolResult, error)
	ListResources(ctx context.Context) ([]Resource, error)
	ReadResource(ctx context.Context, uri string) (*ResourceContent, error)
	ListPrompts(ctx context.Context) ([]Prompt, error)
	GetPrompt(ctx context.Context, name string, args map[string]any) (*PromptResult, error)
}

type ClientConfig struct {
	ID         string            `json:"id"`
	Name       string            `json:"name"`
	Transport  TransportType     `json:"transport"`
	Command    string            `json:"command,omitempty"` // stdio only
	Args       []string          `json:"args,omitempty"`    // stdio only
	URL        string            `json:"url,omitempty"`     // sse/websocket
	Env        map[string]string `json:"env,omitempty"`
	AutoAuth   bool              `json:"auto_auth,omitempty"`
	AuthConfig *AuthConfig       `json:"auth,omitempty"`
}

type ClientStatus string

type MCPManager interface {
	AddClient(ctx context.Context, config ClientConfig) (Client, error)
	RemoveClient(ctx context.Context, clientID string) error
	GetClient(ctx context.Context, clientID string) (Client, error)
	ListClients(ctx context.Context) ([]Client, error)
}

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

type Prompt struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
}

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

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

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

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

type ToolResult struct {
	Content string `json:"content"`
	IsError bool   `json:"is_error"`
}

type TransportType string