plugin

type BasePathProvider interface {
	BasePath() string
}

type CatalogLoader interface {
	Start(ctx context.Context) error
	Stop(ctx context.Context) error
}

type CatalogPlugin interface {
	// Name returns the plugin name (e.g., "model", "dataset").
	// Used for routing and configuration lookup.
	Name() string

	// Version returns the API version (e.g., "v1alpha1").
	Version() string

	// Description returns a human-readable description of the plugin.
	Description() string

	// Init initializes the plugin with its configuration.
	// Called once during server startup before Start.
	Init(ctx context.Context, cfg Config) error

	// Start begins background operations (hot-reload, watchers, etc.).
	// Called after Init and after database migrations.
	Start(ctx context.Context) error

	// Stop gracefully shuts down the plugin.
	// Called during server shutdown.
	Stop(ctx context.Context) error

	// Healthy returns true if the plugin is functioning correctly.
	// Used for health check endpoints.
	Healthy() bool

	// RegisterRoutes mounts the plugin's HTTP routes on the provided router.
	// The router is the server's root router; plugins are responsible for
	// using full path patterns (e.g., "/api/model_catalog/v1alpha1/models").
	RegisterRoutes(router chi.Router) error

	// Migrations returns database migrations for this plugin.
	// Migrations are applied in order during server initialization.
	Migrations() []Migration
}

type Config struct {
	// DB is the shared database connection.
	DB *gorm.DB

	// BasePath is the API base path for this plugin (e.g., "/api/model_catalog/v1alpha1").
	BasePath string

	// ConfigPaths are the paths to all sources.yaml files being used.
	ConfigPaths []string

	// RepoSet is the shared set of repositories from the datastore.
	RepoSet datastore.RepoSet

	// PerformanceMetricsPath holds paths to performance metrics data directories.
	PerformanceMetricsPath []string

	// Logger is a structured logger scoped to this plugin.
	Logger *slog.Logger
}

type DatastoreEntry struct {
	TypeName string
	Category string // "context", "artifact", or "execution"
	Spec     *datastore.SpecType
}

type DatastoreSpecProvider interface {
	DatastoreEntries() []DatastoreEntry
}

type FileWatcher interface {
	Path(ctx context.Context, path string) (<-chan struct{}, error)
}

type FlagProvider interface {
	RegisterFlags(fs *flag.FlagSet)
}

type LeaderAware interface {
	OnBecomeLeader(ctx context.Context) error
}

type LifecycleState interface {
	IsLeader() bool
	SetLeader(bool)
	ShouldWriteDatabase() bool
	SetupFileWatchers(ctx context.Context) (context.Context, error)
	StopFileWatchers()
	WaitForInflightWrites(timeout time.Duration)
	Paths() []string
}

type Migration struct {
	Version     string
	Description string
	Up          func(db *gorm.DB) error
	Down        func(db *gorm.DB) error
}

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

type PluginBaseConfig struct {
	// Name is used for log messages and error wrapping.
	Name string

	// State manages leader election, file watchers, and write tracking.
	State LifecycleState

	// Loader performs config parsing and leader write operations.
	Loader Reloader

	// SourceIDs returns the set of source IDs known to this plugin.
	SourceIDs func() mapset.Set[string]

	// FileWatcher provides file change notifications.
	FileWatcher FileWatcher

	// OnLeaderReady is called after leader operations succeed.
	// Optional — nil means no post-leader hook.
	OnLeaderReady func(ctx context.Context) error
}

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

type Reloader interface {
	ParseAllConfigs() error
	ReloadParsing() error
	PerformLeaderOperations(ctx context.Context, allKnownSourceIDs mapset.Set[string]) error
}

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

type ServerConfig struct {
	DB                     *gorm.DB
	ConfigPaths            []string
	PerformanceMetricsPath []string
	RepoSet                datastore.RepoSet
	Logger                 *slog.Logger
}

type SourceIDProvider interface {
	KnownSourceIDs() mapset.Set[string]
}

type SourceKeyProvider interface {
	SourceKey() string
}