88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/cobra"
|
|
"os"
|
|
"path/filepath"
|
|
"playbookctl/internal/space_worker"
|
|
"playbookctl/internal/types"
|
|
"playbookctl/internal/utils/logger"
|
|
)
|
|
|
|
var (
|
|
flagRestoreTimestamp string
|
|
flagRestoreInventory string
|
|
)
|
|
|
|
func NewCommandRestore() *cobra.Command {
|
|
restoreCmd := &cobra.Command{
|
|
Use: "restore [role1, role2, ...]",
|
|
Short: "восстановить резервную копированию пространства",
|
|
Args: restoreCheckArgsE,
|
|
ValidArgsFunction: ArgRoleCompletion,
|
|
RunE: restoreRunE,
|
|
}
|
|
|
|
restoreCmd.Flags().Uint8Var(&flagAnsibleVerbose, "ansible-verbose", 0, "ansible verbose mode (0-3)")
|
|
restoreCmd.Flags().StringVar(&flagAnsibleBin, "ansible-bin", "/usr/bin/ansible-playbook", "путь к ansible-playbook")
|
|
restoreCmd.Flags().BoolVar(&flagGenOnly, "generate-only", false, "только сгенерировать исполняемую команду")
|
|
restoreCmd.Flags().StringVar(&flagTargetHost, "target", "", "имя целевого хоста")
|
|
restoreCmd.Flags().StringVar(&flagRestoreTimestamp, "timestamp", "latest", "выбор времени бекапа")
|
|
restoreCmd.Flags().StringVar(&flagRestoreInventory, "inventory", "", "выбор хост бекапа")
|
|
|
|
if err := restoreCmd.MarkFlagFilename("ansible-bin"); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
return restoreCmd
|
|
}
|
|
|
|
func restoreCheckArgsE(_ *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 restoreRunE(_ *cobra.Command, args []string) error {
|
|
workDir, err := filepath.Abs(flagWorkdir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
spaceWorker := space_worker.NewSpaceWorker(logger.LogVerbose(flagVerbose), workDir)
|
|
spaceWorker.AnsibleBin = flagAnsibleBin
|
|
spaceWorker.AnsibleVerbose = flagAnsibleVerbose
|
|
|
|
var restoreInventory string
|
|
if flagRestoreInventory != "" {
|
|
restoreInventory = flagRestoreInventory
|
|
} else {
|
|
restoreInventory, err = spaceWorker.GetDefaultHost()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return spaceWorker.Restore(flagTargetHost, flagRestoreTimestamp, restoreInventory, args...)
|
|
}
|