66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package vmware
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mitchellh/multistep"
|
|
"github.com/mitchellh/packer/packer"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// These are the extensions of files that are important for the function
|
|
// of a VMware virtual machine. Any other file is discarded as part of the
|
|
// build.
|
|
var KeepFileExtensions = []string{".nvram", ".vmdk", ".vmsd", ".vmx", ".vmxf"}
|
|
|
|
// This step removes unnecessary files from the final result.
|
|
//
|
|
// Uses:
|
|
// config *config
|
|
// ui packer.Ui
|
|
//
|
|
// Produces:
|
|
// <nothing>
|
|
type stepCleanFiles struct{}
|
|
|
|
func (stepCleanFiles) Run(state multistep.StateBag) multistep.StepAction {
|
|
config := state.Get("config").(*config)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Deleting unnecessary VMware files...")
|
|
visit := func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
// If the file isn't critical to the function of the
|
|
// virtual machine, we get rid of it.
|
|
keep := false
|
|
ext := filepath.Ext(path)
|
|
for _, goodExt := range KeepFileExtensions {
|
|
if goodExt == ext {
|
|
keep = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !keep {
|
|
ui.Message(fmt.Sprintf("Deleting: %s", path))
|
|
return os.Remove(path)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
if err := filepath.Walk(config.OutputDir, visit); err != nil {
|
|
state.Put("error", err)
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (stepCleanFiles) Cleanup(multistep.StateBag) {}
|