koi-net 1.0.0b14__tar.gz → 1.0.0b15__tar.gz

{koi_net-1.0.0b14 → koi_net-1.0.0b15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: koi-net
-Version: 1.0.0b14
+Version: 1.0.0b15
 Summary: Implementation of KOI-net protocol in Python
 Project-URL: Homepage, https://github.com/BlockScience/koi-net/
 Author-email: Luke Miller <luke@block.science>

{koi_net-1.0.0b14 → koi_net-1.0.0b15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "koi-net"
-version = "1.0.0-beta.14"
+version = "1.0.0-beta.15"
 description = "Implementation of KOI-net protocol in Python"
 authors = [
     {name = "Luke Miller", email = "luke@block.science"}

{koi_net-1.0.0b14 → koi_net-1.0.0b15}/src/koi_net/processor/default_handlers.py

@@ -9,6 +9,8 @@ from .handler import KnowledgeHandler, HandlerType, STOP_CHAIN
 from .knowledge_object import KnowledgeObject, KnowledgeSource
 from ..protocol.event import Event, EventType
 from ..protocol.edge import EdgeProfile, EdgeStatus, EdgeType
+from ..protocol.node import NodeProfile
+from ..protocol.helpers import generate_edge_bundle
 
 logger = logging.getLogger(__name__)
 
@@ -124,6 +126,57 @@ def edge_negotiation_handler(processor: ProcessorInterface, kobj: KnowledgeObjec
 
 # Network handlers
 
+@KnowledgeHandler.create(HandlerType.Network, rid_types=[KoiNetNode])
+def coordinator_contact(processor: ProcessorInterface, kobj: KnowledgeObject):
+    node_profile = kobj.bundle.validate_contents(NodeProfile)
+    
+    # looking for event provider of nodes
+    if KoiNetNode not in node_profile.provides.event:
+        return
+    
+    # already have an edge established
+    if processor.network.graph.get_edge_profile(
+        source=kobj.rid,
+        target=processor.identity.rid,
+    ) is not None:
+        return
+    
+    logger.info("Identified a coordinator!")
+    logger.info("Proposing new edge")
+    
+    if processor.identity.profile.node_type == NodeType.FULL:
+        edge_type = EdgeType.WEBHOOK
+    else:
+        edge_type = EdgeType.POLL
+    
+    # queued for processing
+    processor.handle(bundle=generate_edge_bundle(
+        source=kobj.rid,
+        target=processor.identity.rid,
+        edge_type=edge_type,
+        rid_types=[KoiNetNode]
+    ))
+    
+    logger.info("Catching up on network state")
+    
+    payload = processor.network.request_handler.fetch_rids(
+        node=kobj.rid, 
+        rid_types=[KoiNetNode]
+    )
+    for rid in payload.rids:
+        if rid == processor.identity.rid:
+            logger.info("Skipping myself")
+            continue
+        if processor.cache.exists(rid):
+            logger.info(f"Skipping known RID '{rid}'")
+            continue
+        
+        # marked as external since we are handling RIDs from another node
+        # will fetch remotely instead of checking local cache
+        processor.handle(rid=rid, source=KnowledgeSource.External)
+    logger.info("Done")
+    
+
 @KnowledgeHandler.create(HandlerType.Network)
 def basic_network_output_filter(processor: ProcessorInterface, kobj: KnowledgeObject):
     """Default network handler.