spendcap

type AppLoad struct {
	Name        string
	Replicas    int
	MonthToDate float64
	RecentDaily []float64
}

type Breach struct {
	Key  string
	Text string
}

type Cap struct {
	App                string  `json:"app"`
	MonthlyEgressBytes int64   `json:"monthly_egress_bytes"`
	MonthlyCostCeiling float64 `json:"monthly_cost_ceiling"`
	Thresholds         []int   `json:"thresholds"`
	Action             string  `json:"action"`
	ThrottleRPS        int     `json:"throttle_rps"`
	GuardMode          string  `json:"guard_mode"`
}

type Condition struct {
	Healthy bool   `json:"healthy"`
	Reason  string `json:"reason"`
	Message string `json:"message"`
}

type Controller struct {
	Store  Store
	Usage  UsageSource
	Nudge  func(ctx context.Context) error
	Logger *slog.Logger
	Now    func() time.Time
}

type Enforcement struct {
	App         string
	Action      string
	ThrottleRPS int
}

type Eval struct {
	EgressPct float64
	CostPct   float64
	Pct       float64
	Crossed   []int
	Exhausted bool
	// Desired is the action that should be in force right now.
	Desired string
}

type Event struct {
	ID     int64     `json:"id"`
	App    string    `json:"app"`
	Period string    `json:"period"`
	Kind   string    `json:"kind"`
	Detail string    `json:"detail"`
	Pct    float64   `json:"pct"`
	At     time.Time `json:"at"`
}

type Override struct {
	ID        int64     `json:"id"`
	App       string    `json:"app"`
	Period    string    `json:"period"`
	Reason    string    `json:"reason"`
	Actor     string    `json:"actor"`
	CreatedAt time.Time `json:"created_at"`
	ExpiresAt time.Time `json:"expires_at"`
}

type Preview struct {
	Apps            []PreviewApp `json:"apps"`
	TotalReplicas   int          `json:"total_replicas"`
	RemainingDays   float64      `json:"remaining_days"`
	EstimatedSaving float64      `json:"estimated_saving"`
	DryRun          bool         `json:"dry_run"`
}

type PreviewApp struct {
	Name            string  `json:"name"`
	Replicas        int     `json:"replicas"`
	DailyRunRate    float64 `json:"daily_run_rate"`
	EstimatedSaving float64 `json:"estimated_saving"`
}

type State struct {
	App    string
	Period string
	// Enforced is "" or the action ingress must apply; Nudged is the value
	// ingress was last told about.
	Enforced    string
	Nudged      string
	Highest     int
	Pct         float64
	EgressBytes int64
	Cost        float64
	UpdatedAt   time.Time
}

type Status struct {
	Cap
	Period      string    `json:"period"`
	EgressBytes int64     `json:"egress_bytes"`
	Cost        float64   `json:"cost"`
	EgressPct   float64   `json:"egress_pct"`
	CostPct     float64   `json:"cost_pct"`
	Pct         float64   `json:"pct"`
	Enforced    string    `json:"enforced"`
	ResetsAt    time.Time `json:"resets_at"`
	Condition   Condition `json:"condition"`
}

type Store interface {
	ListSpendCaps(ctx context.Context) ([]Cap, error)
	GetSpendCap(ctx context.Context, app string) (Cap, error)
	SaveSpendCap(ctx context.Context, c Cap, ev Event, now time.Time) error
	DeleteSpendCap(ctx context.Context, app string, ev Event) error
	GetSpendCapState(ctx context.Context, app string) (State, bool, error)
	SaveSpendCapState(ctx context.Context, st State, events []Event) error
	ListSpendCapEvents(ctx context.Context, app string, limit int) ([]Event, error)
}

type Usage struct {
	EgressBytes int64
	Cost        float64
}

type UsageSource interface {
	SpendCapUsage(ctx context.Context, now time.Time) (map[string]Usage, error)
}