From d0d15f2bd58324cd89c341e478fca3e40133bda1 Mon Sep 17 00:00:00 2001
From: Unbewohnte <unbewohnte@gmail.com>
Date: Fri, 5 Nov 2021 22:11:14 +0300
Subject: [PATCH] [node] improved mainloop logic

---
 src/node/node.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/node/node.go b/src/node/node.go
index 6f335fc..340790e 100644
--- a/src/node/node.go
+++ b/src/node/node.go
@@ -246,6 +246,12 @@ func (node *Node) Start() {
 
 						protocol.SendPacket(node.Net.Conn, endFilePacket)
 
+						// because there`s still no handling for directories - send
+						// done packet
+						protocol.SendPacket(node.Net.Conn, protocol.Packet{
+							Header: protocol.HeaderDone,
+						})
+
 						node.State.Stopped = true
 					} else {
 						node.State.Stopped = true
@@ -302,7 +308,7 @@ func (node *Node) Start() {
 						panic(err)
 					}
 
-					fmt.Printf("| ID: %d\n| Filename: %s\n| Size: %.2f MB\n| Checksum: %s\n", file.ID, file.Name, float32(file.Size)/1024/1024, file.Checksum)
+					fmt.Printf("| Filename: %s\n| Size: %.2f MB\n| Checksum: %s\n", file.Name, float32(file.Size)/1024/1024, file.Checksum)
 					var answer string
 					fmt.Printf("| Download ? [Y/n]: ")
 					fmt.Scanln(&answer)
@@ -453,7 +459,7 @@ func (node *Node) Start() {
 					}
 				}
 
-				node.State.Stopped = true
+				// node.State.Stopped = true
 
 			case protocol.HeaderEncryptionKey:
 				// retrieve the key
@@ -467,6 +473,9 @@ func (node *Node) Start() {
 
 				node.Net.EncryptionKey = encrKey
 
+			case protocol.HeaderDone:
+				node.State.Stopped = true
+
 			case protocol.HeaderDisconnecting:
 				node.State.Stopped = true