Keep Peer Caching + Resource Verification.

daemons/node/nats.go

@@ -5,16 +5,48 @@ import (
 	"encoding/json"
 	"fmt"
 	"oc-discovery/daemons/node/common"
+	"oc-discovery/daemons/node/stream"
 
 	oclib "cloud.o-forge.io/core/oc-lib"
 	"cloud.o-forge.io/core/oc-lib/config"
 	"cloud.o-forge.io/core/oc-lib/models/peer"
+	"cloud.o-forge.io/core/oc-lib/models/resources"
 	"cloud.o-forge.io/core/oc-lib/tools"
 	pp "github.com/libp2p/go-libp2p/core/peer"
 )
 
-func ListenNATS(n Node) {
+func ListenNATS(n *Node) {
 	tools.NewNATSCaller().ListenNats(map[tools.NATSMethod]func(tools.NATSResponse){
+		tools.VERIFY_RESOURCE: func(resp tools.NATSResponse) {
+			if resp.FromApp == config.GetAppName() {
+				return
+			}
+			if res, err := resources.ToResource(resp.Datatype.EnumIndex(), resp.Payload); err == nil {
+				access := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.PEER), nil)
+				p := access.LoadOne(res.GetCreatorID())
+				realP := p.ToPeer()
+				if realP == nil {
+					return
+				} else if realP.Relation == peer.SELF {
+					pubKey, err := common.PubKeyFromString(realP.PublicKey) // extract pubkey from pubkey str
+					if err != nil {
+						return
+					}
+					ok, _ := pubKey.Verify(resp.Payload, res.GetSignature())
+					if b, err := json.Marshal(stream.Verify{
+						IsVerified: ok,
+					}); err == nil {
+						tools.NewNATSCaller().SetNATSPub(tools.VERIFY_RESOURCE, tools.NATSResponse{
+							FromApp: "oc-discovery",
+							Method:  int(tools.VERIFY_RESOURCE),
+							Payload: b,
+						})
+					}
+				} else if realP.Relation != peer.BLACKLIST {
+					n.StreamService.PublishVerifyResources(&resp.Datatype, resp.User, realP.PeerID, resp.Payload)
+				}
+			}
+		},
 		tools.CREATE_PEER: func(resp tools.NATSResponse) {
 			if resp.FromApp == config.GetAppName() {
 				return
@@ -30,9 +62,16 @@ func ListenNATS(n Node) {
 			p = p.Deserialize(m, p).(*peer.Peer)
 
 			ad, err := pp.AddrInfoFromString(p.PeerID)
+			if err != nil {
+				return
+			}
 			n.StreamService.Mu.Lock()
 			defer n.StreamService.Mu.Unlock()
 
+			n.Mu.Lock()
+			n.Peers[ad.ID] = p.State == peer.ONLINE
+			n.Mu.Unlock()
+
 			if p.Relation == peer.PARTNER {
 				n.StreamService.ConnectToPartner(ad.ID, ad)
 			} else {
@@ -50,6 +89,7 @@ func ListenNATS(n Node) {
 				}
 				n.StreamService.Streams = ps
 			}
+
 		},
 		tools.PROPALGATION_EVENT: func(resp tools.NATSResponse) {
 			var propalgation tools.PropalgationMessage