94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
package iso
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mitchellh/multistep"
|
|
vmwcommon "github.com/mitchellh/packer/builder/vmware/common"
|
|
"github.com/mitchellh/packer/packer"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// This step cleans up the VMX by removing or changing this prior to
|
|
// being ready for use.
|
|
//
|
|
// Uses:
|
|
// ui packer.Ui
|
|
// vmx_path string
|
|
//
|
|
// Produces:
|
|
// <nothing>
|
|
type stepCleanVMX struct{}
|
|
|
|
func (s stepCleanVMX) Run(state multistep.StateBag) multistep.StepAction {
|
|
isoPath := state.Get("iso_path").(string)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
vmxPath := state.Get("vmx_path").(string)
|
|
|
|
ui.Say("Cleaning VMX prior to finishing up...")
|
|
|
|
vmxData, err := s.readVMX(vmxPath)
|
|
if err != nil {
|
|
state.Put("error", fmt.Errorf("Error reading VMX: %s", err))
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if _, ok := state.GetOk("floppy_path"); ok {
|
|
// Delete the floppy0 entries so the floppy is no longer mounted
|
|
ui.Message("Unmounting floppy from VMX...")
|
|
for k, _ := range vmxData {
|
|
if strings.HasPrefix(k, "floppy0.") {
|
|
log.Printf("Deleting key: %s", k)
|
|
delete(vmxData, k)
|
|
}
|
|
}
|
|
vmxData["floppy0.present"] = "FALSE"
|
|
}
|
|
|
|
ui.Message("Detaching ISO from CD-ROM device...")
|
|
devRe := regexp.MustCompile(`^ide\d:\d\.`)
|
|
for k, _ := range vmxData {
|
|
match := devRe.FindString(k)
|
|
if match == "" {
|
|
continue
|
|
}
|
|
|
|
filenameKey := match + "filename"
|
|
if filename, ok := vmxData[filenameKey]; ok {
|
|
if filename == isoPath {
|
|
// Change the CD-ROM device back to auto-detect to eject
|
|
vmxData[filenameKey] = "auto detect"
|
|
vmxData[match+"devicetype"] = "cdrom-raw"
|
|
}
|
|
}
|
|
}
|
|
|
|
// Rewrite the VMX
|
|
if err := vmwcommon.WriteVMX(vmxPath, vmxData); err != nil {
|
|
state.Put("error", fmt.Errorf("Error writing VMX: %s", err))
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (stepCleanVMX) Cleanup(multistep.StateBag) {}
|
|
|
|
func (stepCleanVMX) readVMX(vmxPath string) (map[string]string, error) {
|
|
vmxF, err := os.Open(vmxPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer vmxF.Close()
|
|
|
|
vmxBytes, err := ioutil.ReadAll(vmxF)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return vmwcommon.ParseVMX(string(vmxBytes)), nil
|
|
}
|