mirror of
https://github.com/mulgadc/spinifex.git
synced 2026-10-21 03:36:35 +00:00
benduncan
125 lines
3.2 KiB
Go
125 lines
3.2 KiB
Go
package gateway_ec2_image
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"log/slog"
|
|
"strings"
|
|
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/mulgadc/hive/hive/handlers/ec2/image"
|
|
"github.com/mulgadc/hive/hive/utils"
|
|
)
|
|
|
|
/*
|
|
Sample JSON:
|
|
|
|
"ModifyImageAttribute":{
|
|
"name":"ModifyImageAttribute",
|
|
"http":{
|
|
"method":"POST",
|
|
"requestUri":"/"
|
|
},
|
|
"input":{"shape":"ModifyImageAttributeRequest"}
|
|
},
|
|
|
|
"ModifyImageAttributeRequest":{
|
|
"type":"structure",
|
|
"required":["ImageId"],
|
|
"members":{
|
|
"Attribute":{"shape":"String"},
|
|
"Description":{"shape":"AttributeValue"},
|
|
"ImageId":{"shape":"ImageId"},
|
|
"LaunchPermission":{"shape":"LaunchPermissionModifications"},
|
|
"OperationType":{"shape":"OperationType"},
|
|
"ProductCodes":{
|
|
"shape":"ProductCodeStringList",
|
|
"locationName":"ProductCode"
|
|
},
|
|
"UserGroups":{
|
|
"shape":"UserGroupStringList",
|
|
"locationName":"UserGroup"
|
|
},
|
|
"UserIds":{
|
|
"shape":"UserIdStringList",
|
|
"locationName":"UserId"
|
|
},
|
|
"Value":{"shape":"String"},
|
|
"DryRun":{
|
|
"shape":"Boolean",
|
|
"locationName":"dryRun"
|
|
},
|
|
"OrganizationArns":{
|
|
"shape":"OrganizationArnStringList",
|
|
"locationName":"OrganizationArn"
|
|
},
|
|
"OrganizationalUnitArns":{
|
|
"shape":"OrganizationalUnitArnStringList",
|
|
"locationName":"OrganizationalUnitArn"
|
|
},
|
|
"ImdsSupport":{"shape":"AttributeValue"}
|
|
}
|
|
}
|
|
*/
|
|
|
|
func ValidateModifyImageAttributeInput(input *ec2.ModifyImageAttributeInput) (err error) {
|
|
|
|
// Check required args from JSON above
|
|
// required:["ImageId"]
|
|
|
|
if input == nil {
|
|
return errors.New("MissingParameter")
|
|
}
|
|
|
|
if input.ImageId == nil || *input.ImageId == "" {
|
|
return errors.New("MissingParameter")
|
|
}
|
|
|
|
// Validate ImageId format
|
|
if !strings.HasPrefix(*input.ImageId, "ami-") {
|
|
return errors.New("InvalidAMIID.Malformed")
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func ModifyImageAttribute(input *ec2.ModifyImageAttributeInput) (output ec2.ModifyImageAttributeOutput, err error) {
|
|
|
|
// Validate input
|
|
err = ValidateModifyImageAttributeInput(input)
|
|
|
|
if err != nil {
|
|
return output, err
|
|
}
|
|
|
|
// Marshal to JSON, to send over the wire for processing (NATS)
|
|
jsonData, err := json.Marshal(input)
|
|
if err != nil {
|
|
return output, fmt.Errorf("failed to marshal input to JSON: %w", err)
|
|
}
|
|
|
|
// Run the simulated JSON request via handler, which will return a JSON response
|
|
handler := handlers_ec2_image.NewModifyImageAttributeHandler(handlers_ec2_image.NewMockImageService())
|
|
jsonResp := handler.Process(jsonData)
|
|
|
|
// Validate if the response is successful or an error
|
|
responseError, err := utils.ValidateErrorPayload(jsonResp)
|
|
|
|
if err != nil {
|
|
if responseError.Code != nil {
|
|
slog.Error("ModifyImageAttribute failed", "err", responseError.Code)
|
|
return output, errors.New(*responseError.Code)
|
|
}
|
|
return output, err
|
|
}
|
|
|
|
// Unmarshal the JSON response back into output struct
|
|
err = json.Unmarshal(jsonResp, &output)
|
|
if err != nil {
|
|
return output, fmt.Errorf("failed to unmarshal JSON response to ModifyImageAttributeOutput: %w", err)
|
|
}
|
|
|
|
return output, nil
|
|
}
|