Add decrypt, refactor

cmd/list/main.go

@@ -0,0 +1,110 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+
+	"code.google.com/p/go.crypto/ssh/terminal"
+
+	"github.com/fd0/khepri"
+	"github.com/fd0/khepri/backend"
+)
+
+func read_password(prompt string) string {
+	p := os.Getenv("KHEPRI_PASSWORD")
+	if p != "" {
+		return p
+	}
+
+	fmt.Print(prompt)
+	pw, err := terminal.ReadPassword(int(os.Stdin.Fd()))
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "unable to read password: %v", err)
+		os.Exit(2)
+	}
+	fmt.Println()
+
+	return string(pw)
+}
+
+func list(be backend.Server, key *khepri.Key, t backend.Type) {
+	ids, err := be.List(t)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "failed: %v\n", err)
+		os.Exit(3)
+	}
+
+	for _, id := range ids {
+		buf, err := be.Get(t, id)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "unable to get snapshot %s: %v\n", id, err)
+			continue
+		}
+
+		if t != backend.Key && t != backend.Blob {
+			buf, err = key.Decrypt(buf)
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "error: %v\n", err)
+				continue
+			}
+
+		}
+
+		if t == backend.Snapshot {
+			var sn khepri.Snapshot
+			err = json.Unmarshal(backend.Uncompress(buf), &sn)
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "error: %v\n", err)
+				continue
+			}
+
+			fmt.Printf("%s %s\n", id, sn.String())
+		} else if t == backend.Blob {
+			fmt.Printf("%s %d bytes (encrypted)\n", id, len(buf))
+		} else if t == backend.Tree {
+			fmt.Printf("%s\n", backend.Hash(buf))
+		} else if t == backend.Key {
+			k := &khepri.Key{}
+			err = json.Unmarshal(buf, k)
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "unable to unmashal key: %v\n", err)
+				continue
+			}
+			fmt.Println(key)
+		} else if t == backend.Lock {
+			fmt.Printf("lock: %v\n", id)
+		}
+	}
+}
+
+func main() {
+	if len(os.Args) != 2 {
+		fmt.Fprintf(os.Stderr, "usage: archive REPO\n")
+		os.Exit(1)
+	}
+	repo := os.Args[1]
+
+	be, err := backend.OpenLocal(repo)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "failed: %v\n", err)
+		os.Exit(1)
+	}
+
+	key, err := khepri.SearchKey(be, read_password("Enter Password for Repository: "))
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "failed: %v\n", err)
+		os.Exit(2)
+	}
+
+	fmt.Printf("keys:\n")
+	list(be, key, backend.Key)
+	fmt.Printf("---\nlocks:\n")
+	list(be, key, backend.Lock)
+	fmt.Printf("---\nsnapshots:\n")
+	list(be, key, backend.Snapshot)
+	fmt.Printf("---\ntrees:\n")
+	list(be, key, backend.Tree)
+	fmt.Printf("---\nblobs:\n")
+	list(be, key, backend.Blob)
+}