From 11d6aa53287ab2a7d8665bf505b1bcac606f9119 Mon Sep 17 00:00:00 2001
From: Alexander Neumann <alexander@bumpern.de>
Date: Fri, 1 Aug 2014 22:20:15 +0200
Subject: [PATCH] Fix commands

---
 cmd/khepri/cmd_backup.go  | 2 +-
 cmd/khepri/cmd_restore.go | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/cmd/khepri/cmd_backup.go b/cmd/khepri/cmd_backup.go
index b4770bdcb..71095b7c0 100644
--- a/cmd/khepri/cmd_backup.go
+++ b/cmd/khepri/cmd_backup.go
@@ -70,7 +70,7 @@ func archive_dir(repo *khepri.DirRepository, path string) (khepri.ID, error) {
 
 	var buf bytes.Buffer
 	t.Save(&buf)
-	id, err := repo.PutRaw(buf.Bytes())
+	id, err := repo.PutRaw(khepri.TypeRef, buf.Bytes())
 
 	if err != nil {
 		log.Printf("error saving tree to repo: %v", err)
diff --git a/cmd/khepri/cmd_restore.go b/cmd/khepri/cmd_restore.go
index 29514fa42..bc6bb059c 100644
--- a/cmd/khepri/cmd_restore.go
+++ b/cmd/khepri/cmd_restore.go
@@ -13,7 +13,7 @@ import (
 func restore_file(repo *khepri.DirRepository, node khepri.Node, target string) error {
 	fmt.Printf("  restore file %q\n", target)
 
-	rd, err := repo.Get(node.Content)
+	rd, err := repo.Get(khepri.TypeBlob, node.Content)
 	if err != nil {
 		return err
 	}
@@ -49,7 +49,7 @@ func restore_file(repo *khepri.DirRepository, node khepri.Node, target string) e
 
 func restore_dir(repo *khepri.DirRepository, id khepri.ID, target string) error {
 	fmt.Printf("  restore dir %q\n", target)
-	rd, err := repo.Get(id)
+	rd, err := repo.Get(khepri.TypeRef, id)
 	if err != nil {
 		return err
 	}