package main import ( "errors" "io" "log" "os" "path" "github.com/fd0/khepri" ) func restore_file(repo *khepri.Repository, node khepri.Node, target string) error { log.Printf(" restore file %q\n", target) rd, err := repo.Get(khepri.TYPE_BLOB, node.Content) if err != nil { return err } f, err := os.OpenFile(target, os.O_CREATE|os.O_WRONLY, 0600) defer f.Close() if err != nil { return err } _, err = io.Copy(f, rd) if err != nil { return err } err = f.Chmod(node.Mode) if err != nil { return err } err = f.Chown(int(node.User), int(node.Group)) if err != nil { return err } err = os.Chtimes(target, node.AccessTime, node.ModTime) if err != nil { return err } return nil } func restore_dir(repo *khepri.Repository, id khepri.ID, target string) error { log.Printf(" restore dir %q\n", target) rd, err := repo.Get(khepri.TYPE_BLOB, id) if err != nil { return err } t := khepri.NewTree() err = t.Restore(rd) if err != nil { return err } for _, node := range t.Nodes { name := path.Base(node.Name) if name == "." || name == ".." { return errors.New("invalid path") } nodepath := path.Join(target, name) if node.Mode.IsDir() { err = os.MkdirAll(nodepath, 0700) if err != nil { return err } err = os.Chmod(nodepath, node.Mode) if err != nil { return err } err = os.Chown(nodepath, int(node.User), int(node.Group)) if err != nil { return err } err = os.Chtimes(nodepath, node.AccessTime, node.ModTime) if err != nil { return err } err = restore_dir(repo, node.Content, nodepath) if err != nil { return err } } else { err = restore_file(repo, node, nodepath) if err != nil { return err } } } return nil } func commandRestore(repo *khepri.Repository, args []string) error { if len(args) != 2 { return errors.New("usage: restore ID dir") } id, err := khepri.ParseID(args[0]) if err != nil { errx(1, "invalid id %q: %v", args[0], err) } target := args[1] err = os.MkdirAll(target, 0700) if err != nil { return err } sn, err := khepri.LoadSnapshot(repo, id) if err != nil { log.Fatalf("error loading snapshot %s", id) } err = restore_dir(repo, sn.TreeID, target) if err != nil { return err } log.Printf("%q restored to %q\n", id, target) return nil }