v1alpha1

type AzureBlobStorageDownloadItem struct {
	ConnectionString ValueReference        `json:"connectionString" yaml:"connectionString"`
	Containers       []CloudDownloadBucket `json:"containers"`
}

type CloudDownloadBucket struct {
	Name    string                `json:"name"`
	Files   []CloudDownloadFile   `json:"files,omitempty"`
	Folders []CloudDownloadFolder `json:"folders,omitempty"`
}

type CloudDownloadFile struct {
	Path       string `json:"path"`
	TargetPath string `json:"targetPath" yaml:"targetPath"`
}

type CloudDownloadFolder struct {
	Path       string `json:"path"`
	TargetPath string `json:"targetPath" yaml:"targetPath"`
	Glob       string `json:"glob,omitempty" yaml:"glob,omitempty"`
}

type ClusterQueue struct {
	Name string `json:"name"`

	Spec kueuev1beta1.ClusterQueueSpec `json:"spec,omitempty"`
}

type CommonMetaSpec struct {
	// Name is the name of the workload.
	Name string `json:"name,omitempty"`

	// Namespace defines the namespace in which the workload is deployed.
	Namespace string `json:"namespace,omitempty"`

	// Labels is a map of key-value pairs used for organizing workloads.
	Labels map[string]string `json:"labels,omitempty"`

	// Annotations provides additional metadata for the workload.
	Annotations map[string]string `json:"annotations,omitempty"`
}

type ConfigSource struct {
	// Type specifies the type of configuration source. Valid values are:
	// - "ConfigMap": Configuration is stored in a Kubernetes ConfigMap.
	// - "S3": Configuration is stored in an S3-compatible object storage bucket.
	Type string `json:"type,omitempty"`

	// ConfigMapName specifies the name of the ConfigMap to be mounted.
	// This field is only relevant if Type is "ConfigMap".
	ConfigMapName string `json:"configMapName,omitempty"`

	// S3BucketName specifies the name of the S3 bucket where configuration files are stored.
	// This field is only relevant if Type is "S3".
	S3BucketName string `json:"s3BucketName,omitempty"`

	// S3Path specifies an optional path within the S3 bucket where the configuration files reside.
	// This field is only relevant if Type is "S3".
	S3Path string `json:"s3Path,omitempty"`

	// S3Credentials references a Kubernetes Secret containing credentials for accessing the S3 bucket.
	// The Secret should have standard AWS credential keys (e.g., "AWS_ACCESS_KEY_ID" and "AWS_SECRET_ACCESS_KEY").
	// This field is only relevant if Type is "S3".
	S3Credentials *corev1.SecretReference `json:"s3Credentials,omitempty"`
}

type ConfigSourceItem struct {
	Path  string `json:"mountPath,omitempty"`
	Value string `json:"value,omitempty"`
}

type DataStorageSpec struct {
	// MountPath specifies where the data PVC will be mounted to in each pod
	MountPath string `json:"mountPath,omitempty"`

	// StorageSize specifies the amount of storage allocated to the data PVC
	StorageSize string `json:"storageSize,omitempty"`

	// Download optional object storage pre-downloads
	Download ObjectStorageDownloadSpec `json:"download,omitempty"`
}

type DownloadTaskConfig struct {
	// DownloadRoot specifies the common root for all the downloads
	DownloadRoot string `json:"downloadRoot" yaml:"downloadRoot"`

	// HfHome specifies path for $HF_HOME env variable
	HfHome string `json:"hfHome" yaml:"hfHome"`

	S3        []S3DownloadItem               `json:"s3,omitempty" yaml:"s3,omitempty"`
	GCS       []GCSDownloadItem              `json:"gcs,omitempty" yaml:"gcs,omitempty"`
	HF        []HuggingFaceDownloadItem      `json:"hf,omitempty" yaml:"hf,omitempty"`
	AzureBlob []AzureBlobStorageDownloadItem `json:"azureBlob,omitempty" yaml:"azureBlob,omitempty"`
}

type GCSDownloadItem struct {
	ApplicationCredentials ValueReference        `json:"applicationCredentials" yaml:"applicationCredentials"`
	Buckets                []CloudDownloadBucket `json:"buckets"`
}

type HfStorageSpec struct {
	// MountPath specifies where the data HF will be mounted to in each pod.
	// This is also used to set the HF_HOME environmental variable into each container.
	MountPath string `json:"mountPath,omitempty"`

	// StorageSize specifies the amount of storage allocated to the HF PVC
	StorageSize string `json:"storageSize,omitempty"`

	// PreCacheRepos is a list of repositories (and their files) that should be cached before the workload starts
	PreCacheRepos []HuggingFaceDownloadItem `json:"preCacheRepos,omitempty"`
}

type HuggingFaceDownloadItem struct {
	RepoID string   `json:"repoId" yaml:"repoId"`
	Files  []string `json:"files"`
}

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

	Spec   KaiwoJobSpec   `json:"spec,omitempty"`
	Status KaiwoJobStatus `json:"status,omitempty"`
}

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

type KaiwoJobSpec struct {
	CommonMetaSpec `json:",inline"`

	// User specifies the owner or creator of the workload.
	// If authentication is enabled, this must be email address which is checked against authenticated user for match.
	User string `json:"user"`

	// ClusterQueue is the Kueue ClusterQueue name.
	ClusterQueue string `json:"clusterQueue,omitempty"`

	// PriorityClass specifies the Kubernetes PriorityClass for scheduling.
	PriorityClass string `json:"priorityClass,omitempty"`

	// EntryPoint specifies the command or script executed in a Job or RayJob.
	// Can also be defined inside Job struct as Command in the form of string array or
	// inside RayJob struct as Entrypoint in the form of string
	EntryPoint string `json:"entrypoint,omitempty"`

	// Gpus specifies the total number of GPUs allocated to the workload.
	// Default is 0.
	// +kubebuilder:default=0
	Gpus int `json:"gpus,omitempty"`

	// GpuVendor specifies the GPU vendor (e.g., AMD, NVIDIA, etc.).
	// Default is AMD.
	// +kubebuilder:default=AMD
	GpuVendor string `json:"gpuVendor,omitempty"`

	// Version is an optional field specifying the version of the workload.
	Version string `json:"version,omitempty"`

	// Replicas specifies the number of replicas for the workload.
	// If greater than one, the workload must use Ray.
	// Default is 0.
	// +kubebuilder:default=0
	Replicas int `json:"replicas,omitempty"`

	// GpusPerReplica specifies the number of GPUs allocated per replica.
	GpusPerReplica int `json:"gpus-per-replica,omitempty"`

	// Image defines the container image used for the workload.
	Image string `json:"image,omitempty"`

	// ImagePullSecrets contains the list of secrets used to pull the container image.
	ImagePullSecrets []corev1.LocalObjectReference `json:"imagePullSecrets,omitempty"`

	// EnvVars specifies the environment variables to be passed to the container.
	EnvVars []corev1.EnvVar `json:"envVars,omitempty"`

	// Ray determines whether the operator should use RayCluster for workload execution.
	// Default is false.
	// +kubebuilder:default=false
	Ray bool `json:"ray"`

	// ConfigSource allows mounting from Git
	ConfigSource *ConfigSource `json:"configSource,omitempty"`

	// Storage configuration for the workload.
	Storage StorageSpec `json:"storage,omitempty"`

	// RayJob defines the RayJob configuration.
	RayJob *rayv1.RayJob `json:"rayJob,omitempty"`

	// Job defines the Kubernetes Job configuration.
	Job *batchv1.Job `json:"job,omitempty"`
}

type KaiwoJobStatus struct {
	StartTime          *metav1.Time       `json:"startTime,omitempty"`
	CompletionTime     *metav1.Time       `json:"completionTime,omitempty"`
	Conditions         []metav1.Condition `json:"conditions,omitempty"`
	Status             Status             `json:"Status,omitempty"`
	Duration           int64              `json:"duration,omitempty"`
	ObservedGeneration int64              `json:"observedGeneration,omitempty"`
}

type KaiwoQueueConfig struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`
	Spec              KaiwoQueueConfigSpec   `json:"spec,omitempty"`
	Status            KaiwoQueueConfigStatus `json:"status,omitempty"`
}

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

type KaiwoQueueConfigSpec struct {
	// +kubebuilder:validation:MaxItems=10
	ClusterQueues []ClusterQueue `json:"clusterQueues,omitempty"`

	ResourceFlavors []ResourceFlavorSpec `json:"resourceFlavors,omitempty"`

	// +kubebuilder:validation:MaxItems=5
	WorkloadPriorityClasses []kueuev1beta1.WorkloadPriorityClass `json:"workloadPriorityClasses,omitempty"`
}

type KaiwoQueueConfigStatus struct {
	Conditions []metav1.Condition `json:"conditions,omitempty"`
	Status     Status             `json:"Status,omitempty"`
}

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

	Spec   KaiwoServiceSpec   `json:"spec,omitempty"`
	Status KaiwoServiceStatus `json:"status,omitempty"`
}

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

type KaiwoServiceSpec struct {
	CommonMetaSpec `json:",inline"`

	// User specifies the owner or creator of the workload.
	// If authentication is enabled, this must be email address which is checked against authenticated user for match.
	User string `json:"user"`

	// ClusterQueue is the Kueue ClusterQueue name.
	ClusterQueue string `json:"clusterQueue,omitempty"`

	// PriorityClass specifies the Kubernetes PriorityClass for scheduling.
	PriorityClass string `json:"priorityClass,omitempty"`

	// EntryPoint specifies the command or script executed in a Deployment.
	// Can also be defined inside Deployment struct as regular command in the form of string array
	EntryPoint string `json:"entrypoint,omitempty"`

	// Defines the applications and deployments to deploy, should be a YAML multi-line scalar string.
	// Can also be defined inside RayService struct
	ServeConfigV2 string `json:"serveConfigV2,omitempty"`

	// Gpus specifies the total number of GPUs allocated to the workload.
	// Default is 0.
	// +kubebuilder:default=0
	Gpus int `json:"gpus,omitempty"`

	// GpuVendor specifies the GPU vendor (e.g., AMD, NVIDIA, etc.).
	// Default is AMD.
	// +kubebuilder:default=AMD
	GpuVendor string `json:"gpuVendor,omitempty"`

	// Version is an optional field specifying the version of the workload.
	Version string `json:"version,omitempty"`

	// Replicas specifies the number of replicas for the workload.
	// If greater than one, the workload must use Ray.
	// Default is 0.
	// +kubebuilder:default=0
	Replicas int `json:"replicas,omitempty"`

	// GpusPerReplica specifies the number of GPUs allocated per replica.
	GpusPerReplica int `json:"gpus-per-replica,omitempty"`

	// Image defines the container image used for the workload.
	Image string `json:"image,omitempty"`

	// ImagePullSecrets contains the list of secrets used to pull the container image.
	ImagePullSecrets []corev1.LocalObjectReference `json:"imagePullSecrets,omitempty"`

	// EnvVars specifies the environment variables to be passed to the container.
	EnvVars []corev1.EnvVar `json:"envVars,omitempty"`

	// Ray determines whether the operator should use RayCluster for workload execution.
	// Default is false.
	// +kubebuilder:default=false
	Ray bool `json:"ray"`

	// ConfigSource allows mounting from Git repository
	ConfigSource *ConfigSource `json:"configSource,omitempty"`

	// Storage configuration for the workload.
	Storage StorageSpec `json:"storage,omitempty"`

	// Optional workload-specific configs (Pointers to avoid bloating CRD)
	RayService *rayv1.RayService  `json:"rayService,omitempty"`
	Deployment *appsv1.Deployment `json:"deployment,omitempty"`
}

type KaiwoServiceStatus struct {
	Conditions []metav1.Condition `json:"conditions,omitempty"`
	Status     Status             `json:"Status,omitempty"`
}

type ObjectStorageDownloadSpec struct {
	// S3 lists any S3 downloads
	S3 []S3DownloadItem `json:"s3,omitempty"`

	// GCS lists and Google Cloud Storage downloads
	GCS []GCSDownloadItem `json:"gcs,omitempty"`

	// AzureBlob lists any Azure Blob Storage downloads
	AzureBlob []AzureBlobStorageDownloadItem `json:"azureBlob,omitempty"`
}

type ResourceFlavorSpec struct {
	Name string `json:"name"`
	// NodeLabels must not exceed a reasonable limit to prevent CRD validation failures
	// +kubebuilder:validation:MaxProperties=10
	NodeLabels map[string]string `json:"nodeLabels,omitempty"`

	// +kubebuilder:validation:MaxItems=5
	Taints []corev1.Taint `json:"taints,omitempty"`

	// // +kubebuilder:validation:MaxItems=5
	Tolerations []corev1.Toleration `json:"tolerations,omitempty"`
}

type S3DownloadItem struct {
	// EndpointUrl is the endpoint of the S3 API
	EndpointUrl ValueReference `json:"endpointUrl" yaml:"endpointUrl"`

	// AccessKeyId
	AccessKeyId ValueReference        `json:"accessKeyId" yaml:"accessKeyId"`
	SecretKey   ValueReference        `json:"secretKey" yaml:"secretKey"`
	Buckets     []CloudDownloadBucket `json:"buckets"`
}

type Status string

type StorageSpec struct {
	// StorageEnabled tells whether to enable persistent storage.
	StorageEnabled bool `json:"storageEnabled,omitempty"`

	// StorageClassName specifies the storage class used for PVC.
	StorageClassName string `json:"storageClassName,omitempty"`

	// AccessMode determines the access mode for the storage
	// +kubebuilder:default=ReadWriteMany
	AccessMode corev1.PersistentVolumeAccessMode `json:"accessMode"`

	// Data specifies the main workload PVC and optional object storage pre-downloads
	Data DataStorageSpec `json:"data,omitempty"`

	// HuggingFace specifies any hugging face models that should be cached before the workload starts
	HuggingFace HfStorageSpec `json:"huggingFace,omitempty"`
}

type ValueReference struct {
	// File determines the location of the secret value mounted on the disk
	File string `json:"file,omitempty"`

	// SecretName is the name of the secret where the value is kept
	SecretName string `json:"secretName,omitempty" yaml:"secretName,omitempty"`

	// SecretKey is the name of the key within the secret where the value is kept
	SecretKey string `json:"secretKey,omitempty" yaml:"secretKey,omitempty"`
}