v1alpha1

type Autoscaling struct {
	// MinReplicas is the lower limit for the number of replicas to which the
	// autoscaler can scale down. This value must be greater than zero and less
	// than the MaxReplicas.
	MinReplicas *int32 `json:"minReplicas,omitempty"`
	// MaxReplicas is the upper limit for the number of replicas to which the
	// autoscaler can scale up. This value cannot be less than the replica
	// count for your workload.
	MaxReplicas int32 `json:"maxReplicas"`
	// AverageCPUUtilization is the target value of the average of the resource
	// cpu metric across all relevant pods. This is represented as a percentage
	// of the requested value of the resource for the pods.
	AverageCPUUtilization *int32 `json:"averageCPUUtilization,omitempty"`
	// AverageMemoryUtilization is the target value of the average of the
	// resource memory metric across all relevant pods. This is represented as
	// a percentage of the requested value of the resource for the pods.
	AverageMemoryUtilization *int32 `json:"averageMemoryUtilization,omitempty"`
	// ScaleDownStabilizationWindowSeconds is the number of seconds for which
	// past recommendations should be considered when scaling down. A shorter
	// window will trigger scale down events quicker, but too short a window
	// may cause replica flapping when metrics used for scaling keep fluctuating.
	ScaleDownStabilizationWindowSeconds *int32 `json:"scaleDownStabilizationWindowSeconds,omitempty"`
}

type ClusterConfig struct {
	// IstioConfig overrides for a cluster.
	IstioConfig `json:",inline"`
	// GlobalLabels applied to all resources in addition to stock labels.
	GlobalLabels map[string]string `json:"globalLabels,omitempty"`
	// Image used to launch cluster nodes.
	Image *OCIImageDefinition `json:"image,omitempty"`
	// NetworkPolicy parameters used to IP traffic flow.
	NetworkPolicy NetworkPolicyConfig `json:"networkPolicy,omitempty"`
	// ServiceAccount parameters used to override default behavior.
	ServiceAccount ServiceAccountConfig `json:"serviceAccount,omitempty"`
	// KerberosKeytab parameters used to add kerberos authentication.
	KerberosKeytab *KerberosKeytabConfig `json:"kerberosKeytab,omitempty"`
	// ImagePullSecrets are references to secrets with pull credentials to
	// private registries where cluster image are stored.
	ImagePullSecrets []corev1.LocalObjectReference `json:"imagePullSecrets,omitempty"`
	// PodSecurityContext added to every cluster pod.
	PodSecurityContext *corev1.PodSecurityContext `json:"podSecurityContext,omitempty"`
	// EnvVars added to all every cluster container.
	EnvVars []corev1.EnvVar `json:"envVars,omitempty"`
	// PodSecurityPolicy name can be provided to restrict and/or provide
	// execution permissions to processes running within cluster pods.
	PodSecurityPolicy string `json:"podSecurityPolicy,omitempty"`
}

type ClusterStatusConfig struct {
	ClusterStatus ClusterStatusType `json:"clusterStatus,omitempty"`
	// Reason may contain additional information when status is "Failed"
	Reason    string       `json:"reason,omitempty"`
	StartTime *metav1.Time `json:"startTime,omitempty"`
	// Image is the canonical reference url to the cluster container image.
	Image string `json:"image,omitempty"`
	// Nodes are pods that comprise the cluster.
	Nodes []string `json:"nodes,omitempty"`
	// WorkerReplicas is the `scale.status.replicas` subresource field.
	WorkerReplicas int32 `json:"workerReplicas,omitempty"`
	// WorkerSelector is the `scale.status.selector` subresource field.
	WorkerSelector string `json:"workerSelector,omitempty"`
}

type ClusterStatusType string

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

	Spec   DaskClusterSpec   `json:"spec,omitempty"`
	Status DaskClusterStatus `json:"status,omitempty"`
}

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

type DaskClusterSpec struct {
	ScalableClusterConfig `json:",inline"`

	Scheduler WorkloadConfig    `json:"scheduler,omitempty"`
	Worker    DaskClusterWorker `json:"worker,omitempty"`

	SchedulerPort int32 `json:"schedulerPort,omitempty"`
	DashboardPort int32 `json:"dashboardPort,omitempty"`
	WorkerPort    int32 `json:"workerPort,omitempty"`
	NannyPort     int32 `json:"nannyPort,omitempty"`

	// AdditionalClientPorts are extra ports through which cluster nodes could connect to the client.
	AdditionalClientPorts []corev1.ServicePort `json:"additionalClientPorts,omitempty"`
}

type DaskClusterStatus struct {
	ClusterStatusConfig `json:",inline"`
}

type DaskClusterWorker struct {
	WorkloadConfig `json:",inline"`
	Replicas       *int32 `json:"replicas,omitempty"`
}

type IstioConfig struct {
	// MutualTLSMode will be used to create a workload-specific peer
	// authentication policy that takes precedence over a global and/or
	// namespace-wide policy.
	MutualTLSMode string `json:"istioMutualTLSMode,omitempty"`
}

type KerberosKeytabConfig struct {
	// Contents are the binary data stored in the keytab file.
	Contents []byte `json:"contents,omitempty"`
	// MountPath where the keytab file should be created inside a pod.
	MountPath string `json:"mountPath,omitempty"`
}

type MPICluster struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`
	Spec              MPIClusterSpec      `json:"spec,omitempty"`
	Status            ClusterStatusConfig `json:"status,omitempty"`
}

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

type MPIClusterSpec struct {
	ClusterConfig `json:",inline"`
	Worker        MPIClusterWorker `json:"worker,omitempty"`

	// WorkerPorts specifies the range of ports used by worker processes for communication.
	WorkerPorts []int32 `json:"workerPorts,omitempty"`
	// AdditionalClientPorts are extra ports through which cluster nodes could connect to the client.
	AdditionalClientPorts []corev1.ServicePort `json:"additionalClientPorts,omitempty"`
}

type MPIClusterWorker struct {
	WorkloadConfig  `json:",inline"`
	Replicas        *int32 `json:"replicas,omitempty"`
	SharedSSHSecret string `json:"sharedSSHSecret"`
	UserName        string `json:"userName,omitempty"`
	UserID          *int64 `json:"userID,omitempty"`
	GroupName       string `json:"groupName,omitempty"`
	GroupID         *int64 `json:"groupID,omitempty"`
	HomeDir         string `json:"homeDir,omitempty"`
}

type NetworkPolicyConfig struct {
	// Enabled controls the creation of network policies that limit and provide
	// ingress access to the cluster nodes.
	Enabled *bool `json:"enabled,omitempty"`
	// ClientLabels defines the pod selector clause that grants ingress access
	// to the cluster client port(s).
	ClientLabels map[string]string `json:"clientLabels,omitempty"`
	// Defines the pod selector clause that grants ingress
	// access to the cluster dashboard.
	DashboardLabels map[string]string `json:"dashboardLabels,omitempty"`
	// Defines the namespace selector clause that grants ingress
	// access to the cluster dashboard.
	DashboardNamespaceLabels map[string]string `json:"dashboardNamespaceLabels,omitempty"`
}

type OCIImageDefinition struct {
	// Registry where the container image is hosted.
	Registry string `json:"registry,omitempty"`
	// Repository where the container image is stored.
	Repository string `json:"repository,omitempty"`
	// Tag points to a specific container image variant.
	Tag string `json:"tag,omitempty"`
	// PullPolicy used to fetch container image.
	PullPolicy corev1.PullPolicy `json:"pullPolicy,omitempty"`
}

type PersistentVolumeClaimTemplate struct {
	// Name is the unique metadata ID of the volume claim.
	Name string `json:"name"`
	// Spec describes the storage attributes of the underlying claim.
	Spec corev1.PersistentVolumeClaimSpec `json:"spec"`
}

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

	Spec   RayClusterSpec      `json:"spec,omitempty"`
	Status ClusterStatusConfig `json:"status,omitempty"`
}

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

type RayClusterSpec struct {
	ScalableClusterConfig `json:",inline"`

	// Head node configuration parameters.
	Head WorkloadConfig `json:"head,omitempty"`
	// Worker node configuration parameters.
	Worker RayClusterWorker `json:"worker,omitempty"`

	// Port is the port of the head ray process.
	Port int32 `json:"port,omitempty"`
	// RedisShardPorts is a list of ports for non-primary Redis shards.
	RedisShardPorts []int32 `json:"redisShardPorts,omitempty"`
	// ClientServerPort is the port number to which the ray client server will
	// bind. This port is used by external clients to submit work.
	ClientServerPort int32 `json:"clientServerPort,omitempty"`
	// ObjectManagerPort is the raylet port for the object manager.
	ObjectManagerPort int32 `json:"objectManagerPort,omitempty"`
	// NodeManagerPort is the raylet port for the node manager.
	NodeManagerPort int32 `json:"nodeManagerPort,omitempty"`
	// GCSServerPort is the port for the global control store.
	GCSServerPort int32 `json:"gcsServerPort,omitempty"`
	// WorkerPorts specifies the range of ports used by worker processes.
	WorkerPorts []int32 `json:"workerPorts,omitempty"`
	// ObjectStoreMemoryBytes is initial amount of memory with which to start
	// the object store.
	ObjectStoreMemoryBytes *int64 `json:"objectStoreMemoryBytes,omitempty"`
	// DashboardPort is the port used by the dashboard server.
	DashboardPort int32 `json:"dashboardPort,omitempty"`
	// EnableDashboard starts the dashboard web UI.
	EnableDashboard *bool `json:"enableDashboard,omitempty"`
	// AdditionalClientPorts are extra ports through which cluster nodes could connect to the client.
	AdditionalClientPorts []corev1.ServicePort `json:"additionalClientPorts,omitempty"`
}

type RayClusterWorker struct {
	WorkloadConfig `json:",inline"`

	// Replicas configures the total number of workers in the cluster. This
	// field behaves differently when Autoscaling is enabled. If
	// Autoscaling.MinReplicas is unspecified, then the minimum number of
	// replicas will be set to this value. Additionally, you can specify an
	// "initial cluster size" by setting this field to some value above the
	// minimum number of replicas.
	Replicas *int32 `json:"replicas,omitempty"`
}

type ScalableClusterConfig struct {
	// ClusterConfig base options.
	ClusterConfig `json:",inline"`
	// Autoscaling parameters used to scale up/down cluster nodes.
	Autoscaling *Autoscaling `json:"autoscaling,omitempty"`
}

type ServiceAccountConfig struct {
	// Name of an existing service account used by cluster workloads. This
	// field will disable the creation of a dedicated cluster service account.
	Name string `json:"name,omitempty"`
	// AutomountServiceAccountToken into workload pods. This field is only used
	// when creating a dedicted cluster service account.
	AutomountServiceAccountToken bool `json:"automountServiceAccountToken,omitempty"`
}

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

	Spec   SparkClusterSpec    `json:"spec,omitempty"`
	Status ClusterStatusConfig `json:"status,omitempty"`
}

type SparkClusterDriver struct {
	// Port used for communication by the driver.
	Port int32 `json:"port,omitempty"`
	// UIPort used by the driver.
	UIPort int32 `json:"uiPort,omitempty"`
	// BlockManagerPort used by the driver.
	BlockManagerPort int32 `json:"blockManagerPort,omitempty"`
	// Selector labels for driver pod(s).
	Selector map[string]string `json:"selector,omitempty"`
}

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

type SparkClusterNode struct {
	WorkloadConfig `json:",inline"`

	// DefaultConfiguration can be used to tune the execution environment for
	// your Spark applications. The values provided will be used to construct
	// the spark-defaults.conf file.
	DefaultConfiguration map[string]string `json:"defaultConfiguration,omitempty"`
}

type SparkClusterSpec struct {
	ScalableClusterConfig `json:",inline"`

	// Master node configuration parameters.
	Master SparkClusterNode `json:"master,omitempty"`
	// Worker node configuration parameters.
	Worker SparkClusterWorker `json:"worker,omitempty"`
	// Driver configures the SparkCluster to communicate with the Spark Driver.
	Driver SparkClusterDriver `json:"driver,omitempty"`

	// EnvoyFilterLabels are specific labels that must already exist on the
	// spark-driver so that users can set idle_timeout properly using the
	// EnvoyFilter resource.
	EnvoyFilterLabels map[string]string `json:"envoyFilterLabels,omitempty"`
	// WorkerMemoryLimit configures the SPARK_WORKER_MEMORY envVar.
	WorkerMemoryLimit string `json:"workerMemoryLimit,omitempty"`
	// ClusterPort is the port used for master/worker/driver communication.
	ClusterPort int32 `json:"clusterPort,omitempty"`
	// MasterWebPort is the port for the master web UI.
	MasterWebPort int32 `json:"masterWebPort,omitempty"`
	// WorkerWebPort is the port for the worker web UI.
	WorkerWebPort int32 `json:"workerWebPort,omitempty"`
	// AdditionalClientPorts are extra ports through which cluster nodes could connect to the client.
	AdditionalClientPorts []corev1.ServicePort `json:"additionalClientPorts,omitempty"`
}

type SparkClusterWorker struct {
	SparkClusterNode `json:",inline"`

	// Replicas configures the total number of workers in the cluster. This
	// field behaves differently when Autoscaling is enabled. If
	// Autoscaling.MinReplicas is unspecified, then the minimum number of
	// replicas will be set to this value. Additionally, you can specify an
	// "initial cluster size" by setting this field to some value above the
	// minimum number of replicas.
	Replicas *int32 `json:"replicas,omitempty"`

	// Obsolete value of WorkerMemoryLimit used in previous
	// versions; used here only to check compatibility of CRDs.
	ObsoleteWorkerMemoryLimit string `json:"workerMemoryLimit,omitempty"`
}

type WorkloadConfig struct {
	// Labels applied to cluster pods in addition to stock labels.
	Labels map[string]string `json:"labels,omitempty"`
	// Annotations applied to cluster pods.
	Annotations map[string]string `json:"annotations,omitempty"`
	// NodeSelector applied to cluster pods.
	NodeSelector map[string]string `json:"nodeSelector,omitempty"`
	// Resources are the requests and limits applied to cluster containers.
	Resources corev1.ResourceRequirements `json:"resources,omitempty"`
	// Affinity applied to cluster pods.
	Affinity *corev1.Affinity `json:"affinity,omitempty"`
	// Tolerations applied to cluster pods.
	Tolerations []corev1.Toleration `json:"tolerations,omitempty"`
	// InitContainers added to cluster pods.
	InitContainers []corev1.Container `json:"initContainers,omitempty"`
	// Volumes added to cluster pods.
	Volumes []corev1.Volume `json:"volumes,omitempty"`
	// VolumeMounts added to cluster containers.
	VolumeMounts []corev1.VolumeMount `json:"volumeMounts,omitempty"`
	// VolumeClaimTemplates is a list of claims that cluster pods are allowed
	// to reference. You can enable dynamic provisioning of additional storage
	// on-demand by using a storage class provisioner.
	VolumeClaimTemplates []PersistentVolumeClaimTemplate `json:"volumeClaimTemplates,omitempty"`
	// Customize container security context
	SecurityContext *corev1.SecurityContext `json:"securityContext,omitempty"`
}