Voomra/playbookctl
Archived
1
Fork 0
Code Activity

WIP

Browse Source
This commit is contained in:
Voomra
2024-12-19 19:25:00 +03:00
parent 2186f92263
commit 84be41cf1b
6 changed files with 80 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
internal/space_worker/space_worker.go
View File

@@ -176,9 +176,24 @@ func (app *SpaceWorker) ListRoles() error {
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
func (app *SpaceWorker) Install(generateOnly bool, roles ...string) error {
func (app *SpaceWorker) Install(generateOnly bool, targetHost string, roles ...string) error {
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
ansibleArgs := setupAnsibleArgs(app.AnsibleVerbose, roles)
var host string
if targetHost != "" {
host = targetHost
} else {
var err error
host, err = app.GetDefaultHost()
if err != nil {
return err
}
}
ansibleArgs, err := app.setupAnsibleArgs(app.AnsibleVerbose, host, roles)
if err != nil {
return err
}
command := exec.Command(app.AnsibleBin, ansibleArgs...)
command.Dir = app.workDir
@@ -196,7 +211,7 @@ func (app *SpaceWorker) Install(generateOnly bool, roles ...string) error {
return nil
}
func setupAnsibleArgs(verbose uint8, roles []string) []string {
func (app *SpaceWorker) setupAnsibleArgs(verbose uint8, targetHost string, roles []string) ([]string, error) {
var ansibleArgs []string
switch verbose {
@@ -209,6 +224,7 @@ func setupAnsibleArgs(verbose uint8, roles []string) []string {
}
ansibleArgs = append(ansibleArgs, "-i", "hosts.yml")
ansibleArgs = append(ansibleArgs, "-l", targetHost)
if len(roles) > 0 {
for _, role := range roles {
@@ -218,7 +234,7 @@ func setupAnsibleArgs(verbose uint8, roles []string) []string {
ansibleArgs = append(ansibleArgs, "--extra-vars", "dd_install=true")
}
return append(ansibleArgs, "playbook.yml")
return append(ansibleArgs, "playbook.yml"), nil
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
@@ -232,9 +248,19 @@ func (app *SpaceWorker) ListHosts() error {
return err
}
app.log.Trace("try read default.host.txt")
defaultHost, err := utils.ReadStringFile(filepath.Join(app.workDir, "default.host.txt"))
if err != nil {
return err
}
fmt.Println("Хосты:")
for serverName, props := range *hosts {
fmt.Printf(" - %s [%s:%d] (%s)\n", serverName, props.Host, props.Port, props.User)
defaultMarker := ""
if serverName == defaultHost {
defaultMarker = " *"
}
fmt.Printf(" - %s [%s:%d] (%s)%s\n", serverName, props.Host, props.Port, props.User, defaultMarker)
}
return nil
@@ -302,20 +328,24 @@ func (app *SpaceWorker) HostRemove(name string) error {
func (app *SpaceWorker) SetDefaultHost(name string) error {
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
app.log.Trace("try read playbook.yml")
playbook, err := types.ReadPlaybook(app.workDir)
app.log.Trace("try write default.host.txt")
err := utils.WriteStringFile(filepath.Join(app.workDir, "default.host.txt"), name)
if err != nil {
return err
}
app.log.Trace("set default host")
playbook.Hosts = name
app.log.Trace("write playbook.yml")
if err := types.WritePlaybook(app.workDir, playbook); err != nil {
return err
}
app.log.Info(fmt.Sprintf("Хост '%s' установлен по-умолчанию", name))
return nil
}
func (app *SpaceWorker) GetDefaultHost() (string, error) {
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
app.log.Trace("try read default.host.txt")
value, err := utils.ReadStringFile(filepath.Join(app.workDir, "default.host.txt"))
if err != nil {
return "", err
}
return value, nil
}