forked from TrueCloudLab/restic
74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"sort"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/fd0/khepri/storage"
|
||
|
|
"github.com/jessevdk/go-flags"
|
||
|
|
)
|
||
|
|
|
||
|
|
var Opts struct {
|
||
|
|
Repo string `short:"r" long:"repo" description:"Repository directory to backup to/restor from"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func errmsg(code int, format string, data ...interface{}) {
|
||
|
|
if len(format) > 0 && format[len(format)-1] != '\n' {
|
||
|
|
format += "\n"
|
||
|
|
}
|
||
|
|
fmt.Fprintf(os.Stderr, format, data...)
|
||
|
|
os.Exit(code)
|
||
|
|
}
|
||
|
|
|
||
|
|
type commandFunc func(*storage.DirRepository, []string) error
|
||
|
|
|
||
|
|
var commands map[string]commandFunc
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
commands = make(map[string]commandFunc)
|
||
|
|
commands["backup"] = commandBackup
|
||
|
|
commands["restore"] = commandRestore
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
Opts.Repo = os.Getenv("KHEPRI_REPOSITORY")
|
||
|
|
if Opts.Repo == "" {
|
||
|
|
Opts.Repo = "khepri-backup"
|
||
|
|
}
|
||
|
|
args, err := flags.Parse(&Opts)
|
||
|
|
|
||
|
|
if e, ok := err.(*flags.Error); ok && e.Type == flags.ErrHelp {
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(args) == 0 {
|
||
|
|
cmds := []string{}
|
||
|
|
for k := range commands {
|
||
|
|
cmds = append(cmds, k)
|
||
|
|
}
|
||
|
|
sort.Strings(cmds)
|
||
|
|
fmt.Printf("nothing to do, available commands: [%v]\n", strings.Join(cmds, "|"))
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
cmd := args[0]
|
||
|
|
|
||
|
|
f, ok := commands[cmd]
|
||
|
|
if !ok {
|
||
|
|
errmsg(1, "unknown command: %q\n", cmd)
|
||
|
|
}
|
||
|
|
|
||
|
|
repo, err := storage.NewDirRepository(Opts.Repo)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
errmsg(1, "unable to create/open repo: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = f(repo, args[1:])
|
||
|
|
if err != nil {
|
||
|
|
errmsg(1, "error executing command %q: %v", cmd, err)
|
||
|
|
}
|
||
|
|
}
|