autorestic/internal/backend.go

208 lines
5.1 KiB
Go
Raw Normal View History

2021-04-09 01:55:10 +02:00
package internal
import (
2021-04-12 00:02:35 +02:00
"crypto/rand"
"encoding/base64"
2021-04-09 01:55:10 +02:00
"fmt"
"net/url"
2021-04-15 21:55:35 +02:00
"os"
2021-04-12 00:02:35 +02:00
"strings"
2021-04-12 16:15:17 +02:00
"github.com/cupcakearmy/autorestic/internal/colors"
2022-02-16 21:42:54 +01:00
"github.com/cupcakearmy/autorestic/internal/flags"
2021-04-09 01:55:10 +02:00
)
type BackendRest struct {
User string `yaml:"user,omitempty"`
Password string `yaml:"password,omitempty"`
}
2021-04-09 01:55:10 +02:00
type Backend struct {
2021-05-17 22:34:14 +02:00
name string
Type string `yaml:"type,omitempty"`
Path string `yaml:"path,omitempty"`
Key string `yaml:"key,omitempty"`
Env map[string]string `yaml:"env,omitempty"`
Rest BackendRest `yaml:"rest,omitempty"`
Options Options `yaml:"options,omitempty"`
2021-04-09 01:55:10 +02:00
}
func GetBackend(name string) (Backend, bool) {
2021-04-16 22:02:25 +02:00
b, ok := GetConfig().Backends[name]
b.name = name
return b, ok
}
2021-04-09 01:55:10 +02:00
func (b Backend) generateRepo() (string, error) {
switch b.Type {
case "local":
2021-04-11 18:17:21 +02:00
return GetPathRelativeToConfig(b.Path)
case "rest":
parsed, err := url.Parse(b.Path)
if err != nil {
return "", err
}
if b.Rest.User != "" {
if b.Rest.Password == "" {
parsed.User = url.User(b.Rest.User)
} else {
parsed.User = url.UserPassword(b.Rest.User, b.Rest.Password)
}
}
return fmt.Sprintf("%s:%s", b.Type, parsed.String()), nil
2021-04-24 16:40:53 +02:00
case "b2", "azure", "gs", "s3", "sftp", "rclone":
2021-04-09 01:55:10 +02:00
return fmt.Sprintf("%s:%s", b.Type, b.Path), nil
default:
return "", fmt.Errorf("backend type \"%s\" is invalid", b.Type)
}
}
2021-04-11 18:17:21 +02:00
func (b Backend) getEnv() (map[string]string, error) {
2021-04-09 01:55:10 +02:00
env := make(map[string]string)
2021-10-30 13:48:44 +02:00
// Key
2021-10-25 17:36:06 +02:00
if b.Key != "" {
env["RESTIC_PASSWORD"] = b.Key
}
2021-10-30 13:48:44 +02:00
// From config file
2021-04-09 01:55:10 +02:00
repo, err := b.generateRepo()
env["RESTIC_REPOSITORY"] = repo
2021-04-17 13:04:40 +02:00
for key, value := range b.Env {
env[strings.ToUpper(key)] = value
}
2021-10-30 13:48:44 +02:00
// From Envfile and passed as env
var prefix = "AUTORESTIC_" + strings.ToUpper(b.name) + "_"
for _, variable := range os.Environ() {
var splitted = strings.SplitN(variable, "=", 2)
if strings.HasPrefix(splitted[0], prefix) {
env[strings.TrimPrefix(splitted[0], prefix)] = splitted[1]
}
}
2021-04-11 18:17:21 +02:00
return env, err
2021-04-09 01:55:10 +02:00
}
2021-04-12 00:02:35 +02:00
func generateRandomKey() string {
b := make([]byte, 64)
rand.Read(b)
key := base64.StdEncoding.EncodeToString(b)
key = strings.ReplaceAll(key, "=", "")
key = strings.ReplaceAll(key, "+", "")
key = strings.ReplaceAll(key, "/", "")
return key
}
2021-04-09 01:55:10 +02:00
func (b Backend) validate() error {
2021-04-12 00:02:35 +02:00
if b.Type == "" {
2021-04-16 22:02:25 +02:00
return fmt.Errorf(`Backend "%s" has no "type"`, b.name)
2021-04-12 00:02:35 +02:00
}
if b.Path == "" {
2021-04-16 22:02:25 +02:00
return fmt.Errorf(`Backend "%s" has no "path"`, b.name)
2021-04-12 00:02:35 +02:00
}
if b.Key == "" {
2021-10-25 17:36:06 +02:00
// Check if key is set in environment
2021-10-30 13:48:44 +02:00
env, _ := b.getEnv()
if _, found := env["RESTIC_PASSWORD"]; !found {
// No key set in config file or env => generate random key and save file
2021-10-25 17:36:06 +02:00
key := generateRandomKey()
b.Key = key
c := GetConfig()
tmp := c.Backends[b.name]
tmp.Key = key
c.Backends[b.name] = tmp
if err := c.SaveConfig(); err != nil {
return err
}
2021-04-12 00:02:35 +02:00
}
}
2021-04-11 18:17:21 +02:00
env, err := b.getEnv()
if err != nil {
return err
}
options := ExecuteOptions{Envs: env}
2021-04-09 01:55:10 +02:00
// Check if already initialized
2021-04-11 18:17:21 +02:00
_, err = ExecuteResticCommand(options, "snapshots")
2021-04-09 01:55:10 +02:00
if err == nil {
return nil
} else {
// If not initialize
2021-04-17 13:04:40 +02:00
colors.Body.Printf("Initializing backend \"%s\"...\n", b.name)
2021-04-09 01:55:10 +02:00
out, err := ExecuteResticCommand(options, "init")
2022-02-16 21:42:54 +01:00
if flags.VERBOSE {
2021-04-17 13:04:40 +02:00
colors.Faint.Println(out)
}
2021-04-09 01:55:10 +02:00
return err
}
}
func (b Backend) Exec(args []string) error {
2021-04-11 18:17:21 +02:00
env, err := b.getEnv()
if err != nil {
return err
}
options := ExecuteOptions{Envs: env}
2021-04-09 01:55:10 +02:00
out, err := ExecuteResticCommand(options, args...)
if err != nil {
colors.Error.Println(out)
return err
}
2022-02-16 21:42:54 +01:00
if flags.VERBOSE {
2021-04-12 16:15:17 +02:00
colors.Faint.Println(out)
}
return nil
2021-04-09 01:55:10 +02:00
}
2021-04-15 21:55:35 +02:00
func (b Backend) ExecDocker(l Location, args []string) (string, error) {
2021-04-15 21:55:35 +02:00
env, err := b.getEnv()
if err != nil {
return "", err
2021-04-15 21:55:35 +02:00
}
2021-10-31 22:32:55 +01:00
volume := l.From[0]
2021-04-15 21:55:35 +02:00
options := ExecuteOptions{
Command: "docker",
Envs: env,
}
2021-10-31 22:32:55 +01:00
dir := "/data"
2021-11-01 00:16:54 +01:00
args = append([]string{"restic"}, args...)
2021-04-15 21:55:35 +02:00
docker := []string{
"run", "--rm",
2021-11-01 00:16:54 +01:00
"--pull", "always",
2021-04-15 21:55:35 +02:00
"--entrypoint", "ash",
2021-10-31 22:32:55 +01:00
"--workdir", dir,
"--volume", volume + ":" + dir,
2021-04-15 21:55:35 +02:00
}
2021-10-31 22:32:55 +01:00
// Use of docker host, not the container host
2021-04-15 21:55:35 +02:00
if hostname, err := os.Hostname(); err == nil {
docker = append(docker, "--hostname", hostname)
}
2021-10-31 22:32:55 +01:00
switch b.Type {
case "local":
2021-04-15 21:55:35 +02:00
actual := env["RESTIC_REPOSITORY"]
docker = append(docker, "--volume", actual+":"+"/repo")
env["RESTIC_REPOSITORY"] = "/repo"
2021-10-31 22:32:55 +01:00
case "b2":
case "s3":
2021-11-01 00:19:32 +01:00
case "azure":
case "gs":
2021-10-31 22:32:55 +01:00
// No additional setup needed
2021-11-01 00:16:54 +01:00
case "rclone":
// Read host rclone config and mount it into the container
configFile, err := ExecuteCommand(ExecuteOptions{Command: "rclone"}, "config", "file")
if err != nil {
return "", err
}
splitted := strings.Split(strings.TrimSpace(configFile), "\n")
configFilePath := splitted[len(splitted)-1]
docker = append(docker, "--volume", configFilePath+":"+"/root/.config/rclone/rclone.conf:ro")
2021-10-31 22:32:55 +01:00
default:
return "", fmt.Errorf("Backend type \"%s\" is not supported as volume endpoint", b.Type)
2021-04-15 21:55:35 +02:00
}
for key, value := range env {
docker = append(docker, "--env", key+"="+value)
}
2021-12-06 11:59:58 +01:00
docker = append(docker, "cupcakearmy/autorestic:"+VERSION, "-c", strings.Join(args, " "))
2021-04-15 21:55:35 +02:00
out, err := ExecuteCommand(options, docker...)
return out, err
2021-04-15 21:55:35 +02:00
}