Archived
1
This commit is contained in:
2024-12-20 00:37:30 +03:00
parent d5d27c1037
commit fd73510b86
5 changed files with 160 additions and 19 deletions

View File

@@ -3,22 +3,67 @@ package cmd
import (
"fmt"
"github.com/spf13/cobra"
"os"
"path/filepath"
"playbookctl/internal/space_worker"
"playbookctl/internal/types"
"playbookctl/internal/utils/logger"
)
func NewCommandBackup() *cobra.Command {
spaceCmd := &cobra.Command{
Use: "backup",
Short: "выполнить резервное копирование пространства",
RunE: backupRunE,
backupCmd := &cobra.Command{
Use: "backup [role1, role2, ...]",
Short: "выполнить резервное копирование пространства",
Args: backupCheckArgsE,
ValidArgsFunction: ArgRoleCompletion,
RunE: backupRunE,
}
return spaceCmd
backupCmd.Flags().Uint8Var(&flagAnsibleVerbose, "ansible-verbose", 0, "ansible verbose mode (0-3)")
backupCmd.Flags().StringVar(&flagAnsibleBin, "ansible-bin", "/usr/bin/ansible-playbook", "путь к ansible-playbook")
backupCmd.Flags().BoolVar(&flagGenOnly, "generate-only", false, "только сгенерировать исполняемую команду")
backupCmd.Flags().StringVar(&flagTargetHost, "target", "", "имя целевого хоста")
if err := backupCmd.MarkFlagFilename("ansible-bin"); err != nil {
fmt.Println(err)
os.Exit(1)
}
return backupCmd
}
func backupCheckArgsE(_ *cobra.Command, args []string) error {
if len(args) == 0 {
return nil
}
workDir, err := filepath.Abs(flagWorkdir)
if err != nil {
return err
}
playbook, err := types.ReadPlaybook(workDir)
if err != nil {
return err
}
for _, roleName := range args {
if !containsInSlice(roleName, playbook.Roles) {
return fmt.Errorf("роли \"%s\" не существует или она не добавлена в playbook.yml", roleName)
}
}
return nil
}
func backupRunE(_ *cobra.Command, args []string) error {
fmt.Println("[dummy] backup")
for _, elm := range args {
fmt.Printf("- %s\n", elm)
workDir, err := filepath.Abs(flagWorkdir)
if err != nil {
return err
}
return nil
spaceWorker := space_worker.NewSpaceWorker(logger.LogVerbose(flagVerbose), workDir)
spaceWorker.AnsibleBin = flagAnsibleBin
spaceWorker.AnsibleVerbose = flagAnsibleVerbose
return spaceWorker.Backup(flagGenOnly, flagTargetHost, args...)
}

View File

@@ -10,13 +10,6 @@ import (
"playbookctl/internal/utils/logger"
)
var (
flagAnsibleVerbose uint8
flagAnsibleBin string
flagGenOnly bool
flagTargetHost string
)
func NewCommandInstall() *cobra.Command {
installCmd := &cobra.Command{
Use: "install [role1, role2, ...]",

View File

@@ -11,6 +11,13 @@ var (
flagWorkdir string
)
var (
flagAnsibleVerbose uint8
flagAnsibleBin string
flagGenOnly bool
flagTargetHost string
)
var rootCmd = &cobra.Command{
Use: "playbookctl",
Short: "Ansible Playbook Dedic2 Control",
@@ -34,7 +41,7 @@ func init() {
rootCmd.AddCommand(NewCommandInstall())
rootCmd.AddCommand(NewCommandHosts())
//mainCmd.AddCommand(cmd.NewCommandOpenports())
//mainCmd.AddCommand(cmd.NewCommandBackup())
rootCmd.AddCommand(NewCommandBackup())
//mainCmd.AddCommand(cmd.NewCommandRestore())
}