mirror of
https://github.com/thegeeklab/wp-ansible.git
synced 2024-11-23 06:10:39 +00:00
268 lines
5.5 KiB
Go
268 lines
5.5 KiB
Go
package ansible
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
plugin_exec "github.com/thegeeklab/wp-plugin-go/v3/exec"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
const (
|
|
AnsibleForksDefault = 5
|
|
|
|
ansibleBin = "/usr/local/bin/ansible"
|
|
ansibleGalaxyBin = "/usr/local/bin/ansible-galaxy"
|
|
ansiblePlaybookBin = "/usr/local/bin/ansible-playbook"
|
|
)
|
|
|
|
var ErrAnsiblePlaybookNotFound = errors.New("no playbook found")
|
|
|
|
type Ansible struct {
|
|
GalaxyRequirements string
|
|
Inventories cli.StringSlice
|
|
Playbooks cli.StringSlice
|
|
Limit string
|
|
SkipTags string
|
|
StartAtTask string
|
|
Tags string
|
|
ExtraVars cli.StringSlice
|
|
ModulePath cli.StringSlice
|
|
Check bool
|
|
Diff bool
|
|
FlushCache bool
|
|
ForceHandlers bool
|
|
ListHosts bool
|
|
ListTags bool
|
|
ListTasks bool
|
|
SyntaxCheck bool
|
|
Forks int
|
|
VaultID string
|
|
VaultPasswordFile string
|
|
Verbose int
|
|
PrivateKeyFile string
|
|
User string
|
|
Connection string
|
|
Timeout int
|
|
SSHCommonArgs string
|
|
SFTPExtraArgs string
|
|
SCPExtraArgs string
|
|
SSHExtraArgs string
|
|
Become bool
|
|
BecomeMethod string
|
|
BecomeUser string
|
|
}
|
|
|
|
// Version runs the Ansible binary with the --version flag to retrieve the current version.
|
|
func (a *Ansible) Version() *plugin_exec.Cmd {
|
|
args := []string{
|
|
"--version",
|
|
}
|
|
|
|
cmd := plugin_exec.Command(ansibleBin, args...)
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
return cmd
|
|
}
|
|
|
|
// GetPlaybooks retrieves the list of Ansible playbook files based on the configured playbook patterns.
|
|
func (a *Ansible) GetPlaybooks() error {
|
|
var playbooks []string
|
|
|
|
for _, pb := range a.Playbooks.Value() {
|
|
files, err := filepath.Glob(pb)
|
|
if err != nil {
|
|
playbooks = append(playbooks, pb)
|
|
|
|
continue
|
|
}
|
|
|
|
playbooks = append(playbooks, files...)
|
|
}
|
|
|
|
if len(playbooks) == 0 {
|
|
log.Debug().Strs("patterns", a.Playbooks.Value()).Msg("no playbooks found")
|
|
|
|
return ErrAnsiblePlaybookNotFound
|
|
}
|
|
|
|
a.Playbooks = *cli.NewStringSlice(playbooks...)
|
|
|
|
return nil
|
|
}
|
|
|
|
// GalaxyInstall runs the ansible-galaxy install command with the configured options.
|
|
func (a *Ansible) GalaxyInstall() *plugin_exec.Cmd {
|
|
args := []string{
|
|
"install",
|
|
"--force",
|
|
"--role-file",
|
|
a.GalaxyRequirements,
|
|
}
|
|
|
|
if a.Verbose > 0 {
|
|
args = append(args, fmt.Sprintf("-%s", strings.Repeat("v", a.Verbose)))
|
|
}
|
|
|
|
cmd := plugin_exec.Command(ansibleGalaxyBin, args...)
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
return cmd
|
|
}
|
|
|
|
// Play runs the Ansible playbook with the configured options.
|
|
//
|
|
//nolint:gocyclo
|
|
func (a *Ansible) Play() *plugin_exec.Cmd {
|
|
args := make([]string, 0)
|
|
|
|
for _, inventory := range a.Inventories.Value() {
|
|
args = append(args, "--inventory", inventory)
|
|
}
|
|
|
|
if len(a.ModulePath.Value()) > 0 {
|
|
args = append(args, "--module-path", strings.Join(a.ModulePath.Value(), ":"))
|
|
}
|
|
|
|
if a.VaultID != "" {
|
|
args = append(args, "--vault-id", a.VaultID)
|
|
}
|
|
|
|
if a.VaultPasswordFile != "" {
|
|
args = append(args, "--vault-password-file", a.VaultPasswordFile)
|
|
}
|
|
|
|
for _, v := range a.ExtraVars.Value() {
|
|
args = append(args, "--extra-vars", v)
|
|
}
|
|
|
|
if a.ListHosts {
|
|
args = append(args, "--list-hosts")
|
|
args = append(args, a.Playbooks.Value()...)
|
|
|
|
cmd := plugin_exec.Command(ansiblePlaybookBin, args...)
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
return cmd
|
|
}
|
|
|
|
if a.SyntaxCheck {
|
|
args = append(args, "--syntax-check")
|
|
args = append(args, a.Playbooks.Value()...)
|
|
|
|
cmd := plugin_exec.Command(ansiblePlaybookBin, args...)
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
return cmd
|
|
}
|
|
|
|
if a.Check {
|
|
args = append(args, "--check")
|
|
}
|
|
|
|
if a.Diff {
|
|
args = append(args, "--diff")
|
|
}
|
|
|
|
if a.FlushCache {
|
|
args = append(args, "--flush-cache")
|
|
}
|
|
|
|
if a.ForceHandlers {
|
|
args = append(args, "--force-handlers")
|
|
}
|
|
|
|
if a.Forks != AnsibleForksDefault {
|
|
args = append(args, "--forks", strconv.Itoa(a.Forks))
|
|
}
|
|
|
|
if a.Limit != "" {
|
|
args = append(args, "--limit", a.Limit)
|
|
}
|
|
|
|
if a.ListTags {
|
|
args = append(args, "--list-tags")
|
|
}
|
|
|
|
if a.ListTasks {
|
|
args = append(args, "--list-tasks")
|
|
}
|
|
|
|
if a.SkipTags != "" {
|
|
args = append(args, "--skip-tags", a.SkipTags)
|
|
}
|
|
|
|
if a.StartAtTask != "" {
|
|
args = append(args, "--start-at-task", a.StartAtTask)
|
|
}
|
|
|
|
if a.Tags != "" {
|
|
args = append(args, "--tags", a.Tags)
|
|
}
|
|
|
|
if a.PrivateKeyFile != "" {
|
|
args = append(args, "--private-key", a.PrivateKeyFile)
|
|
}
|
|
|
|
if a.User != "" {
|
|
args = append(args, "--user", a.User)
|
|
}
|
|
|
|
if a.Connection != "" {
|
|
args = append(args, "--connection", a.Connection)
|
|
}
|
|
|
|
if a.Timeout != 0 {
|
|
args = append(args, "--timeout", strconv.Itoa(a.Timeout))
|
|
}
|
|
|
|
if a.SSHCommonArgs != "" {
|
|
args = append(args, "--ssh-common-args", a.SSHCommonArgs)
|
|
}
|
|
|
|
if a.SFTPExtraArgs != "" {
|
|
args = append(args, "--sftp-extra-args", a.SFTPExtraArgs)
|
|
}
|
|
|
|
if a.SCPExtraArgs != "" {
|
|
args = append(args, "--scp-extra-args", a.SCPExtraArgs)
|
|
}
|
|
|
|
if a.SSHExtraArgs != "" {
|
|
args = append(args, "--ssh-extra-args", a.SSHExtraArgs)
|
|
}
|
|
|
|
if a.Become {
|
|
args = append(args, "--become")
|
|
}
|
|
|
|
if a.BecomeMethod != "" {
|
|
args = append(args, "--become-method", a.BecomeMethod)
|
|
}
|
|
|
|
if a.BecomeUser != "" {
|
|
args = append(args, "--become-user", a.BecomeUser)
|
|
}
|
|
|
|
if a.Verbose > 0 {
|
|
args = append(args, fmt.Sprintf("-%s", strings.Repeat("v", a.Verbose)))
|
|
}
|
|
|
|
args = append(args, a.Playbooks.Value()...)
|
|
|
|
cmd := plugin_exec.Command(ansiblePlaybookBin, args...)
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
return cmd
|
|
}
|