server

type CatalogID string

type ClientInfo struct {
	Name    string
	Version string
}

type ClientNotification struct {
	Server ServerID
	Kind   ClientNotificationKind

	ResourceURI string

	LoggerName   string
	LoggingLevel string
	LogData      any

	Progress float64
	Total    float64
	Message  string
}

type ClientNotificationKind string

type ClientNotificationSink interface {
	OnClientNotification(ctx context.Context, event ClientNotification)
}

type Digest string

type InvokeMCPToolRequest struct {
	Server ServerID `json:"server" required:"true"`
	Body   *InvokeMCPToolRequestBody
}

type InvokeMCPToolRequestBody struct {
	Source           MCPInvocationSource `json:"source"                     required:"true"`
	ToolName         string              `json:"toolName"                   required:"true"`
	ProviderToolName string              `json:"providerToolName,omitempty"`
	ChoiceID         string              `json:"choiceID,omitempty"`
	ToolDigest       string              `json:"toolDigest,omitempty"`

	Arguments map[string]any `json:"arguments,omitempty"`

	ApprovalID    string `json:"approvalID,omitempty"`
	ApprovalToken string `json:"approvalToken,omitempty"`

	ConversationID string `json:"conversationID,omitempty"`
	MessageID      string `json:"messageID,omitempty"`
	ToolUseID      string `json:"toolUseID,omitempty"`

	AppInstanceID string `json:"appInstanceID,omitempty"`
}

type InvokeMCPToolResponse struct {
	Body *InvokeMCPToolResponseBody
}

type InvokeMCPToolResponseBody struct {
	Server ServerID `json:"server"`

	ToolName         string `json:"toolName"`
	ProviderToolName string `json:"providerToolName,omitempty"`

	Content           []MCPContent `json:"content,omitempty"`
	StructuredContent any          `json:"structuredContent,omitempty"`
	IsError           bool         `json:"isError,omitempty"`

	Provenance MCPToolCallProvenance `json:"provenance"`
	App        *MCPToolAppRenderInfo `json:"app,omitempty"`
}

type MCPApprovalDecision string

type MCPApprovalEvaluation struct {
	Decision   MCPApprovalDecision `json:"decision"`
	Reason     string              `json:"reason,omitempty"`
	ApprovalID string              `json:"approvalID,omitempty"`
	Summary    *MCPApprovalSummary `json:"summary,omitempty"`
}

type MCPApprovalResolution string

type MCPApprovalResolutionResult struct {
	ApprovalID string                `json:"approvalID"`
	Resolution MCPApprovalResolution `json:"resolution"`
	Decision   MCPApprovalDecision   `json:"decision"`

	RememberedForSession bool   `json:"rememberedForSession,omitempty"`
	Token                string `json:"token,omitempty"`
	ExpiresAt            string `json:"expiresAt,omitempty"`
}

type MCPApprovalSummary struct {
	Server            ServerID               `json:"server"`
	ServerDisplayName string                 `json:"serverDisplayName,omitempty"`
	Source            MCPInvocationSource    `json:"source"`
	AppInstanceID     string                 `json:"appInstanceID,omitempty"`
	ToolName          string                 `json:"toolName"`
	ToolDigest        string                 `json:"toolDigest,omitempty"`
	Risk              MCPToolRisk            `json:"risk"`
	Arguments         jsonutil.JSONRawString `json:"arguments,omitempty"`
}

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

type MCPCompleteArgumentRequest struct {
	Server ServerID `json:"server" required:"true"`
	Body   *MCPCompleteArgumentRequestBody
}

type MCPCompleteArgumentRequestBody struct {
	RefType       string            `json:"refType"                 required:"true"` // resource | prompt
	Name          string            `json:"name"                    required:"true"`
	ArgumentName  string            `json:"argumentName"            required:"true"`
	ArgumentValue string            `json:"argumentValue,omitempty"`
	Context       map[string]string `json:"context,omitempty"`
}

type MCPCompletionResult struct {
	Values  []string `json:"values,omitempty"`
	Total   int      `json:"total,omitempty"`
	HasMore bool     `json:"hasMore,omitempty"`
}

type MCPContent struct {
	Type MCPContentType `json:"type"`

	Text     string `json:"text,omitempty"`
	Data     []byte `json:"data,omitempty"`
	MIMEType string `json:"mimeType,omitempty"`

	URI         string `json:"uri,omitempty"`
	Name        string `json:"name,omitempty"`
	Title       string `json:"title,omitempty"`
	Description string `json:"description,omitempty"`
	Size        *int64 `json:"size,omitempty"`

	Resource *MCPResourceContents `json:"resource,omitempty"`

	Annotations map[string]any `json:"annotations,omitempty"`
	Meta        map[string]any `json:"_meta,omitempty"`
	Icons       []MCPIcon      `json:"icons,omitempty"`
}

type MCPContentType string

type MCPDiscoveryPageToken struct {
	Server         ServerID `json:"server"`
	SnapshotDigest string   `json:"dig"`
	Kind           string   `json:"k"`
	PageSize       int      `json:"ps"`
	Index          int      `json:"i"`
}

type MCPDiscoverySnapshot struct {
	Server ServerID `json:"server"`

	NegotiatedProtocolVersion string                        `json:"negotiatedProtocolVersion,omitempty"`
	ServerInfo                *MCPImplementationInfo        `json:"serverInfo,omitempty"`
	ServerCapabilities        *MCPServerCapabilitiesSummary `json:"serverCapabilities,omitempty"`
	Instructions              string                        `json:"instructions,omitempty"`

	Tools             []MCPToolCapability      `json:"tools,omitempty"`
	Resources         []MCPResourceRef         `json:"resources,omitempty"`
	ResourceTemplates []MCPResourceTemplateRef `json:"resourceTemplates,omitempty"`
	Prompts           []MCPPromptRef           `json:"prompts,omitempty"`

	Digest   string `json:"digest,omitempty"`
	SyncedAt string `json:"syncedAt,omitempty"`
}

type MCPGetPromptRequest struct {
	Server ServerID `json:"server" required:"true"`
	Body   *MCPGetPromptRequestBody
}

type MCPGetPromptRequestBody struct {
	PromptName string            `json:"promptName"          required:"true"`
	Arguments  map[string]string `json:"arguments,omitempty"`
}

type MCPGetPromptResponse struct {
	Body *MCPGetPromptResponseBody
}

type MCPGetPromptResponseBody struct {
	Server      ServerID           `json:"server"`
	PromptName  string             `json:"promptName"`
	Description string             `json:"description,omitempty"`
	Messages    []MCPPromptMessage `json:"messages,omitempty"`
}

type MCPHTTPAuthMode string

type MCPIcon struct {
	Source   string   `json:"src"`
	MIMEType string   `json:"mimeType,omitempty"`
	Sizes    []string `json:"sizes,omitempty"`
	Theme    string   `json:"theme,omitempty"`
}

type MCPImplementationInfo struct {
	Name    string `json:"name,omitempty"`
	Version string `json:"version,omitempty"`
}

type MCPInvocationSource string

type MCPPromptMessage struct {
	Role    string     `json:"role"`
	Content MCPContent `json:"content"`
}

type MCPPromptRef struct {
	Server      ServerID                         `json:"server"`
	PromptName  string                           `json:"promptName"`
	Title       string                           `json:"title,omitempty"`
	DisplayName string                           `json:"displayName"`
	Description string                           `json:"description,omitempty"`
	Arguments   map[string]MCPArgumentDefinition `json:"arguments,omitempty"`
	Digest      string                           `json:"digest,omitempty"`
}

type MCPReadResourceRequest struct {
	Server ServerID `json:"server" required:"true"`
	Body   *MCPReadResourceRequestBody
}

type MCPReadResourceRequestBody struct {
	URI string `json:"uri" required:"true"`
}

type MCPReadResourceResponse struct {
	Body *MCPReadResourceResponseBody
}

type MCPReadResourceResponseBody struct {
	Server   ServerID     `json:"server"`
	URI      string       `json:"uri"`
	Contents []MCPContent `json:"contents,omitempty"`
}

type MCPResourceContents struct {
	URI      string         `json:"uri"`
	MIMEType string         `json:"mimeType,omitempty"`
	Text     string         `json:"text,omitempty"`
	Blob     []byte         `json:"blob,omitempty"`
	Meta     map[string]any `json:"_meta,omitempty"`
}

type MCPResourceRef struct {
	Server      ServerID       `json:"server"`
	URI         string         `json:"uri"`
	Name        string         `json:"name,omitempty"`
	Title       string         `json:"title,omitempty"`
	DisplayName string         `json:"displayName"`
	Description string         `json:"description,omitempty"`
	MimeType    string         `json:"mimeType,omitempty"`
	Size        int64          `json:"size,omitempty"`
	Annotations map[string]any `json:"annotations,omitempty"`
	Digest      string         `json:"digest,omitempty"`
}

type MCPResourceTemplateRef struct {
	Server      ServerID                         `json:"server"`
	URITemplate string                           `json:"uriTemplate"`
	Name        string                           `json:"name,omitempty"`
	Title       string                           `json:"title,omitempty"`
	DisplayName string                           `json:"displayName"`
	Description string                           `json:"description,omitempty"`
	MimeType    string                           `json:"mimeType,omitempty"`
	Arguments   map[string]MCPArgumentDefinition `json:"arguments,omitempty"`
	Annotations map[string]any                   `json:"annotations,omitempty"`
	Digest      string                           `json:"digest,omitempty"`
}

type MCPRuntimeStdioConfig struct {
	Command          string            `json:"command"`
	Args             []string          `json:"args,omitempty"`
	Env              map[string]string `json:"env,omitempty"`
	StartupTimeoutMS int               `json:"startupTimeoutMS,omitempty"`
}

type MCPRuntimeStreamableHTTPConfig struct {
	URL       string          `json:"url"`
	TimeoutMS int             `json:"timeoutMS,omitempty"`
	AuthMode  MCPHTTPAuthMode `json:"authMode"`

	Headers map[string]string `json:"headers,omitempty"`

	ClientCredentialRef         string `json:"clientCredentialRef,omitempty"`
	ClientIDMetadataDocumentURL string `json:"clientIDMetadataDocumentURL,omitempty"`
}

type MCPServerCapabilitiesSummary struct {
	Tools                bool           `json:"tools,omitempty"`
	ToolsListChanged     bool           `json:"toolsListChanged,omitempty"`
	Resources            bool           `json:"resources,omitempty"`
	ResourcesSubscribe   bool           `json:"resourcesSubscribe,omitempty"`
	ResourcesListChanged bool           `json:"resourcesListChanged,omitempty"`
	Prompts              bool           `json:"prompts,omitempty"`
	PromptsListChanged   bool           `json:"promptsListChanged,omitempty"`
	Completions          bool           `json:"completions,omitempty"`
	Experimental         map[string]any `json:"experimental,omitempty"`
	Extensions           map[string]any `json:"extensions,omitempty"`
}

type MCPServerRuntimeSnapshot struct {
	Server     ServerID        `json:"server"`
	Collection CatalogID       `json:"collection"`
	Status     MCPServerStatus `json:"status"`

	NegotiatedProtocolVersion string                        `json:"negotiatedProtocolVersion,omitempty"`
	ServerInfo                *MCPImplementationInfo        `json:"serverInfo,omitempty"`
	ServerCapabilities        *MCPServerCapabilitiesSummary `json:"serverCapabilities,omitempty"`
	Instructions              string                        `json:"instructions,omitempty"`

	LastError       string `json:"lastError,omitempty"`
	LastConnectedAt string `json:"lastConnectedAt,omitempty"`
	LastSyncedAt    string `json:"lastSyncedAt,omitempty"`

	ToolCount             int `json:"toolCount"`
	ResourceCount         int `json:"resourceCount"`
	ResourceTemplateCount int `json:"resourceTemplateCount"`
	PromptCount           int `json:"promptCount"`

	SnapshotDigest string `json:"snapshotDigest,omitempty"`
}

type MCPServerStatus string

type MCPTaskSupport string

type MCPToolAnnotations struct {
	DestructiveHint *bool  `json:"destructiveHint,omitempty"`
	IdempotentHint  bool   `json:"idempotentHint"`
	OpenWorldHint   *bool  `json:"openWorldHint,omitempty"`
	ReadOnlyHint    bool   `json:"readOnlyHint"`
	Title           string `json:"title,omitempty"`
}

type MCPToolAppInfo struct {
	ResourceURI string   `json:"resourceUri,omitempty"`
	Visibility  []string `json:"visibility,omitempty"`
}

type MCPToolAppRenderInfo struct {
	ResourceURI       string       `json:"resourceUri,omitempty"`
	MimeType          string       `json:"mimeType,omitempty"`
	Content           []MCPContent `json:"content,omitempty"`
	StructuredContent any          `json:"structuredContent,omitempty"`
	IsError           bool         `json:"isError,omitempty"`
}

type MCPToolCallProvenance struct {
	Server     ServerID
	Collection CatalogID

	ServerDisplayName string `json:"serverDisplayName,omitempty"`

	ToolName         string `json:"toolName"`
	ProviderToolName string `json:"providerToolName"`
	ToolDigest       string `json:"toolDigest,omitempty"`
	ChoiceID         string `json:"choiceID,omitempty"`

	ToolUseID  string `json:"toolUseID,omitempty"`
	ApprovalID string `json:"approvalID,omitempty"`

	AppResourceURI string `json:"appResourceUri,omitempty"`
	AppInstanceID  string `json:"appInstanceID,omitempty"`
}

type MCPToolCapability struct {
	Server           ServerID `json:"server"`
	ToolName         string   `json:"toolName"`
	ProviderToolName string   `json:"providerToolName"`
	ChoiceID         string   `json:"choiceID"`
	Title            string   `json:"title,omitempty"`
	DisplayName      string   `json:"displayName"`
	Description      string   `json:"description,omitempty"`

	InputSchema  map[string]any `json:"inputSchema,omitempty"`
	OutputSchema map[string]any `json:"outputSchema,omitempty"`

	Annotations  *MCPToolAnnotations `json:"annotations,omitempty"`
	InferredRisk MCPToolRisk         `json:"inferredRisk"`

	ApprovalRule  mcpPolicy.MCPApprovalRule  `json:"approvalRule"`
	ExecutionMode mcpPolicy.MCPExecutionMode `json:"executionMode"`

	TaskSupport MCPTaskSupport `json:"taskSupport"`

	App *MCPToolAppInfo `json:"app,omitempty"`

	Digest  string `json:"digest"`
	Enabled bool   `json:"enabled"`
	Stale   bool   `json:"stale,omitempty"`
}

type MCPToolRisk string

type MCPTransportType string

type ResolvedServer struct {
	Server  ServerID
	Catalog CatalogID
	Version Digest
	Config  RuntimeConfig
}

type RuntimeConfig struct {
	Server  ServerID
	Catalog CatalogID

	LogicalName string
	DisplayName string

	Transport                 MCPTransportType
	Stdio                     *MCPRuntimeStdioConfig
	StreamableHTTP            *MCPRuntimeStreamableHTTPConfig
	OAuthClientSecretRequired bool

	Policy mcpPolicy.MCPPolicy

	SensitiveValues []string
}

type ServerID string

type ServerSource interface {
	ResolveServer(
		ctx context.Context,
		server ServerID,
	) (ResolvedServer, error)
}