restic/cmd/khepri/cmd_restore.go

133 lines
2.2 KiB
Go
Raw Normal View History

2014-04-27 22:00:15 +00:00
package main
import (
"errors"
"fmt"
"io"
"os"
"path"
2014-07-28 18:20:32 +00:00
"github.com/fd0/khepri"
2014-04-27 22:00:15 +00:00
)
2014-07-28 18:20:32 +00:00
func restore_file(repo *khepri.DirRepository, node khepri.Node, target string) error {
2014-04-27 22:00:15 +00:00
fmt.Printf(" restore file %q\n", target)
2014-08-01 20:20:15 +00:00
rd, err := repo.Get(khepri.TypeBlob, node.Content)
2014-04-27 22:00:15 +00:00
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
}
2014-07-28 18:20:32 +00:00
func restore_dir(repo *khepri.DirRepository, id khepri.ID, target string) error {
2014-04-27 22:00:15 +00:00
fmt.Printf(" restore dir %q\n", target)
2014-08-01 20:20:15 +00:00
rd, err := repo.Get(khepri.TypeRef, id)
2014-04-27 22:00:15 +00:00
if err != nil {
return err
}
2014-07-28 18:20:32 +00:00
t := khepri.NewTree()
2014-04-27 22:00:15 +00:00
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
}
2014-07-28 18:20:32 +00:00
func commandRestore(repo *khepri.DirRepository, args []string) error {
2014-04-27 22:00:15 +00:00
if len(args) != 2 {
return errors.New("usage: restore ID dir")
}
2014-07-28 18:20:32 +00:00
id, err := khepri.ParseID(args[0])
2014-04-27 22:00:15 +00:00
if err != nil {
errmsg(1, "invalid id %q: %v", args[0], err)
}
target := args[1]
err = os.MkdirAll(target, 0700)
if err != nil {
return err
}
err = restore_dir(repo, id, target)
if err != nil {
return err
}
fmt.Printf("%q restored to %q\n", id, target)
return nil
}