From b8d2e1bede3b0ba6cb216b8e02fd9b727923412d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Camille=20Eyri=C3=A8s?= <camille.eyries@gmail.com>
Date: Sun, 22 Oct 2017 12:14:46 +0200
Subject: [PATCH] receive directories to \'croc_received\'

---
 connect.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/connect.go b/connect.go
index d5f32bf4..39fb7ece 100644
--- a/connect.go
+++ b/connect.go
@@ -42,6 +42,7 @@ type FileMetaData struct {
 }
 
 const (
+	crocReceiveDir   = "croc_received"
 	tmpTarGzFileName = "to_send.tmp.tar.gz"
 )
 
@@ -76,6 +77,7 @@ func NewConnection(flags *Flags) *Connection {
 			flags.File = tmpTarGzFileName
 			// we set the value IsDir to true
 			c.File.IsDir = true
+
 			fmt.Println("Done !")
 			c.File.Name = path.Base(tmpTarGzFileName)
 		} else {
@@ -390,6 +392,18 @@ func (c *Connection) runClient() error {
 			fmt.Printf("\nReceived file written to %s\n", c.File.Name)
 		}
 
+		if c.File.IsDir { // if the file was originally a dir
+			fmt.Print("Since the receive file was originally a directory, uncompressing... ")
+			tarinator.UnTarinate(crocReceiveDir, tmpTarGzFileName)
+			fmt.Println("Done !\nDirectory written into " + crocReceiveDir)
+
+			// we remove the old tar.gz file
+			err := os.RemoveAll(tmpTarGzFileName)
+			if err != nil {
+				log.Error(err)
+			}
+		}
+
 	}
 	return nil
 }