mcp

type Args map[string]any

type AuthConfig struct {
	Type     string // "bearer", "oauth"
	Token    string
	TokenEnv string
}

type CallToolParams struct {
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments"`
}

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

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

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

type ClusterConfig struct {
	ServiceName         string
	Discovery           ServiceDiscovery
	Strategy            LoadBalancerStrategy
	HealthCheckInterval time.Duration
	HealthCheckTimeout  time.Duration
	MaxFailures         int
	RetryAttempts       int
	RetryDelay          time.Duration
	TLS                 *TLSConfig
}

type ClusterNode struct {
	Instance        *ServiceInstance
	State           NodeState
	LastHealthCheck time.Time
	FailureCount    int
	Connections     int64
	Transport       *GRPCTransport
	// contains filtered or unexported fields
}

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

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

type ConsulDiscoveryConfig struct {
	Address string
	Token   string
}

type Content struct {
	Type string `json:"type"` // "text", "image", "resource"
	Text string `json:"text,omitempty"`
	Data string `json:"data,omitempty"`
}

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

type DNSDiscoveryConfig struct {
	CacheTTL    time.Duration
	DefaultPort int
	Resolver    *net.Resolver
}

type GRPCServer struct {
	pb.UnimplementedMCPServiceServer
	// contains filtered or unexported fields
}

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

type GRPCTransportConfig struct {
	Address string
	TLS     *TLSConfig
}

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

type KubernetesDiscoveryConfig struct {
	Namespace  string
	LabelKey   string
	LabelValue string
}

type LoadBalancerStrategy string

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

type NodeState string

type Schema map[string]SchemaField

type SchemaField struct {
	Type        string   `json:"type"`
	Description string   `json:"description,omitempty"`
	Required    bool     `json:"required,omitempty"`
	Default     any      `json:"default,omitempty"`
	Pattern     string   `json:"pattern,omitempty"`   // Regex pattern for strings
	MaxLength   int      `json:"maxLength,omitempty"` // Max length for strings
	MinLength   int      `json:"minLength,omitempty"` // Min length for strings
	Enum        []any    `json:"enum,omitempty"`      // Allowed values
	Minimum     *float64 `json:"minimum,omitempty"`   // Minimum for numbers
	Maximum     *float64 `json:"maximum,omitempty"`   // Maximum for numbers
}

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

type ServerConfig struct {
	Name      string
	Transport string // "local" or "grpc"
	Address   string
	TLS       bool
	Auth      *AuthConfig
}

type ServerOption func(*Server)

type ServiceDiscovery interface {
	// Discover returns available service instances
	Discover(ctx context.Context, serviceName string) ([]*ServiceInstance, error)
	// Register registers a service instance (for dynamic discovery)
	Register(ctx context.Context, instance *ServiceInstance) error
	// Deregister removes a service instance
	Deregister(ctx context.Context, instanceID string) error
	// Watch returns a channel that emits updates when services change
	Watch(ctx context.Context, serviceName string) (<-chan []*ServiceInstance, error)
	// Close cleans up resources
	Close() error
}

type ServiceInstance struct {
	ID       string            `json:"id"`
	Name     string            `json:"name"`
	Address  string            `json:"address"`
	Port     int               `json:"port"`
	Metadata map[string]string `json:"metadata,omitempty"`
	Healthy  bool              `json:"healthy"`
	Weight   int               `json:"weight,omitempty"`
}

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

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

type TLSConfig struct {
	Enabled            bool
	CertFile           string
	KeyFile            string
	CAFile             string
	InsecureSkipVerify bool
	ServerName         string
	ClientAuth         tls.ClientAuthType
}

type Tool struct {
	Name               string              `json:"name"`
	Description        string              `json:"description"`
	Handler            ToolHandler         `json:"-"`
	Schema             Schema              `json:"input_schema"`
	RequiredPermission security.Permission `json:"-"` // Required permission to execute this tool
	AllowedRoles       []string            `json:"-"` // Allowed roles to execute this tool
}

type ToolHandler func(context.Context, Args) (any, error)

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

type Transport interface {
	Send(ctx context.Context, method string, params any) (any, error)
	Close() error
}

type TypedTool[I, O any] struct {
	// contains filtered or unexported fields
}