store

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

type Store interface {
	// Deployments
	ListDeployments(ctx context.Context, search string) ([]*pb.Deployment, error)
	GetDeployment(ctx context.Context, id string) (*pb.Deployment, error)
	CreateDeployment(ctx context.Context, req *pb.CreateDeploymentRequest) (*pb.Deployment, error)
	DeleteDeployment(ctx context.Context, id string) error

	// Jobs — Planner-owned state-machine snapshot of each job, persisted as models.Job.
	UpsertJob(ctx context.Context, rec *models.Job) error
	GetJob(ctx context.Context, id string) (*models.Job, error) // (nil, nil) when not found
	ListJobs(ctx context.Context, ids []string) (map[string]*models.Job, error)
	ListJobsByBatchIDs(ctx context.Context, batchIDs []string) (map[string]*models.Job, error)
	ListJobsByDatasetID(ctx context.Context, fileID string) ([]*models.Job, error)
	DeleteJob(ctx context.Context, id string) error

	ListNonTerminalJobs(ctx context.Context) ([]*models.Job, error)
	// ListAllJobs lists all jobs with cursor-based pagination.
	// after is the job ID cursor (empty string for first page).
	// Returns jobs sorted by created_at descending, hasMore indicates if more results exist.
	ListAllJobs(ctx context.Context, after string, limit int) ([]*models.Job, bool, error)

	// Models
	ListModels(ctx context.Context, search, category string) ([]*pb.Model, error)
	GetModel(ctx context.Context, id string) (*pb.Model, error)
	CreateModel(ctx context.Context, m *pb.Model) (*pb.Model, error)

	// Model Deployment Templates. modelID is the parent — Get/Update/Delete
	// validate that the addressed template actually belongs to it, returning
	// NotFound otherwise so URLs are canonical.
	//
	// ResolveModelDeploymentTemplate looks up by (modelID, name, version);
	// version="" means "latest active". This is the path used by clients
	// that pin templates from outside the UI (e.g. batch SDK callers passing
	// model_template / model_template_version).
	ListModelDeploymentTemplates(ctx context.Context, modelID, statusFilter, name string) ([]*pb.ModelDeploymentTemplate, error)
	GetModelDeploymentTemplate(ctx context.Context, modelID, id string) (*pb.ModelDeploymentTemplate, error)
	CreateModelDeploymentTemplate(ctx context.Context, req *pb.CreateModelDeploymentTemplateRequest) (*pb.ModelDeploymentTemplate, error)
	UpdateModelDeploymentTemplate(ctx context.Context, req *pb.UpdateModelDeploymentTemplateRequest) (*pb.ModelDeploymentTemplate, error)
	DeleteModelDeploymentTemplate(ctx context.Context, modelID, id string) error
	ResolveModelDeploymentTemplate(ctx context.Context, modelID, name, version string) (*pb.ModelDeploymentTemplate, error)

	// API Keys
	ListAPIKeys(ctx context.Context) ([]*pb.APIKey, error)
	CreateAPIKey(ctx context.Context, name string) (*pb.APIKey, string, error) // returns key + full secret
	DeleteAPIKey(ctx context.Context, id string) error

	// Secrets
	ListSecrets(ctx context.Context, search string) ([]*pb.Secret, error)
	CreateSecret(ctx context.Context, name, value string) (*pb.Secret, error)
	DeleteSecret(ctx context.Context, id string) error

	// Quotas
	ListQuotas(ctx context.Context, search string) ([]*pb.Quota, error)

	// Provision
	// GetProvision retrieves a stored provision result by provision ID.
	// Returns nil and a NotFound error if the key doesn't exist.
	GetProvision(ctx context.Context, provisionId string) (*types.ProvisionResult, error)

	// GetProvisionByIdempotencyKey retrieves a stored provision result by idempotency key.
	// Returns nil and a NotFound error if the key doesn't exist.
	GetProvisionByIdempotencyKey(ctx context.Context, idempotencyKey string) (*types.ProvisionResult, error)

	// UpsertProvision inserts/updates a provision result with the given idempotency key.
	UpsertProvision(ctx context.Context, result *types.ProvisionResult) error

	// UpdateProvisionStatus updates the status of a provision result.
	UpdateProvisionStatus(ctx context.Context, provisionId string, status types.ProvisionStatus) error

	// DeleteProvision marks a stored provision result by provision ID as deleted.
	DeleteProvision(ctx context.Context, provisionId string) error

	// ExistsProvision checks if a provision exists for the given provision ID.
	ExistsProvision(ctx context.Context, provisionId string) (bool, error)

	// ListProvisions lists stored provision results.
	ListProvisions(ctx context.Context, options *types.ListOptions) ([]*types.ProvisionResult, error)

	// Close closes the store.
	Close() error
}