forked from TrueCloudLab/restic
138 lines
2.3 KiB
Go
138 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/fd0/khepri"
|
|
)
|
|
|
|
func restore_file(repo *khepri.Repository, node khepri.Node, target string) error {
|
|
fmt.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 {
|
|
fmt.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.Tree, target)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("%q restored to %q\n", id, target)
|
|
|
|
return nil
|
|
}
|