bc/peer/p2plib/messages.go

package p2plib

import (
	"encoding/json"
	"fmt"
	"log"
	"time"

	"github.com/fatih/color"
)

type Msg struct {
	Type      string    `json:"type"`
	Date      time.Time `json:"date"`
	Content   string    `json:"content"`
	PeersList PeersList `json:"peerslist"`
	Data      []byte    `json:"data"`
}

func MessageHandler(peer Peer, msg Msg) {

	log.Println("[New msg]")
	log.Println(msg)

	switch msg.Type {
	case "Hi":
		color.Yellow(msg.Type)
		color.Green(msg.Content)
		break
	case "PeersList":
		color.Blue("newPeerslist")
		fmt.Println(msg.PeersList)
		color.Red("PeersList")

		UpdateNetworkPeersList(peer.Conn, msg.PeersList)
		PropagatePeersList(peer)
		PrintPeersList()
		break
	case "PeersList_Response":
		//for the moment is not beeing used
		color.Blue("newPeerslist, from PeersList_Response")
		fmt.Println(msg.PeersList)
		color.Red("PeersList_Response")

		UpdateNetworkPeersList(peer.Conn, msg.PeersList)
		PropagatePeersList(peer)
		PrintPeersList()
		break
	case "Block":
		/*//TODO check if the block is signed by an autorized emitter
		if !blockchain.blockExists(msg.Block) {
			blockchain.addBlock(msg.Block)
			propagateBlock(msg.Block)
		}*/
		break
	default:
		log.Println("Msg.Type not supported")
		break
	}

}
func (msg *Msg) Construct(msgtype string, msgcontent string) {
	msg.Type = msgtype
	msg.Content = msgcontent
	msg.Date = time.Now()
}
func (msg Msg) ToBytes() []byte {
	msgS, err := json.Marshal(msg)
	check(err)
	l := string(msgS) + "\n"
	r := []byte(l)
	return r
}
func (msg Msg) CreateFromBytes(bytes []byte) Msg {
	err := json.Unmarshal(bytes, &msg)
	check(err)
	return msg
}