internal

type BatchStrategy struct {
	GroupSize int32 `yaml:"groupSize"`
	AutoPause bool  `yaml:"autoPause"`
}

type Container struct {
	Docker *DockerContainer `yaml:"docker"`
}

type DockerContainer struct {
	Name string `yaml:"name"`
	Tag  string `yaml:"tag"`
}

type Executor struct {
	Name string `yaml:"name"`
	Data string `yaml:"data"`
}

type InstanceRange struct {
	First int32 `yaml:"first"`
	Last  int32 `yaml:"last"`
}

type Job struct {
	Environment         string            `yaml:"environment"`
	Role                string            `yaml:"role"`
	Name                string            `yaml:"name"`
	CPU                 float64           `yaml:"cpu"`
	RAM                 int64             `yaml:"ram"`
	Disk                int64             `yaml:"disk"`
	Port                int64             `yaml:"port"`
	GPU                 int64             `yaml:"gpu"`
	Executor            Executor          `yaml:"executor"`
	Instances           int32             `yaml:"instances"`
	MaxFailures         int32             `yaml:"maxFailures"`
	URIs                []URI             `yaml:"uris"`
	Metadata            map[string]string `yaml:"labels"`
	Service             bool              `yaml:"service"`
	Tier                string            `yaml:"tier,omitempty" default:"preemptible"`
	Priority            int32             `yaml:"priority"`
	Production          bool              `yaml:"production"`
	Thermos             []ThermosProcess  `yaml:",flow,omitempty"`
	Container           *Container        `yaml:"container,omitempty"`
	CronSchedule        *string           `yaml:"cronSchedule,omitempty"`
	CronCollisionPolicy *string           `yaml:"cronCollisionPolicy,omitempty"`
	ValueConstraints    []ValueConstraint `yaml:"valueConstraints,flow,omitempty"`
	LimitConstraints    []LimitConstraint `yaml:"limitConstraints,flow,omitempty"`
}

type LimitConstraint struct {
	Name  string `yaml:"name"`
	Limit int32  `yaml:"limit"`
}

type MonitorCmdConfig struct {
	Cmd                             *cobra.Command
	MonitorInterval, MonitorTimeout time.Duration
	StatusList                      []string
}

type QueueStrategy struct {
	GroupSize int32 `yaml:"groupSize"`
}

type ThermosProcess struct {
	Name string `yaml:"name"`
	Cmd  string `yaml:"cmd"`
}

type URI struct {
	URI     string `yaml:"uri"`
	Extract bool   `yaml:"extract"`
	Cache   bool   `yaml:"cache"`
}

type UpdateJob struct {
	JobConfig      Job            `yaml:"jobConfig"`
	UpdateSettings UpdateSettings `yaml:"updateSettings"`
}

type UpdateSettings struct {
	MaxPerInstanceFailures int32           `yaml:"maxPerInstanceFailures"`
	MaxFailedInstances     int32           `yaml:"maxFailedInstances"`
	MinTimeInRunning       time.Duration   `yaml:"minTimeInRunning"`
	RollbackOnFailure      bool            `yaml:"rollbackOnFailure"`
	InstanceRanges         []InstanceRange `yaml:"instanceRanges"`
	InstanceCount          int32           `yaml:"instanceCount"`
	PulseTimeout           time.Duration   `yaml:"pulseTimeout"`
	SLAAware               bool            `yaml:"slaAware"`
	Strategy               UpdateStrategy  `yaml:"strategy"`
}

type UpdateStrategy struct {
	VariableBatch *VariableBatchStrategy `yaml:"variableBatch"`
	Batch         *BatchStrategy         `yaml:"batch"`
	Queue         *QueueStrategy         `yaml:"queue"`
}

type ValueConstraint struct {
	Name    string   `yaml:"name"`
	Values  []string `yaml:"values"`
	Negated bool     `yaml:"negated"`
}

type VariableBatchStrategy struct {
	GroupSizes []int32 `yaml:"groupSizes"`
	AutoPause  bool    `yaml:"autoPause"`
}