adding force_deregister (#58)

nutanix-cloud-native · Nov 7, 2022 · 8e24259 · 8e24259
1 parent bf659ad
commit 8e24259
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 5 deletions.
diff --git a/builder/nutanix/config.go b/builder/nutanix/config.go
@@ -29,11 +29,13 @@ const (
 
 type Config struct {
 	common.PackerConfig            `mapstructure:",squash"`
-	CommConfig                     communicator.Config `mapstructure:",squash"`
+	CommConfig communicator.Config `mapstructure:",squash"`
 	commonsteps.CDConfig           `mapstructure:",squash"`
 	shutdowncommand.ShutdownConfig `mapstructure:",squash"`
 	ClusterConfig                  `mapstructure:",squash"`
 	VmConfig                       `mapstructure:",squash"`
+	ForceDeregister         bool   `mapstructure:"force_deregister" json:"force_deregister" required:"false"`
+
 
 	ctx interpolate.Context
 }

diff --git a/builder/nutanix/config.hcl2spec.go b/builder/nutanix/config.hcl2spec.go
diff --git a/builder/nutanix/driver.go b/builder/nutanix/driver.go
@@ -24,7 +24,7 @@ type Driver interface {
 	PowerOff(string) error
 	UploadImage(string, VmConfig) (*nutanixImage, error)
 	DeleteImage(string) error
-	SaveVMDisk(string, string) (*nutanixImage, error)
+	SaveVMDisk(string, string, bool) (*nutanixImage, error)
 	WaitForShutdown(string, <-chan struct{}) bool
 }
 
@@ -640,7 +640,7 @@ func (d *NutanixDriver) PowerOff(vmUUID string) error {
 	log.Printf("PowerOff task: %s", taskUUID)
 	return nil
 }
-func (d *NutanixDriver) SaveVMDisk(diskUUID string, imageName string) (*nutanixImage, error) {
+func (d *NutanixDriver) SaveVMDisk(diskUUID string, imageName string, ForceDeregister bool) (*nutanixImage, error) {
 
 	configCreds := client.Credentials{
 		URL:      fmt.Sprintf("%s:%d", d.ClusterConfig.Endpoint, d.ClusterConfig.Port),
@@ -656,6 +656,41 @@ func (d *NutanixDriver) SaveVMDisk(diskUUID string, imageName string) (*nutanixI
 		return nil, fmt.Errorf("error while NewV3Client, %s", err.Error())
 	}
 
+	// When force_deregister, check if image already exists
+	if ForceDeregister {
+		log.Println("force_deregister is set, check if image already exists")
+		ImageList, err:=conn.V3.ListAllImage(fmt.Sprintf("name==%s",imageName))
+		if err != nil {
+			return nil, fmt.Errorf("error while ListAllImage, %s", err.Error())
+		}
+		if *ImageList.Metadata.TotalMatches==0 {
+			log.Println("Image with given Name not found, no need to deregister")
+		} else if *ImageList.Metadata.TotalMatches>1 {
+			log.Println("More than one image with given Name found, will not deregister")
+		} else if *ImageList.Metadata.TotalMatches==1 {
+			log.Println("Exactly one image with given Name found, will deregister")
+			resp,err:= conn.V3.DeleteImage(*ImageList.Entities[0].Metadata.UUID)
+			if err != nil {
+				return nil, fmt.Errorf("error while DeleteImage, %s", err.Error())
+			}
+			taskUUID := resp.Status.ExecutionContext.TaskUUID.(string)
+			log.Printf("Wait until delete Image %s is finished, %s\n",*ImageList.Entities[0].Metadata.UUID,taskUUID)
+			// Wait for the Image to be deleted
+			for i := 0; i < 1200; i++ {
+				resp, err := conn.V3.GetTask(taskUUID)
+				if err != nil || *resp.Status != "SUCCEEDED" {
+					<-time.After(1 * time.Second)
+					continue
+				}
+				if *resp.Status == "SUCCEEDED" {
+					break
+				}
+				return nil, fmt.Errorf("error while Image Delete getting Task Status, %s", err.Error())
+			}
+
+
+		}
+	}
 	req := &v3.ImageIntentInput{
 		Spec: &v3.Image{
 			Name: &imageName,

diff --git a/builder/nutanix/step_copy_image.go b/builder/nutanix/step_copy_image.go
@@ -40,7 +40,7 @@ func (s *stepCopyImage) Run(ctx context.Context, state multistep.StateBag) multi
 		return multistep.ActionHalt
 	}
 
-	imageResponse, err := d.SaveVMDisk(diskToCopy, s.Config.VmConfig.ImageName)
+	imageResponse, err := d.SaveVMDisk(diskToCopy, s.Config.VmConfig.ImageName, s.Config.ForceDeregister)
 	if err != nil {
 		ui.Error("Unexpected Nutanix Task status: " + err.Error())
 		state.Put("error", err)

diff --git a/example/source.nutanix.pkr.hcl b/example/source.nutanix.pkr.hcl
@@ -17,7 +17,8 @@ source "nutanix" "centos" {
     subnet_name       = var.nutanix_subnet
   }
 
-  image_name        ="centos-{{isotime `Jan-_2-15:04:05`}}"
+  image_name        = "centos-packer-image"
+  force_deregister  = true
   user_data         = "I2Nsb3VkLWNvbmZpZwp1c2VyczoKICAtIG5hbWU6IGNlbnRvcwogICAgc3VkbzogWydBTEw9KEFMTCkgTk9QQVNTV0Q6QUxMJ10KY2hwYXNzd2Q6CiAgbGlzdDogfAogICAgY2VudG9zOnBhY2tlcgogIGV4cGlyZTogRmFsc2UKc3NoX3B3YXV0aDogVHJ1ZQ=="
 
   shutdown_command  = "echo 'packer' | sudo -S shutdown -P now"