119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/antihax/optional"
|
|
"github.com/hashicorp/packer-plugin-sdk/multistep"
|
|
packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
|
|
"github.com/hashicorp/packer-plugin-sdk/template/interpolate"
|
|
"github.com/outscale/osc-sdk-go/osc"
|
|
)
|
|
|
|
type StepUpdateOMIAttributes struct {
|
|
AccountIds []string
|
|
SnapshotAccountIds []string
|
|
RawRegion string
|
|
Ctx interpolate.Context
|
|
}
|
|
|
|
func (s *StepUpdateOMIAttributes) Run(_ context.Context, state multistep.StateBag) multistep.StepAction {
|
|
config := state.Get("accessConfig").(*AccessConfig)
|
|
ui := state.Get("ui").(packersdk.Ui)
|
|
omis := state.Get("omis").(map[string]string)
|
|
snapshots := state.Get("snapshots").(map[string][]string)
|
|
|
|
// Determine if there is any work to do.
|
|
valid := false
|
|
valid = valid || (s.AccountIds != nil && len(s.AccountIds) > 0)
|
|
valid = valid || (s.SnapshotAccountIds != nil && len(s.SnapshotAccountIds) > 0)
|
|
|
|
if !valid {
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
s.Ctx.Data = extractBuildInfo(s.RawRegion, state)
|
|
|
|
updateSnapshoptRequest := osc.UpdateSnapshotRequest{
|
|
PermissionsToCreateVolume: osc.PermissionsOnResourceCreation{
|
|
Additions: osc.PermissionsOnResource{
|
|
AccountIds: s.AccountIds,
|
|
GlobalPermission: false,
|
|
},
|
|
},
|
|
}
|
|
|
|
updateImageRequest := osc.UpdateImageRequest{
|
|
PermissionsToLaunch: osc.PermissionsOnResourceCreation{
|
|
Additions: osc.PermissionsOnResource{
|
|
AccountIds: s.AccountIds,
|
|
GlobalPermission: false,
|
|
},
|
|
},
|
|
}
|
|
|
|
// Updating image attributes
|
|
for region, omi := range omis {
|
|
ui.Say(fmt.Sprintf("Updating attributes on OMI (%s)...", omi))
|
|
regionconn := config.NewOSCClientByRegion(region)
|
|
|
|
// newConfig := &osc.Configuration{
|
|
// UserAgent: config.UserAgent,
|
|
// AccessKey: config.AccessKey,
|
|
// SecretKey: config.SecretKey,
|
|
// Service: config.Service,
|
|
// Region: region, //New region
|
|
// URL: config.URL,
|
|
// }
|
|
|
|
// skipClient := &http.Client{
|
|
// Transport: &http.Transport{
|
|
// TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
// },
|
|
// }
|
|
|
|
//regionconn := oapi.NewClient(newConfig, skipClient)
|
|
|
|
ui.Message(fmt.Sprintf("Updating: %s", omi))
|
|
updateImageRequest.ImageId = omi
|
|
_, _, err := regionconn.ImageApi.UpdateImage(context.Background(), &osc.UpdateImageOpts{
|
|
UpdateImageRequest: optional.NewInterface(updateImageRequest),
|
|
})
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf("Error updating OMI: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
}
|
|
|
|
// Updating snapshot attributes
|
|
for region, region_snapshots := range snapshots {
|
|
for _, snapshot := range region_snapshots {
|
|
ui.Say(fmt.Sprintf("Updating attributes on snapshot (%s)...", snapshot))
|
|
regionconn := config.NewOSCClientByRegion(region)
|
|
|
|
ui.Message(fmt.Sprintf("Updating: %s", snapshot))
|
|
updateSnapshoptRequest.SnapshotId = snapshot
|
|
_, _, err := regionconn.SnapshotApi.UpdateSnapshot(context.Background(), &osc.UpdateSnapshotOpts{
|
|
UpdateSnapshotRequest: optional.NewInterface(updateSnapshoptRequest),
|
|
})
|
|
if err != nil {
|
|
err := fmt.Errorf("Error updating snapshot: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepUpdateOMIAttributes) Cleanup(state multistep.StateBag) {
|
|
// No cleanup...
|
|
}
|