From ecfdb2d440a14b3217ed0266af2063492cf53e6d Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Wed, 1 May 2019 13:09:47 -0600
Subject: [PATCH] stdout should erase file

---
 src/croc/croc.go | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/croc/croc.go b/src/croc/croc.go
index 9351f24a..317c94a3 100644
--- a/src/croc/croc.go
+++ b/src/croc/croc.go
@@ -8,6 +8,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"io"
+	"io/ioutil"
 	"math"
 	"os"
 	"path"
@@ -367,16 +368,24 @@ func (c *Client) transfer(options TransferOptions) (err error) {
 		var done bool
 		data, err = c.conn[0].Receive()
 		if err != nil {
-			return
+			break
 		}
 		done, err = c.processMessage(data)
 		if err != nil {
-			return
+			break
 		}
 		if done {
 			break
 		}
 	}
+
+	if c.Options.Stdout && !c.Options.IsSender {
+		pathToFile := path.Join(
+			c.FilesToTransfer[c.FilesToTransferCurrentNum].FolderRemote,
+			c.FilesToTransfer[c.FilesToTransferCurrentNum].Name,
+		)
+		os.Remove(pathToFile)
+	}
 	return
 }
 
@@ -720,6 +729,14 @@ func (c *Client) receiveData(i int) {
 		if c.TotalChunksTransfered == len(c.CurrentFileChunks) || c.TotalSent == c.FilesToTransfer[c.FilesToTransferCurrentNum].Size {
 			log.Debug("finished receiving!")
 			c.CurrentFile.Close()
+			if c.Options.Stdout {
+				pathToFile := path.Join(
+					c.FilesToTransfer[c.FilesToTransferCurrentNum].FolderRemote,
+					c.FilesToTransfer[c.FilesToTransferCurrentNum].Name,
+				)
+				b, _ := ioutil.ReadFile(pathToFile)
+				fmt.Print(string(b))
+			}
 			log.Debug("sending close-sender")
 			err = message.Send(c.conn[0], c.Key, message.Message{
 				Type: "close-sender",