56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/mitchellh/multistep"
|
|
"log"
|
|
)
|
|
|
|
// 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) {}
|