make sdk generation external pluggable/configurable

castai · Sep 3, 2024 · 582c8a4 · 582c8a4
1 parent a1dc9a3
commit 582c8a4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/Makefile b/Makefile
@@ -1,5 +1,8 @@
 SHELL := /bin/bash
 
+export API_TAGS ?= ExternalClusterAPI,PoliciesAPI,NodeConfigurationAPI,NodeTemplatesAPI,AuthTokenAPI,ScheduledRebalancingAPI,InventoryAPI,UsersAPI,OperationsAPI,EvictorAPI,SSOAPI,CommitmentsAPI,WorkloadOptimizationAPI
+export SWAGGER_LOCATION ?= https://api.cast.ai/v1/spec/openapi.json
+
 default: build
 
 .PHONY: init-examples
@@ -26,8 +29,8 @@ format-tf:
 
 .PHONY: generate-sdk
 generate-sdk:
-	@echo "==> Generating castai sdk client"
-	@API_TAGS=ExternalClusterAPI,PoliciesAPI,NodeConfigurationAPI,NodeTemplatesAPI,AuthTokenAPI,ScheduledRebalancingAPI,InventoryAPI,UsersAPI,OperationsAPI,EvictorAPI,SSOAPI,CommitmentsAPI,WorkloadOptimizationAPI go generate castai/sdk/generate.go
+	echo "==> Generating castai sdk client"
+	go generate castai/sdk/generate.go
 
 # The following command also rewrites existing documentation
 .PHONY: generate-docs

diff --git a/castai/sdk/generate.go b/castai/sdk/generate.go
@@ -1,8 +1,8 @@
 package sdk
 
-//go:generate echo "generating sdk for: ${API_TAGS}"
+//go:generate echo "generating sdk for: ${API_TAGS} from ${SWAGGER_LOCATION}"
 //go:generate go install github.com/deepmap/oapi-codegen/cmd/oapi-codegen@v1.11.0
-//go:generate oapi-codegen -o api.gen.go --old-config-style -generate types -include-tags $API_TAGS -package sdk https://api.cast.ai/v1/spec/openapi.json
-//go:generate oapi-codegen -o client.gen.go --old-config-style -templates codegen/templates -generate client -include-tags $API_TAGS -package sdk https://api.cast.ai/v1/spec/openapi.json
+//go:generate oapi-codegen -o api.gen.go --old-config-style -generate types -include-tags $API_TAGS -package sdk $SWAGGER_LOCATION
+//go:generate oapi-codegen -o client.gen.go --old-config-style -templates codegen/templates -generate client -include-tags $API_TAGS -package sdk $SWAGGER_LOCATION
 //go:generate go install github.com/golang/mock/mockgen
 //go:generate mockgen -source client.gen.go -destination mock/client.go . ClientInterface