engine

type AlertSpec struct {
	Type     string            `yaml:"type"`
	Endpoint string            `yaml:"endpoint,omitempty"`
	URL      string            `yaml:"url,omitempty"`
	Headers  map[string]string `yaml:"headers,omitempty"`
}

type BackupConfig struct {
	Tool   string     `yaml:"tool"`
	Stanza string     `yaml:"stanza"`
	Source string     `yaml:"source"`
	Repo   RepoConfig `yaml:"repo"`
	Target string     `yaml:"-"`
}

type Check struct {
	Type   string   `yaml:"type"`
	Name   string   `yaml:"name"`
	SQL    string   `yaml:"sql,omitempty"`
	Keys   []string `yaml:"keys,omitempty"`
	Expect string   `yaml:"expect"`
}

type CheckResult struct {
	Name     string
	Type     string
	Expected string
	Actual   string
	Passed   bool
	Duration time.Duration
	Error    error
}

type Config struct {
	Drills    []DrillConfig `yaml:"drills"`
	Metrics   MetricsConfig `yaml:"metrics"`
	Reporting ReportConfig  `yaml:"reporting"`
}

type Container interface {
	// ID returns the container identifier.
	ID() string

	// Host returns the hostname or IP for connecting to the container.
	Host() string

	// Port returns the mapped port for the given container port.
	Port(containerPort int) int
}

type ContainerConf struct {
	Image     string            `yaml:"image"`
	Env       map[string]string `yaml:"env,omitempty"`
	Resources ResourceConf      `yaml:"resources"`
}

type ContainerSpec struct {
	Image       string
	Env         map[string]string
	Ports       []int
	MemoryLimit int64
	CPULimit    int64
	Cmd         []string
}

type DrillConfig struct {
	Name     string       `yaml:"name"`
	Provider string       `yaml:"provider"`
	Schedule string       `yaml:"schedule"`
	Backup   BackupConfig `yaml:"backup"`
	Restore  RestoreSpec  `yaml:"restore"`
	Validate []Check      `yaml:"checks"`
	Alerts   []AlertSpec  `yaml:"alerts"`
}

type DrillResult struct {
	Name             string
	Provider         string
	StartedAt        time.Time
	Duration         time.Duration
	BackupTimestamp  time.Time
	BackupAge        time.Duration
	ValidationPassed bool
	Checks           []CheckResult
	Error            error
	CleanupSkipped   bool
	TargetID         string
	TargetHost       string
	TargetPorts      map[int]int
}

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

type MetricsConfig struct {
	Prometheus struct {
		Enabled     bool              `yaml:"enabled"`
		Pushgateway string            `yaml:"pushgateway"`
		Labels      map[string]string `yaml:"labels"`
	} `yaml:"prometheus"`
}

type PreflightProvider interface {
	Preflight(ctx context.Context, rt Runtime, cfg BackupConfig, target Container, checks []Check) error
}

type Provider interface {
	// Name returns the provider identifier (e.g., "postgres", "mysql").
	Name() string

	// Restore pulls a backup and restores it into the target container.
	Restore(ctx context.Context, rt Runtime, cfg BackupConfig, target Container) (*RestoreResult, error)

	// Validate runs provider-specific checks against the restored data.
	Validate(ctx context.Context, rt Runtime, target Container, checks []Check) (*ValidationResult, error)

	// Cleanup performs provider-specific teardown.
	Cleanup(ctx context.Context, target Container) error
}

type RepoConfig struct {
	Type     string `yaml:"type"`
	Bucket   string `yaml:"bucket"`
	Endpoint string `yaml:"endpoint"`
	Prefix   string `yaml:"prefix"`
	Region   string `yaml:"region"`
}

type ReportConfig struct {
	Format    []string `yaml:"format"`
	Output    string   `yaml:"output"`
	Retention string   `yaml:"retention"`
}

type Reporter interface {
	// Report publishes results to the configured output(s).
	Report(ctx context.Context, results []DrillResult) error
}

type ResourceConf struct {
	Memory string `yaml:"memory"`
	CPU    string `yaml:"cpu"`
}

type RestoreResult struct {
	BackupTimestamp string
	BackupSize      int64
	Duration        int64 // milliseconds
}

type RestoreSpec struct {
	Target    string        `yaml:"target"`
	Container ContainerConf `yaml:"container"`
	Timeout   string        `yaml:"timeout"`
}

type Runtime interface {
	// Create provisions a container with the given spec.
	Create(ctx context.Context, spec ContainerSpec) (Container, error)

	// Exec runs a command inside the container.
	Exec(ctx context.Context, c Container, cmd []string) ([]byte, error)

	// CopyTo copies data into the container filesystem.
	CopyTo(ctx context.Context, c Container, dest string, src io.Reader) error

	// Destroy tears down the container and cleans up.
	Destroy(ctx context.Context, c Container) error

	// Logs returns the container's log stream.
	Logs(ctx context.Context, c Container) (io.ReadCloser, error)
}

type ValidationResult struct {
	Checks []CheckResult
}