Add ability to generate MultiNodeEnvironment CRD, client, and deepcopy

Signed-off-by: Kevin Klues <[email protected]>

Author: klueska

Makefile

@@ -101,26 +101,87 @@ coverage: test
 	cat $(COVERAGE_FILE) | grep -v "_mock.go" > $(COVERAGE_FILE).no-mocks
 	go tool cover -func=$(COVERAGE_FILE).no-mocks
 
-generate: generate-deepcopy fmt
+generate: generate-crds fmt
 
-generate-deepcopy: .remove-deepcopy
+generate-crds: generate-deepcopy
+	for dir in $(CLIENT_SOURCES); do \
+		controller-gen crd:crdVersions=v1 \
+			paths=$(CURDIR)/$${dir} \
+			output:crd:dir=$(CURDIR)/deployments/helm/$(DRIVER_NAME)/crds; \
+	done
+
+
+generate-deepcopy: generate-informers
 	for dir in $(DEEPCOPY_SOURCES); do \
 		controller-gen \
 			object:headerFile=$(CURDIR)/hack/boilerplate.go.txt,year=$(shell date +"%Y") \
 			paths=$(CURDIR)/$${dir}/ \
 			output:object:dir=$(CURDIR)/$${dir}; \
 	done
 
+generate-informers: vendor generate-listers
+	informer-gen \
+		--go-header-file=$(CURDIR)/hack/boilerplate.go.txt \
+		--output-package "$(MODULE)/$(PKG_BASE)/informers" \
+        --input-dirs "$(shell for api in $(CLIENT_APIS); do echo -n "$(MODULE)/$(API_BASE)/$$api,"; done | sed 's/,$$//')" \
+		--output-base "$(CURDIR)/pkg/tmp_informers" \
+		--versioned-clientset-package "$(MODULE)/$(PKG_BASE)/clientset/versioned" \
+		--listers-package "$(MODULE)/$(PKG_BASE)/listers"
+	mkdir -p $(CURDIR)/$(PKG_BASE)
+	mv $(CURDIR)/pkg/tmp_informers/$(MODULE)/$(PKG_BASE)/informers \
+	   $(CURDIR)/$(PKG_BASE)/informers
+	rm -rf $(CURDIR)/pkg/tmp_informers
+
+generate-listers: generate-clientset
+	lister-gen \
+		--go-header-file=$(CURDIR)/hack/boilerplate.go.txt \
+		--output-package "$(MODULE)/$(PKG_BASE)/listers" \
+        --input-dirs "$(shell for api in $(CLIENT_APIS); do echo -n "$(MODULE)/$(API_BASE)/$$api,"; done | sed 's/,$$//')" \
+		--output-base "$(CURDIR)/pkg/tmp_listers"
+	mkdir -p $(CURDIR)/$(PKG_BASE)
+	mv $(CURDIR)/pkg/tmp_listers/$(MODULE)/$(PKG_BASE)/listers \
+	   $(CURDIR)/$(PKG_BASE)/listers
+	rm -rf $(CURDIR)/pkg/tmp_listers
+
+generate-clientset: .remove-informers .remove-listers .remove-clientset .remove-deepcopy .remove-crds
+	client-gen \
+		--go-header-file=$(CURDIR)/hack/boilerplate.go.txt \
+		--clientset-name "versioned" \
+		--build-tag "ignore_autogenerated" \
+		--output-package "$(MODULE)/$(PKG_BASE)/clientset" \
+		--input-base "$(MODULE)/$(API_BASE)" \
+		--output-base "$(CURDIR)/pkg/tmp_clientset" \
+		--input "$(shell echo $(CLIENT_APIS) | tr ' ' ',')" \
+		--plural-exceptions "$(shell echo $(PLURAL_EXCEPTIONS) | tr ' ' ',')"
+	mkdir -p $(CURDIR)/$(PKG_BASE)
+	mv $(CURDIR)/pkg/tmp_clientset/$(MODULE)/$(PKG_BASE)/clientset \
+       $(CURDIR)/$(PKG_BASE)/clientset
+	rm -rf $(CURDIR)/pkg/tmp_clientset
+
+.remove-crds:
+	rm -rf $(CURDIR)/deployments/helm/$(DRIVER_NAME)/crds
+
 .remove-deepcopy:
 	for dir in $(DEEPCOPY_SOURCES); do \
 		rm -f $(CURDIR)/$${dir}/zz_generated.deepcopy.go; \
 	done
 
+.remove-clientset:
+	rm -rf $(CURDIR)/$(PKG_BASE)/clientset
+
+.remove-listers:
+	rm -rf $(CURDIR)/$(PKG_BASE)/listers
+
+.remove-informers:
+	rm -rf $(CURDIR)/$(PKG_BASE)/informers
+
 build-image:
 	$(DOCKER) build \
 		--progress=plain \
 		--build-arg GOLANG_VERSION="$(GOLANG_VERSION)" \
 		--build-arg CLIENT_GEN_VERSION="$(CLIENT_GEN_VERSION)" \
+		--build-arg LISTER_GEN_VERSION="$(LISTER_GEN_VERSION)" \
+		--build-arg INFORMER_GEN_VERSION="$(INFORMER_GEN_VERSION)" \
 		--build-arg CONTROLLER_GEN_VERSION="$(CONTROLLER_GEN_VERSION)" \
 		--build-arg GOLANGCI_LINT_VERSION="$(GOLANGCI_LINT_VERSION)" \
 		--build-arg MOQ_VERSION="$(MOQ_VERSION)" \

common.mk

@@ -21,3 +21,5 @@ CLIENT_APIS := gpu/v1alpha1
 CLIENT_SOURCES += $(patsubst %, $(API_BASE)/%, $(CLIENT_APIS))
 
 DEEPCOPY_SOURCES = $(CLIENT_SOURCES)
+
+PLURAL_EXCEPTIONS = ""

versions.mk

@@ -27,6 +27,8 @@ CUDA_VERSION ?= 12.3.2
 
 # These variables are only needed when building a local image
 CLIENT_GEN_VERSION ?= v0.29.2
+LISTER_GEN_VERSION ?= v0.29.2
+INFORMER_GEN_VERSION ?= v0.29.2
 CONTROLLER_GEN_VERSION ?= v0.14.0
 GOLANGCI_LINT_VERSION ?= v1.52.0
 MOQ_VERSION ?= v0.4.0