@myagentroam/node 0.9.68 → 0.9.69

package/dist/connector.js

@@ -1,5 +1,5 @@
 import { dirname, resolve } from 'node:path';
-import { createEnvelope } from '@myagentroam/protocol';
+import { createEnvelope, directNodeCommandSchema } from '@myagentroam/protocol';
 import { hostedConversationStorePath, loadNodeConfig, directTransferUdpPortRange, nodeDataDirectory, nodeDatabasePath } from './config.js';
 import { NodeMetrics, nodeLog } from './operational.js';
 import { assertLocalSqlitePath } from './storage.js';
@@ -1075,36 +1075,6 @@ export class NodeConnector {
                     void this.terminalManager.shutdown();
             }
         });
-        this.directTransferService = new DirectTransferService({
-            nodeGeneration: () => this.nodeGeneration,
-            send: (type, payload) => this.send(type, payload),
-            udpPortRange: () => directTransferUdpPortRange(this.config),
-            openUpload: async (descriptor) => {
-                if (descriptor.resource.kind !== 'WORKSPACE_FILE')
-                    throw new Error('DIRECT_RESOURCE_UNSUPPORTED');
-                const workspace = this.workspaceCoordinator.require(descriptor.resource.workspaceId);
-                const upload = await this.workspaceUploadService.createDirect(workspace, {
-                    path: descriptor.resource.path,
-                    size: descriptor.size,
-                    sha256: descriptor.sha256,
-                    overwrite: descriptor.overwrite === true,
-                    ...(descriptor.composerAttachment === undefined
-                        ? {}
-                        : { composerAttachment: descriptor.composerAttachment })
-                });
-                return {
-                    offset: upload.receivedBytes,
-                    write: (bytes) => this.workspaceUploadService.writeDirect(upload.uploadId, bytes),
-                    complete: (size, sha256) => this.workspaceUploadService.completeDirect(upload.uploadId, size, sha256),
-                    cancel: () => this.workspaceUploadService.cancel(upload.uploadId)
-                };
-            },
-            openDownload: async (descriptor) => {
-                if (this.workspaceWorkbenchService === undefined)
-                    throw new Error('DIRECT_RESOURCE_UNAVAILABLE');
-                return this.workspaceWorkbenchService.directDownload(descriptor);
-            }
-        });
         this.controlMessages = new NodeControlMessageService({
             config: () => this.config,
             register: (nodeId, credential) => this.connectionLifecycle.register(nodeId, credential),
@@ -1113,12 +1083,7 @@ export class NodeConnector {
             inspectWorkspace: (envelope) => this.workspaceCoordinator.inspect(envelope),
             handleNodeRequest: (envelope) => this.nodeRequests.handle(envelope),
             handleRuntimeCredentialResponse: (envelope) => this.runtimeCredentials.handle(envelope),
-            handleDirectTransfer: (envelope) => this.directTransferService.handle({
-                type: envelope.type,
-                ...(typeof envelope.payload === 'object' && envelope.payload !== null
-                    ? envelope.payload
-                    : {})
-            }),
+            handleDirectTransfer: (envelope) => this.handleDirectTransfer(envelope),
             reconcileRuns: (envelope) => this.runEventBridge.reconcile(envelope),
             runners: this.runners,
             deliverSessionChannelMessage: (envelope) => this.runnerChannelMessages.handle(envelope),
@@ -1419,9 +1384,72 @@ export class NodeConnector {
         this.database ??= new NodeDatabase(nodeDatabasePath(this.config, this.configPath), () => this.config?.nodeId ?? 'unregistered');
         this.database.closeRunningTerminalSessions();
         await this.initializeConversationHosting();
+        this.directTransferService ??= this.createDirectTransferService();
         this.connectionLifecycle.start();
         this.controlChannel.connect();
     }
+    createDirectTransferService() {
+        try {
+            return new DirectTransferService({
+                nodeGeneration: () => this.nodeGeneration,
+                send: (type, payload) => this.send(type, payload),
+                udpPortRange: () => directTransferUdpPortRange(this.config),
+                openUpload: async (descriptor) => {
+                    if (descriptor.resource.kind !== 'WORKSPACE_FILE')
+                        throw new Error('DIRECT_RESOURCE_UNSUPPORTED');
+                    const workspace = this.workspaceCoordinator.require(descriptor.resource.workspaceId);
+                    const upload = await this.workspaceUploadService.createDirect(workspace, {
+                        path: descriptor.resource.path,
+                        size: descriptor.size,
+                        sha256: descriptor.sha256,
+                        overwrite: descriptor.overwrite === true,
+                        ...(descriptor.composerAttachment === undefined
+                            ? {}
+                            : { composerAttachment: descriptor.composerAttachment })
+                    });
+                    return {
+                        offset: upload.receivedBytes,
+                        write: (bytes) => this.workspaceUploadService.writeDirect(upload.uploadId, bytes),
+                        complete: (size, sha256) => this.workspaceUploadService.completeDirect(upload.uploadId, size, sha256),
+                        cancel: () => this.workspaceUploadService.cancel(upload.uploadId)
+                    };
+                },
+                openDownload: async (descriptor) => {
+                    if (this.workspaceWorkbenchService === undefined)
+                        throw new Error('DIRECT_RESOURCE_UNAVAILABLE');
+                    return this.workspaceWorkbenchService.directDownload(descriptor);
+                }
+            });
+        }
+        catch (error) {
+            if (!(error instanceof Error) || error.message !== 'DIRECT_UDP_PORT_UNAVAILABLE')
+                throw error;
+            const range = directTransferUdpPortRange(this.config);
+            nodeLog('direct-transfer.udp-unavailable', {
+                portRangeBegin: range.begin,
+                portRangeEnd: range.end
+            });
+            return undefined;
+        }
+    }
+    handleDirectTransfer(envelope) {
+        const command = {
+            type: envelope.type,
+            ...(typeof envelope.payload === 'object' && envelope.payload !== null ? envelope.payload : {})
+        };
+        if (this.directTransferService !== undefined)
+            return this.directTransferService.handle(command);
+        const parsed = directNodeCommandSchema.safeParse(command);
+        if (!parsed.success)
+            return false;
+        this.send('direct.error', {
+            type: 'direct.error',
+            directSessionId: parsed.data.directSessionId,
+            code: 'DIRECT_UDP_PORT_UNAVAILABLE',
+            message: 'The configured direct-transfer UDP port range is unavailable.'
+        });
+        return true;
+    }
     clearMissionRetry(sessionId) {
         const retry = this.missionRetryTimers.get(sessionId);
         if (retry !== undefined)
@@ -1600,7 +1628,8 @@ export class NodeConnector {
         const terminalDrain = this.terminalManager.shutdown();
         this.terminalChannel.stop();
         this.runtimeCredentials.close();
-        this.directTransferService.close();
+        this.directTransferService?.dispose();
+        this.directTransferService = undefined;
         this.controlChannel.stop();
         const database = this.database;
         this.database = undefined;

package/dist/service/direct-transfer-service.js

@@ -1,4 +1,5 @@
 import { directDataControlFrameSchema, directNodeCommandSchema } from '@myagentroam/protocol';
+import { IceUdpMuxListener as NativeIceUdpMuxListener } from 'node-datachannel';
 import { RTCPeerConnection } from 'node-datachannel/polyfill';
 const MAX_HOST_CANDIDATES = 32;
 const MAX_ACTIVE_SESSIONS = 32;
@@ -11,21 +12,27 @@ export class DirectTransferService {
     sessions = new Map();
     now;
     createPeerConnection;
+    iceUdpMuxListener;
+    peerConfiguration;
     leaseGraceMs;
     leaseCheckMs;
+    disposed = false;
     constructor(options) {
         this.options = options;
         this.now = options.now ?? Date.now;
+        const range = options.udpPortRange?.() ?? { begin: 49_152, end: 49_215 };
+        const createIceUdpMuxListener = options.createIceUdpMuxListener ?? ((port) => new NativeIceUdpMuxListener(port));
+        this.iceUdpMuxListener = bindIceUdpMuxListener(range, createIceUdpMuxListener);
+        const udpPort = this.iceUdpMuxListener.port();
+        this.peerConfiguration = {
+            iceServers: [],
+            enableIceUdpMux: true,
+            portRangeBegin: udpPort,
+            portRangeEnd: udpPort
+        };
         this.createPeerConnection =
             options.createPeerConnection ??
-                (() => {
-                    const range = options.udpPortRange?.() ?? { begin: 49_152, end: 49_215 };
-                    return new RTCPeerConnection({
-                        iceServers: [],
-                        portRangeBegin: range.begin,
-                        portRangeEnd: range.end
-                    });
-                });
+                ((configuration) => new RTCPeerConnection(configuration));
         this.leaseGraceMs = options.leaseGraceMs ?? LEASE_GRACE_MS;
         this.leaseCheckMs = options.leaseCheckMs ?? LEASE_CHECK_MS;
     }
@@ -49,6 +56,13 @@ export class DirectTransferService {
         for (const session of [...this.sessions.values()])
             void this.remove(session, 'DIRECT_NODE_STOPPED');
     }
+    dispose() {
+        if (this.disposed)
+            return;
+        this.disposed = true;
+        this.close();
+        this.iceUdpMuxListener.stop();
+    }
     async dispatch(command) {
         if (command.type === 'direct.prepare') {
             this.prepare(command);
@@ -86,7 +100,7 @@ export class DirectTransferService {
             throw new Error('DIRECT_SESSION_CONFLICT');
         if (this.sessions.size >= MAX_ACTIVE_SESSIONS)
             throw new Error('DIRECT_SESSION_LIMIT_REACHED');
-        const peer = this.createPeerConnection();
+        const peer = this.createPeerConnection(this.peerConfiguration);
         const session = {
             command,
             peer,
@@ -360,6 +374,18 @@ export class DirectTransferService {
             });
     }
 }
+function bindIceUdpMuxListener(range, create) {
+    let lastError;
+    for (let port = range.begin; port <= range.end; port += 1) {
+        try {
+            return create(port);
+        }
+        catch (error) {
+            lastError = error;
+        }
+    }
+    throw new Error('DIRECT_UDP_PORT_UNAVAILABLE', { cause: lastError });
+}
 export function fingerprint(sdp) {
     const match = sdp?.match(/^a=fingerprint:(sha-256 [A-F0-9:]+)\r?$/imu);
     return match?.[1];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@myagentroam/node",
-  "version": "0.9.68",
+  "version": "0.9.69",
   "description": "MyAgentRoam Node runtime CLI.",
   "license": "Apache-2.0",
   "type": "module",
@@ -28,8 +28,8 @@
     "node-pty": "1.1.0",
     "ws": "^8.21.3",
     "zod": "4.4.3",
-    "@myagentroam/agent": "0.9.68",
-    "@myagentroam/protocol": "0.9.68"
+    "@myagentroam/agent": "0.9.69",
+    "@myagentroam/protocol": "0.9.69"
   },
   "devDependencies": {
     "@types/ws": "^8.18.1"