@olane/o-node 0.7.12-alpha.3 → 0.7.12-alpha.31

package/dist/src/o-node.js

@@ -9,6 +9,9 @@ import { oNodeConnectionManager } from './connection/o-node-connection.manager.j
 import { oNodeResolver } from './router/resolvers/o-node.resolver.js';
 import { oMethodResolver, oToolBase } from '@olane/o-tool';
 import { oLeaderResolverFallback } from './router/index.js';
+import { oNodeNotificationManager } from './o-node.notification-manager.js';
+import { oConnectionHeartbeatManager } from './managers/o-connection-heartbeat.manager.js';
+import { oReconnectionManager } from './managers/o-reconnection.manager.js';
 export class oNode extends oToolBase {
     constructor(config) {
         super(config);
@@ -43,6 +46,9 @@ export class oNode extends oToolBase {
         });
         this.router = new oNodeRouter();
     }
+    createNotificationManager() {
+        return new oNodeNotificationManager(this.p2pNode, this.hierarchyManager, this.address);
+    }
     get staticAddress() {
         return this.config.address;
     }
@@ -60,6 +66,30 @@ export class oNode extends oToolBase {
             this.logger.debug('Skipping unregistration, node is leader');
             return;
         }
+        // Notify parent we're stopping (best-effort, 2s timeout)
+        if (this.config.parent) {
+            try {
+                await Promise.race([
+                    this.use(this.config.parent, {
+                        method: 'notify',
+                        params: {
+                            eventType: 'node:stopping',
+                            eventData: {
+                                address: this.address.toString(),
+                                reason: 'graceful_shutdown',
+                                expectedDowntime: null,
+                            },
+                            source: this.address.toString(),
+                        },
+                    }),
+                    new Promise((_, reject) => setTimeout(() => reject(new Error('timeout')), 2000)),
+                ]);
+                this.logger.debug('Notified parent of shutdown');
+            }
+            catch (error) {
+                this.logger.warn('Failed to notify parent (will be detected by heartbeat):', error instanceof Error ? error.message : error);
+            }
+        }
         if (!this.config.leader) {
             this.logger.debug('No leader found, skipping unregistration');
             return;
@@ -79,11 +109,21 @@ export class oNode extends oToolBase {
             this.logger.debug('Skipping parent registration, node is leader');
             return;
         }
+        if (!this.parent?.libp2pTransports?.length) {
+            this.logger.debug('Parent has no transports, waiting for reconnection & leader ack');
+            if (this.parent?.toString() === oAddress.leader().toString()) {
+                this.parent.setTransports(this.leader?.libp2pTransports || []);
+            }
+            else {
+                this.logger.debug('Waiting for parent and reconnecting...');
+                await this.reconnectionManager?.waitForParentAndReconnect();
+            }
+        }
         // if no parent transports, register with the parent to get them
         // TODO: should we remove the transports check to make this more consistent?
-        if (this.config.parent && this.config.parent.transports.length === 0) {
-            this.logger.debug('Registering node with parent...', this.config.parent);
-            const parentRegistration = await this.use(this.config.parent, {
+        if (this.config.parent) {
+            this.logger.debug('Registering node with parent...', this.config.parent?.toString());
+            await this.use(this.config.parent, {
                 method: 'child_register',
                 params: {
                     address: this.address.toString(),
@@ -92,11 +132,22 @@ export class oNode extends oToolBase {
                     _token: this.config.joinToken,
                 },
             });
-            const { parentTransports } = parentRegistration.result.data;
-            // update the parent transports
-            this.config.parent.setTransports(parentTransports.map((t) => new oNodeTransport(t)));
         }
     }
+    async registerLeader() {
+        const address = oAddress.registry();
+        const params = {
+            method: 'commit',
+            params: {
+                peerId: this.peerId.toString(),
+                address: this.address.toString(),
+                protocols: this.p2pNode.getProtocols(),
+                transports: this.transports,
+                staticAddress: this.staticAddress.toString(),
+            },
+        };
+        await this.use(address, params);
+    }
     async register() {
         if (this.type === NodeType.LEADER) {
             this.logger.debug('Skipping registration, node is leader');
@@ -117,18 +168,7 @@ export class oNode extends oToolBase {
             this.logger.debug('Registering node with leader...');
         }
         await this.registerParent();
-        const address = oAddress.registry();
-        const params = {
-            method: 'commit',
-            params: {
-                peerId: this.peerId.toString(),
-                address: this.address.toString(),
-                protocols: this.p2pNode.getProtocols(),
-                transports: this.transports,
-                staticAddress: this.staticAddress.toString(),
-            },
-        };
-        await this.use(address, params);
+        await this.registerLeader();
         this.logger.debug('Registration successful');
     }
     extractMethod(address) {
@@ -136,11 +176,10 @@ export class oNode extends oToolBase {
     }
     async start() {
         await super.start();
-        // await NetworkUtils.advertiseToNetwork(
-        //   this.address,
-        //   this.staticAddress,
-        //   this.p2pNode,
-        // );
+        // Start heartbeat after node is running
+        if (this.connectionHeartbeatManager) {
+            await this.connectionHeartbeatManager.start();
+        }
     }
     async validateJoinRequest(request) {
         return true;
@@ -157,6 +196,7 @@ export class oNode extends oToolBase {
             listeners: (this.config.network?.listeners ||
                 defaultLibp2pConfig.listeners ||
                 []).concat(`/memory/${uuidv4()}`), // ensure we allow for local in-memory communication
+            prometheusRegistry: this.config.network?.prometheusRegistry,
         };
         // if the seed is provided, use it to generate the private key
         if (this.config.seed) {
@@ -190,12 +230,32 @@ export class oNode extends oToolBase {
             // ];
             //   // let's make sure we only allow communication through the parent transports
             params.connectionGater = {
+                denyDialPeer: (peerId) => {
+                    // we can call the leader
+                    if (this.config.leader?.libp2pTransports.some((t) => t.toPeerId() === peerId.toString())) {
+                        return false;
+                    }
+                    // we can call our parent
+                    if (this.parentPeerId === peerId.toString()) {
+                        return false;
+                    }
+                    // we can call our children
+                    if (this.hierarchyManager.children.some((c) => c.libp2pTransports.some((t) => t.toPeerId() === peerId.toString()))) {
+                        return false;
+                    }
+                    // check for standalone node
+                    if (!this.config.parent && !this.config.leader) {
+                        return false;
+                    }
+                    return true;
+                },
                 // who can call us?
                 denyInboundEncryptedConnection: (peerId, maConn) => {
                     // deny all inbound connections unless they are from a parent transport
                     if (this.parentPeerId === peerId.toString()) {
                         return false;
                     }
+                    // allow connections from children (for ping)
                     if (this.hierarchyManager.children.some((c) => c.libp2pTransports.some((t) => t.toPeerId() === peerId.toString()))) {
                         return false;
                     }
@@ -223,17 +283,13 @@ export class oNode extends oToolBase {
         this.p2pNode = await createNode(params);
         return this.p2pNode;
     }
-    async connect(nextHopAddress, targetAddress) {
+    async connect(config) {
         if (!this.connectionManager) {
             this.logger.error('Connection manager not initialized');
             throw new Error('Node is not ready to connect to other nodes');
         }
         const connection = await this.connectionManager
-            .connect({
-            address: targetAddress,
-            nextHopAddress,
-            callerAddress: this.address,
-        })
+            .connect(config)
             .catch((error) => {
             // TODO: we need to handle this better and document
             if (error.message === 'Can not dial self') {
@@ -246,6 +302,42 @@ export class oNode extends oToolBase {
         }
         return connection;
     }
+    async initConnectionManager() {
+        this.connectionManager = new oNodeConnectionManager({
+            p2pNode: this.p2pNode,
+            defaultReadTimeoutMs: this.config.connectionTimeouts?.readTimeoutMs,
+            defaultDrainTimeoutMs: this.config.connectionTimeouts?.drainTimeoutMs,
+        });
+    }
+    async initReconnectionManager() {
+        // Initialize reconnection manager
+        if (this.config.reconnection?.enabled !== false) {
+            this.reconnectionManager = new oReconnectionManager(this, {
+                enabled: true,
+                maxAttempts: this.config.reconnection?.maxAttempts ?? 10,
+                baseDelayMs: this.config.reconnection?.baseDelayMs ?? 5000,
+                maxDelayMs: this.config.reconnection?.maxDelayMs ?? 60000,
+                useLeaderFallback: this.config.reconnection?.useLeaderFallback ?? true,
+                parentDiscoveryIntervalMs: this.config.reconnection?.parentDiscoveryIntervalMs ?? 10000,
+                parentDiscoveryMaxDelayMs: this.config.reconnection?.parentDiscoveryMaxDelayMs ?? 60000,
+            });
+        }
+    }
+    async hookInitializeFinished() { }
+    async hookStartFinished() {
+        // Initialize connection heartbeat manager
+        this.connectionHeartbeatManager = new oConnectionHeartbeatManager(this, {
+            enabled: this.config.connectionHeartbeat?.enabled ?? true,
+            intervalMs: this.config.connectionHeartbeat?.intervalMs ?? 15000,
+            timeoutMs: this.config.connectionHeartbeat?.timeoutMs ?? 15000,
+            failureThreshold: this.config.connectionHeartbeat?.failureThreshold ?? 3,
+            checkChildren: this.config.connectionHeartbeat?.checkChildren ?? false,
+            checkParent: this.config.connectionHeartbeat?.checkParent ?? true,
+            checkLeader: true,
+        });
+        this.logger.info(`Connection heartbeat config: leader=${this.connectionHeartbeatManager.getConfig().checkLeader}, ` +
+            `parent=${this.connectionHeartbeatManager.getConfig().checkParent}`);
+    }
     async initialize() {
         this.logger.debug('Initializing node...');
         if (this.p2pNode && this.state !== NodeState.STOPPED) {
@@ -256,13 +348,13 @@ export class oNode extends oToolBase {
         }
         await this.createNode();
         await this.initializeRouter();
+        // need to wait until our libpp2 node is initialized before calling super.initialize
+        await super.initialize();
         this.logger.debug('Node initialized!', this.transports.map((t) => t.toString()));
         this.address.setTransports(this.transports);
         this.peerId = this.p2pNode.peerId;
         // initialize connection manager
-        this.connectionManager = new oNodeConnectionManager({
-            p2pNode: this.p2pNode,
-        });
+        await this.initConnectionManager();
         // initialize address resolution
         this.router.addResolver(new oMethodResolver(this.address));
         this.router.addResolver(new oNodeResolver(this.address));
@@ -271,12 +363,49 @@ export class oNode extends oToolBase {
             this.logger.debug('Adding leader resolver fallback...');
             this.router.addResolver(new oLeaderResolverFallback(this.address));
         }
+        // initialize reconnection manager
+        await this.initReconnectionManager();
+        await this.hookInitializeFinished();
     }
+    /**
+     * Override use() to wrap leader/registry requests with retry logic
+     */
+    // async use(
+    //   address: oAddress,
+    //   data?: {
+    //     method?: string;
+    //     params?: { [key: string]: any };
+    //     id?: string;
+    //   },
+    //   options?: UseOptions,
+    // ): Promise<any> {
+    //   // Wrap leader/registry requests with retry logic
+    //   return super.use(address, data, options),
+    //     address,
+    //     data?.method,
+    // }
     async teardown() {
+        // Stop heartbeat before parent teardown
+        if (this.connectionHeartbeatManager) {
+            await this.connectionHeartbeatManager.stop();
+        }
         await this.unregister();
         await super.teardown();
         if (this.p2pNode) {
             await this.p2pNode.stop();
         }
     }
+    // IHeartbeatableNode interface methods
+    getLeaders() {
+        return [this.leader];
+    }
+    getParents() {
+        return this.hierarchyManager.getParents();
+    }
+    getChildren() {
+        return this.hierarchyManager.getChildren();
+    }
+    removeChild(childAddress) {
+        this.hierarchyManager.removeChild(childAddress);
+    }
 }

package/dist/src/o-node.notification-manager.d.ts

@@ -0,0 +1,52 @@
+import { Libp2p } from '@olane/o-config';
+import { oNotificationManager } from '@olane/o-core';
+import { oNodeAddress } from './router/o-node.address.js';
+import { oNodeHierarchyManager } from './o-node.hierarchy-manager.js';
+/**
+ * libp2p-specific implementation of oNotificationManager
+ * Wraps libp2p events and enriches them with Olane context
+ */
+export declare class oNodeNotificationManager extends oNotificationManager {
+    private p2pNode;
+    private hierarchyManager;
+    private address;
+    constructor(p2pNode: Libp2p, hierarchyManager: oNodeHierarchyManager, address: oNodeAddress);
+    /**
+     * Wire up libp2p event listeners
+     */
+    protected setupListeners(): void;
+    /**
+     * Handle peer connect event from libp2p
+     */
+    private handlePeerConnect;
+    /**
+     * Handle peer disconnect event from libp2p
+     */
+    private handlePeerDisconnect;
+    /**
+     * Handle peer discovery event from libp2p
+     */
+    private handlePeerDiscovery;
+    /**
+     * Handle connection open event from libp2p
+     */
+    private handleConnectionOpen;
+    /**
+     * Handle connection close event from libp2p
+     */
+    private handleConnectionClose;
+    /**
+     * Try to resolve a libp2p peer ID to an Olane address
+     * Checks hierarchy manager for known peers
+     */
+    private peerIdToAddress;
+    /**
+     * Check if an address is a direct child
+     */
+    private isChild;
+    /**
+     * Check if an address is a parent
+     */
+    private isParent;
+}
+//# sourceMappingURL=o-node.notification-manager.d.ts.map

package/dist/src/o-node.notification-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"o-node.notification-manager.d.ts","sourceRoot":"","sources":["../../src/o-node.notification-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AACzC,OAAO,EACL,oBAAoB,EAQrB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAEtE;;;GAGG;AACH,qBAAa,wBAAyB,SAAQ,oBAAoB;IAE9D,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,OAAO;gBAFP,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,qBAAqB,EACvC,OAAO,EAAE,YAAY;IAK/B;;OAEG;IACH,SAAS,CAAC,cAAc,IAAI,IAAI;IAgChC;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAkDzB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAoD5B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAmB3B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAI5B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAI7B;;;OAGG;IACH,OAAO,CAAC,eAAe;IAkCvB;;OAEG;IACH,OAAO,CAAC,OAAO;IAMf;;OAEG;IACH,OAAO,CAAC,QAAQ;CAKjB"}

package/dist/src/o-node.notification-manager.js

@@ -0,0 +1,185 @@
+import { oNotificationManager, NodeConnectedEvent, NodeDisconnectedEvent, NodeDiscoveredEvent, ChildJoinedEvent, ChildLeftEvent, ParentConnectedEvent, ParentDisconnectedEvent, } from '@olane/o-core';
+/**
+ * libp2p-specific implementation of oNotificationManager
+ * Wraps libp2p events and enriches them with Olane context
+ */
+export class oNodeNotificationManager extends oNotificationManager {
+    constructor(p2pNode, hierarchyManager, address) {
+        super();
+        this.p2pNode = p2pNode;
+        this.hierarchyManager = hierarchyManager;
+        this.address = address;
+    }
+    /**
+     * Wire up libp2p event listeners
+     */
+    setupListeners() {
+        this.logger.debug('Setting up libp2p event listeners...');
+        // Peer connection events
+        this.p2pNode.addEventListener('peer:connect', this.handlePeerConnect.bind(this));
+        this.p2pNode.addEventListener('peer:disconnect', this.handlePeerDisconnect.bind(this));
+        // Peer discovery events
+        this.p2pNode.addEventListener('peer:discovery', this.handlePeerDiscovery.bind(this));
+        // Connection events
+        this.p2pNode.addEventListener('connection:open', this.handleConnectionOpen.bind(this));
+        this.p2pNode.addEventListener('connection:close', this.handleConnectionClose.bind(this));
+        this.logger.debug('libp2p event listeners configured');
+    }
+    /**
+     * Handle peer connect event from libp2p
+     */
+    handlePeerConnect(evt) {
+        const peerId = evt.detail;
+        // this.logger.debug(`Peer connected: ${peerId.toString()}`);
+        // Try to resolve peer ID to Olane address
+        const nodeAddress = this.peerIdToAddress(peerId.toString());
+        if (!nodeAddress) {
+            // this.logger.debug(
+            //   `Could not resolve peer ID ${peerId.toString()} to address`,
+            // );
+            return;
+        }
+        // Emit generic node connected event
+        this.emit(new NodeConnectedEvent({
+            source: this.address,
+            nodeAddress,
+            connectionMetadata: {
+                peerId: peerId.toString(),
+                transport: 'libp2p',
+            },
+        }));
+        // Check if this is a child node
+        if (this.isChild(nodeAddress)) {
+            // this.logger.debug(`Child node connected: ${nodeAddress.toString()}`);
+            this.emit(new ChildJoinedEvent({
+                source: this.address,
+                childAddress: nodeAddress,
+                parentAddress: this.address,
+            }));
+        }
+        // Check if this is a parent node
+        if (this.isParent(nodeAddress)) {
+            // this.logger.debug(`Parent node connected: ${nodeAddress.toString()}`);
+            this.emit(new ParentConnectedEvent({
+                source: this.address,
+                parentAddress: nodeAddress,
+            }));
+        }
+    }
+    /**
+     * Handle peer disconnect event from libp2p
+     */
+    handlePeerDisconnect(evt) {
+        const peerId = evt.detail;
+        // this.logger.debug(`Peer disconnected: ${peerId.toString()}`);
+        // Try to resolve peer ID to Olane address
+        const nodeAddress = this.peerIdToAddress(peerId.toString());
+        if (!nodeAddress) {
+            // this.logger.debug(
+            //   `Could not resolve peer ID ${peerId.toString()} to address`,
+            // );
+            return;
+        }
+        // Emit generic node disconnected event
+        this.emit(new NodeDisconnectedEvent({
+            source: this.address,
+            nodeAddress,
+            reason: 'peer_disconnected',
+        }));
+        // Check if this is a child node
+        if (this.isChild(nodeAddress)) {
+            this.logger.debug(`Child node disconnected: ${nodeAddress.toString()}`);
+            this.emit(new ChildLeftEvent({
+                source: this.address,
+                childAddress: nodeAddress,
+                parentAddress: this.address,
+                reason: 'peer_disconnected',
+            }));
+            // Optionally remove from hierarchy (auto-cleanup)
+            // this.hierarchyManager.removeChild(nodeAddress);
+        }
+        // Check if this is a parent node
+        if (this.isParent(nodeAddress)) {
+            this.logger.debug(`Parent node disconnected: ${nodeAddress.toString()}`);
+            this.emit(new ParentDisconnectedEvent({
+                source: this.address,
+                parentAddress: nodeAddress,
+                reason: 'peer_disconnected',
+            }));
+        }
+    }
+    /**
+     * Handle peer discovery event from libp2p
+     */
+    handlePeerDiscovery(evt) {
+        const peerInfo = evt.detail;
+        // this.logger.debug(`Peer discovered: ${peerInfo.id.toString()}`);
+        // Try to resolve peer ID to Olane address
+        const nodeAddress = this.peerIdToAddress(peerInfo.id.toString());
+        if (!nodeAddress) {
+            return;
+        }
+        this.emit(new NodeDiscoveredEvent({
+            source: this.address,
+            nodeAddress,
+        }));
+    }
+    /**
+     * Handle connection open event from libp2p
+     */
+    handleConnectionOpen(evt) {
+        // do nothing for now
+    }
+    /**
+     * Handle connection close event from libp2p
+     */
+    handleConnectionClose(evt) {
+        // do nothing for now
+    }
+    /**
+     * Try to resolve a libp2p peer ID to an Olane address
+     * Checks hierarchy manager for known peers
+     */
+    peerIdToAddress(peerId) {
+        // Check children
+        for (const child of this.hierarchyManager.children) {
+            const childTransports = child.transports;
+            for (const transport of childTransports) {
+                if (transport.toString().includes(peerId)) {
+                    return child;
+                }
+            }
+        }
+        // Check parents
+        for (const parent of this.hierarchyManager.parents) {
+            const parentTransports = parent.transports;
+            for (const transport of parentTransports) {
+                if (transport.toString().includes(peerId)) {
+                    return parent;
+                }
+            }
+        }
+        // Check leaders
+        for (const leader of this.hierarchyManager.leaders) {
+            const leaderTransports = leader.transports;
+            for (const transport of leaderTransports) {
+                if (transport.toString().includes(peerId)) {
+                    return leader;
+                }
+            }
+        }
+        return null;
+    }
+    /**
+     * Check if an address is a direct child
+     */
+    isChild(address) {
+        return this.hierarchyManager.children.some((child) => child.toString() === address.toString());
+    }
+    /**
+     * Check if an address is a parent
+     */
+    isParent(address) {
+        return this.hierarchyManager.parents.some((parent) => parent.toString() === address.toString());
+    }
+}

package/dist/src/o-node.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"o-node.tool.d.ts","sourceRoot":"","sources":["../../src/o-node.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,QAAQ,EAGR,QAAQ,EAET,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;;AAGrD;;;;GAIG;AACH,qBAAa,SAAU,SAAQ,cAAkB;IACzC,cAAc,CAAC,OAAO,EAAE,QAAQ;IAQhC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAW3B,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IA2CnE,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC;IAQ9B,oBAAoB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;CAa5D"}
+{"version":3,"file":"o-node.tool.d.ts","sourceRoot":"","sources":["../../src/o-node.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,QAAQ,EAGR,QAAQ,EAIT,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;;AAIrD;;;;GAIG;AACH,qBAAa,SAAU,SAAQ,cAAkB;IACzC,cAAc,CAAC,OAAO,EAAE,QAAQ;IAQhC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAW3B,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAuCnE,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC;IAQ9B,oBAAoB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;CA2B5D"}

package/dist/src/o-node.tool.js

@@ -1,7 +1,8 @@
-import { CoreUtils, oAddress, oError, oErrorCodes, oRequest, } from '@olane/o-core';
+import { CoreUtils, oRequest, ResponseBuilder, ChildJoinedEvent, } from '@olane/o-core';
 import { oTool } from '@olane/o-tool';
 import { oServerNode } from './nodes/server.node.js';
 import { oNodeTransport } from './router/o-node.transport.js';
+import { oNodeAddress } from './router/o-node.address.js';
 /**
  * oTool is a mixin that extends the base class and implements the oTool interface
  * @param Base - The base class to extend
@@ -24,35 +25,32 @@ export class oNodeTool extends oTool(oServerNode) {
         }
     }
     async handleStream(stream, connection) {
-        stream.addEventListener('message', async (event) => {
+        // CRITICAL: Attach message listener immediately to prevent buffer overflow (libp2p v3)
+        // Per libp2p migration guide: "If no message event handler is added, streams will
+        // buffer incoming data until a pre-configured limit is reached, after which the stream will be reset."
+        const messageHandler = async (event) => {
             if (!event.data) {
                 this.logger.warn('Malformed event data');
                 return;
             }
             const requestConfig = await CoreUtils.processStream(event);
             const request = new oRequest(requestConfig);
-            let success = true;
-            const result = await this.execute(request, stream).catch((error) => {
-                this.logger.error('Error executing tool: ', request.toString(), error, typeof error);
-                success = false;
-                const responseError = error instanceof oError
-                    ? error
-                    : new oError(oErrorCodes.UNKNOWN, error.message);
-                return {
-                    error: responseError.toJSON(),
-                };
-            });
-            if (success) {
-                this.metrics.successCount++;
+            // Use ResponseBuilder with automatic error handling and metrics tracking
+            const responseBuilder = ResponseBuilder.create().withMetrics(this.metrics);
+            let response;
+            try {
+                const result = await this.execute(request, stream);
+                response = await responseBuilder.build(request, result, null);
             }
-            else {
-                this.metrics.errorCount++;
+            catch (error) {
+                this.logger.error('Error executing tool: ', request.toString(), error, typeof error);
+                response = await responseBuilder.buildError(request, error);
             }
-            // compose the response & add the expected connection + request fields
-            const response = CoreUtils.buildResponse(request, result, result?.error);
-            // add the request method to the response
+            // Send the response
             await CoreUtils.sendResponse(response, stream);
-        });
+        };
+        // Attach listener synchronously before any async operations
+        stream.addEventListener('message', messageHandler);
     }
     async _tool_identify() {
         return {
@@ -64,8 +62,17 @@ export class oNodeTool extends oTool(oServerNode) {
     async _tool_child_register(request) {
         this.logger.debug('Child register: ', request.params);
         const { address, transports } = request.params;
-        const childAddress = new oAddress(address, transports.map((t) => new oNodeTransport(t)));
+        const childAddress = new oNodeAddress(address, transports.map((t) => new oNodeTransport(t)));
+        // Add child to hierarchy
         this.hierarchyManager.addChild(childAddress);
+        // Emit child joined event
+        if (this.notificationManager) {
+            this.notificationManager.emit(new ChildJoinedEvent({
+                source: this.address,
+                childAddress,
+                parentAddress: this.address,
+            }));
+        }
         return {
             message: `Child node registered with parent! ${childAddress.toString()}`,
             parentTransports: this.parentTransports.map((t) => t.toString()),

package/dist/src/router/o-node.router.d.ts

@@ -17,6 +17,7 @@ export declare class oNodeRouter extends oToolRouter {
     protected forward(address: oNodeAddress, request: oRouterRequest, node: oNode): Promise<any>;
     /**
      * Executes a request locally when routing to self.
+     * Now uses ResponseBuilder for consistency with useSelf() behavior.
      */
     private executeSelfRouting;
     /**

package/dist/src/router/o-node.router.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"o-node.router.d.ts","sourceRoot":"","sources":["../../../src/router/o-node.router.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAML,cAAc,EACd,aAAa,EACd,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAK5C,qBAAa,WAAY,SAAQ,WAAW;IAC1C,OAAO,CAAC,aAAa,CAAqB;;IAO1C;;;;;;;;OAQG;cACa,OAAO,CACrB,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,KAAK,GACV,OAAO,CAAC,GAAG,CAAC;IA0Bf;;OAEG;YACW,kBAAkB;IAgBhC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAS5B;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAUhC;;OAEG;YACW,eAAe;IA2B7B;;;OAGG;IACG,SAAS,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC;IAyB3E;;;OAGG;IACH,UAAU,CAAC,qBAAqB,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,GAAG,OAAO;CAGtE"}
+{"version":3,"file":"o-node.router.d.ts","sourceRoot":"","sources":["../../../src/router/o-node.router.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAML,cAAc,EAGd,aAAa,EACd,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAO5C,qBAAa,WAAY,SAAQ,WAAW;IAC1C,OAAO,CAAC,aAAa,CAAqB;;IAO1C;;;;;;;;OAQG;cACa,OAAO,CACrB,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,KAAK,GACV,OAAO,CAAC,GAAG,CAAC;IA6Bf;;;OAGG;YACW,kBAAkB;IA8DhC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAS5B;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAWhC;;OAEG;YACW,eAAe;IA2C7B;;;OAGG;IACG,SAAS,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC;IAyB3E;;;OAGG;IACH,UAAU,CAAC,qBAAqB,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,GAAG,OAAO;CAGtE"}