OMNI-496: update RegisterCluster API model

Author: fra98

Makefile

@@ -7,7 +7,9 @@ export SWAGGER_LOCATION ?= https://$(API_HOST)/v1/spec/openapi.json
 #  To add a new SDK, add a line here in the format: package_name:ApiTagName:spec_location
 SDK_SPECS := \
 	cluster_autoscaler:HibernationSchedulesAPI:https://$(API_HOST)/spec/cluster-autoscaler/openapi.yaml \
-	organization_management:EnterpriseAPI:https://$(API_HOST)/spec/organization-management/openapi.yaml \
+	organization_management:EnterpriseAPI:https://$(API_HOST)/spec/organization-management/openapi.yaml
+
+OMNI_SDK_SPECS := \
 	omni:EdgeLocationsAPI,ClustersAPI:https://$(API_HOST)/spec/omni/openapi.yaml
 
 default: build
@@ -16,17 +18,25 @@ default: build
 format-tf:
 	terraform fmt -recursive -list=false
 
-.PHONY: generate-sdk 
-generate-sdk: generate-sdk-new
+.PHONY: generate-sdk
+generate-sdk:
 	@echo "==> Generating main sdk client"
+	$(MAKE) generate-sdk-new SPECS="$(SDK_SPECS)"
 	go generate castai/sdk/generate.go
 
+.PHONY: generate-omni-sdk
+generate-omni-sdk:
+	@echo "==> Generating omni sdk client"
+	$(MAKE) generate-sdk-new SPECS="$(OMNI_SDK_SPECS)"
+
 .PHONY: generate-sdk-new
+# Internal target: run oapi-codegen for the given SPECS variable.
+# Not meant to be called directly; use generate-sdk or generate-omni-sdk.
 generate-sdk-new:
 	@echo "==> Generating api sdk clients"
 	@go install github.com/oapi-codegen/oapi-codegen/v2/cmd/oapi-codegen@v2.4.1
 	@go install github.com/golang/mock/mockgen
-	@cd castai/sdk && for spec in $(SDK_SPECS); do \
+	@cd castai/sdk && for spec in $(SPECS); do \
 		IFS=':' read -r pkg tag loc <<< "$$spec"; \
 		[ -z "$$pkg" ] && continue; \
 		echo "generating sdk for: $$tag from $$loc"; \

castai/resource_omni_cluster.go

@@ -10,6 +10,8 @@ import (
 	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
 	"github.com/hashicorp/terraform-plugin-framework/resource/schema/stringplanmodifier"
 	"github.com/hashicorp/terraform-plugin-framework/types"
+
+	"github.com/castai/terraform-provider-castai/castai/sdk/omni"
 )
 
 var (
@@ -22,9 +24,11 @@ type omniClusterResource struct {
 }
 
 type omniClusterModel struct {
-	ID             types.String `tfsdk:"id"`
-	OrganizationID types.String `tfsdk:"organization_id"`
-	ClusterID      types.String `tfsdk:"cluster_id"`
+	ID               types.String `tfsdk:"id"`
+	OrganizationID   types.String `tfsdk:"organization_id"`
+	ClusterID        types.String `tfsdk:"cluster_id"`
+	OmniAgentVersion types.String `tfsdk:"omni_agent_version"`
+	PodCIDR          types.String `tfsdk:"pod_cidr"`
 }
 
 func newOmniClusterResource() resource.Resource {
@@ -60,6 +64,20 @@ func (r *omniClusterResource) Schema(_ context.Context, _ resource.SchemaRequest
 					stringplanmodifier.RequiresReplace(),
 				},
 			},
+			"omni_agent_version": schema.StringAttribute{
+				Optional:    true,
+				Description: "Version of the omni agent running on the cluster",
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
+			"pod_cidr": schema.StringAttribute{
+				Optional:    true,
+				Description: "Pod CIDR of the cluster",
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
 		},
 	}
 }
@@ -89,11 +107,31 @@ func (r *omniClusterResource) Create(ctx context.Context, req resource.CreateReq
 		return
 	}
 
+	agentVersion := plan.OmniAgentVersion.ValueString()
+	podCIDR := plan.PodCIDR.ValueString()
+
+	// Both fields must be set or both must be empty
+	if (agentVersion == "") != (podCIDR == "") {
+		resp.Diagnostics.AddError(
+			"Invalid configuration",
+			"all cluster status fields (omni_agent_version and pod_cidr) must be set",
+		)
+		return
+	}
+
+	body := omni.RegisteredCluster{}
+	if agentVersion != "" {
+		body.Status = &omni.RegisteredClusterStatus{
+			OmniAgentVersion: agentVersion,
+			PodCidr:          podCIDR,
+		}
+	}
+
 	client := r.client.omniAPI
 	organizationID := plan.OrganizationID.ValueString()
 	clusterID := plan.ClusterID.ValueString()
 
-	apiResp, err := client.ClustersAPIRegisterClusterWithResponse(ctx, organizationID, clusterID)
+	apiResp, err := client.ClustersAPIRegisterClusterWithResponse(ctx, organizationID, clusterID, body)
 	if err != nil {
 		resp.Diagnostics.AddError("Failed to register omni cluster", err.Error())
 		return