63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
package vmware
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// OutputDir is an interface type that abstracts the creation and handling
|
|
// of the output directory for VMware-based products. The abstraction is made
|
|
// so that the output directory can be properly made on remote (ESXi) based
|
|
// VMware products as well as local.
|
|
type OutputDir interface {
|
|
DirExists() (bool, error)
|
|
ListFiles() ([]string, error)
|
|
MkdirAll() error
|
|
Remove(string) error
|
|
RemoveAll() error
|
|
SetOutputDir(string)
|
|
}
|
|
|
|
// localOutputDir is an OutputDir implementation where the directory
|
|
// is on the local machine.
|
|
type localOutputDir struct {
|
|
dir string
|
|
}
|
|
|
|
func (d *localOutputDir) DirExists() (bool, error) {
|
|
_, err := os.Stat(d.dir)
|
|
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)
|
|
}
|
|
|
|
func (d *localOutputDir) SetOutputDir(path string) {
|
|
d.dir = path
|
|
}
|