Add main binary

cmd_backup.go

@@ -0,0 +1,99 @@
+package main
+
+import (
+	"bytes"
+	"crypto/sha256"
+	"errors"
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+
+	"github.com/fd0/khepri/storage"
+)
+
+func hash(filename string) (storage.ID, error) {
+	h := sha256.New()
+	f, err := os.Open(filename)
+	if err != nil {
+		return nil, err
+	}
+
+	io.Copy(h, f)
+	return h.Sum([]byte{}), nil
+}
+
+func archive_dir(repo *storage.DirRepository, path string) (storage.ID, error) {
+	log.Printf("archiving dir %q", path)
+	dir, err := os.Open(path)
+	if err != nil {
+		log.Printf("open(%q): %v\n", path, err)
+		return nil, err
+	}
+
+	entries, err := dir.Readdir(-1)
+	if err != nil {
+		log.Printf("readdir(%q): %v\n", path, err)
+		return nil, err
+	}
+
+	// use nil ID for empty directories
+	if len(entries) == 0 {
+		return nil, nil
+	}
+
+	t := storage.NewTree()
+	for _, e := range entries {
+		node := storage.NodeFromFileInfo(e)
+
+		var id storage.ID
+		var err error
+
+		if e.IsDir() {
+			id, err = archive_dir(repo, filepath.Join(path, e.Name()))
+		} else {
+			id, err = repo.PutFile(filepath.Join(path, e.Name()))
+		}
+
+		node.Content = id
+
+		t.Nodes = append(t.Nodes, node)
+
+		if err != nil {
+			log.Printf("  error storing %q: %v\n", e.Name(), err)
+			continue
+		}
+	}
+
+	log.Printf("  dir %q: %v entries", path, len(t.Nodes))
+
+	var buf bytes.Buffer
+	t.Save(&buf)
+	id, err := repo.PutRaw(buf.Bytes())
+
+	if err != nil {
+		log.Printf("error saving tree to repo: %v", err)
+	}
+
+	log.Printf("tree for %q saved at %s", path, id)
+
+	return id, nil
+}
+
+func commandBackup(repo *storage.DirRepository, args []string) error {
+	if len(args) != 1 {
+		return errors.New("usage: backup dir")
+	}
+
+	target := args[0]
+
+	id, err := archive_dir(repo, target)
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("%q archived as %v\n", target, id)
+
+	return nil
+}