imageswitch

type Applier interface {
	// Plan returns what Apply would change, as JSON-ready data; nothing is written.
	Plan(ctx context.Context, env map[string]string) (any, error)
	// Apply writes the lines, regenerates, and undoes the lines if it fails.
	Apply(ctx context.Context, env map[string]string) error
}

type Check struct {
	Name   string `json:"name"`
	OK     bool   `json:"ok"`
	Detail string `json:"detail,omitempty"`
}

type Config struct {
	ProjectDir  string
	ProjectName string
	// Service is the compose service name of postgres ("postgres").
	Service string
	// Container is the postgres container name (<project>_postgres).
	Container string
	// User is POSTGRES_USER; Password is POSTGRES_PASSWORD (EnvPass only).
	User     string
	Password string
	// TargetFor returns the pgvector image of the given Postgres major, or an
	// error when there is none.
	TargetFor func(major int) (string, error)
	// PrevEnv holds the current raw values of EnvImage and EnvDataVolume in the
	// project (empty when unset); rollback writes them back.
	PrevEnv map[string]string
}

type ContainerInfo struct {
	Image          string
	Running        bool
	DataVolume     string // name of the volume mounted at /var/lib/postgresql/data
	Networks       []string
	ComposeProject string // label com.docker.compose.project
}

type Deps struct {
	Runtime Runtime
	Stack   Stack
	Applier Applier
	// Out receives one progress line per step; nil discards.
	Out io.Writer
	// Now is the clock; nil uses time.Now.
	Now func() time.Time
	// FreeBytes reports the free bytes under a host path; nil uses statfs.
	FreeBytes func(path string) (uint64, error)
}

type DockerRuntime struct{}

type Phase string

type Report struct {
	// Status: planned, switched, already-pgvector, rolled-back.
	Status    string `json:"status"`
	Phase     Phase  `json:"phase,omitempty"`
	OldImage  string `json:"old_image,omitempty"`
	NewImage  string `json:"new_image,omitempty"`
	OldVolume string `json:"old_volume,omitempty"`
	NewVolume string `json:"new_volume,omitempty"`
	// Preflight lists every check; a failed one stops the switch with E509.
	Preflight []Check `json:"preflight,omitempty"`
	// Steps names what a run does, in order.
	Steps []string `json:"steps,omitempty"`
	// Stage is the directory the dump is written to (mode 0700); Dumps names the dump
	// files in it, one per database. Both are for --plan; the stage is removed at the end.
	Stage     string   `json:"stage,omitempty"`
	Dumps     []string `json:"dumps,omitempty"`
	Databases int      `json:"databases,omitempty"`
	Tables    int      `json:"tables,omitempty"`
	Sequences int      `json:"sequences,omitempty"`
	DataBytes int64    `json:"data_bytes,omitempty"`
	// DowntimeSeconds estimates the window from stopping services to a healthy stack.
	DowntimeSeconds int `json:"downtime_seconds,omitempty"`
	// Tolerated lists the bootstrap statements the restore skipped, each logged.
	Tolerated []string `json:"tolerated,omitempty"`
	// Build is the compose and lock change the build plan path would make.
	Build     any      `json:"build,omitempty"`
	NextSteps []string `json:"next_steps,omitempty"`
	Warnings  []string `json:"warnings,omitempty"`
}

type RunOptions struct {
	// Resume continues an incomplete switch from its state.
	Resume bool
}

type Runtime interface {
	// Inspect returns the container, or an error when it does not exist.
	Inspect(ctx context.Context, name string) (ContainerInfo, error)
	// ExecSQL runs sql in the running container with psql over the local socket.
	ExecSQL(ctx context.Context, container, user, db, sql string) (string, error)
	// RunOneShot runs a throwaway container to completion.
	RunOneShot(ctx context.Context, spec docker.RunSpec) (stdout, stderr string, err error)
	// RemoveContainer force-removes a container; absent is success.
	RemoveContainer(ctx context.Context, name string) error
	// VolumeLabels returns a volume's labels; exists is false when it is absent.
	VolumeLabels(ctx context.Context, name string) (labels map[string]string, exists bool, err error)
	// VolumeCreate, VolumeExists and VolumeRemove manage named volumes.
	VolumeCreate(ctx context.Context, name string, labels map[string]string) error
	VolumeExists(ctx context.Context, name string) (bool, error)
	VolumeRemove(ctx context.Context, name string) error
}

type Stack interface {
	// Running lists the service names that are running now.
	Running(ctx context.Context) ([]string, error)
	// Stop stops the services; Up starts them (none: every service).
	Stop(ctx context.Context, services ...string) error
	Up(ctx context.Context, services ...string) error
	// WaitHealthy returns when the service reports healthy, or an error on timeout.
	WaitHealthy(ctx context.Context, service string) error
}

type State struct {
	Version int    `json:"version"`
	RunID   string `json:"run_id"`
	Phase   Phase  `json:"phase"`

	Container      string   `json:"container"`
	Network        string   `json:"network"`
	ComposeProject string   `json:"compose_project"`
	BootUser       string   `json:"boot_user"`
	Major          int      `json:"major"`
	OldImage       string   `json:"old_image"`
	NewImage       string   `json:"new_image"`
	OldVolume      string   `json:"old_volume"`
	NewVolume      string   `json:"new_volume"`
	Databases      []string `json:"databases"`
	// ReadOnlySet is true once ALTER DATABASE ... default_transaction_read_only
	// was set on the old cluster by this switch.
	ReadOnlySet bool `json:"read_only_set"`
	// StoppedServices are the services this switch stopped; abort and rollback restart them.
	StoppedServices []string `json:"stopped_services"`
	// NewVolumeCreated is true once this switch created the new volume (it may remove it).
	NewVolumeCreated bool `json:"new_volume_created"`
	// PrevEnv is the raw value of each env key the switch writes, before the switch.
	PrevEnv   map[string]string `json:"prev_env"`
	Tolerated []string          `json:"tolerated,omitempty"`
	Tables    int               `json:"tables,omitempty"`
	Sequences int               `json:"sequences,omitempty"`
	DataBytes int64             `json:"data_bytes,omitempty"`

	StartedAt time.Time `json:"started_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

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