From 7513702e9ca2ff2fcd13e70f21981fe184cd5737 Mon Sep 17 00:00:00 2001 From: Zack Scholl Date: Sat, 22 Sep 2018 06:29:11 -0700 Subject: [PATCH] add transfer complete --- src/sender/sender.go | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/sender/sender.go b/src/sender/sender.go index f85cd10..83d830c 100644 --- a/src/sender/sender.go +++ b/src/sender/sender.go @@ -37,11 +37,6 @@ func Send(done chan struct{}, c *websocket.Conn, fname string, codephrase string if strings.HasPrefix(err.Error(), "websocket: close 100") { err = nil } - if err == nil { - fmt.Fprintf(os.Stderr, "Transfer complete") - } else { - fmt.Fprintf(os.Stderr, err.Error()) - } } done <- struct{}{} } @@ -204,8 +199,10 @@ func send(c *websocket.Conn, fname string, codephrase string) (err error) { log.Debugf("[%d] determing whether it went ok", step) if bytes.Equal(message, []byte("ok")) { log.Debug("file transfered successfully") + fmt.Fprintf(os.Stderr, "\nTransfer complete") return nil } else { + fmt.Fprintf(os.Stderr, "\nTransfer corrupted") return errors.New("file not transfered succesfully") } default: