koi-net 1.2.0b1__py3-none-any.whl → 1.2.0b3__py3-none-any.whl

koi_net/behaviors.py

@@ -1,51 +0,0 @@
-from logging import getLogger
-from rid_lib.ext import Cache
-from rid_lib.types import KoiNetNode
-from rid_lib import RIDType
-from koi_net.identity import NodeIdentity
-from koi_net.network.event_queue import EventQueue
-from koi_net.network.request_handler import RequestHandler
-from koi_net.network.resolver import NetworkResolver
-from koi_net.processor.kobj_queue import KobjQueue
-from koi_net.protocol.api_models import ErrorResponse
-from .protocol.event import Event, EventType
-
-
-logger = getLogger(__name__)
-
-
-
-class Behaviors:
-    def __init__(self, cache: Cache, identity: NodeIdentity, event_queue: EventQueue, resolver: NetworkResolver, request_handler: RequestHandler, kobj_queue: KobjQueue):
-        self.cache = cache
-        self.identity = identity
-        self.event_queue = event_queue
-        self.resolver = resolver
-        self.request_handler = request_handler
-        self.kobj_queue = kobj_queue
-
-    def identify_coordinators(self) -> list[KoiNetNode]:
-        """Returns node's providing state for `orn:koi-net.node`."""
-        return self.resolver.get_state_providers(KoiNetNode)
-
-    def catch_up_with(self, target: KoiNetNode, rid_types: list[RIDType] = []):
-        """Fetches and processes knowledge objects from target node.
-        Args:
-            target: Node to catch up with
-            rid_types: RID types to fetch from target (all types if list is empty)
-        """
-        logger.debug(f"catching up with {target} on {rid_types or 'all types'}")
-        payload = self.request_handler.fetch_manifests(
-            node=target,
-            rid_types=rid_types
-        )
-        if type(payload) == ErrorResponse:
-            logger.debug("failed to reach node")
-            return
-        for manifest in payload.manifests:
-            if manifest.rid == self.identity.rid:
-                continue
-            self.kobj_queue.put_kobj(
-                manifest=manifest,
-                source=target
-            )

koi_net/config.py

@@ -1,161 +0,0 @@
-import os
-from rid_lib import RIDType
-from ruamel.yaml import YAML
-from pydantic import BaseModel, Field, PrivateAttr
-from dotenv import load_dotenv
-from rid_lib.ext.utils import sha256_hash
-from rid_lib.types import KoiNetNode
-from .protocol.secure import PrivateKey
-from .protocol.node import NodeProfile, NodeType
-
-
-class ServerConfig(BaseModel):
-    """Config for the node server (full node only)."""
-    
-    host: str = "127.0.0.1"
-    port: int = 8000
-    path: str | None = "/koi-net"
-    
-    @property
-    def url(self) -> str:
-        return f"http://{self.host}:{self.port}{self.path or ''}"
-
-class NodeContact(BaseModel):
-    rid: KoiNetNode | None = None
-    url: str | None = None
-
-class KoiNetConfig(BaseModel):
-    """Config for KOI-net."""
-    
-    node_name: str
-    node_rid: KoiNetNode | None = None
-    node_profile: NodeProfile
-    
-    rid_types_of_interest: list[RIDType] = Field(
-        default_factory=lambda: [KoiNetNode])
-    
-    cache_directory_path: str = ".rid_cache"
-    event_queues_path: str = "event_queues.json"
-    private_key_pem_path: str = "priv_key.pem"
-    polling_interval: int = 5
-    
-    first_contact: NodeContact = Field(default_factory=NodeContact)
-    
-    _priv_key: PrivateKey | None = PrivateAttr(default=None)
-
-class EnvConfig(BaseModel):
-    """Config for environment variables.
-    
-    Values set in the config are the variables names, and are loaded
-    from the environment at runtime. For example, if the config YAML
-    sets `priv_key_password: PRIV_KEY_PASSWORD` accessing 
-    `priv_key_password` would retrieve the value of `PRIV_KEY_PASSWORD`
-    from the environment.
-    """
-    
-    priv_key_password: str | None = "PRIV_KEY_PASSWORD"
-    
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-        load_dotenv()
-    
-    def __getattribute__(self, name):
-        value = super().__getattribute__(name)
-        if name in type(self).model_fields:
-            env_val = os.getenv(value)
-            if env_val is None:
-                raise ValueError(f"Required environment variable {value} not set")
-            return env_val
-        return value
-
-class NodeConfig(BaseModel):
-    """Base configuration class for all nodes.
-    
-    Designed to be extensible for custom node implementations. Classes
-    inheriting from `NodeConfig` may add additional config groups.
-    """
-    
-    server: ServerConfig = Field(default_factory=ServerConfig)
-    koi_net: KoiNetConfig
-    env: EnvConfig = Field(default_factory=EnvConfig)
-    
-    _file_path: str = PrivateAttr(default="config.yaml")
-    _file_content: str | None = PrivateAttr(default=None)
-    
-    @classmethod
-    def load_from_yaml(
-        cls, 
-        file_path: str = "config.yaml", 
-        generate_missing: bool = True
-    ):
-        """Loads config state from YAML file.
-        
-        Defaults to `config.yaml`. If `generate_missing` is set to 
-        `True`, a private key and RID will be generated if not already
-        present in the config.
-        """
-        yaml = YAML()
-        
-        try:
-            with open(file_path, "r") as f:
-                file_content = f.read()
-            config_data = yaml.load(file_content)
-            config = cls.model_validate(config_data)
-            config._file_content = file_content
-            
-        except FileNotFoundError:
-            # empty_fields = {}
-            # for name, field in cls.model_fields.items():
-                
-            #     if field.default is None or field.default_factory is None:
-            # print(empty_fields)
-            config = cls()
-            
-            
-        config._file_path = file_path
-        
-        if generate_missing:
-            if not config.koi_net.node_rid:
-                priv_key = PrivateKey.generate()
-                pub_key = priv_key.public_key()
-                
-                config.koi_net.node_rid = KoiNetNode(
-                    config.koi_net.node_name,
-                    sha256_hash(pub_key.to_der())
-                )
-                
-                with open(config.koi_net.private_key_pem_path, "w") as f:
-                    f.write(
-                        priv_key.to_pem(config.env.priv_key_password)
-                    )
-                
-                config.koi_net.node_profile.public_key = pub_key.to_der()
-            
-            if config.koi_net.node_profile.node_type == NodeType.FULL:
-                config.koi_net.node_profile.base_url = (
-                    config.koi_net.node_profile.base_url or config.server.url
-                )
-                
-            config.save_to_yaml()
-                    
-        return config
-    
-    def save_to_yaml(self):
-        """Saves config state to YAML file.
-        
-        File path is set by `load_from_yaml` class method.
-        """
-        
-        yaml = YAML()
-        
-        with open(self._file_path, "w") as f:
-            try:
-                config_data = self.model_dump(mode="json")
-                yaml.dump(config_data, f)
-            except Exception as e:
-                if self._file_content:
-                    f.seek(0)
-                    f.truncate()
-                    f.write(self._file_content)
-                raise e
-                

koi_net/models.py

@@ -1,14 +0,0 @@
-from pydantic import BaseModel
-from rid_lib.types import KoiNetNode
-from koi_net.protocol.event import Event
-
-class End:
-    """Class for a sentinel value by knowledge handlers."""
-    pass
-
-END = End()
-
-
-class QueuedEvent(BaseModel):
-    event: Event
-    target: KoiNetNode

koi_net/poll_event_buffer.py

@@ -1,17 +0,0 @@
-from rid_lib.types import KoiNetNode
-
-from koi_net.protocol.event import Event
-
-
-class PollEventBuffer:
-    buffers: dict[KoiNetNode, list[Event]]
-    
-    def __init__(self):
-        self.buffers = dict()
-        
-    def put(self, node: KoiNetNode, event: Event):
-        event_buf = self.buffers.setdefault(node, [])
-        event_buf.append(event)
-        
-    def flush(self, node: KoiNetNode):
-        return self.buffers.pop(node, [])

koi_net/server.py

@@ -1,145 +0,0 @@
-import logging
-import uvicorn
-from contextlib import asynccontextmanager
-from fastapi import FastAPI, APIRouter
-from fastapi.responses import JSONResponse
-
-from koi_net.poll_event_buffer import PollEventBuffer
-from .network.response_handler import ResponseHandler
-from .processor.kobj_queue import KobjQueue
-from .protocol.api_models import (
-    PollEvents,
-    FetchRids,
-    FetchManifests,
-    FetchBundles,
-    EventsPayload,
-    RidsPayload,
-    ManifestsPayload,
-    BundlesPayload,
-    ErrorResponse
-)
-from .protocol.errors import ProtocolError
-from .protocol.envelope import SignedEnvelope
-from .protocol.consts import (
-    BROADCAST_EVENTS_PATH,
-    POLL_EVENTS_PATH,
-    FETCH_RIDS_PATH,
-    FETCH_MANIFESTS_PATH,
-    FETCH_BUNDLES_PATH
-)
-from .secure import Secure
-from .lifecycle import NodeLifecycle
-from .config import NodeConfig
-
-logger = logging.getLogger(__name__)
-
-
-class NodeServer:
-    """Manages FastAPI server and event handling for full nodes."""
-    config: NodeConfig
-    lifecycle: NodeLifecycle
-    secure: Secure
-    kobj_queue: KobjQueue
-    poll_event_buf: PollEventBuffer
-    response_handler: ResponseHandler
-    app: FastAPI
-    router: APIRouter
-    
-    def __init__(
-        self,
-        config: NodeConfig,
-        lifecycle: NodeLifecycle,
-        secure: Secure,
-        kobj_queue: KobjQueue,
-        poll_event_buf: PollEventBuffer,
-        response_handler: ResponseHandler
-    ):
-        self.config = config
-        self.lifecycle = lifecycle
-        self.secure = secure
-        self.kobj_queue = kobj_queue
-        self.poll_event_buf = poll_event_buf
-        self.response_handler = response_handler
-        self._build_app()
-        
-    def _build_app(self):
-        """Builds FastAPI app and adds endpoints."""
-        @asynccontextmanager
-        async def lifespan(*args, **kwargs):
-            async with self.lifecycle.async_run():
-                yield
-        
-        self.app = FastAPI(
-            lifespan=lifespan, 
-            title="KOI-net Protocol API",
-            version="1.0.0"
-        )
-        
-        self.router = APIRouter(prefix="/koi-net")
-        self.app.add_exception_handler(ProtocolError, self.protocol_error_handler)
-        
-        def _add_endpoint(path, func):
-            self.router.add_api_route(
-                path=path,
-                endpoint=self.secure.envelope_handler(func),
-                methods=["POST"],
-                response_model_exclude_none=True
-            )
-        
-        _add_endpoint(BROADCAST_EVENTS_PATH, self.broadcast_events)
-        _add_endpoint(POLL_EVENTS_PATH, self.poll_events)
-        _add_endpoint(FETCH_RIDS_PATH, self.fetch_rids)
-        _add_endpoint(FETCH_MANIFESTS_PATH, self.fetch_manifests)
-        _add_endpoint(FETCH_BUNDLES_PATH, self.fetch_bundles)
-        
-        self.app.include_router(self.router)
-    
-    def run(self):
-        """Starts FastAPI server and event handler."""
-        uvicorn.run(
-            app=self.app,
-            host=self.config.server.host,
-            port=self.config.server.port
-        )
-        
-    def protocol_error_handler(self, request, exc: ProtocolError):
-        """Catches `ProtocolError` and returns as `ErrorResponse`."""
-        logger.info(f"caught protocol error: {exc}")
-        resp = ErrorResponse(error=exc.error_type)
-        logger.info(f"returning error response: {resp}")
-        return JSONResponse(
-            status_code=400,
-            content=resp.model_dump(mode="json")
-        )
-
-    async def broadcast_events(self, req: SignedEnvelope[EventsPayload]):
-        """Handles events broadcast endpoint."""
-        logger.info(f"Request to {BROADCAST_EVENTS_PATH}, received {len(req.payload.events)} event(s)")
-        for event in req.payload.events:
-            self.kobj_queue.put_kobj(event=event, source=req.source_node)
-        
-    async def poll_events(
-        self, req: SignedEnvelope[PollEvents]
-    ) -> SignedEnvelope[EventsPayload] | ErrorResponse:
-        """Handles poll events endpoint."""
-        logger.info(f"Request to {POLL_EVENTS_PATH}")
-        events = self.poll_event_buf.flush(req.source_node)
-        return EventsPayload(events=events)
-
-    async def fetch_rids(
-        self, req: SignedEnvelope[FetchRids]
-    ) -> SignedEnvelope[RidsPayload] | ErrorResponse:
-        """Handles fetch RIDs endpoint."""
-        return self.response_handler.fetch_rids(req.payload, req.source_node)
-
-    async def fetch_manifests(
-        self, req: SignedEnvelope[FetchManifests]
-    ) -> SignedEnvelope[ManifestsPayload] | ErrorResponse:
-        """Handles fetch manifests endpoint."""
-        return self.response_handler.fetch_manifests(req.payload, req.source_node)
-
-    async def fetch_bundles(
-        self, req: SignedEnvelope[FetchBundles]
-    ) -> SignedEnvelope[BundlesPayload] | ErrorResponse:
-        """Handles fetch bundles endpoint."""
-        return self.response_handler.fetch_bundles(req.payload, req.source_node)

koi_net-1.2.0b1.dist-info/RECORD

@@ -1,49 +0,0 @@
-koi_net/__init__.py,sha256=S6Ew8CDhwaysGFJGX19tgkXLcJiJR2R5fwz449gaXFY,31
-koi_net/behaviors.py,sha256=_2yeX3IdDQWiOsXoehiFjr2jPj58oZTnBX28M32LM4w,1955
-koi_net/config.py,sha256=yV53GFQrJchyKKRs3sOhj27pyM6rwiUhIgDZ81rvZ48,5238
-koi_net/context.py,sha256=z24-pwYE3YDIznqNkRBIZuqT-JpksHnkuUfDOH034j0,1429
-koi_net/core.py,sha256=qUq87KfL-DKUu0NVfJPFBiNYpD35ODzvP6bV7F5ijd4,7252
-koi_net/default_actions.py,sha256=y9oaDeb3jJGSvVbuonmyGWkKWlzpEX3V32MjgEuYWc8,622
-koi_net/effector.py,sha256=iFX34Tq9Byev22CUbGnQU1MUccyNH5ebmQzoovwY_aE,4738
-koi_net/handshaker.py,sha256=e5bzWGWvU7w5gQ4kBEiU6ShCXwc-XXgfvx3c1RxMguA,1238
-koi_net/identity.py,sha256=FvIWksGTqwM7HCevIwmo_6l-t-2tnYkaaR4CanZatL4,569
-koi_net/kobj_worker.py,sha256=p_1QJDDoz-cIDoTQ3qCptPTbuvqou5BAPEQqZjH8niE,1306
-koi_net/lifecycle.py,sha256=WP3XiiCtSG8ZGa4oBnNBxwsmQ35QRQEgeYqwnGRxcsI,4379
-koi_net/models.py,sha256=_hL-ohrlH-kZRV9B0PWjYyZbb_Ef1z14HOLI5RK_Dto,274
-koi_net/poll_event_buffer.py,sha256=DMuEnaoH7nZccXY0LwHaGEyecrWfW1z0s7AbVHMGM88,441
-koi_net/poller.py,sha256=zBHWPXvca-Ia9ggE5emu8rJx5c8Jdi1ZAYB7lTyezv8,1395
-koi_net/secure.py,sha256=1t1WtHeqmoHfi4siW1jFCmIhKFsQGactITkEmVYKqnc,4919
-koi_net/server.py,sha256=iP9agexIjqmNlCOXs50cLkyEWHth7cGlQRN593rBOhU,4955
-koi_net/utils.py,sha256=4K6BqtifExDUfPP8zJAICgPV1mcabz_MaFB0P0fSnic,411
-koi_net/worker.py,sha256=G3JLlILN0cyY7Fv3eA5y8uf9rImy4e21OQ1mEKFoVVo,191
-koi_net/cli/__init__.py,sha256=Vpq0yLN0ROyF0qQYsyiuWFbjs-zsa90G1DpAv13qaiw,25
-koi_net/cli/commands.py,sha256=5eVsvJAnHjwJdt8MNYrk5zSzWWm1ooiP35bJeQxYIq4,2592
-koi_net/cli/models.py,sha256=ON-rqMgHDng3J2hbBbGYzkynaWc9Tq4QfOJIu_6pFuA,1190
-koi_net/network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-koi_net/network/error_handler.py,sha256=8XgOljFBJ4lzXdEO5I-spND9iRln9309lwgpPV1pMnM,1802
-koi_net/network/event_queue.py,sha256=rvTuMuG-QicKHGSfB2FOyFMngmnESovec1TOPIowvss,780
-koi_net/network/graph.py,sha256=8hweEivCwxnagFoUrefE7Hnz06MWScF3XJHeXU2Qqnw,4178
-koi_net/network/request_handler.py,sha256=1Sp0-OCqcQXM-fAE5sFH4_M9FP1EbJh69P8gB56BSKg,7102
-koi_net/network/resolver.py,sha256=MqUMwG5snZp4QLV7d-gZ7tRJDQ0uJZ_IhssozTc9JkU,5309
-koi_net/network/response_handler.py,sha256=tbTqHmwXUEZIqiYqVL2vi4q6eDH0bPRrAO6G4XpbMic,2123
-koi_net/processor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-koi_net/processor/default_handlers.py,sha256=siRZl_Ww7QA9MhJR4_nmrSHOG8ejfS0NjiF6braA338,11020
-koi_net/processor/event_worker.py,sha256=bS_Xgkg4s1e_5dG0qtHY3d79JsYEyGzbI6F7JSZbSws,4297
-koi_net/processor/handler.py,sha256=PJlQCPfKm3zYsJrwmGIw5xZmc6vArKz_UoeJuXBj16Y,2356
-koi_net/processor/knowledge_object.py,sha256=0VyBOqkCoINMK3fqYhLlXU-Ri7gKEpJloDHfhaGITxg,4172
-koi_net/processor/knowledge_pipeline.py,sha256=sz8T12nAcJ1tbF2jHf0Wmi6ffnQPMrIl64xif4WP7fc,8784
-koi_net/processor/kobj_queue.py,sha256=LIh9Sz3EaEZj9qJBd4iLPlzYBO6aH-6qt_JHCxAzAJU,1839
-koi_net/protocol/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-koi_net/protocol/api_models.py,sha256=AIFT49fyNAnOsHep9bgjga2M0S38-PouBaqfQI2zcKY,1795
-koi_net/protocol/consts.py,sha256=bisbVEojPIHlLhkLafBzfIhH25TjNfvTORF1g6YXzIM,243
-koi_net/protocol/edge.py,sha256=PzdEhC43T1KO5iMSEu7I4tiz-7sZxtz41dJfWf-oHA0,1034
-koi_net/protocol/envelope.py,sha256=UVHlO2BDyDiP5eixqx9xD6xUsCfFRi0kZyzC4BC-DOw,1886
-koi_net/protocol/errors.py,sha256=uKPQ-TGLouZuK0xd2pXuCQoRTyu_JFsydSCLml13Cz8,595
-koi_net/protocol/event.py,sha256=HxzLN-iCXPyr2YzrswMIkgZYeUdFbBpa5v98dAB06lQ,1328
-koi_net/protocol/node.py,sha256=KH_SjHDzW-V4pn2tqpTDhIzfuCgKgM1YqCafyofLN3k,466
-koi_net/protocol/secure.py,sha256=6sRLWxG5EDF0QLBj29gk3hPmZnPXATrTTFdwx39wQfY,5127
-koi_net-1.2.0b1.dist-info/METADATA,sha256=reMf5147lyI17Gu5Ca98sQFZsoMk9pL5mV2BHZKtpV0,37314
-koi_net-1.2.0b1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-koi_net-1.2.0b1.dist-info/entry_points.txt,sha256=l7He_JTyXrfKIHkttnPWXHI717v8WpLLfCduL5QcEWA,40
-koi_net-1.2.0b1.dist-info/licenses/LICENSE,sha256=03mgCL5qth2aD9C3F3qNVs4sFJSpK9kjtYCyOwdSp7s,1069
-koi_net-1.2.0b1.dist-info/RECORD,,