blockchainIDsystem/peer/peers.go

package main

import (
	"fmt"
	"net"
	"time"

	"github.com/fatih/color"
)

type PeersList struct {
	PeerID string
	Peers  []Peer    `json:"peerslist"`
	Date   time.Time `json:"date"`
}

var outcomingPeersList PeersList
var incomingPeersList PeersList
var networkPeersList PeersList //the peers that have been received in the lists from other peers

/*
a future option is to put:
	type PeersList struct {
		Incoming	PeersList
		Outcoming	PeersList
		Network	PeersList
	}
*/

func peerIsInPeersList(p Peer, pl []Peer) int {
	r := -1
	for i, peer := range pl {
		if peer.IP+":"+peer.Port == p.IP+":"+p.Port {
			r = i
		}
	}
	return r
}

func deletePeerFromPeersList(p Peer, pl *PeersList) {
	i := peerIsInPeersList(p, pl.Peers)
	if i != -1 {
		//delete peer from pl.Peers
		pl.Peers = append(pl.Peers[:i], pl.Peers[i+1:]...)
	}
}
func appendPeerIfNoExist(pl PeersList, p Peer) PeersList {
	i := peerIsInPeersList(p, pl.Peers)
	if i == -1 {
		pl.Peers = append(pl.Peers, p)
	}
	return pl
}
func updateNetworkPeersList(conn net.Conn, newPeersList PeersList) {
	for _, peer := range newPeersList.Peers {
		if getIPPortFromConn(conn) == peer.IP+":"+peer.Port {
			peer.ID = newPeersList.PeerID
			color.Yellow(peer.ID)
		}
		i := peerIsInPeersList(peer, networkPeersList.Peers)
		if i == -1 {
			networkPeersList.Peers = append(networkPeersList.Peers, peer)
		} else {
			fmt.Println(networkPeersList.Peers[i])
			networkPeersList.Peers[i].ID = peer.ID
		}
	}
}
func searchPeerAndUpdate(p Peer) {
	for _, peer := range outcomingPeersList.Peers {
		color.Red(p.IP + ":" + p.Port)
		color.Yellow(peer.IP + ":" + peer.Port)
		if p.IP+":"+p.Port == peer.IP+":"+peer.Port {
			peer.ID = p.ID
		}
	}
}

//send the outcomingPeersList to all the peers except the peer p that has send the outcomingPeersList
func propagatePeersList(p Peer) {
	for _, peer := range networkPeersList.Peers {
		if peer.Conn != nil {
			if peer.ID != p.ID && p.ID != "" {
				color.Yellow(peer.ID + " - " + p.ID)
				var msg Msg
				msg.construct("PeersList", "here my outcomingPeersList")
				msg.PeersList = outcomingPeersList
				msgB := msg.toBytes()
				_, err := peer.Conn.Write(msgB)
				check(err)
			} else {
				/*
					for the moment, this is not being called, due that in the IncomingPeersList,
					there is no peer.ID, so in the comparation wih the peer that has send the
					peersList, is comparing ID with "", so nevere enters this 'else' section

					maybe it's not needed. TODO check if it's needed the PeerList_Response
					For the moment is working without it
				*/
				//to the peer that has sent the peerList, we send our PeersList

				var msg Msg
				msg.construct("PeersList_Response", "here my outcomingPeersList")
				msg.PeersList = outcomingPeersList
				msgB := msg.toBytes()
				_, err := peer.Conn.Write(msgB)
				check(err)
			}
		} else {
			//connect to peer
			if peer.ID != p.ID && peer.ID != runningPeer.ID {
				if peerIsInPeersList(peer, outcomingPeersList.Peers) == -1 {
					color.Red("no connection, connecting to peer: " + peer.Port)
					connectToPeer(peer)
				}
			}

		}
	}
}
func printPeersList() {
	fmt.Println("")
	color.Blue("runningPeer.ID: " + runningPeer.ID)
	color.Green("OUTCOMING PEERSLIST:")
	for _, peer := range outcomingPeersList.Peers {
		fmt.Println(peer)
	}
	color.Green("INCOMING PEERSLIST:")
	for _, peer := range incomingPeersList.Peers {
		fmt.Println(peer)
	}

	color.Green("NETWORK PEERSLIST:")
	for _, peer := range networkPeersList.Peers {
		fmt.Println(peer)
	}
	fmt.Println("")
}

//send the block to all the peers of the outcomingPeersList
func propagateBlock(b Block) {
	//prepare the msg to send to all connected peers
	var msg Msg
	msg.construct("Block", "new block")
	msg.Block = b
	msgB := msg.toBytes()
	for _, peer := range outcomingPeersList.Peers {
		if peer.Conn != nil {
			_, err := peer.Conn.Write(msgB)
			check(err)
		}
	}
}