You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
1.6 KiB
78 lines
1.6 KiB
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
|
|
}
|