registry

type DiscoverySignature struct {
	Name        string `json:"name"`
	Description string `json:"description"`
}

type EventType int

type HealthStatus string

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

type LifecycleManager interface {
	Start(ctx context.Context, config ServerConfig) (ServerHandle, error)
	Stop(ctx context.Context, id string) error
	Kill(ctx context.Context, id string) error
	Restart(ctx context.Context, id string) error
	Status(ctx context.Context, id string) (ServerStatus, error)
	List(ctx context.Context) ([]ServerStatus, error)
	Close()
}

type MatchCriteria struct {
	Capabilities []string
	Transport    TransportType
	MinHealth    HealthStatus
	MaxLoad      float64
}

type Namespace struct {
	Name           string                `yaml:"name"`
	Description    string                `yaml:"description,omitempty"`
	Servers        []string              `yaml:"servers"`
	Children       map[string]*Namespace `yaml:"children,omitempty"`
	AllowedClients []string              `yaml:"allowed_clients,omitempty"`
}

type NamespaceConfig struct {
	Namespaces map[string]*Namespace `yaml:"namespaces,omitempty"`
}

type NamespaceManager interface {
	Load(ctx context.Context, r io.Reader) error
	GetNamespace(ctx context.Context, name string) (*Namespace, error)
	GetToolsForNamespace(ctx context.Context, ns string) ([]string, error)
	CheckAccess(ctx context.Context, ns, clientID string) error
	GetAllNamespaces(ctx context.Context) []*Namespace
	GetServerNamespace(ctx context.Context, serverID string) (string, error)
	GetChildNamespaces(ctx context.Context, parentName string) ([]string, error)
	ListToolsInNamespace(ctx context.Context, nsName string) ([]ToolEntry, error)
}

type Registry interface {
	Register(ctx context.Context, entry ServerEntry) error
	Unregister(ctx context.Context, id string) error
	Update(ctx context.Context, entry ServerEntry) error

	Get(ctx context.Context, id string) (*ServerEntry, error)
	List(ctx context.Context) ([]*ServerEntry, error)
	FindByCapability(ctx context.Context, capability string) ([]*ServerEntry, error)
	FindByTransport(ctx context.Context, transport TransportType) ([]*ServerEntry, error)
	FindBest(ctx context.Context, criteria MatchCriteria) (*ServerEntry, error)

	UpdateHealth(ctx context.Context, id string, health HealthStatus) error
	ListUnhealthy(ctx context.Context) ([]*ServerEntry, error)

	Save(ctx context.Context) error
	Load(ctx context.Context) error

	Subscribe(ch chan<- RegistryEvent) func()
}

type RegistryConfig struct {
	CompactByDefault  bool `yaml:"compact_by_default"`
	MaxSignatureBytes int  `yaml:"max_signature_bytes"`
}

type RegistryEvent struct {
	Type    EventType
	Server  *ServerEntry
	Details string
}

type SchemaCache interface {
	Get(key string) (json.RawMessage, bool)
	Set(key string, schema json.RawMessage)
	Delete(key string)
	Clear()
}

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

type ServerConfig struct {
	ID      string
	Name    string
	Command []string
	Env     []string
	Dir     string
	Port    int
}

type ServerEntry struct {
	ID           string
	Config       *ServerConfig
	Address      string
	Transport    TransportType
	Capabilities []string
	Health       HealthStatus
	Stats        ServerStats
	RegisteredAt time.Time
	LastSeenAt   time.Time
}

type ServerHandle struct {
	ID     string
	Config ServerConfig
	PID    int
}

type ServerState string

type ServerStats struct {
	RequestCount int64
	ErrorCount   int64
	AvgLatencyMs float64
	Load         float64
}

type ServerStatus struct {
	ID         string
	State      ServerState
	Uptime     time.Duration
	MemoryMB   float64
	CPUPercent float64
	ExitCode   int
	Error      string
}

type Tool struct {
	Signature  DiscoverySignature `json:"signature"`
	FullSchema json.RawMessage    `json:"full_schema,omitempty"`
	ServerID   string             `json:"server_id"`
}

type ToolEntry struct {
	Name      string
	Namespace string
	ServerID  string
}

type ToolEvent struct {
	Type     ToolEventType
	ServerID string
	ToolName string
	Tool     ToolEntry
}

type ToolEventType int

type ToolMatch struct {
	Tool    ToolEntry
	Score   float64
	MatchOn string
}

type ToolRegistry interface {
	RegisterTool(ctx context.Context, serverID, toolName string) error
	UnregisterTool(ctx context.Context, serverID, toolName string) error
	GetToolServer(ctx context.Context, toolName string) (string, error)
	SearchTools(ctx context.Context, query string) []ToolMatch
	ListAllTools(ctx context.Context) []ToolEntry
	SubscribeTools(ch chan<- ToolEvent) func()
}

type ToolSchema struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	InputSchema json.RawMessage `json:"inputSchema"`
	ServerID    string          `json:"serverId"`
}

type ToolSchemaRegistry interface {
	RegisterTool(ctx context.Context, tool Tool) error
	UnregisterTool(ctx context.Context, serverID, toolName string) error
	GetDiscoverySignatures() []DiscoverySignature
	GetFullSchema(ctx context.Context, toolName string) (json.RawMessage, error)
	RefreshManifest(ctx context.Context) error
}

type ToolStub struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Category    string `json:"category,omitempty"`
}

type TransportType string