autorestic/internal/location.go

package internal

import (
	"fmt"
)

type HookArray = []string

type Hooks struct {
	Before HookArray `mapstructure:"before"`
	After  HookArray `mapstructure:"after"`
}

type Options map[string]map[string][]string

type Location struct {
	From    string   `mapstructure:"from"`
	To      []string `mapstructure:"to"`
	Hooks   Hooks    `mapstructure:"hooks"`
	Cron    string   `mapstructure:"cron"`
	Options Options  `mapstructure:"options"`
}

func (l Location) validate(c Config) error {
	// Check if backends are all valid
	for _, to := range l.To {
		_, ok := c.Backends[to]
		if !ok {
			return fmt.Errorf("invalid backend `%s`", to)
		}
	}
	return nil
}

func (l Location) getOptions(key string) []string {
	var options []string
	saved := l.Options[key]
	for k, values := range saved {
		for _, value := range values {
			options = append(options, fmt.Sprintf("--%s", k), value)
		}
	}
	return options
}

func ExecuteHooks(commands []string, options ExecuteOptions) error {
	for _, command := range commands {
		out, err := ExecuteCommand(options, "-c", command)
		fmt.Println(out)
		return err
	}
	return nil
}

func (l Location) Backup() error {
	c := GetConfig()
	from := GetPathRelativeToConfig(l.From)
	for _, to := range l.To {
		backend := c.Backends[to]
		options := ExecuteOptions{
			Command: "bash",
			Envs:    backend.getEnv(),
			Dir:     from,
		}

		if err := ExecuteHooks(l.Hooks.Before, options); err != nil {
			return nil
		}
		{
			flags := l.getOptions("backup")
			cmd := []string{"backup"}
			cmd = append(cmd, flags...)
			cmd = append(cmd, ".")
			out, err := ExecuteResticCommand(options, cmd...)
			fmt.Println(out)
			if err != nil {
				return err
			}
		}
		if err := ExecuteHooks(l.Hooks.After, options); err != nil {
			return nil
		}
	}
	return nil
}

func (l Location) Forget(prune bool) error {
	c := GetConfig()
	from := GetPathRelativeToConfig(l.From)
	for _, to := range l.To {
		backend := c.Backends[to]
		options := ExecuteOptions{
			Envs: backend.getEnv(),
			Dir:  from,
		}
		flags := l.getOptions("forget")
		cmd := []string{"forget", "--path", from}
		if prune {
			cmd = append(cmd, "--prune")
		}
		cmd = append(cmd, flags...)
		out, err := ExecuteResticCommand(options, cmd...)
		fmt.Println(out)
		if err != nil {
			return err
		}
	}
	return nil
}