mcp

type AuthFlow struct {
	Server  string // MCP server name
	State   string // opaque anti-CSRF token, also the registry key
	AuthURL string // URL to open in the browser
	// contains filtered or unexported fields
}

type CallResult struct {
	Output  string
	IsError bool
}

type OAuthConfig struct {
	ClientID     string   `json:"client_id"`
	ClientSecret string   `json:"client_secret,omitempty"`
	Scopes       []string `json:"scopes,omitempty"`
	AuthURL      string   `json:"auth_url,omitempty"`
	TokenURL     string   `json:"token_url,omitempty"`
	// RedirectPort pins the loopback callback port (0 = an ephemeral free port).
	RedirectPort int `json:"redirect_port,omitempty"`
}

type Option func(*options)

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

type ServerConfig struct {
	// Name is a unique identifier used to prefix tool names: mcp__<name>__<tool>.
	Name string `json:"name"`
	// Type is the transport: "stdio" (default) or "url".
	Type string `json:"type,omitempty"`
	// Command and Args are used for stdio servers.
	Command string   `json:"command,omitempty"`
	Args    []string `json:"args,omitempty"`
	// Env adds extra environment variables for the stdio child process.
	Env map[string]string `json:"env,omitempty"`
	// URL is used for HTTP/SSE servers.
	URL string `json:"url,omitempty"`
	// Headers are sent with every HTTP request (e.g. Authorization).
	// Values of the form "${VAR}" are expanded from the environment at connect time.
	Headers map[string]string `json:"headers,omitempty"`
	// OAuth, when set on a url server, enables an OAuth 2.0 authorization-code
	// flow: the access token is injected as an Authorization: Bearer header and
	// refreshed automatically. Endpoints are auto-discovered (RFC 9728 + RFC
	// 8414) unless AuthURL/TokenURL are set explicitly.
	OAuth *OAuthConfig `json:"oauth,omitempty"`
	// AllowedTools, when non-empty, restricts which tools from this server are
	// exposed to the LLM. Unlisted tools are silently dropped after tools/list.
	AllowedTools []string `json:"allowed_tools,omitempty"`
	// RequireConfirmation makes every tool call from this server require explicit
	// user approval via the standard confirm_request / thread.confirm flow.
	RequireConfirmation bool `json:"require_confirmation,omitempty"`
	// Enabled gates whether the server is connected at all. A nil pointer (field
	// omitted from settings.json) means enabled — the opt-out default — so
	// existing configs keep working. The MCP tab's Space toggle writes this.
	Enabled *bool `json:"enabled,omitempty"`
}

type ServerStatus struct {
	Name      string
	Type      string // "stdio" or "url"
	Connected bool
	ToolCount int
	Error     string
}

type TokenStore interface {
	Load(server string) (*oauth2.Token, error)
	Save(server string, tok *oauth2.Token) error
	Delete(server string) error
}

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