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 }