restic/main.go

74 lines
1.4 KiB
Go
Raw Normal View History

2014-04-27 22:00:15 +00:00
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)
}
}