protocol

type BatchInfo struct {
	Total     int    `json:"total"`
	Current   int    `json:"current"`
	BatchID   string `json:"batchId"`
	LastBatch bool   `json:"lastBatch"`
}

type CacheConfig struct {
	Enabled     bool          `json:"enabled"`
	TTL         time.Duration `json:"ttl,omitempty"`
	Refreshable bool          `json:"refreshable"`
	Tags        []string      `json:"tags,omitempty"`
	Key         string        `json:"key,omitempty"`
}

type CachedResource struct {
	Resource    *Resource `json:"resource"`
	CachedAt    time.Time `json:"cachedAt"`
	ExpiresAt   time.Time `json:"expiresAt"`
	AccessCount int64     `json:"accessCount"`
	LastAccess  time.Time `json:"lastAccess"`
	Tags        []string  `json:"tags"`
}

type CapabilitiesOpts struct {
	Resources *ResourcesOpts `json:"resources,omitempty"`
	Tools     *ToolsOpts     `json:"tools,omitempty"`
	Prompts   *PromptsOpts   `json:"prompts,omitempty"`
	Sampling  *SamplingOpts  `json:"sampling,omitempty"`
	Logging   *LoggingOpts   `json:"logging,omitempty"`
	Roots     *RootsOpts     `json:"roots,omitempty"`
}

type Capability string

type ChangeNotification struct {
	JSONRPC string       `json:"jsonrpc"`
	Method  string       `json:"method"` // "notifications/tools/list_changed" or "notifications/prompts/list_changed"
	Params  ChangeParams `json:"params"`
}

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

type ChangeParams struct {
}

type ChangeSubscriber struct {
	ClientID   string
	SessionID  string
	NotifyFunc func(*ChangeNotification) error
	Subscribed time.Time
	LastNotify time.Time
}

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

type ClientSubscriptions struct {
	ClientID      string
	SessionID     string
	Subscriptions map[string]*ResourceSubscription
	NotifyFunc    func(*ResourceUpdateNotification) error
	LastSeen      time.Time
}

type DefaultTextTransformer struct{}

type EmbeddedPromptResource struct {
	Type      string                 `json:"type"` // "resource"
	Resource  *Resource              `json:"resource"`
	Inline    bool                   `json:"inline"`
	Transform string                 `json:"transform,omitempty"`
	Context   map[string]interface{} `json:"context,omitempty"`
}

type EmbeddedResource struct {
	URI         string                 `json:"uri"`
	Name        string                 `json:"name,omitempty"`
	Description string                 `json:"description,omitempty"`
	Content     string                 `json:"content,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

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

type HumanControlConfig struct {
	RequireApproval     bool     `json:"requireApproval"`
	AutoApprovePatterns []string `json:"autoApprovePatterns,omitempty"`
	BlockPatterns       []string `json:"blockPatterns,omitempty"`
	MaxTokens           int      `json:"maxTokens,omitempty"`
	AllowedModels       []string `json:"allowedModels,omitempty"`
	TimeoutSeconds      int      `json:"timeoutSeconds,omitempty"`
}

type HumanReviewResult struct {
	Approved      bool                   `json:"approved"`
	Reviewer      string                 `json:"reviewer"`
	ReviewTime    time.Time              `json:"reviewTime"`
	Comments      string                 `json:"comments,omitempty"`
	Modifications map[string]interface{} `json:"modifications,omitempty"`
}

type IndexingConfig struct {
	Enabled    bool     `json:"enabled"`
	FullText   bool     `json:"fullText"`
	Keywords   []string `json:"keywords,omitempty"`
	Summary    string   `json:"summary,omitempty"`
	Searchable bool     `json:"searchable"`
}

type InitializeParams struct {
	ProtocolVersion string           `json:"protocolVersion"`
	Capabilities    CapabilitiesOpts `json:"capabilities"`
	ClientInfo      ClientInfo       `json:"clientInfo"`
	// Add roots support
	Roots []Root `json:"roots,omitempty"`
}

type InitializeResult struct {
	ProtocolVersion string           `json:"protocolVersion"`
	ServerInfo      ServerInfo       `json:"serverInfo"`
	Capabilities    CapabilitiesOpts `json:"capabilities"`
	Instructions    string           `json:"instructions,omitempty"`
	// Server can announce its roots
	Roots []Root `json:"roots,omitempty"`
}

type LoggingOpts struct {
}

type MCPError struct {
	Code    int                    `json:"code"`
	Message string                 `json:"message"`
	Data    map[string]interface{} `json:"data,omitempty"`
}

type MCPMessage struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      interface{}     `json:"id,omitempty"`
	Method  string          `json:"method,omitempty"`
	Params  json.RawMessage `json:"params,omitempty"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *MCPError       `json:"error,omitempty"`
	// Progress support
	ProgressToken string `json:"progressToken,omitempty"`
}

type MCPNotification struct {
	JSONRPC string          `json:"jsonrpc"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
}

type MCPRequest struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      interface{}     `json:"id"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
	// Progress support
	ProgressToken string `json:"progressToken,omitempty"`
}

type MCPResponse struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      interface{}     `json:"id"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *MCPError       `json:"error,omitempty"`
	// Progress support
	ProgressToken string `json:"progressToken,omitempty"`
}

type MCPTransport interface {
	Transport
	// GetType returns the transport type
	GetType() string
	// IsConnected returns true if transport is connected
	IsConnected() bool
	// GetLastActivity returns the last activity timestamp
	GetLastActivity() time.Time
}

type MessageType string

type ModelHint struct {
	Name string `json:"name"` // e.g., "claude-3-sonnet", "gpt-4"
}

type ModelPreferences struct {
	Hints                []ModelHint `json:"hints,omitempty"`
	CostPriority         string      `json:"costPriority,omitempty"`         // "low", "medium", "high"
	SpeedPriority        string      `json:"speedPriority,omitempty"`        // "low", "medium", "high"
	IntelligencePriority string      `json:"intelligencePriority,omitempty"` // "low", "medium", "high"
}

type ProgressListener func(token string, progress ProgressParams)

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

type ProgressNotification struct {
	JSONRPC string         `json:"jsonrpc"`
	Method  string         `json:"method"` // "notifications/progress"
	Params  ProgressParams `json:"params"`
}

type ProgressParams struct {
	ProgressToken string      `json:"progressToken"`
	Progress      float64     `json:"progress"` // 0.0 to 1.0
	Total         *int64      `json:"total,omitempty"`
	Current       *int64      `json:"current,omitempty"`
	Message       string      `json:"message,omitempty"`
	Details       interface{} `json:"details,omitempty"`
}

type ProgressToken struct {
	Token     string      `json:"token"`
	RequestID interface{} `json:"requestId"`
	Created   time.Time   `json:"created"`
}

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

type PromptContentItem struct {
	Type     string                  `json:"type"` // "text", "image", "resource"
	Text     string                  `json:"text,omitempty"`
	ImageURL string                  `json:"imageUrl,omitempty"`
	Resource *EmbeddedPromptResource `json:"resource,omitempty"`
}

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

type PromptMessage struct {
	Role      string                    `json:"role"`
	Content   []PromptContentItem       `json:"content"`
	Resources []*EmbeddedPromptResource `json:"resources,omitempty"`
	Metadata  map[string]interface{}    `json:"metadata,omitempty"`
}

type PromptsOpts struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type Resource struct {
	URI         string                 `json:"uri"`
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	MimeType    string                 `json:"mimeType,omitempty"`
	Size        int64                  `json:"size,omitempty"`
	Content     *ResourceContentData   `json:"content,omitempty"`
	Metadata    *ResourceMetadata      `json:"metadata,omitempty"`
	Annotations map[string]interface{} `json:"annotations,omitempty"`
	Template    *URITemplate           `json:"template,omitempty"`
	Embedding   *ResourceEmbedding     `json:"embedding,omitempty"`
	Cache       *CacheConfig           `json:"cache,omitempty"`
	Created     time.Time              `json:"created"`
	Modified    time.Time              `json:"modified"`
	Accessed    time.Time              `json:"accessed"`
}

type ResourceContent struct {
	Type     string `json:"type"`           // "text", "blob"
	Data     string `json:"data,omitempty"` // Base64 for blob, direct for text
	MimeType string `json:"mimeType,omitempty"`
	Size     int64  `json:"size,omitempty"`
	Encoding string `json:"encoding,omitempty"` // "utf-8", "base64"
}

type ResourceContentData struct {
	Type         string    `json:"type"`                  // "text", "blob"
	Data         string    `json:"data"`                  // Base64 for blob, direct for text
	Encoding     string    `json:"encoding"`              // "utf-8", "base64", etc.
	Hash         string    `json:"hash,omitempty"`        // Content hash
	Compression  string    `json:"compression,omitempty"` // "gzip", "deflate", etc.
	LastModified time.Time `json:"lastModified"`
}

type ResourceEmbedding struct {
	Strategy  string                 `json:"strategy"` // "inline", "reference", "summary"
	MaxSize   int64                  `json:"maxSize,omitempty"`
	Transform string                 `json:"transform,omitempty"` // "text", "markdown", "json"
	Context   map[string]interface{} `json:"context,omitempty"`
	Fallback  string                 `json:"fallback,omitempty"` // Fallback strategy if embedding fails
}

type ResourceFilter struct {
	Type     string      `json:"type"`     // "glob", "regex", "exact", "prefix"
	Pattern  string      `json:"pattern"`  // Pattern to match
	Property string      `json:"property"` // Resource property to filter on
	Value    interface{} `json:"value"`    // Expected value
}

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

type ResourceMetadata struct {
	Author      string                 `json:"author,omitempty"`
	Version     string                 `json:"version,omitempty"`
	Tags        []string               `json:"tags,omitempty"`
	Language    string                 `json:"language,omitempty"`
	Encoding    string                 `json:"encoding,omitempty"`
	ContentType string                 `json:"contentType,omitempty"`
	Permissions *ResourcePermissions   `json:"permissions,omitempty"`
	Indexing    *IndexingConfig        `json:"indexing,omitempty"`
	Custom      map[string]interface{} `json:"custom,omitempty"`
}

type ResourcePermissions struct {
	Read    bool     `json:"read"`
	Write   bool     `json:"write"`
	Execute bool     `json:"execute"`
	Delete  bool     `json:"delete"`
	Share   bool     `json:"share"`
	Roles   []string `json:"roles,omitempty"`
	Users   []string `json:"users,omitempty"`
}

type ResourceSubscription struct {
	ID           string              `json:"id"`
	ClientID     string              `json:"clientId"`
	SessionID    string              `json:"sessionId"`
	URI          string              `json:"uri"`
	IsTemplate   bool                `json:"isTemplate"`
	Filters      []ResourceFilter    `json:"filters,omitempty"`
	Options      SubscriptionOptions `json:"options"`
	Created      time.Time           `json:"created"`
	LastNotified time.Time           `json:"lastNotified,omitempty"`
}

type ResourceTemplate struct {
	URI         string                 `json:"uri"`         // Template URI
	Name        string                 `json:"name"`        // Resource name template
	Description string                 `json:"description"` // Description template
	MimeType    string                 `json:"mimeType,omitempty"`
	Annotations map[string]interface{} `json:"annotations,omitempty"`
	// Template-specific fields
	Variables []TemplateVariable `json:"variables,omitempty"` // Template variables
	Examples  []TemplateExample  `json:"examples,omitempty"`  // Usage examples
}

type ResourceTransformer interface {
	// Transform transforms resource content
	Transform(resource *Resource, targetFormat string, options map[string]interface{}) (*ResourceContentData, error)
	// GetSupportedFormats returns supported transformation formats
	GetSupportedFormats() []string
	// GetTransformationOptions returns available options for transformation
	GetTransformationOptions(fromFormat, toFormat string) map[string]interface{}
}

type ResourceUpdate struct {
	URI       string                 `json:"uri"`
	Type      string                 `json:"type"` // "created", "updated", "deleted"
	Content   *ResourceContent       `json:"content,omitempty"`
	Metadata  map[string]interface{} `json:"metadata,omitempty"`
	Timestamp time.Time              `json:"timestamp"`
}

type ResourceUpdateNotification struct {
	JSONRPC string               `json:"jsonrpc"`
	Method  string               `json:"method"` // "notifications/resources/updated"
	Params  ResourceUpdateParams `json:"params"`
}

type ResourceUpdateParams struct {
	Resources []ResourceUpdate `json:"resources"`
	BatchInfo *BatchInfo       `json:"batchInfo,omitempty"`
}

type ResourcesOpts struct {
	ListChanged bool `json:"listChanged,omitempty"`
	Subscribe   bool `json:"subscribe,omitempty"`
}

type Root struct {
	URI  string `json:"uri"`
	Name string `json:"name,omitempty"`
}

type RootEntry struct {
	Root        Root             `json:"root"`
	Added       time.Time        `json:"added"`
	LastUsed    time.Time        `json:"lastUsed"`
	Permissions *RootPermissions `json:"permissions,omitempty"`
}

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

type RootPermissions struct {
	Read  bool `json:"read"`
	Write bool `json:"write"`
	List  bool `json:"list"`
	Watch bool `json:"watch"`
}

type RootWatcher struct {
	RootURI    string
	Callback   func(uri string, event string)
	LastChange time.Time
	Active     bool
}

type RootsListRequest struct {
}

type RootsListResponse struct {
	Roots []Root `json:"roots"`
}

type RootsOpts struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

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

type SamplingCapabilities struct {
	Models            []string `json:"models"`
	MaxTokens         int      `json:"maxTokens"`
	SupportsImages    bool     `json:"supportsImages"`
	SupportsTools     bool     `json:"supportsTools"`
	SupportsStreaming bool     `json:"supportsStreaming"`
}

type SamplingContent struct {
	Type      string            `json:"type"` // "text", "image", "resource"
	Text      string            `json:"text,omitempty"`
	ImageData string            `json:"imageData,omitempty"` // base64
	MimeType  string            `json:"mimeType,omitempty"`
	Resource  *EmbeddedResource `json:"resource,omitempty"`
}

type SamplingContext struct {
	ServerInfo      map[string]interface{} `json:"serverInfo,omitempty"`
	ToolContext     []string               `json:"toolContext,omitempty"`     // Recently used tools
	ResourceContext []string               `json:"resourceContext,omitempty"` // Recently accessed resources
	SessionContext  map[string]interface{} `json:"sessionContext,omitempty"`
}

type SamplingHandler interface {
	// HandleSamplingRequest processes a sampling request
	HandleSamplingRequest(request *SamplingRequest) (*SamplingResponse, error)
	// GetSupportedModels returns supported model names
	GetSupportedModels() []string
	// GetCapabilities returns handler capabilities
	GetCapabilities() SamplingCapabilities
}

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

type SamplingMessage struct {
	Role     string                 `json:"role"` // "system", "user", "assistant"
	Content  SamplingContent        `json:"content"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type SamplingOpts struct {
}

type SamplingRequest struct {
	ID           string             `json:"id"`
	ServerName   string             `json:"serverName"`
	Messages     []SamplingMessage  `json:"messages"`
	ModelPrefs   ModelPreferences   `json:"modelPrefs,omitempty"`
	MaxTokens    int                `json:"maxTokens,omitempty"`
	StopSequence []string           `json:"stopSequence,omitempty"`
	Temperature  float64            `json:"temperature,omitempty"`
	Context      SamplingContext    `json:"context,omitempty"`
	Created      time.Time          `json:"created"`
	Status       string             `json:"status"` // "pending", "approved", "rejected", "completed", "failed"
	HumanReview  *HumanReviewResult `json:"humanReview,omitempty"`
}

type SamplingResponse struct {
	Content    SamplingContent        `json:"content"`
	Model      string                 `json:"model,omitempty"`
	StopReason string                 `json:"stopReason,omitempty"`
	Usage      SamplingUsage          `json:"usage,omitempty"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
}

type SamplingUsage struct {
	InputTokens  int `json:"inputTokens,omitempty"`
	OutputTokens int `json:"outputTokens,omitempty"`
	TotalTokens  int `json:"totalTokens,omitempty"`
}

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

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

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

type SubscribeRequest struct {
	URI     string              `json:"uri"`
	Filters []ResourceFilter    `json:"filters,omitempty"`
	Options SubscriptionOptions `json:"options,omitempty"`
}

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

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

type SubscriptionOptions struct {
	IncludeContent  bool     `json:"includeContent,omitempty"`  // Include resource content in notifications
	Debounce        int      `json:"debounce,omitempty"`        // Debounce period in milliseconds
	BatchSize       int      `json:"batchSize,omitempty"`       // Max resources per notification
	ContentTypes    []string `json:"contentTypes,omitempty"`    // Filter by content types
	MaxSize         int64    `json:"maxSize,omitempty"`         // Max resource size to include
	IncludeMetadata bool     `json:"includeMetadata,omitempty"` // Include resource metadata
}

type TemplateExample struct {
	Description string                 `json:"description,omitempty"`
	Variables   map[string]interface{} `json:"variables"`
	Result      string                 `json:"result"` // Expected URI result
}

type TemplateExpression struct {
	Operator  string         // "", "+", "#", ".", "/", ";", "?", "&"
	Variables []VariableSpec // Variable specifications
	Raw       string         // Raw expression text
	StartPos  int            // Position in template
	EndPos    int            // End position in template
}

type TemplateVariable struct {
	Name        string      `json:"name"`
	Description string      `json:"description,omitempty"`
	Type        string      `json:"type,omitempty"`    // "string", "number", "boolean"
	Default     interface{} `json:"default,omitempty"` // Default value
	Required    bool        `json:"required,omitempty"`
	Pattern     string      `json:"pattern,omitempty"` // Regex pattern for validation
	Enum        []string    `json:"enum,omitempty"`    // Allowed values
}

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

type ToolsOpts struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type Transport interface {
	// Send sends an MCP message
	Send(msg MCPMessage) error
	// Receive receives an MCP message
	Receive() (MCPMessage, error)
	// Close closes the transport
	Close() error
	// SupportsProgress returns true if transport supports progress notifications
	SupportsProgress() bool
	// SendProgress sends a progress notification
	SendProgress(notification *ProgressNotification) error
}

type URITemplate struct {
	Template    string
	Expressions []TemplateExpression
}

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

type VariableSpec struct {
	Name      string
	Modifier  string // ":" for prefix, "*" for explode
	MaxLength int    // For prefix modifier
}

type WebSocketConnection struct {
	ID           string
	Transport    *WebSocketTransport
	ClientInfo   *ClientInfo
	ServerInfo   *ServerInfo
	Capabilities *CapabilitiesOpts
	Initialized  bool
	LastActivity time.Time
	Context      map[string]interface{}
}

type WebSocketHandler interface {
	OnConnect(transport *WebSocketTransport) error
	OnDisconnect(transport *WebSocketTransport) error
	OnMessage(transport *WebSocketTransport, message MCPMessage) error
}

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

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