feat: ami type for mgmt cluster

konstructio · Jan 30, 2025 · 2b9be1d · 2b9be1d
1 parent 3cbfc64
commit 2b9be1d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/internal/controller/controller.go b/internal/controller/controller.go
@@ -208,6 +208,7 @@ func (clctrl *ClusterController) InitController(def *types.ClusterDefinition) er
 	clctrl.DNSProvider = def.DNSProvider
 	clctrl.ClusterType = def.Type
 	clctrl.Client = http.DefaultClient
+	clctrl.AMIType = def.AMIType
 	clctrl.NodeType = def.NodeType
 	clctrl.NodeCount = def.NodeCount
 	clctrl.PostInstallCatalogApps = def.PostInstallCatalogApps
@@ -434,6 +435,7 @@ func (clctrl *ClusterController) InitController(def *types.ClusterDefinition) er
 		VultrAuth:              clctrl.VultrAuth,
 		K3sAuth:                clctrl.K3sAuth,
 		CloudflareAuth:         clctrl.CloudflareAuth,
+		AMIType:                clctrl.AMIType,
 		NodeType:               clctrl.NodeType,
 		NodeCount:              clctrl.NodeCount,
 		LogFileName:            def.LogFileName,

diff --git a/pkg/types/cluster.go b/pkg/types/cluster.go
@@ -101,6 +101,7 @@ type Cluster struct {
 	AtlantisWebhookSecret string `bson:"atlantis_webhook_secret" json:"atlantis_webhook_secret"`
 	AtlantisWebhookURL    string `bson:"atlantis_webhook_url" json:"atlantis_webhook_url"`
 	KubefirstTeam         string `bson:"kubefirst_team" json:"kubefirst_team"`
+	AMIType               string `bson:"ami_type,omitempty" json:"ami_type,omitempty"`
 	NodeType              string `bson:"node_type" json:"node_type" binding:"required"`
 	NodeCount             int    `bson:"node_count" json:"node_count" binding:"required"`
 	LogFileName           string `bson:"log_file,omitempty" json:"log_file,omitempty"`