mirror of
https://github.com/cupcakearmy/autorestic.git
synced 2024-08-28 23:27:18 +02:00
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
type Backend struct {
|
|
Name string `mapstructure:"name"`
|
|
Type string `mapstructure:"type"`
|
|
Path string `mapstructure:"path"`
|
|
Key string `mapstructure:"key"`
|
|
Env map[string]string `mapstructure:"env"`
|
|
}
|
|
|
|
func GetBackend(name string) (Backend, bool) {
|
|
c := GetConfig()
|
|
for _, b := range c.Backends {
|
|
if b.Name == name {
|
|
return b, true
|
|
}
|
|
}
|
|
return Backend{}, false
|
|
}
|
|
|
|
func (b Backend) generateRepo() (string, error) {
|
|
switch b.Type {
|
|
case "local":
|
|
return GetPathRelativeToConfig(b.Path), nil
|
|
case "b2", "azure", "gs", "s3", "sftp", "rest":
|
|
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 {
|
|
env := make(map[string]string)
|
|
env["RESTIC_PASSWORD"] = b.Key
|
|
repo, err := b.generateRepo()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
env["RESTIC_REPOSITORY"] = repo
|
|
return env
|
|
}
|
|
|
|
func (b Backend) validate() error {
|
|
options := ExecuteOptions{Envs: b.getEnv()}
|
|
// Check if already initialized
|
|
_, err := ExecuteResticCommand(options, "snapshots")
|
|
if err == nil {
|
|
return nil
|
|
} else {
|
|
// If not initialize
|
|
out, err := ExecuteResticCommand(options, "init")
|
|
fmt.Println(out)
|
|
return err
|
|
}
|
|
}
|
|
|
|
func (b Backend) Exec(args []string) error {
|
|
options := ExecuteOptions{Envs: b.getEnv()}
|
|
out, err := ExecuteResticCommand(options, args...)
|
|
fmt.Println(out)
|
|
return err
|
|
}
|