registry

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

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

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

	// 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)

	// SetMiddleware sets the middleware chain for tool handlers.
	SetMiddleware(chain *middleware.Chain)

	// Close releases resources.
	Close() error
}

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

type ToolkitFactory func(name string, config map[string]interface{}) (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"`
}