registry

type AggregateToolkitFactory func(defaultName string, instances map[string]map[string]any) (Toolkit, error)

type ConnectionResolver interface {
	ConnectionForTool(toolName string) string
}

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

type LoaderConfig struct {
	Toolkits map[string]ToolkitKindConfig `yaml:"toolkits"`
}

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

type PromptDescriber interface {
	PromptInfos() []PromptInfo
}

type PromptInfo struct {
	Name        string               `json:"name"`
	Description string               `json:"description"`
	Category    string               `json:"category,omitempty"`
	Content     string               `json:"content,omitempty"`
	Arguments   []PromptArgumentInfo `json:"arguments,omitempty"`
}

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

type Toolkit interface {
	// Kind returns the toolkit type (e.g., "trino", "datahub", "s3").
	Kind() string

	// Name returns the instance name from config.
	Name() string

	// Connection returns the connection name for audit logging.
	// This identifies the specific backend connection (e.g., "prod-trino", "main-datahub").
	Connection() string

	// RegisterTools registers all tools with the MCP server.
	RegisterTools(s *mcp.Server)

	// Tools returns a list of tool names provided by this toolkit.
	Tools() []string

	// SetSemanticProvider sets the semantic metadata provider for enrichment.
	SetSemanticProvider(provider semantic.Provider)

	// SetQueryProvider sets the query execution provider for enrichment.
	SetQueryProvider(provider query.Provider)

	// Close releases resources.
	Close() error
}

type ToolkitConfig struct {
	Kind    string
	Name    string
	Enabled bool
	Config  map[string]any
	Default bool
}

type ToolkitFactory func(name string, config map[string]any) (Toolkit, error)

type ToolkitKindConfig struct {
	Enabled   bool                      `yaml:"enabled"`
	Instances map[string]map[string]any `yaml:"instances"`
	Default   string                    `yaml:"default"`
	Config    map[string]any            `yaml:"config"`
}

type ToolkitMatch struct {
	Kind       string
	Name       string
	Connection string
	Found      bool

	// ConnectionResolved is true when Connection was determined by the
	// toolkit's ConnectionResolver (per-tool routing) rather than from
	// the toolkit's default Connection(). When true, downstream
	// middleware MUST NOT override Connection from request arguments —
	// the toolkit owns the routing and a caller-supplied "connection"
	// arg is either ignored by the toolkit or attempts to spoof audit.
	ConnectionResolved bool
}