packer-cn/builder/vmware/step_run.go

package vmware

import (
	"fmt"
	"github.com/mitchellh/multistep"
	"github.com/mitchellh/packer/packer"
	"os/exec"
	"time"
)

// This step runs the created virtual machine.
//
// Uses:
//   config *config
//   ui     packer.Ui
//   vmx_path string
//
// Produces:
//   <nothing>
type stepRun struct {
	bootTime time.Time
	vmxPath  string
}

func (s *stepRun) Run(state map[string]interface{}) multistep.StepAction {
	config := state["config"].(*config)
	ui := state["ui"].(packer.Ui)
	vmxPath := state["vmx_path"].(string)

	vmrun_path := "/Applications/VMware Fusion.app/Contents/Library/vmrun"

	// Set the VMX path so that we know we started the machine
	s.bootTime = time.Now()
	s.vmxPath = vmxPath

	ui.Say("Starting virtual machine...")
	cmd := exec.Command(vmrun_path, "-T", "fusion", "start", s.vmxPath, "gui")
	if err := cmd.Run(); err != nil {
		ui.Error(fmt.Sprintf("Error starting VM: %s", err))
		return multistep.ActionHalt
	}

	// Wait the wait amount
	if config.BootWait > 0 {
		ui.Say(fmt.Sprintf("Waiting %d seconds for boot...", config.BootWait))
		time.Sleep(time.Duration(config.BootWait) * time.Second)
	}

	return multistep.ActionContinue
}

func (s *stepRun) Cleanup(state map[string]interface{}) {
	ui := state["ui"].(packer.Ui)

	vmrun_path := "/Applications/VMware Fusion.app/Contents/Library/vmrun"

	// If we started the machine... stop it.
	if s.vmxPath != "" {
		// If we started it less than 5 seconds ago... wait.
		sinceBootTime := time.Since(s.bootTime)
		waitBootTime := 5 * time.Second
		if sinceBootTime < waitBootTime {
			time.Sleep(waitBootTime - sinceBootTime)
		}

		ui.Say("Stopping virtual machine...")
		cmd := exec.Command(vmrun_path, "-T", "fusion", "stop", s.vmxPath, "hard")
		if err := cmd.Run(); err != nil {
			ui.Error(fmt.Sprintf("Error stopping VM: %s", err))
		}
	}
}