FULL OC-DISCOVERY LOGIC

daemons/node/stream/publish.go

@@ -0,0 +1,150 @@
+package stream
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"oc-discovery/daemons/node/common"
+	"time"
+
+	oclib "cloud.o-forge.io/core/oc-lib"
+	"cloud.o-forge.io/core/oc-lib/models/peer"
+	"cloud.o-forge.io/core/oc-lib/tools"
+	"github.com/libp2p/go-libp2p/core/network"
+	pp "github.com/libp2p/go-libp2p/core/peer"
+	"github.com/libp2p/go-libp2p/core/protocol"
+)
+
+func (ps *StreamService) PublishResources(dt *tools.DataType, user string, toPeerID string, resource []byte) error {
+	access := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.PEER), nil)
+	p := access.LoadOne(toPeerID)
+	if p.Err != "" {
+		return errors.New(p.Err)
+	} else {
+		ad, err := pp.AddrInfoFromString(p.Data.(*peer.Peer).StreamAddress)
+		if err != nil {
+			return err
+		}
+		ps.write(tools.PB_SEARCH, toPeerID, ad, dt, user, resource, ProtocolSearchResource)
+	}
+	return nil
+}
+
+func (ps *StreamService) SearchKnownPublishEvent(dt *tools.DataType, user string, search string) error {
+	access := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.PEER), nil)
+	peers := access.Search(nil, search, false)
+	if peers.Err != "" {
+		return errors.New(peers.Err)
+	} else {
+		b, err := json.Marshal(map[string]string{"search": search})
+		if err != nil {
+			return err
+		}
+		for _, p := range peers.Data {
+			ad, err := pp.AddrInfoFromString(p.(*peer.Peer).StreamAddress)
+			if err != nil {
+				continue
+			}
+			ps.write(tools.PB_SEARCH, p.GetID(), ad, dt, user, b, ProtocolSearchResource)
+		}
+	}
+	return nil
+}
+
+func (ps *StreamService) SearchPartnersPublishEvent(dt *tools.DataType, user string, search string) error {
+	if peers, err := ps.searchPeer(fmt.Sprintf("%v", peer.PARTNER.EnumIndex())); err != nil {
+		return err
+	} else {
+		b, err := json.Marshal(map[string]string{"search": search})
+		if err != nil {
+			return err
+		}
+		for _, p := range peers {
+			ad, err := pp.AddrInfoFromString(p.StreamAddress)
+			if err != nil {
+				continue
+			}
+			ps.write(tools.PB_SEARCH, p.GetID(), ad, dt, user, b, ProtocolSearchResource)
+		}
+	}
+	return nil
+}
+
+func (ps *StreamService) ToPartnerPublishEvent(
+	ctx context.Context, action tools.PubSubAction, dt *tools.DataType, user string, payload []byte) error {
+	if peers, err := ps.searchPeer(fmt.Sprintf("%v", peer.PARTNER.EnumIndex())); err != nil {
+		return err
+	} else {
+		for _, p := range peers {
+			for _, protocol := range protocols {
+				ad, err := pp.AddrInfoFromString(p.StreamAddress)
+				if err != nil {
+					continue
+				}
+				ps.write(action, p.GetID(), ad, dt, user, payload, protocol)
+			}
+		}
+	}
+	return nil
+}
+
+func (s *StreamService) write(
+	action tools.PubSubAction,
+	did string,
+	peerID *pp.AddrInfo,
+	dt *tools.DataType,
+	user string,
+	payload []byte,
+	proto protocol.ID) error {
+	logger := oclib.GetLogger()
+
+	name := action.String() + "#" + peerID.ID.String()
+	if dt != nil {
+		name = action.String() + "." + (*dt).String() + "#" + peerID.ID.String()
+	}
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	if s.Streams[proto] == nil {
+		s.Streams[proto] = map[pp.ID]*common.Stream{}
+	}
+
+	if s.Streams[proto][peerID.ID] == nil {
+		// should create a very temp stream
+		ctxTTL, err := context.WithTimeout(context.Background(), 10*time.Second)
+		if err == nil {
+			if s.Host.Network().Connectedness(peerID.ID) != network.Connected {
+				_ = s.Host.Connect(ctxTTL, *peerID)
+				str, err := s.Host.NewStream(ctxTTL, peerID.ID, ProtocolHeartbeatPartner)
+				if err == nil {
+					s.Streams[ProtocolHeartbeatPartner][peerID.ID] = &common.Stream{
+						DID:    did,
+						Stream: str,
+						Expiry: time.Now().UTC().Add(5 * time.Second),
+					}
+					str2, err := s.Host.NewStream(ctxTTL, peerID.ID, proto)
+					if err == nil {
+						s.Streams[proto][peerID.ID] = &common.Stream{
+							DID:    did,
+							Stream: str2,
+							Expiry: time.Now().UTC().Add(5 * time.Second),
+						}
+					}
+				}
+
+			}
+		}
+		return errors.New("no stream available for protocol " + fmt.Sprintf("%v", proto) + " from PID " + peerID.ID.String())
+	}
+	stream := s.Streams[proto][peerID.ID]
+
+	enc := json.NewEncoder(stream.Stream)
+
+	evt := common.NewEvent(name, peerID.ID.String(), dt, user, payload)
+	if err := enc.Encode(evt); err != nil {
+		stream.Stream.Close()
+		logger.Err(err)
+		return nil
+	}
+	return nil
+}