koi-net 1.1.0b3__tar.gz → 1.1.0b4__tar.gz

{koi_net-1.1.0b3 → koi_net-1.1.0b4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: koi-net
-Version: 1.1.0b3
+Version: 1.1.0b4
 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.1.0b3 → koi_net-1.1.0b4}/pyproject.toml

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

{koi_net-1.1.0b3 → koi_net-1.1.0b4}/src/koi_net/context.py

@@ -1,6 +1,6 @@
-
-from koi_net.effector import Effector
 from rid_lib.ext import Cache
+from .config import NodeConfig
+from .effector import Effector
 from .network.graph import NetworkGraph
 from .network.event_queue import NetworkEventQueue
 from .network.request_handler import RequestHandler
@@ -23,6 +23,7 @@ class ActionContext:
 
 class HandlerContext:
     identity: NodeIdentity
+    config: NodeConfig
     cache: Cache
     event_queue: NetworkEventQueue
     graph: NetworkGraph
@@ -33,6 +34,7 @@ class HandlerContext:
     def __init__(
         self,
         identity: NodeIdentity,
+        config: NodeConfig,
         cache: Cache,
         event_queue: NetworkEventQueue,
         graph: NetworkGraph,
@@ -40,6 +42,7 @@ class HandlerContext:
         effector: Effector
     ):
         self.identity = identity
+        self.config = config
         self.cache = cache
         self.event_queue = event_queue
         self.graph = graph

{koi_net-1.1.0b3 → koi_net-1.1.0b4}/src/koi_net/core.py

@@ -132,6 +132,7 @@ class NodeInterface(Generic[T]):
         
         self.handler_context = (HandlerContextOverride or HandlerContext)(
             identity=self.identity,
+            config=self.config,
             cache=self.cache,
             event_queue=self.event_queue,
             graph=self.graph,

{koi_net-1.1.0b3 → koi_net-1.1.0b4}/src/koi_net/network/event_queue.py

@@ -7,7 +7,7 @@ from rid_lib.ext import Cache
 from rid_lib.types import KoiNetNode
 
 from .graph import NetworkGraph
-from .request_handler import RequestHandler
+from .request_handler import NodeNotFoundError, RequestHandler
 from ..protocol.node import NodeProfile, NodeType
 from ..protocol.edge import EdgeProfile, EdgeType
 from ..protocol.event import Event
@@ -187,11 +187,13 @@ class NetworkEventQueue:
         
         try:  
             self.request_handler.broadcast_events(node, events=events)
-            return True
+
+        except NodeNotFoundError:
+            logger.warning("Broadcast failed (node not found)")
+            
         except httpx.ConnectError:
-            logger.warning("Broadcast failed")
+            logger.warning("Broadcast failed (couldn't connect)")
             
             if requeue_on_fail:
                 for event in events:
-                    self.push_event_to(event, node)
-            return False
+                    self.push_event_to(event, node)

{koi_net-1.1.0b3 → koi_net-1.1.0b4}/src/koi_net/network/request_handler.py

@@ -37,6 +37,23 @@ if TYPE_CHECKING:
 logger = logging.getLogger(__name__)
 
 
+# Custom error types for request handling
+class SelfRequestError(Exception):
+    """Raised when a node tries to request itself."""
+    pass
+
+class PartialNodeQueryError(Exception):
+    """Raised when attempting to query a partial node."""
+    pass
+
+class NodeNotFoundError(Exception):
+    """Raised when a node URL cannot be found."""
+    pass
+
+class UnknownPathError(Exception):
+    """Raised when an unknown path is requested."""
+    pass
+
 class RequestHandler:
     """Handles making requests to other KOI nodes."""
     
@@ -65,7 +82,7 @@ class RequestHandler:
         node_url = None
         
         if node_rid == self.identity.rid:
-            raise Exception("Don't talk to yourself")
+            raise SelfRequestError("Don't talk to yourself")
         
         node_bundle = self.effector.deref(node_rid)
                 
@@ -73,7 +90,7 @@ class RequestHandler:
             node_profile = node_bundle.validate_contents(NodeProfile)
             logger.debug(f"Found node profile: {node_profile}")
             if node_profile.node_type != NodeType.FULL:
-                raise Exception("Can't query partial node")
+                raise PartialNodeQueryError("Can't query partial node")
             node_url = node_profile.base_url
         
         else:
@@ -82,7 +99,7 @@ class RequestHandler:
                 node_url = self.identity.config.koi_net.first_contact.url
         
         if not node_url:
-            raise Exception("Node not found")
+            raise NodeNotFoundError("Node not found")
         
         logger.debug(f"Resolved {node_rid!r} to {node_url}")
         return node_url
@@ -124,7 +141,7 @@ class RequestHandler:
         elif path == FETCH_BUNDLES_PATH:
             EnvelopeModel = SignedEnvelope[BundlesPayload]
         else:
-            raise Exception(f"Unknown path '{path}'")
+            raise UnknownPathError(f"Unknown path '{path}'")
         
         resp_envelope = EnvelopeModel.model_validate_json(result.text)
         self.secure.validate_envelope(resp_envelope)

{koi_net-1.1.0b3 → koi_net-1.1.0b4}/src/koi_net/processor/default_handlers.py

@@ -233,4 +233,17 @@ def basic_network_output_filter(ctx: HandlerContext, kobj: KnowledgeObject):
         kobj.network_targets.update(subscribers)
         
     return kobj
-            
+
+@KnowledgeHandler.create(HandlerType.Final, rid_types=[KoiNetNode])
+def forget_edge_on_node_deletion(ctx: HandlerContext, kobj: KnowledgeObject):
+    if kobj.normalized_event_type != EventType.FORGET:
+        return
+    
+    for edge_rid in ctx.graph.get_edges():
+        edge_bundle = ctx.cache.read(edge_rid)
+        if not edge_bundle: continue
+        edge_profile = edge_bundle.validate_contents(EdgeProfile)
+        
+        if kobj.rid in (edge_profile.source, edge_profile.target):
+            logger.debug("Identified edge with forgotten node")
+            ctx.handle(rid=edge_rid, event_type=EventType.FORGET)