package common

import (
	"fmt"
	"log"

	"github.com/hashicorp/packer/packer"
	"github.com/mitchellh/multistep"
)

// This step compacts the virtual disk for the VM unless the "skip_compaction"
// boolean is true.
//
// Uses:
//   driver Driver
//   full_disk_path string
//   ui     packer.Ui
//
// Produces:
//   <nothing>
type StepCompactDisk struct {
	Skip bool
}

func (s StepCompactDisk) Run(state multistep.StateBag) multistep.StepAction {
	driver := state.Get("driver").(Driver)
	ui := state.Get("ui").(packer.Ui)
	full_disk_path := state.Get("full_disk_path").(string)

	if s.Skip {
		log.Println("Skipping disk compaction step...")
		return multistep.ActionContinue
	}

	ui.Say("Compacting the disk image")
	if err := driver.CompactDisk(full_disk_path); err != nil {
		state.Put("error", fmt.Errorf("Error compacting disk: %s", err))
		return multistep.ActionHalt
	}

	if state.Get("additional_disk_paths") != nil {
		if moreDisks := state.Get("additional_disk_paths").([]string); len(moreDisks) > 0 {
			for i, path := range moreDisks {
				ui.Say(fmt.Sprintf("Compacting additional disk image %d", i+1))
				if err := driver.CompactDisk(path); err != nil {
					state.Put("error", fmt.Errorf("Error compacting additional disk %d: %s", i+1, err))
					return multistep.ActionHalt
				}
			}
		}
	}

	return multistep.ActionContinue
}

func (StepCompactDisk) Cleanup(multistep.StateBag) {}