builder/vmware: Add ListFiles and Remove methods to OutputDir interface

- Fixes builder/vmware when using RemoteType esx5,
  as there is no longer a local OutputDir

builder/vmware/builder.go

@@ -10,7 +10,6 @@ import (
 	"log"
 	"math/rand"
 	"os"
-	"path/filepath"
 	"strings"
 	"text/template"
 	"time"
@@ -459,20 +458,8 @@ func (b *Builder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packe
 	}
 
 	// Compile the artifact list
-	files := make([]string, 0, 10)
-	visit := func(path string, info os.FileInfo, err error) error {
-		if err != nil {
-			return err
-		}
-
-		if !info.IsDir() {
-			files = append(files, path)
-		}
-
-		return nil
-	}
-
-	if err := filepath.Walk(b.config.OutputDir, visit); err != nil {
+	files, err := state.Get("dir").(OutputDir).ListFiles()
+	if err != nil {
 		return nil, err
 	}
 

builder/vmware/driver_esx5.go

@@ -1,6 +1,7 @@
 package vmware
 
 import (
+	"bufio"
 	"bytes"
 	gossh "code.google.com/p/go.crypto/ssh"
 	"encoding/csv"
@@ -214,10 +215,37 @@ func (d *ESX5Driver) DirExists() (bool, error) {
 	return err == nil, nil
 }
 
+func (d *ESX5Driver) ListFiles() ([]string, error) {
+	stdout, err := d.ssh("ls -1p "+d.outputDir, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	files := make([]string, 0, 10)
+	reader := bufio.NewReader(stdout)
+	for {
+		line, _, err := reader.ReadLine()
+		if err == io.EOF {
+			break
+		}
+		if line[len(line)-1] == '/' {
+			continue
+		}
+
+		files = append(files, filepath.Join(d.outputDir, string(line)))
+	}
+
+	return files, nil
+}
+
 func (d *ESX5Driver) MkdirAll() error {
 	return d.mkdir(d.outputDir)
 }
 
+func (d *ESX5Driver) Remove(path string) error {
+	return d.sh("rm", path)
+}
+
 func (d *ESX5Driver) RemoveAll() error {
 	return d.sh("rm", "-rf", d.outputDir)
 }

builder/vmware/output_dir.go

@@ -2,6 +2,7 @@ package vmware
 
 import (
 	"os"
+	"path/filepath"
 )
 
 // OutputDir is an interface type that abstracts the creation and handling
@@ -10,7 +11,9 @@ import (
 // VMware products as well as local.
 type OutputDir interface {
 	DirExists() (bool, error)
+	ListFiles() ([]string, error)
 	MkdirAll() error
+	Remove(string) error
 	RemoveAll() error
 	SetOutputDir(string)
 }
@@ -26,10 +29,30 @@ func (d *localOutputDir) DirExists() (bool, error) {
 	return err == nil, nil
 }
 
+func (d *localOutputDir) ListFiles() ([]string, error) {
+	files := make([]string, 0, 10)
+
+	visit := func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			files = append(files, path)
+		}
+		return nil
+	}
+
+	return files, filepath.Walk(d.dir, visit)
+}
+
 func (d *localOutputDir) MkdirAll() error {
 	return os.MkdirAll(d.dir, 0755)
 }
 
+func (d *localOutputDir) Remove(path string) error {
+	return os.Remove(path)
+}
+
 func (d *localOutputDir) RemoveAll() error {
 	return os.RemoveAll(d.dir)
 }

builder/vmware/step_clean_files.go

@@ -4,7 +4,6 @@ import (
 	"fmt"
 	"github.com/mitchellh/multistep"
 	"github.com/mitchellh/packer/packer"
-	"os"
 	"path/filepath"
 )
 
@@ -16,7 +15,7 @@ var KeepFileExtensions = []string{".nvram", ".vmdk", ".vmsd", ".vmx", ".vmxf"}
 // This step removes unnecessary files from the final result.
 //
 // Uses:
-//   config *config
+//   dir    OutputDir
 //   ui     packer.Ui
 //
 // Produces:
@@ -24,41 +23,37 @@ var KeepFileExtensions = []string{".nvram", ".vmdk", ".vmsd", ".vmx", ".vmxf"}
 type stepCleanFiles struct{}
 
 func (stepCleanFiles) Run(state multistep.StateBag) multistep.StepAction {
-	config := state.Get("config").(*config)
+	dir := state.Get("dir").(OutputDir)
 	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 {
+	files, err := dir.ListFiles()
+	if err != nil {
 		state.Put("error", err)
 		return multistep.ActionHalt
 	}
 
+	for _, path := range files {
+		// 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))
+			if err = dir.Remove(path); err != nil {
+				state.Put("error", err)
+				return multistep.ActionHalt
+			}
+		}
+	}
+
 	return multistep.ActionContinue
 }
 

builder/vmware/step_prepare_output_dir.go

@@ -36,6 +36,8 @@ func (s *stepPrepareOutputDir) Run(state multistep.StateBag) multistep.StepActio
 
 	s.dir = dir
 
+	state.Put("dir", dir)
+
 	return multistep.ActionContinue
 }