74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"backup-tool/s3helper"
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
func main() {
|
|
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
log.Fatalf("error loading .env file: (%s)", err.Error())
|
|
}
|
|
|
|
path, err := exec.LookPath("restic")
|
|
if err != nil {
|
|
log.Fatalf("could not find restic in PATH %s", err)
|
|
}
|
|
|
|
cmd := exec.Command(path, "backup", "--dry-run", "/media/piwigo")
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
log.Fatalf("error connecting to stdout: %q: %s", cmd.String(), err)
|
|
}
|
|
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
log.Fatalf("error running command: %s", err)
|
|
}
|
|
|
|
cmd = exec.Command(path, "forget", "--dry-run", "--keep-last", "7")
|
|
stdout, err = cmd.StdoutPipe()
|
|
if err != nil {
|
|
log.Fatalf("error connecting to stdout: %s", err)
|
|
}
|
|
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
log.Fatalf("error running command %q: %s", cmd.String(), err)
|
|
}
|
|
|
|
log.Print(stdout)
|
|
|
|
if len(os.Args) > 1 {
|
|
// only do this if we have been given a file to upload (for now)
|
|
// TODO: have this tool take the backup of the DB itself
|
|
b2Creds, err := getCredentials([]string{"B2_KEY_ID", "B2_APPLICATION_KEY"})
|
|
if err != nil {
|
|
log.Fatalf("could not get B2 credentials: %s", err)
|
|
}
|
|
s3helper.UploadFile(context.Background(), b2Creds)
|
|
}
|
|
|
|
}
|
|
|
|
func getCredentials(credNames []string) (map[string]string, error) {
|
|
creds := map[string]string{}
|
|
for _, name := range credNames {
|
|
value := os.Getenv(name)
|
|
if value == "" {
|
|
return nil, fmt.Errorf("missing or empty ENV var: %s", name)
|
|
}
|
|
creds[name] = value
|
|
}
|
|
|
|
return creds, nil
|
|
}
|