agentupgrade

type Actor struct {
	UserID uint
	Name   string
	IP     string
}

type Batch struct {
	Percent            int `json:"percent"`
	MinDurationSeconds int `json:"min_duration_seconds"`
}

type BatchView struct {
	Index              int            `json:"index"`
	Percent            int            `json:"percent"`
	MinDurationSeconds int            `json:"min_duration_seconds"`
	Nodes              int            `json:"nodes"`
	States             map[string]int `json:"states"`
	// Offered and Failed are the batch's failure ratio's terms (H19):
	// skipped and pending nodes do not count.
	Offered int `json:"offered"`
	Failed  int `json:"failed"`
}

type CampaignView struct {
	model.AgentUpgradeCampaign
	Artifacts []agentcontrol.UpgradeArtifact `json:"artifacts"`
	Batches   []BatchView                    `json:"batches"`
	Exclude   Exclude                        `json:"exclude"`
	Total     int                            `json:"total"`
	States    map[string]int                 `json:"states"`
	// BatchEndsAt is the earliest the current batch can pass.
	BatchEndsAt *time.Time `json:"batch_ends_at"`
	Nodes       []NodeView `json:"nodes,omitempty"`
}

type Exclude struct {
	Nodes []string `json:"nodes"`
	Tags  []string `json:"tags"`
}

type NodeView struct {
	model.AgentUpgradeNode
	Node string `json:"node"`
}

type Service struct {
	DB *gorm.DB
	// Now is the clock; time.Now when nil.
	Now func() time.Time
}

type StartRequest struct {
	TargetVersion string
	// ControlVersion is the running Control's release (H25: the Agent
	// release is Control's); a newer target is refused.
	ControlVersion string
	// Artifacts are the verified release assets
	// (agentinstall.UpgradeArtifacts).
	Artifacts []agentcontrol.UpgradeArtifact
	// Batches default to DefaultBatches.
	Batches []Batch
	Exclude Exclude
	Reason  string
	Actor   Actor
}

type Worker struct {
	Service *Service
	// Streams returns the Agent Control streams; a nil result skips the
	// evaluation.
	Streams func() agentstreams.Streams
	// Interval defaults to DefaultInterval.
	Interval time.Duration
}