72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package space_worker
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
func (app *SpaceWorker) Install(generateOnly bool, targetHost string, roles ...string) error {
|
|
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
|
|
|
|
var host string
|
|
if targetHost != "" {
|
|
host = targetHost
|
|
} else {
|
|
var err error
|
|
host, err = app.GetDefaultHost()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
ansibleArgs, err := app.setupInstallAnsibleArgs(app.AnsibleVerbose, host, roles)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
command := exec.Command(app.AnsibleBin, ansibleArgs...)
|
|
command.Dir = app.workDir
|
|
command.Stdin = os.Stdin
|
|
command.Stdout = os.Stdout
|
|
command.Stderr = os.Stderr
|
|
|
|
app.log.Info(fmt.Sprintf("Target Host: %s", host))
|
|
|
|
if generateOnly {
|
|
fmt.Println(strings.Join(command.Args, " "))
|
|
} else {
|
|
app.log.Debug(fmt.Sprintf("ansible command line: %s", command.Args))
|
|
_ = command.Run()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (app *SpaceWorker) setupInstallAnsibleArgs(verbose uint8, targetHost string, roles []string) ([]string, error) {
|
|
var ansibleArgs []string
|
|
|
|
switch verbose {
|
|
case 1:
|
|
ansibleArgs = append(ansibleArgs, "-v")
|
|
case 2:
|
|
ansibleArgs = append(ansibleArgs, "-vv")
|
|
case 3:
|
|
ansibleArgs = append(ansibleArgs, "-vvv")
|
|
}
|
|
|
|
ansibleArgs = append(ansibleArgs, "-i", "hosts.yml")
|
|
ansibleArgs = append(ansibleArgs, "-l", targetHost)
|
|
|
|
if len(roles) > 0 {
|
|
for _, role := range roles {
|
|
ansibleArgs = append(ansibleArgs, "--extra-vars", fmt.Sprintf("dd_install_%s=true", role))
|
|
}
|
|
} else {
|
|
ansibleArgs = append(ansibleArgs, "--extra-vars", "dd_install=true")
|
|
}
|
|
|
|
return append(ansibleArgs, "playbook.yml"), nil
|
|
}
|