From ccaaf07dad84ebc9ea1d4487d5da6ecf47ec77b9 Mon Sep 17 00:00:00 2001
From: anthdm <anthony@academiclabs.co>
Date: Sat, 27 Jan 2018 13:39:07 +0100
Subject: [PATCH] wip refactor

---
 pkg/network/server.go | 34 ++++++++++++++++++++++++++++++----
 1 file changed, 30 insertions(+), 4 deletions(-)

diff --git a/pkg/network/server.go b/pkg/network/server.go
index ea3e3fc8e..cb9880a8a 100644
--- a/pkg/network/server.go
+++ b/pkg/network/server.go
@@ -132,6 +132,8 @@ func (s *Server) loop() {
 		case peer := <-s.register:
 			s.logger.Printf("peer registered from address %s", peer.conn.RemoteAddr())
 
+			s.peers[peer] = true
+
 			// only respond with the version mesage if the peer initiated the connection.
 			if peer.initiator {
 				resp, err := s.handlePeerConnected()
@@ -148,6 +150,7 @@ func (s *Server) loop() {
 		case tuple := <-s.message:
 			if err := s.processMessage(tuple.msg, tuple.peer); err != nil {
 				s.logger.Fatalf("failed to process message: %s", err)
+				s.disconnect(tuple.peer)
 			}
 		case <-s.quit:
 			s.shutdown()
@@ -166,9 +169,10 @@ func (s *Server) processMessage(msg *Message, peer *Peer) error {
 		if err != nil {
 			return err
 		}
-		s.handleVersionCmd(v.(*Version), peer)
+		return s.handleVersionCmd(v.(*Version), peer)
 	case cmdVerack:
 	case cmdGetAddr:
+		return s.handleGetAddrCmd(msg, peer)
 	case cmdAddr:
 	case cmdGetHeaders:
 	case cmdHeaders:
@@ -198,21 +202,43 @@ func (s *Server) handlePeerConnected() (*Message, error) {
 }
 
 // Version declares the server's version.
-func (s *Server) handleVersionCmd(v *Version, peer *Peer) {
+func (s *Server) handleVersionCmd(v *Version, peer *Peer) error {
 	// TODO: check version and verify to trust that node.
 
 	payload := newVersionPayload(s.port, s.userAgent, 0, s.relay)
 	b, err := payload.encode()
 	if err != nil {
-		s.disconnect(peer)
-		return
+		return err
 	}
+
+	// we respond with our version.
 	versionMsg := newMessage(s.net, cmdVersion, b)
 	peer.send <- versionMsg
 
+	// we respond with a verack, we successfully received peer's version
+	// at this point.
 	peer.verack = true
 	verackMsg := newMessage(s.net, cmdVerack, nil)
 	peer.send <- verackMsg
+
+	return nil
+}
+
+// After receiving the "getaddr" the server needs to respond with an "addr" message.
+// providing information about the other nodes in the network.
+// e.g. this server's connected peers.
+func (s *Server) handleGetAddrCmd(msg *Message, peer *Peer) error {
+	// payload := NewAddrPayload()
+	// b, err := payload.encode()
+	// if err != nil {
+	// 	return err
+	// }
+	var addrList []AddrWithTimestamp
+	for peer := range s.peers {
+		addrList = append(addrList, newAddrWithTimestampFromPeer(peer))
+	}
+
+	return nil
 }
 
 func logo() string {