@@ -22,7 +22,7 @@ import (
2222 "math/big"
2323 "strconv"
2424
25- admin20231115002 "go.mongodb.org/atlas-sdk/v20231115002 /admin"
25+ "go.mongodb.org/atlas-sdk/v20250312013 /admin"
2626
2727 "github.com/aws-cloudformation/cloudformation-cli-go-plugin/cfn/handler"
2828 "github.com/aws/aws-sdk-go-v2/aws"
@@ -57,7 +57,7 @@ func Create(req handler.Request, prevModel *Model, currentModel *Model) (handler
5757 return * pe , nil
5858 }
5959
60- _ , resp , err := client .Atlas20231115002 .EncryptionAtRestUsingCustomerKeyManagementApi .UpdateEncryptionAtRest (context .Background (), * currentModel .ProjectId , currentModel .getParams ()).Execute ()
60+ _ , resp , err := client .AtlasSDK .EncryptionAtRestUsingCustomerKeyManagementApi .UpdateEncryptionAtRest (context .Background (), * currentModel .ProjectId , currentModel .getParams ()).Execute ()
6161 if err != nil {
6262 return progressevent .GetFailedEventByResponse (err .Error (), resp ), nil
6363 }
@@ -82,7 +82,7 @@ func Read(req handler.Request, prevModel *Model, currentModel *Model) (handler.P
8282 return * pe , nil
8383 }
8484
85- info , resp , err := client .Atlas20231115002 .EncryptionAtRestUsingCustomerKeyManagementApi .GetEncryptionAtRest (context .Background (), * currentModel .ProjectId ).Execute ()
85+ info , resp , err := client .AtlasSDK .EncryptionAtRestUsingCustomerKeyManagementApi .GetEncryptionAtRest (context .Background (), * currentModel .ProjectId ).Execute ()
8686 if err != nil {
8787 return progressevent .GetFailedEventByResponse (err .Error (), resp ), nil
8888 }
@@ -91,10 +91,17 @@ func Read(req handler.Request, prevModel *Model, currentModel *Model) (handler.P
9191 return * pe , nil
9292 }
9393
94+ if currentModel .AwsKmsConfig == nil {
95+ currentModel .AwsKmsConfig = & AwsKmsConfig {}
96+ }
9497 currentModel .AwsKmsConfig .CustomerMasterKeyID = info .AwsKms .CustomerMasterKeyID
9598 currentModel .AwsKmsConfig .Enabled = info .AwsKms .Enabled
9699 currentModel .AwsKmsConfig .RoleID = info .AwsKms .RoleId
97100 currentModel .AwsKmsConfig .Region = info .AwsKms .Region
101+ currentModel .AwsKmsConfig .Valid = info .AwsKms .Valid
102+ currentModel .AwsKmsConfig .RequirePrivateNetworking = info .AwsKms .RequirePrivateNetworking
103+
104+ currentModel .EnabledForSearchNodes = info .EnabledForSearchNodes
98105
99106 return handler.ProgressEvent {
100107 OperationStatus : handler .Success ,
@@ -115,7 +122,7 @@ func Update(req handler.Request, prevModel *Model, currentModel *Model) (handler
115122 return * pe , nil
116123 }
117124
118- info , resp , err := client .Atlas20231115002 .EncryptionAtRestUsingCustomerKeyManagementApi .GetEncryptionAtRest (context .Background (), * currentModel .ProjectId ).Execute ()
125+ info , resp , err := client .AtlasSDK .EncryptionAtRestUsingCustomerKeyManagementApi .GetEncryptionAtRest (context .Background (), * currentModel .ProjectId ).Execute ()
119126 if err != nil {
120127 return progressevent .GetFailedEventByResponse (err .Error (), resp ), nil
121128 }
@@ -124,7 +131,7 @@ func Update(req handler.Request, prevModel *Model, currentModel *Model) (handler
124131 return * pe , nil
125132 }
126133
127- _ , resp , err = client .Atlas20231115002 .EncryptionAtRestUsingCustomerKeyManagementApi .UpdateEncryptionAtRest (context .Background (), * currentModel .ProjectId , currentModel .getParams ()).Execute ()
134+ _ , resp , err = client .AtlasSDK .EncryptionAtRestUsingCustomerKeyManagementApi .UpdateEncryptionAtRest (context .Background (), * currentModel .ProjectId , currentModel .getParams ()).Execute ()
128135 if err != nil {
129136 return progressevent .GetFailedEventByResponse (err .Error (), resp ), nil
130137 }
@@ -148,7 +155,7 @@ func Delete(req handler.Request, prevModel *Model, currentModel *Model) (handler
148155 return * pe , nil
149156 }
150157
151- info , resp , err := client .Atlas20231115002 .EncryptionAtRestUsingCustomerKeyManagementApi .GetEncryptionAtRest (context .Background (), * currentModel .ProjectId ).Execute ()
158+ info , resp , err := client .AtlasSDK .EncryptionAtRestUsingCustomerKeyManagementApi .GetEncryptionAtRest (context .Background (), * currentModel .ProjectId ).Execute ()
152159 if err != nil {
153160 return progressevent .GetFailedEventByResponse (err .Error (), resp ), nil
154161 }
@@ -157,10 +164,10 @@ func Delete(req handler.Request, prevModel *Model, currentModel *Model) (handler
157164 return * pe , nil
158165 }
159166
160- params := & admin20231115002 .EncryptionAtRest {
161- AwsKms : & admin20231115002 .AWSKMSConfiguration {Enabled : aws .Bool (false )},
167+ params := & admin .EncryptionAtRest {
168+ AwsKms : & admin .AWSKMSConfiguration {Enabled : aws .Bool (false )},
162169 }
163- _ , resp , err = client .Atlas20231115002 .EncryptionAtRestUsingCustomerKeyManagementApi .UpdateEncryptionAtRest (context .Background (), * currentModel .ProjectId , params ).Execute ()
170+ _ , resp , err = client .AtlasSDK .EncryptionAtRestUsingCustomerKeyManagementApi .UpdateEncryptionAtRest (context .Background (), * currentModel .ProjectId , params ).Execute ()
164171 if err != nil {
165172 return progressevent .GetFailedEventByResponse (err .Error (), resp ), nil
166173 }
@@ -175,7 +182,7 @@ func List(req handler.Request, prevModel *Model, currentModel *Model) (handler.P
175182 return handler.ProgressEvent {}, errors .New ("not implemented: List" )
176183}
177184
178- func validateExist (info * admin20231115002 .EncryptionAtRest ) * handler.ProgressEvent {
185+ func validateExist (info * admin .EncryptionAtRest ) * handler.ProgressEvent {
179186 if info != nil && info .AwsKms != nil && aws .ToBool (info .AwsKms .Enabled ) {
180187 return nil
181188 }
@@ -193,13 +200,22 @@ func randInt64() int64 {
193200 return val .Int64 ()
194201}
195202
196- func (m * Model ) getParams () * admin20231115002.EncryptionAtRest {
197- return & admin20231115002.EncryptionAtRest {
198- AwsKms : & admin20231115002.AWSKMSConfiguration {
199- Enabled : m .AwsKmsConfig .Enabled ,
200- CustomerMasterKeyID : m .AwsKmsConfig .CustomerMasterKeyID ,
201- RoleId : m .AwsKmsConfig .RoleID ,
202- Region : m .AwsKmsConfig .Region ,
203- },
203+ func (m * Model ) getParams () * admin.EncryptionAtRest {
204+ params := & admin.EncryptionAtRest {}
205+
206+ if m .EnabledForSearchNodes != nil {
207+ params .EnabledForSearchNodes = m .EnabledForSearchNodes
204208 }
209+
210+ if m .AwsKmsConfig != nil {
211+ params .AwsKms = & admin.AWSKMSConfiguration {
212+ Enabled : m .AwsKmsConfig .Enabled ,
213+ CustomerMasterKeyID : m .AwsKmsConfig .CustomerMasterKeyID ,
214+ Region : m .AwsKmsConfig .Region ,
215+ RoleId : m .AwsKmsConfig .RoleID ,
216+ RequirePrivateNetworking : m .AwsKmsConfig .RequirePrivateNetworking ,
217+ }
218+ }
219+
220+ return params
205221}
0 commit comments