Source code for koi_net.core

import sys
from pathlib import Path

from .infra import LogSystem
from .base import BaseAssembly
from .config.base import BaseNodeConfig
from .config.full_node import FullNodeConfig
from .config.partial_node import PartialNodeConfig
from .components import (
    Cache,
    Effector,
    Handshaker,
    SyncManager,
    PortManager,
    NodeIdentity,
    KnowledgeProcessingWorker, 
    EventProcessingWorker,
    ErrorHandler,
    EventQueue,
    NetworkGraph,
    RequestHandler,
    NetworkResolver,
    ResponseHandler,
    EventBuffer,
    KnowledgePipeline,
    KobjQueue,
    SecureManager,
    ProfileMonitor,
    NodeServer, 
    NodePoller,
    BasicManifestHandler,
    BasicNetworkOutputFilter,
    BasicRidHandler,
    ForgetNodeHandler,
    NodeContactHandler,
    EdgeNegotiationHandler,
    SecureProfileHandler,
    ConfigProvider
)




class BaseNode(BaseAssembly):
    _log_system: LogSystem = LogSystem
    kobj_queue: KobjQueue = KobjQueue
    event_queue: EventQueue = EventQueue
    poll_event_buf: EventBuffer = EventBuffer
    broadcast_event_buf: EventBuffer = EventBuffer
    config_schema: BaseNodeConfig = BaseNodeConfig
    config: ConfigProvider | BaseNodeConfig = ConfigProvider
    cache: Cache = Cache
    identity: NodeIdentity = NodeIdentity
    graph: NetworkGraph = NetworkGraph
    secure_manager: SecureManager = SecureManager
    handshaker: Handshaker = Handshaker
    error_handler: ErrorHandler = ErrorHandler
    request_handler: RequestHandler = RequestHandler
    sync_manager: SyncManager = SyncManager
    response_handler: ResponseHandler = ResponseHandler
    resolver: NetworkResolver = NetworkResolver
    effector: Effector = Effector
    pipeline: KnowledgePipeline = KnowledgePipeline
    kobj_worker: KnowledgeProcessingWorker = KnowledgeProcessingWorker
    event_worker: EventProcessingWorker = EventProcessingWorker
    profile_monitor: ProfileMonitor = ProfileMonitor
    
    # knowledge handlers
    
    basic_manifest_handler: BasicManifestHandler = BasicManifestHandler
    basic_network_output_filter: BasicNetworkOutputFilter = BasicNetworkOutputFilter
    basic_rid_handler: BasicRidHandler = BasicRidHandler
    forget_node_handler: ForgetNodeHandler = ForgetNodeHandler
    node_contact_handler: NodeContactHandler = NodeContactHandler
    edge_negotiation_handler: EdgeNegotiationHandler = EdgeNegotiationHandler
    secure_profile_handler: SecureProfileHandler = SecureProfileHandler
    
    def __new__(cls, *args, root_dir: Path | None = None, **kwargs):
        if root_dir is None:
            if len(sys.argv) > 1:
                root_dir = Path(sys.argv[1])
                print(f"Root dir was set by CLI to '{root_dir}'")
            else:
                root_dir = Path.cwd()
        
        cls._log_system()
        return super().__new__(cls, *args, root_dir=root_dir, **kwargs)




class FullNode(BaseNode):
    config: FullNodeConfig
    server: NodeServer = NodeServer
    port_manager: PortManager = PortManager




class PartialNode(BaseNode):
    config: PartialNodeConfig
    poller: NodePoller = NodePoller