mcpserver

type Annotated struct {
	Audience []Role  `json:"audience,omitempty"`
	Priority float64 `json:"priority,omitempty"`
}

type BaseNotification struct {
	Meta map[string]interface{} `json:"_meta,omitempty"`
}

type BlobResourceContents struct {
	URI      string `json:"uri"`
	Blob     []byte `json:"blob"`
	MimeType string `json:"mimeType,omitempty"`
}

type CallToolRequest struct {
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments,omitempty"`
}

type CallToolResult struct {
	Content []interface{} `json:"content"` // Can contain TextContent, ImageContent, or EmbeddedResource
	IsError bool          `json:"isError,omitempty"`
}

type CancelledNotification struct {
	BaseNotification
	RequestID string `json:"requestId"`
	Reason    string `json:"reason,omitempty"`
}

type ClientCapabilities struct {
	Roots *struct {
		ListChanged bool `json:"listChanged"`
	} `json:"roots,omitempty"`
	Sampling     *struct{}              `json:"sampling,omitempty"`
	Experimental map[string]interface{} `json:"experimental,omitempty"`
}

type CompleteRequest struct {
	Argument struct {
		Name  string `json:"name"`
		Value string `json:"value"`
	} `json:"argument"`
	Ref interface{} `json:"ref"` // Can be PromptReference or ResourceReference
}

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

type ContentValidationOptions struct {
	MaxTextSize      int64    // Maximum size for text content in bytes
	MaxBlobSize      int64    // Maximum size for blob content in bytes
	AllowedMIMETypes []string // List of allowed MIME types (empty means all allowed)
}

type CreateMessageRequest struct {
	Messages         []SamplingMessage `json:"messages"`
	MaxTokens        int               `json:"maxTokens"`
	Temperature      float64           `json:"temperature,omitempty"`
	StopSequences    []string          `json:"stopSequences,omitempty"`
	SystemPrompt     string            `json:"systemPrompt,omitempty"`
	IncludeContext   string            `json:"includeContext,omitempty"`
	ModelPreferences ModelPreferences  `json:"modelPreferences,omitempty"`
	Metadata         interface{}       `json:"metadata,omitempty"`
}

type CreateMessageResult struct {
	Content    interface{} `json:"content"` // Can be TextContent or ImageContent
	Model      string      `json:"model"`
	Role       Role        `json:"role"`
	StopReason string      `json:"stopReason,omitempty"`
}

type EmbeddedResource struct {
	Type        string      `json:"type"`     // Must be "resource"
	Resource    interface{} `json:"resource"` // Can be TextResourceContents or BlobResourceContents
	Annotations *Annotated  `json:"annotations,omitempty"`
}

type GetPromptRequest struct {
	Name      string            `json:"name"`
	Arguments map[string]string `json:"arguments,omitempty"`
}

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

type ImageContent struct {
	Type        string     `json:"type"` // Must be "image"
	Data        []byte     `json:"data"`
	MimeType    string     `json:"mimeType"`
	Annotations *Annotated `json:"annotations,omitempty"`
}

type Implementation struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type InitializeResult struct {
	ServerInfo      Implementation     `json:"serverInfo"`
	Capabilities    ServerCapabilities `json:"capabilities"`
	ProtocolVersion string             `json:"protocolVersion"`
	Instructions    string             `json:"instructions,omitempty"`
}

type InitializedNotification struct {
	BaseNotification
}

type JSONRPCError struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      interface{} `json:"id"`
	Error   struct {
		Code    int         `json:"code"`
		Message string      `json:"message"`
		Data    interface{} `json:"data,omitempty"`
	} `json:"error"`
}

type JSONRPCNotification struct {
	JSONRPC string      `json:"jsonrpc"`
	Method  string      `json:"method"`
	Params  interface{} `json:"params"`
}

type JSONRPCRequest struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      interface{}     `json:"id"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
}

type JSONRPCResponse struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      interface{} `json:"id"`
	Result  interface{} `json:"result"`
}

type ListPromptsRequest struct {
	Cursor string `json:"cursor,omitempty"`
}

type ListPromptsResult struct {
	Prompts    []Prompt `json:"prompts"`
	NextCursor string   `json:"nextCursor,omitempty"`
}

type ListResourcesRequest struct {
	Cursor string `json:"cursor,omitempty"`
}

type ListResourcesResult struct {
	Resources  []Resource `json:"resources"`
	NextCursor string     `json:"nextCursor,omitempty"`
}

type ListToolsRequest struct {
	Cursor string `json:"cursor,omitempty"`
}

type ListToolsResult struct {
	Tools      []Tool `json:"tools"`
	NextCursor string `json:"nextCursor,omitempty"`
}

type LoggingLevel string

type LoggingMessageNotification struct {
	BaseNotification
	Level  LoggingLevel `json:"level"`
	Data   interface{}  `json:"data"`
	Logger string       `json:"logger,omitempty"`
}

type ModelHint struct {
	Name string `json:"name,omitempty"`
}

type ModelPreferences struct {
	SpeedPriority        float64     `json:"speedPriority,omitempty"`
	CostPriority         float64     `json:"costPriority,omitempty"`
	IntelligencePriority float64     `json:"intelligencePriority,omitempty"`
	Hints                []ModelHint `json:"hints,omitempty"`
}

type NotificationHandler func(context.Context, interface{}) error

type NotificationSender interface {
	SendResourceUpdate(ctx context.Context, uri string) error
	SendResourceListChanged(ctx context.Context) error
	SendPromptListChanged(ctx context.Context) error
	SendToolListChanged(ctx context.Context) error
	SendProgress(ctx context.Context, token string, progress, total float64) error
	SendLoggingMessage(ctx context.Context, level LoggingLevel, data interface{}, logger string) error
}

type NotificationType string

type ProgressNotification struct {
	BaseNotification
	Progress      float64 `json:"progress"`
	ProgressToken string  `json:"progressToken"`
	Total         float64 `json:"total,omitempty"`
}

type Prompt struct {
	Name        string           `json:"name"`
	Description string           `json:"description,omitempty"`
	Arguments   []PromptArgument `json:"arguments,omitempty"`
}

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

type PromptListChangedNotification struct {
	BaseNotification
}

type PromptMessage struct {
	Role    Role        `json:"role"`
	Content interface{} `json:"content"` // Can be TextContent, ImageContent, or EmbeddedResource
}

type PromptReference struct {
	Type string `json:"type"` // Must be "ref/prompt"
	Name string `json:"name"`
}

type ReadResourceRequest struct {
	URI string `json:"uri"`
}

type ReadResourceResult struct {
	Contents []interface{} `json:"contents"` // Can contain TextResourceContents or BlobResourceContents
}

type RequestError struct {
	Code    int
	Message string
}

type Resource struct {
	Name        string     `json:"name"`
	URI         string     `json:"uri"`
	MimeType    string     `json:"mimeType,omitempty"`
	Description string     `json:"description,omitempty"`
	Annotations *Annotated `json:"annotations,omitempty"`
}

type ResourceListChangedNotification struct {
	BaseNotification
}

type ResourceReference struct {
	Type string `json:"type"` // Must be "ref/resource"
	URI  string `json:"uri"`
}

type ResourceSubscriber struct {
	URI      string
	Callback func(string)
}

type ResourceTemplate struct {
	Name        string     `json:"name"`
	URITemplate string     `json:"uriTemplate"`
	MimeType    string     `json:"mimeType,omitempty"`
	Description string     `json:"description,omitempty"`
	Annotations *Annotated `json:"annotations,omitempty"`
}

type ResourceUpdatedNotification struct {
	BaseNotification
	URI string `json:"uri"`
}

type Role string

type SamplingMessage struct {
	Role    Role        `json:"role"`
	Content interface{} `json:"content"` // Can be TextContent or ImageContent
}

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

type ServerCapabilities struct {
	Logging   *struct{} `json:"logging,omitempty"`
	Resources *struct {
		ListChanged bool `json:"listChanged"`
		Subscribe   bool `json:"subscribe"`
	} `json:"resources,omitempty"`
	Prompts *struct {
		ListChanged bool `json:"listChanged"`
	} `json:"prompts,omitempty"`
	Tools *struct {
		ListChanged bool `json:"listChanged"`
	} `json:"tools,omitempty"`
	Experimental map[string]interface{} `json:"experimental,omitempty"`
}

type ServerConfig struct {
	Implementation      Implementation
	Capabilities        ServerCapabilities
	Handler             ServerHandler
	NotificationHandler NotificationHandler
	ValidationOptions   *ContentValidationOptions
}

type ServerHandler interface {
	// Resource methods
	ListResources(ctx context.Context, cursor string) ([]Resource, string, error)
	ReadResource(ctx context.Context, uri string) ([]interface{}, error)
	SubscribeResource(ctx context.Context, uri string, callback func(string)) error
	UnsubscribeResource(ctx context.Context, uri string) error

	// Prompt methods
	ListPrompts(ctx context.Context, cursor string) ([]Prompt, string, error)
	GetPrompt(ctx context.Context, name string, args map[string]string) (*GetPromptResult, error)

	// Tool methods
	ListTools(ctx context.Context, cursor string) ([]Tool, string, error)
	CallTool(ctx context.Context, name string, args map[string]interface{}) (*CallToolResult, error)

	// Completion method
	Complete(ctx context.Context, req CompleteRequest) (*CompleteResult, error)

	// Sampling method
	CreateMessage(ctx context.Context, req CreateMessageRequest) (*CreateMessageResult, error)
}

type SetLevelRequest struct {
	Level LoggingLevel `json:"level"`
}

type SubscribeRequest struct {
	URI string `json:"uri"`
}

type SubscriptionResponse struct {
	SubscriptionID string `json:"subscriptionId"`
}

type TextContent struct {
	Type        string     `json:"type"` // Must be "text"
	Text        string     `json:"text"`
	Annotations *Annotated `json:"annotations,omitempty"`
}

type TextResourceContents struct {
	URI      string `json:"uri"`
	Text     string `json:"text"`
	MimeType string `json:"mimeType,omitempty"`
}

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

type ToolListChangedNotification struct {
	BaseNotification
}

type UnsubscribeRequest struct {
	URI string `json:"uri"`
}

type ValidationError struct {
	Field   string
	Message string
}