Add DeviceSelector types

Signed-off-by: Fredrik Lönnegren <[email protected]>
rancher · Nov 2, 2023 · 758eb92 · 758eb92
1 parent de0d098
commit 758eb92
Show file tree

Hide file tree

Showing 4 changed files with 118 additions and 0 deletions.
diff --git a/api/v1beta1/types.go b/api/v1beta1/types.go
@@ -26,6 +26,8 @@ type Install struct {
 	// +optional
 	Device string `json:"device,omitempty" yaml:"device,omitempty"`
 	// +optional
+	DeviceSelector DeviceSelector `json:"device-selector,omitempty" yaml:"device-selector,omitempty"`
+	// +optional
 	NoFormat bool `json:"no-format,omitempty" yaml:"no-format,omitempty"`
 	// +optional
 	ConfigURLs []string `json:"config-urls,omitempty" yaml:"config-urls,omitempty"`
@@ -120,3 +122,31 @@ type Config struct {
 	// +optional
 	CloudConfig map[string]runtime.RawExtension `json:"cloud-config,omitempty" yaml:"cloud-config,omitempty"`
 }
+
+type DeviceSelector []DeviceSelectorRequirement
+
+type DeviceSelectorRequirement struct {
+	// +required
+	// +kubebuilder:validation:Required
+	// +kubebuilder:validation:Enum=Name;Size
+	Key DeviceSelectorKey `json:"key"`
+	// +required
+	// +kubebuilder:validation:Required
+	// +kubebuilder:validation:Enum=In;NotIn;Gt;Lt
+	Operator DeviceSelectorOperator `json:"operator"`
+	// +optional
+	Values []string `json:"values,omitempty"`
+}
+
+type DeviceSelectorKey string
+type DeviceSelectorOperator string
+
+const (
+	DeviceSelectorOpIn    DeviceSelectorOperator = "In"
+	DeviceSelectorOpNotIn DeviceSelectorOperator = "NotIn"
+	DeviceSelectorOpGt    DeviceSelectorOperator = "Gt"
+	DeviceSelectorOpLt    DeviceSelectorOperator = "Lt"
+
+	DeviceSelectorKeyName DeviceSelectorKey = "Name"
+	DeviceSelectorKeySize DeviceSelectorKey = "Size"
+)
diff --git a/api/v1beta1/zz_generated.deepcopy.go b/api/v1beta1/zz_generated.deepcopy.go
diff --git a/charts/crds/templates/crds.yaml b/charts/crds/templates/crds.yaml
@@ -669,6 +669,22 @@ spec:
                             type: boolean
                           device:
                             type: string
+                          device-selector:
+                            items:
+                              properties:
+                                key:
+                                  type: string
+                                operator:
+                                  type: string
+                                values:
+                                  items:
+                                    type: string
+                                  type: array
+                              required:
+                              - key
+                              - operator
+                              type: object
+                            type: array
                           disable-boot-entry:
                             type: boolean
                           eject-cd:

diff --git a/config/crd/bases/elemental.cattle.io_machineregistrations.yaml b/config/crd/bases/elemental.cattle.io_machineregistrations.yaml
@@ -53,6 +53,30 @@ spec:
                             type: boolean
                           device:
                             type: string
+                          device-selector:
+                            items:
+                              properties:
+                                key:
+                                  enum:
+                                  - Name
+                                  - Size
+                                  type: string
+                                operator:
+                                  enum:
+                                  - In
+                                  - NotIn
+                                  - Gt
+                                  - Lt
+                                  type: string
+                                values:
+                                  items:
+                                    type: string
+                                  type: array
+                              required:
+                              - key
+                              - operator
+                              type: object
+                            type: array
                           disable-boot-entry:
                             type: boolean
                           eject-cd: