73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package provisioner
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
const UnixOSType = "unix"
|
|
const WindowsOSType = "windows"
|
|
const DefaultOSType = UnixOSType
|
|
|
|
type guestOSTypeCommand struct {
|
|
chmod string
|
|
mkdir string
|
|
removeDir string
|
|
}
|
|
|
|
var guestOSTypeCommands = map[string]guestOSTypeCommand{
|
|
UnixOSType: {
|
|
chmod: "chmod %s '%s'",
|
|
mkdir: "mkdir -p '%s'",
|
|
removeDir: "rm -rf '%s'",
|
|
},
|
|
WindowsOSType: {
|
|
chmod: "echo 'skipping chmod %s %s'", // no-op
|
|
mkdir: "powershell.exe -Command \"New-Item -ItemType directory -Force -ErrorAction SilentlyContinue -Path %s\"",
|
|
removeDir: "powershell.exe -Command \"rm %s -recurse -force\"",
|
|
},
|
|
}
|
|
|
|
type GuestCommands struct {
|
|
GuestOSType string
|
|
Sudo bool
|
|
}
|
|
|
|
func NewGuestCommands(osType string, sudo bool) (*GuestCommands, error) {
|
|
_, ok := guestOSTypeCommands[osType]
|
|
if !ok {
|
|
return nil, fmt.Errorf("Invalid osType: \"%s\"", osType)
|
|
}
|
|
return &GuestCommands{GuestOSType: osType, Sudo: sudo}, nil
|
|
}
|
|
|
|
func (g *GuestCommands) Chmod(path string, mode string) string {
|
|
return g.sudo(fmt.Sprintf(g.commands().chmod, mode, g.escapePath(path)))
|
|
}
|
|
|
|
func (g *GuestCommands) CreateDir(path string) string {
|
|
return g.sudo(fmt.Sprintf(g.commands().mkdir, g.escapePath(path)))
|
|
}
|
|
|
|
func (g *GuestCommands) RemoveDir(path string) string {
|
|
return g.sudo(fmt.Sprintf(g.commands().removeDir, g.escapePath(path)))
|
|
}
|
|
|
|
func (g *GuestCommands) commands() guestOSTypeCommand {
|
|
return guestOSTypeCommands[g.GuestOSType]
|
|
}
|
|
|
|
func (g *GuestCommands) escapePath(path string) string {
|
|
if g.GuestOSType == WindowsOSType {
|
|
return strings.Replace(path, " ", "` ", -1)
|
|
}
|
|
return path
|
|
}
|
|
|
|
func (g *GuestCommands) sudo(cmd string) string {
|
|
if g.GuestOSType == UnixOSType && g.Sudo {
|
|
return "sudo " + cmd
|
|
}
|
|
return cmd
|
|
}
|