Files
kube-forge/internal/kubespray/playbook.go

77 lines
2.1 KiB
Go

package kubespray
import (
"context"
"fmt"
"kube-forge/internal/config"
"kube-forge/internal/logging"
"kube-forge/internal/python"
"os"
"path/filepath"
"github.com/apenella/go-ansible/v2/pkg/execute"
"github.com/apenella/go-ansible/v2/pkg/execute/stdoutcallback"
"github.com/apenella/go-ansible/v2/pkg/playbook"
)
func runPlaybook(playbookPath string, tags string) {
config := config.GetConfig()
var callbackExecute execute.Executor
epExec := python.NewPythonExec()
envVars := map[string]string{
"ANSIBLE_DISPLAY_SKIPPED_HOSTS": "false",
"ANSIBLE_DISPLAY_OK_HOSTS": "false",
"ANSIBLE_FORCE_COLOR": "true",
"ANSIBLE_HOST_KEY_CHECKING": "false",
"ANSIBLE_FORKS": config.Forks,
}
playbookOptions := getPlaybookParameters(tags)
playbookCmd := playbook.NewAnsiblePlaybookCmd(
playbook.WithPlaybooks(
filepath.Join(epExec.ResourcesFsPath, playbookPath),
),
playbook.WithPlaybookOptions(&playbookOptions),
playbook.WithBinary(
filepath.Join(epExec.PythonLibFsPath, "bin", "ansible-playbook"),
),
)
execute := execute.NewDefaultExecute(
execute.WithCmd(playbookCmd),
execute.WithErrorEnrich(playbook.NewAnsiblePlaybookErrorEnrich()),
execute.WithExecutable(epExec),
)
if config.Verbose > 0 {
envVars["ANSIBLE_STDOUT_CALLBACK"] = "unixy"
envVars["ANSIBLE_VERBOSITY"] = fmt.Sprintf("%d", config.Verbose)
envVars["ANSIBLE_DISPLAY_SKIPPED_HOSTS"] = "true"
envVars["ANSIBLE_DISPLAY_OK_HOSTS"] = "true"
} else {
envVars["ANSIBLE_STDOUT_CALLBACK"] = "dense"
envVars["ANSIBLE_ACTION_WARNINGS"] = "false"
envVars["ANSIBLE_DEVEL_WARNING"] = "false"
envVars["ANSIBLE_DEPRECATION_WARNINGS"] = "false"
envVars["ANSIBLE_DUPLICATE_YAML_DICT_KEY"] = "ignore"
envVars["ANSIBLE_HOST_PATTERN_MISMATCH"] = "ignore"
envVars["ANSIBLE_SYSTEM_WARNINGS"] = "false"
}
if config.LogFile != "" {
envVars["ANSIBLE_LOG_PATH"] = config.LogFile
}
callbackExecute = stdoutcallback.NewDefaultStdoutCallbackExecute(execute)
for key, value := range envVars {
os.Setenv(key, value)
}
err := callbackExecute.Execute(context.Background())
if err != nil {
logging.Log.Fatal(err)
}
}