v1alpha1

type ETReplicaSpec struct {
	// Replicas is the desired number of replicas of the given template.
	// If unspecified, defaults to 1.
	Replicas *int32 `json:"Replicas,omitempty"`

	// MaxReplicas is the desired max number of replicas of the given template.
	// If unspecified, MaxReplicas defaults to infinite.
	MaxReplicas *int32 `json:"maxReplicas,omitempty"`

	// MinReplicas is the desired min number of replicas of the given template.
	// If unspecified, MinReplicas defaults to InitReplicas
	MinReplicas *int32 `json:"minReplicas,omitempty"`

	// Template is the object that describes the pod that
	// will be created for this replica. RestartPolicy in PodTemplateSpec
	// will be overide by RestartPolicy in ReplicaSpec
	Template v1.PodTemplateSpec `json:"template,omitempty"`

	// Restart policy for all replicas within the job.
	// One of Always, OnFailure, Never and ExitCode.
	// Default to Never.
	RestartPolicy *common.RestartPolicy `json:"restartPolicy,omitempty"`
}

type ETReplicaSpecs struct {
	Launcher *common.ReplicaSpec `json:"Launcher"`
	Worker   *ETReplicaSpec      `json:"Worker"`
}

type ETReplicaType common.ReplicaType

type EnvSpec struct {
	Name  string `json:"name,omitempty"`
	Value string `json:"value,omitempty"`
}

type ScaleIn struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`

	Spec ScaleInSpec `json:"spec,omitempty"`
	// Most recently observed status of the PyTorchJob.
	// Read-only (modified by the system).
	Status common.JobStatus `json:"status,omitempty"`
}

type ScaleInList struct {
	metav1.TypeMeta `json:",inline"`
	metav1.ListMeta `json:"metadata,omitempty"`
	Items           []ScaleIn `json:"items"`
}

type ScaleInSpec struct {
	Script string `json:"script,omitempty"`

	// Optional number of retries to execute script.
	// +optional
	BackoffLimit *int32 `json:"backoffLimit,omitempty"`

	// Optional number of timeout to execute script.
	// +optional
	Timeout *int32 `json:"timeout,omitempty"`

	ToDelete *ToDeleteSpec `json:"toDelete,omitempty"`

	Selector Selector `json:"selector,omitempty"`

	Env []EnvSpec `json:"env,omitempty"`
}

type ScaleInStatus struct {
}

type ScaleOut struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`

	Spec ScaleOutSpec `json:"spec,omitempty"`
	// Most recently observed status of the PyTorchJob.
	// Read-only (modified by the system).
	Status common.JobStatus `json:"status,omitempty"`
}

type ScaleOutList struct {
	metav1.TypeMeta `json:",inline"`
	metav1.ListMeta `json:"metadata,omitempty"`
	Items           []ScaleOut `json:"items"`
}

type ScaleOutSpec struct {
	Script string `json:"script,omitempty"`

	// Optional number of retries to execute script.
	// +optional
	BackoffLimit *int32 `json:"backoffLimit,omitempty"`

	// Optional number of timeout to execute script.
	// +optional
	Timeout *int32 `json:"timeout,omitempty"`

	ToAdd *ToAddSpec `json:"toAdd,omitempty"`

	Selector Selector `json:"selector,omitempty"`

	Env []EnvSpec `json:"env,omitempty"`
}

type ScaleOutStatus struct {
	// INSERT ADDITIONAL STATUS FIELD - define observed state of cluster
	// Important: Run "make" to regenerate code after modifying this file
	ScaleOutStatus string `json:"scaleOutStatus,omitempty"`
}

type Selector struct {
	Name string `json:"name,omitempty"`
}

type ToAddSpec struct {
	Count *int32 `json:"count,omitempty"`
}

type ToDeleteSpec struct {
	Count    int      `json:"count,omitempty"`
	PodNames []string `json:"podNames,omitempty"`
}

type TrainingJob struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`

	Spec TrainingJobSpec `json:"spec,omitempty"`

	// Most recently observed status of the PyTorchJob.
	// Read-only (modified by the system).
	Status common.JobStatus `json:"status,omitempty"`
}

type TrainingJobList struct {
	metav1.TypeMeta `json:",inline"`
	metav1.ListMeta `json:"metadata,omitempty"`
	Items           []TrainingJob `json:"items"`
}

type TrainingJobSpec struct {

	// CleanPodPolicy defines the policy that whether to kill pods after the job completes.
	// Defaults to None.
	CleanPodPolicy *common.CleanPodPolicy `json:"cleanPodPolicy,omitempty"`

	// `ETReplicaSpecs` contains maps from `ETReplicaType` to `ReplicaSpec` that
	// specify the ET replicas to run.
	ETReplicaSpecs ETReplicaSpecs `json:"etReplicaSpecs"`

	// Specifies the number of slots per worker used in hostfile.
	// Defaults to 1.
	// +optional
	SlotsPerWorker *int32 `json:"slotsPerWorker,omitempty"`
}

type TrainingJobStatus struct {
}