abra/cli/app/restore.go

package app

import (
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path"
	"strings"

	"coopcloud.tech/abra/cli/internal"
	"coopcloud.tech/abra/pkg/config"
	"github.com/sirupsen/logrus"
	"github.com/urfave/cli"
)

var restoreAllServices bool
var restoreAllServicesFlag = &cli.BoolFlag{
	Name:        "all, a",
	Destination: &restoreAllServices,
	Usage:       "Restore all services",
}

var appRestoreCommand = cli.Command{
	Name:    "restore",
	Aliases: []string{"rs"},
	Usage:   "Restore an app from a backup",
	Flags: []cli.Flag{
		internal.DebugFlag,
		internal.NoInputFlag,
		restoreAllServicesFlag,
	},
	Before:    internal.SubCommandBefore,
	ArgsUsage: "<service> [<backup file>]",
	Action: func(c *cli.Context) error {
		app := internal.ValidateApp(c)

		if len(c.Args()) > 1 && restoreAllServices {
			internal.ShowSubcommandHelpAndError(c, errors.New("cannot use <service>/<backup file> and '--all' together"))
		}

		abraSh := path.Join(config.RECIPES_DIR, app.Type, "abra.sh")
		if _, err := os.Stat(abraSh); err != nil {
			if os.IsNotExist(err) {
				logrus.Fatalf("%s does not exist?", abraSh)
			}
			logrus.Fatal(err)
		}

		sourceCmd := fmt.Sprintf("source %s", abraSh)
		execCmd := "abra_restore"
		if !restoreAllServices {
			serviceName := c.Args().Get(1)
			if serviceName == "" {
				internal.ShowSubcommandHelpAndError(c, errors.New("no service(s) target provided"))
			}
			execCmd = fmt.Sprintf("abra_restore_%s", serviceName)
		}

		bytes, err := ioutil.ReadFile(abraSh)
		if err != nil {
			logrus.Fatal(err)
		}
		if !strings.Contains(string(bytes), execCmd) {
			logrus.Fatalf("%s doesn't have a %s function", app.Type, execCmd)
		}

		backupFile := c.Args().Get(2)
		if backupFile != "" {
			execCmd = fmt.Sprintf("%s %s", execCmd, backupFile)
		}

		sourceAndExec := fmt.Sprintf("%s; %s", sourceCmd, execCmd)
		cmd := exec.Command("bash", "-c", sourceAndExec)
		if err := internal.RunCmd(cmd); err != nil {
			logrus.Fatal(err)
		}

		return nil
	},
}