package uhost import ( "context" "fmt" "strings" "time" ucloudcommon "github.com/hashicorp/packer/builder/ucloud/common" "github.com/hashicorp/packer/packer-plugin-sdk/retry" "github.com/hashicorp/packer/packer" "github.com/hashicorp/packer/packer-plugin-sdk/multistep" "github.com/ucloud/ucloud-sdk-go/ucloud" ) type stepCopyUCloudImage struct { ImageDestinations []ucloudcommon.ImageDestination RegionId string ProjectId string WaitImageReadyTimeout int } func (s *stepCopyUCloudImage) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction { if len(s.ImageDestinations) == 0 { return multistep.ActionContinue } client := state.Get("client").(*ucloudcommon.UCloudClient) conn := client.UHostConn ui := state.Get("ui").(packer.Ui) srcImageId := state.Get("image_id").(string) artifactImages := state.Get("ucloud_images").(*ucloudcommon.ImageInfoSet) expectedImages := ucloudcommon.NewImageInfoSet(nil) ui.Say(fmt.Sprintf("Copying images from %q...", srcImageId)) for _, v := range s.ImageDestinations { if v.ProjectId == s.ProjectId && v.Region == s.RegionId { continue } req := conn.NewCopyCustomImageRequest() req.TargetProjectId = ucloud.String(v.ProjectId) req.TargetRegion = ucloud.String(v.Region) req.SourceImageId = ucloud.String(srcImageId) req.TargetImageName = ucloud.String(v.Name) req.TargetImageDescription = ucloud.String(v.Description) resp, err := conn.CopyCustomImage(req) if err != nil { return ucloudcommon.Halt(state, err, fmt.Sprintf("Error on copying image %q to %s:%s", srcImageId, v.ProjectId, v.Region)) } image := ucloudcommon.ImageInfo{ Region: v.Region, ProjectId: v.ProjectId, ImageId: resp.TargetImageId, } expectedImages.Set(image) artifactImages.Set(image) ui.Message(fmt.Sprintf("Copying image from %s:%s:%s to %s:%s:%s", s.ProjectId, s.RegionId, srcImageId, v.ProjectId, v.Region, resp.TargetImageId)) } ui.Message("Waiting for the copied images to become available...") err := retry.Config{ StartTimeout: time.Duration(s.WaitImageReadyTimeout) * time.Second, ShouldRetry: func(err error) bool { return ucloudcommon.IsNotCompleteError(err) }, RetryDelay: (&retry.Backoff{InitialBackoff: 2 * time.Second, MaxBackoff: 12 * time.Second, Multiplier: 2}).Linear, }.Run(ctx, func(ctx context.Context) error { for _, v := range expectedImages.GetAll() { imageSet, err := client.DescribeImageByInfo(v.ProjectId, v.Region, v.ImageId) if err != nil { return fmt.Errorf("reading copied image %s:%s:%s failed, %s", v.ProjectId, v.Region, v.ImageId, err) } if imageSet.State == ucloudcommon.ImageStateAvailable { expectedImages.Remove(v.Id()) continue } if imageSet.State == ucloudcommon.ImageStateUnavailable { return fmt.Errorf("the copied image %s:%s:%s got %q error", v.ProjectId, v.Region, v.ImageId, ucloudcommon.ImageStateUnavailable) } } if len(expectedImages.GetAll()) != 0 { return ucloudcommon.NewNotCompletedError("copying image") } return nil }) if err != nil { var s []string for _, v := range expectedImages.GetAll() { s = append(s, fmt.Sprintf("%s:%s:%s", v.ProjectId, v.Region, v.ImageId)) } return ucloudcommon.Halt(state, err, fmt.Sprintf("Error on waiting for copying images %q to become available", strings.Join(s, ","))) } ui.Message(fmt.Sprintf("Copying image complete")) return multistep.ActionContinue } func (s *stepCopyUCloudImage) Cleanup(state multistep.StateBag) { _, cancelled := state.GetOk(multistep.StateCancelled) _, halted := state.GetOk(multistep.StateHalted) if !cancelled && !halted { return } srcImageId := state.Get("image_id").(string) ucloudImages := state.Get("ucloud_images").(*ucloudcommon.ImageInfoSet) imageInfos := ucloudImages.GetAll() if len(imageInfos) == 0 { return } else if len(imageInfos) == 1 && imageInfos[0].ImageId == srcImageId { return } ui := state.Get("ui").(packer.Ui) client := state.Get("client").(*ucloudcommon.UCloudClient) conn := client.UHostConn ui.Say(fmt.Sprintf("Deleting copied image because of cancellation or error...")) for _, v := range imageInfos { if v.ImageId == srcImageId { continue } req := conn.NewTerminateCustomImageRequest() req.ProjectId = ucloud.String(v.ProjectId) req.Region = ucloud.String(v.Region) req.ImageId = ucloud.String(v.ImageId) _, err := conn.TerminateCustomImage(req) if err != nil { ui.Error(fmt.Sprintf("Error on deleting copied image %q", v.ImageId)) } } ui.Message("Deleting copied image complete") }