mirror of
https://github.com/cupcakearmy/autorestic.git
synced 2024-08-28 15:17:18 +02:00
167 lines
3.8 KiB
Go
167 lines
3.8 KiB
Go
package internal
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
|
)
|
|
|
|
type BackendRest struct {
|
|
User string `yaml:"user,omitempty"`
|
|
Password string `yaml:"password,omitempty"`
|
|
}
|
|
|
|
type Backend struct {
|
|
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"`
|
|
}
|
|
|
|
func GetBackend(name string) (Backend, bool) {
|
|
b, ok := GetConfig().Backends[name]
|
|
b.name = name
|
|
return b, ok
|
|
}
|
|
|
|
func (b Backend) generateRepo() (string, error) {
|
|
switch b.Type {
|
|
case "local":
|
|
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
|
|
case "b2", "azure", "gs", "s3", "sftp", "rclone":
|
|
return fmt.Sprintf("%s:%s", b.Type, b.Path), nil
|
|
default:
|
|
return "", fmt.Errorf("backend type \"%s\" is invalid", b.Type)
|
|
}
|
|
}
|
|
|
|
func (b Backend) getEnv() (map[string]string, error) {
|
|
env := make(map[string]string)
|
|
env["RESTIC_PASSWORD"] = b.Key
|
|
repo, err := b.generateRepo()
|
|
env["RESTIC_REPOSITORY"] = repo
|
|
for key, value := range b.Env {
|
|
env[strings.ToUpper(key)] = value
|
|
}
|
|
return env, err
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func (b Backend) validate() error {
|
|
if b.Type == "" {
|
|
return fmt.Errorf(`Backend "%s" has no "type"`, b.name)
|
|
}
|
|
if b.Path == "" {
|
|
return fmt.Errorf(`Backend "%s" has no "path"`, b.name)
|
|
}
|
|
if b.Key == "" {
|
|
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
|
|
}
|
|
}
|
|
env, err := b.getEnv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
options := ExecuteOptions{Envs: env}
|
|
// Check if already initialized
|
|
_, err = ExecuteResticCommand(options, "snapshots")
|
|
if err == nil {
|
|
return nil
|
|
} else {
|
|
// If not initialize
|
|
colors.Body.Printf("Initializing backend \"%s\"...\n", b.name)
|
|
out, err := ExecuteResticCommand(options, "init")
|
|
if VERBOSE {
|
|
colors.Faint.Println(out)
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
|
|
func (b Backend) Exec(args []string) error {
|
|
env, err := b.getEnv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
options := ExecuteOptions{Envs: env}
|
|
out, err := ExecuteResticCommand(options, args...)
|
|
if err != nil {
|
|
colors.Error.Println(out)
|
|
return err
|
|
}
|
|
if VERBOSE {
|
|
colors.Faint.Println(out)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b Backend) ExecDocker(l Location, args []string) (string, error) {
|
|
env, err := b.getEnv()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
volume := l.getVolumeName()
|
|
path, _ := l.getPath()
|
|
options := ExecuteOptions{
|
|
Command: "docker",
|
|
Envs: env,
|
|
}
|
|
docker := []string{
|
|
"run", "--rm",
|
|
"--entrypoint", "ash",
|
|
"--workdir", path,
|
|
"--volume", volume + ":" + path,
|
|
}
|
|
if hostname, err := os.Hostname(); err == nil {
|
|
docker = append(docker, "--hostname", hostname)
|
|
}
|
|
if b.Type == "local" {
|
|
actual := env["RESTIC_REPOSITORY"]
|
|
docker = append(docker, "--volume", actual+":"+"/repo")
|
|
env["RESTIC_REPOSITORY"] = "/repo"
|
|
}
|
|
for key, value := range env {
|
|
docker = append(docker, "--env", key+"="+value)
|
|
}
|
|
docker = append(docker, "restic/restic", "-c", "restic "+strings.Join(args, " "))
|
|
out, err := ExecuteCommand(options, docker...)
|
|
return out, err
|
|
}
|