mcp

type AuthServerMetadata struct {
	Issuer                            string   `json:"issuer"`
	AuthorizationEndpoint             string   `json:"authorization_endpoint"`
	TokenEndpoint                     string   `json:"token_endpoint"`
	RegistrationEndpoint              string   `json:"registration_endpoint,omitempty"`
	ScopesSupported                   []string `json:"scopes_supported,omitempty"`
	ResponseTypesSupported            []string `json:"response_types_supported,omitempty"`
	GrantTypesSupported               []string `json:"grant_types_supported,omitempty"`
	CodeChallengeMethodsSupported     []string `json:"code_challenge_methods_supported,omitempty"`
	TokenEndpointAuthMethodsSupported []string `json:"token_endpoint_auth_methods_supported,omitempty"`
}

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

type Client interface {
	// Initialize sends the MCP initialize handshake and returns the server's
	// instructions string (empty if none) for injection into the system prompt.
	Initialize(ctx context.Context) (string, error)
	// ListTools fetches the server's tool list.
	ListTools(ctx context.Context) ([]ToolDef, error)
	// CallTool invokes a tool and returns its result.
	CallTool(ctx context.Context, name string, input json.RawMessage) (CallResult, error)
	// ListResources fetches the server's resource list (MCP resources/list).
	ListResources(ctx context.Context) ([]ResourceDef, error)
	// ReadResource reads the contents of one resource (MCP resources/read).
	ReadResource(ctx context.Context, uri string) ([]ResourceContent, error)
	// Close shuts down the transport.
	Close() error
}

type ClientRegistration struct {
	ClientID                string   `json:"client_id"`
	ClientSecret            string   `json:"client_secret,omitempty"`
	ClientIDIssuedAt        int64    `json:"client_id_issued_at,omitempty"`
	ClientSecretExpiresAt   int64    `json:"client_secret_expires_at,omitempty"`
	RedirectURIs            []string `json:"redirect_uris,omitempty"`
	TokenEndpointAuthMethod string   `json:"token_endpoint_auth_method,omitempty"`
}

type ConnectedServer struct {
	Name         string
	Config       ServerConfig
	Status       ServerStatus
	Disabled     bool // true when the server is in disabledMcpServers
	Tools        []ToolDef
	Instructions string // server-provided instructions from initialize response
	Error        string // set when Status == StatusFailed
	// contains filtered or unexported fields
}

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

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

type McpJSON struct {
	McpServers map[string]ServerConfig `json:"mcpServers"`
}

type NamedTool struct {
	ServerName    string
	Prefix        string
	Def           ToolDef
	QualifiedName string
}

type OAuthTokens struct {
	AccessToken  string    `json:"access_token"`
	RefreshToken string    `json:"refresh_token,omitempty"`
	TokenType    string    `json:"token_type"`
	ExpiresAt    time.Time `json:"expires_at"`
	Scope        string    `json:"scope,omitempty"`
	// Embedded so token refresh can re-auth without another DCR roundtrip.
	Client        ClientRegistration `json:"client"`
	TokenEndpoint string             `json:"token_endpoint"`
}

type ProtectedResourceMetadata struct {
	Resource             string   `json:"resource"`
	AuthorizationServers []string `json:"authorization_servers"`
	ScopesSupported      []string `json:"scopes_supported,omitempty"`
}

type ResourceContent struct {
	URI      string `json:"uri"`
	MimeType string `json:"mimeType,omitempty"`
	Text     string `json:"text,omitempty"`
	Blob     string `json:"blob,omitempty"` // base64
}

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

type ServerConfig struct {
	// Type is "stdio" | "sse" | "http". Empty means "stdio".
	Type string `json:"type,omitempty"`

	// Stdio fields
	Command string            `json:"command,omitempty"`
	Args    []string          `json:"args,omitempty"`
	Env     map[string]string `json:"env,omitempty"`

	// SSE / HTTP fields
	URL     string            `json:"url,omitempty"`
	Headers map[string]string `json:"headers,omitempty"`

	// Source is the config file path this entry was loaded from.
	// Not serialized — set at load time by LoadConfigs.
	Source string `json:"-"`

	// Scope is "user" | "local" | "project" | "plugin".
	// Not serialized — set at load time by LoadConfigs.
	Scope string `json:"-"`

	// PluginName is set when the server was defined by an installed plugin.
	// Not serialized.
	PluginName string `json:"-"`
}

type ServerStatus string

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