NVIPAMPool CRD

Signed-off-by: Fred Rolland <[email protected]>

Makefile

@@ -127,11 +127,13 @@ ENVTEST ?= $(LOCALBIN)/setup-envtest
 GOLANGCILINT ?= $(LOCALBIN)/golangci-lint
 GCOV2LCOV ?= $(LOCALBIN)/gcov2lcov
 MOCKERY ?= $(LOCALBIN)/mockery
+CONTROLLER_GEN ?= $(LOCALBIN)/controller-gen
 
 ## Tool Versions
 GOLANGCILINT_VERSION ?= v1.52.2
 GCOV2LCOV_VERSION ?= v1.0.5
 MOCKERY_VERSION ?= v2.27.1
+CONTROLLER_GEN_VERSION ?= v0.11.1
 
 .PHONY: envtest
 envtest: $(ENVTEST) ## Download envtest-setup locally if necessary.
@@ -153,6 +155,19 @@ mockery: $(MOCKERY) ## Download mockery locally if necessary.
 $(MOCKERY): | $(LOCALBIN)
 	GOBIN=$(LOCALBIN) go install github.com/vektra/mockery/v2@$(MOCKERY_VERSION)
 
+.PHONY: controller-gen
+controller-gen: $(CONTROLLER_GEN) ## Download controller-gen locally if necessary
+$(CONTROLLER_GEN): | $(LOCALBIN)
+	GOBIN=$(LOCALBIN) go install sigs.k8s.io/controller-tools/cmd/controller-gen@$(CONTROLLER_GEN_VERSION)
+
+.PHONY: generate
+generate: controller-gen
+	$(CONTROLLER_GEN) object object:headerFile="./boilerplate.go.txt" paths="./api/..."
+
+.PHONY: manifests
+manifests: controller-gen
+	$(CONTROLLER_GEN) crd paths="./api/..." output:dir="./deploy"
+
 .PHONY: clean
 clean: ## Remove downloaded tools and compiled binaries
 	@rm -rf $(LOCALBIN)

api/v1alpha1/groupversion_info.go

@@ -0,0 +1,32 @@
+/*
+ Copyright 2023, NVIDIA CORPORATION & AFFILIATES
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+     http://www.apache.org/licenses/LICENSE-2.0
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+*/
+
+//+kubebuilder:object:generate=true
+//+groupName=nvipam.nvidia.com
+package v1alpha1
+
+import (
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"sigs.k8s.io/controller-runtime/pkg/scheme"
+)
+
+var (
+	// GroupVersion is group version used to register these objects
+	GroupVersion = schema.GroupVersion{Group: "nvipam.nvidia.com", Version: "v1alpha1"}
+
+	// SchemeBuilder is used to add go types to the GroupVersionKind scheme
+	SchemeBuilder = &scheme.Builder{GroupVersion: GroupVersion}
+
+	// AddToScheme adds the types in this group-version to the given scheme.
+	AddToScheme = SchemeBuilder.AddToScheme
+)

api/v1alpha1/ipampool_type.go

@@ -0,0 +1,64 @@
+/*
+ Copyright 2023, NVIDIA CORPORATION & AFFILIATES
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+     http://www.apache.org/licenses/LICENSE-2.0
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+*/
+
+package v1alpha1
+
+import metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+import corev1 "k8s.io/api/core/v1"
+
+//+kubebuilder:object:root=true
+//+kubebuilder:subresource:status
+// IPAMPool contains configuration for IPAM controller
+type IPAMPool struct {
+	metav1.TypeMeta   `json:",inline"`
+	metav1.ObjectMeta `json:"metadata,omitempty"`
+	Spec              IPAMPoolSpec `json:"spec,omitempty"`
+	Status            IPAMPoolStatus `json:"status,omitempty"`
+}
+
+// PoolConfig contains configuration for IP pool
+type IPAMPoolSpec struct {
+	// subnet of the pool
+	Subnet string `json:"subnet"`
+	// amount of IPs to allocate for each node,
+	// must be less than amount of available IPs in the subnet
+	PerNodeBlockSize int `json:"perNodeBlockSize"`
+	// gateway for the pool, defaults to the first IP of the Subnet if not set
+	Gateway string `json:"gateway"`
+		// selector for nodes, if empty match all nodes
+	NodeSelectorTerms []corev1.NodeSelectorTerm `json:"nodeSelectorTerms"`
+}
+
+type IPAMPoolStatus struct {
+	// IP allocations for Nodes
+	Allocations []Allocation `json:"allocations"`
+}
+
+// Allocation contains IP Allocation for a specific Node
+type Allocation struct {
+	Name    string `json:"NodeName"`
+	StartIP string `json:"startIP"`
+	EndIP   string `json:"endIP"`
+}
+
+//+kubebuilder:object:root=true
+// IPAMPoolList contains a list of IPAMPool
+type IPAMPoolList struct {
+	metav1.TypeMeta `json:",inline"`
+	metav1.ListMeta `json:"metadata,omitempty"`
+	Items           []IPAMPool `json:"items"`
+}
+
+func init() {
+	SchemeBuilder.Register(&IPAMPool{}, &IPAMPoolList{})
+}

boilerplate.go.txt

@@ -0,0 +1,12 @@
+/*
+ Copyright 2023, NVIDIA CORPORATION & AFFILIATES
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+     http://www.apache.org/licenses/LICENSE-2.0
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+*/