package datastore

import "context"

type AgentRepository interface {
	Create(ctx context.Context, remoteID string, state AgentStatus) (*Agent, error)
	Get(ctx context.Context, id AgentID) (*Agent, error)
	Update(ctx context.Context, id AgentID, updates ...AgentUpdateOptionFunc) (*Agent, error)
	Query(ctx context.Context, opts ...AgentQueryOptionFunc) ([]*Agent, int, error)
	Delete(ctx context.Context, id AgentID) error

	UpdateSpec(ctx context.Context, id AgentID, name string, revision int, data map[string]any) (*Spec, error)
	GetSpecs(ctx context.Context, id AgentID) ([]*Spec, error)
	DeleteSpec(ctx context.Context, id AgentID, name string) error
}

type AgentQueryOptionFunc func(*AgentQueryOptions)

type AgentQueryOptions struct {
	Limit     *int
	Offset    *int
	RemoteIDs []string
	IDs       []AgentID
	Statuses  []AgentStatus
}

func WithAgentQueryLimit(limit int) AgentQueryOptionFunc {
	return func(opts *AgentQueryOptions) {
		opts.Limit = &limit
	}
}

func WithAgentQueryOffset(offset int) AgentQueryOptionFunc {
	return func(opts *AgentQueryOptions) {
		opts.Offset = &offset
	}
}

func WithAgentQueryRemoteID(remoteIDs ...string) AgentQueryOptionFunc {
	return func(opts *AgentQueryOptions) {
		opts.RemoteIDs = remoteIDs
	}
}

func WithAgentQueryID(ids ...AgentID) AgentQueryOptionFunc {
	return func(opts *AgentQueryOptions) {
		opts.IDs = ids
	}
}

func WithAgentQueryStatus(statuses ...AgentStatus) AgentQueryOptionFunc {
	return func(opts *AgentQueryOptions) {
		opts.Statuses = statuses
	}
}

type AgentUpdateOptionFunc func(*AgentUpdateOptions)

type AgentUpdateOptions struct {
	Status *AgentStatus
}

func WithAgentUpdateStatus(status AgentStatus) AgentUpdateOptionFunc {
	return func(opts *AgentUpdateOptions) {
		opts.Status = &status
	}
}