db

type Catalog struct {
	Ref         string          `db:"ref"`
	Digest      string          `db:"digest"`
	Title       string          `db:"title"`
	Source      string          `db:"source"`
	LastUpdated *time.Time      `db:"last_updated"`
	Servers     []CatalogServer `db:"-"`
}

type CatalogDAO interface {
	GetCatalog(ctx context.Context, ref string) (*Catalog, error)
	UpsertCatalog(ctx context.Context, catalog Catalog) error
	DeleteCatalog(ctx context.Context, ref string) error
	ListCatalogs(ctx context.Context) ([]Catalog, error)
}

type CatalogServer struct {
	ID         *int64   `db:"id" json:"id"`
	ServerType string   `db:"server_type" json:"server_type"`
	Tools      ToolList `db:"tools" json:"tools"`
	Source     string   `db:"source" json:"source"`
	Image      string   `db:"image" json:"image"`
	Endpoint   string   `db:"endpoint" json:"endpoint"`
	CatalogRef string   `db:"catalog_ref" json:"catalog_ref"`

	Snapshot *ServerSnapshot `db:"snapshot" json:"snapshot"`
}

type DAO interface {
	WorkingSetDAO
	CatalogDAO
	MigrationStatusDAO
	PullRecordDAO

	// Normally unnecessary to call this
	Close() error
}

type MigrationStatus struct {
	ID          *int64     `db:"id"`
	Status      string     `db:"status"`
	Logs        string     `db:"logs"`
	LastUpdated *time.Time `db:"last_updated"`
}

type MigrationStatusDAO interface {
	GetMigrationStatus(ctx context.Context) (*MigrationStatus, error)
	TryAcquireMigration(ctx context.Context, status string) (*MigrationStatus, bool, error)
	UpdateMigrationStatus(ctx context.Context, status MigrationStatus) error
}

type Option func(o *options) error

type PullRecord struct {
	ID          *int64     `db:"id"`
	Ref         string     `db:"ref"`
	FirstPulled *time.Time `db:"first_pulled"`
}

type PullRecordDAO interface {
	RecordPull(ctx context.Context, ref string) error
	CheckPullRecord(ctx context.Context, ref string) (bool, error)
}

type Secret struct {
	Provider string `json:"provider"`
}

type SecretMap map[string]Secret

type Server struct {
	Type     string         `json:"type"`
	Config   map[string]any `json:"config,omitempty"`
	Secrets  string         `json:"secrets,omitempty"`
	Tools    []string       `json:"tools"`
	Source   string         `json:"source,omitempty"`
	Image    string         `json:"image,omitempty"`
	Endpoint string         `json:"endpoint,omitempty"`

	// CatalogRef is the catalog reference that this server was sourced from.
	CatalogRef string `json:"catalog_ref,omitempty"`

	// Optional snapshot of the server schema
	Snapshot *ServerSnapshot `json:"snapshot,omitempty"`
}

type ServerList []Server

type ServerSnapshot struct {
	// TODO(cody): hacky reference to the same type that we use elsewhere
	Server catalog.Server `json:"server"`
}

type ToolList []string

type WorkingSet struct {
	ID      string     `db:"id"`
	Name    string     `db:"name"`
	Servers ServerList `db:"servers"`
	Secrets SecretMap  `db:"secrets"`
}

type WorkingSetDAO interface {
	GetWorkingSet(ctx context.Context, id string) (*WorkingSet, error)
	ListWorkingSets(ctx context.Context) ([]WorkingSet, error)
	CreateWorkingSet(ctx context.Context, workingSet WorkingSet) error
	UpdateWorkingSet(ctx context.Context, workingSet WorkingSet) error
	RemoveWorkingSet(ctx context.Context, id string) error
	SearchWorkingSets(ctx context.Context, query string, workingSetID string) ([]WorkingSet, error)
}