Add restore-dir command

1 files changed, 7 insertions, 2 deletions

diff --git a/main.go b/main.go
index 9057231..b0f3f6b 100644
--- a/main.go
+++ b/main.go
@@ -12,7 +12,7 @@ type subcmd func(args []string) int
 
 var subcmds = map[string]subcmd{
 	"write-dir":       WriteDir,
-	"restore-dir":     notImplementedYet,
+	"restore-dir":     RestoreDir,
 	"take-snapshot":   notImplementedYet,
 	"create-snapshot": notImplementedYet,
 	"list-snapshots":  notImplementedYet,
@@ -56,7 +56,12 @@ func Main() int {
 	}
 	defer objectstore.Close()
 
-	remaining := flag.Args()
+	remaining := make([]string, 0)
+	for _, arg := range flag.Args() {
+		if arg != "" {
+			remaining = append(remaining, arg)
+		}
+	}
 
 	var cmd subcmd
 	if len(remaining) > 0 {