feat: ami type for mgmt cluster (#573)

* feat: ami type for mgmt cluster * chore: update swager docs
konstructio · Jan 30, 2025 · d8d50d8 · d8d50d8
1 parent 3cbfc64
commit d8d50d8
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 0 deletions.
diff --git a/docs/docs.go b/docs/docs.go
@@ -1091,6 +1091,9 @@ const docTemplate = `{
                     "description": "Identifiers",
                     "type": "string"
                 },
+                "ami_type": {
+                    "type": "string"
+                },
                 "argocd_auth_token": {
                     "type": "string"
                 },

diff --git a/docs/swagger.json b/docs/swagger.json
@@ -1085,6 +1085,9 @@
                     "description": "Identifiers",
                     "type": "string"
                 },
+                "ami_type": {
+                    "type": "string"
+                },
                 "argocd_auth_token": {
                     "type": "string"
                 },

diff --git a/docs/swagger.yaml b/docs/swagger.yaml
@@ -66,6 +66,8 @@ definitions:
       alerts_email:
         description: Identifiers
         type: string
+      ami_type:
+        type: string
       argocd_auth_token:
         type: string
       argocd_create_registry_check:

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"`