@bolloon/bolloon-agent 0.1.2 → 0.1.4

package/dist/web/components/p2p/p2p-tools.js

@@ -0,0 +1,248 @@
+/**
+ * P2P 工具调用模块
+ * 支持文件和本地信息的 P2P 传递
+ */
+// 工具类型枚举
+export var P2PToolType;
+(function (P2PToolType) {
+    P2PToolType["FILE_TRANSFER"] = "file_transfer";
+    P2PToolType["LOCAL_INFO"] = "local_info";
+    P2PToolType["SYSTEM_INFO"] = "system_info";
+    P2PToolType["FILE_LIST"] = "file_list";
+})(P2PToolType || (P2PToolType = {}));
+// 文件传输
+export async function transferFile(targetDid, fileInfo, messageId) {
+    try {
+        const payload = {
+            type: 'file',
+            tool: P2PToolType.FILE_TRANSFER,
+            data: {
+                name: fileInfo.name,
+                size: fileInfo.size,
+                mimeType: fileInfo.mimeType,
+                content: fileInfo.content
+            }
+        };
+        const res = await fetch('/api/message-p2p', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                targetDid,
+                content: JSON.stringify(payload),
+                type: 'file'
+            })
+        });
+        if (res.ok) {
+            const data = await res.json();
+            return {
+                success: true,
+                messageId: data.messageId || messageId
+            };
+        }
+        else {
+            return { success: false, error: '文件传输失败' };
+        }
+    }
+    catch (e) {
+        return { success: false, error: e.message };
+    }
+}
+// 本地信息查询
+export async function queryLocalInfo(targetDid, query) {
+    try {
+        const payload = {
+            type: 'info_query',
+            tool: P2PToolType.LOCAL_INFO,
+            query: query
+        };
+        const res = await fetch('/api/message-p2p', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                targetDid,
+                content: JSON.stringify(payload),
+                type: 'ai-dialogue'
+            })
+        });
+        if (res.ok) {
+            const data = await res.json();
+            return {
+                success: true,
+                data: data.response,
+                messageId: data.messageId
+            };
+        }
+        else {
+            return { success: false, error: '信息查询失败' };
+        }
+    }
+    catch (e) {
+        return { success: false, error: e.message };
+    }
+}
+// 系统信息请求
+export async function getSystemInfo(targetDid) {
+    try {
+        const payload = {
+            type: 'system_info_request',
+            tool: P2PToolType.SYSTEM_INFO
+        };
+        const res = await fetch('/api/message-p2p', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                targetDid,
+                content: JSON.stringify(payload),
+                type: 'ai-dialogue'
+            })
+        });
+        if (res.ok) {
+            const data = await res.json();
+            return {
+                success: true,
+                data: data.response,
+                messageId: data.messageId
+            };
+        }
+        else {
+            return { success: false, error: '系统信息获取失败' };
+        }
+    }
+    catch (e) {
+        return { success: false, error: e.message };
+    }
+}
+// 文件列表请求
+export async function listFiles(targetDid, path = '/') {
+    try {
+        const payload = {
+            type: 'file_list_request',
+            tool: P2PToolType.FILE_LIST,
+            path: path
+        };
+        const res = await fetch('/api/message-p2p', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                targetDid,
+                content: JSON.stringify(payload),
+                type: 'ai-dialogue'
+            })
+        });
+        if (res.ok) {
+            const data = await res.json();
+            return {
+                success: true,
+                data: data.response,
+                messageId: data.messageId
+            };
+        }
+        else {
+            return { success: false, error: '文件列表获取失败' };
+        }
+    }
+    catch (e) {
+        return { success: false, error: e.message };
+    }
+}
+// 工具调用入口
+export async function executeP2PTool(request) {
+    const { toolName, payload, targetDid } = request;
+    if (!targetDid) {
+        return { success: false, error: '未指定目标节点' };
+    }
+    switch (toolName) {
+        case P2PToolType.FILE_TRANSFER:
+            return transferFile(targetDid, payload.data);
+        case P2PToolType.LOCAL_INFO:
+            return queryLocalInfo(targetDid, payload.data);
+        case P2PToolType.SYSTEM_INFO:
+            return getSystemInfo(targetDid);
+        case P2PToolType.FILE_LIST:
+            return listFiles(targetDid, payload.data?.path || '/');
+        default:
+            return { success: false, error: `未知工具: ${toolName}` };
+    }
+}
+// 本地系统信息获取 (用于响应来自远程的请求)
+export function getLocalSystemInfo() {
+    const memUsage = process.memoryUsage();
+    const cpuCores = require('os').cpus();
+    return {
+        platform: process.platform,
+        arch: process.arch,
+        nodeVersion: process.version,
+        memory: {
+            total: memUsage.heapTotal,
+            free: memUsage.heapTotal - memUsage.heapUsed,
+            used: memUsage.heapUsed
+        },
+        cpu: {
+            cores: cpuCores.length,
+            model: cpuCores[0]?.model || 'Unknown'
+        },
+        uptime: process.uptime()
+    };
+}
+// 本地文件列表 (用于响应来自远程的请求)
+export function getLocalFileList(dirPath) {
+    try {
+        // Dynamic import for ESM compatibility
+        const pathModule = require('path');
+        const fs = require('fs');
+        const fullPath = pathModule.resolve(dirPath);
+        const files = fs.readdirSync(fullPath);
+        const fileList = files.map((name) => {
+            const fullFilePath = pathModule.join(fullPath, name);
+            let stat;
+            try {
+                stat = fs.statSync(fullFilePath);
+            }
+            catch {
+                return null;
+            }
+            return {
+                name: name,
+                size: stat.size,
+                isDirectory: stat.isDirectory(),
+                modified: stat.mtime.toISOString()
+            };
+        }).filter(Boolean);
+        return {
+            success: true,
+            path: fullPath,
+            files: fileList
+        };
+    }
+    catch (e) {
+        return {
+            success: false,
+            path: dirPath,
+            error: e.message
+        };
+    }
+}
+// 导出工具名称列表
+export const P2P_TOOLS = [
+    {
+        name: P2PToolType.FILE_TRANSFER,
+        description: '传输文件到远程节点',
+        parameters: ['targetDid', 'fileInfo']
+    },
+    {
+        name: P2PToolType.LOCAL_INFO,
+        description: '查询远程节点本地信息',
+        parameters: ['targetDid', 'query']
+    },
+    {
+        name: P2PToolType.SYSTEM_INFO,
+        description: '获取远程节点系统信息',
+        parameters: ['targetDid']
+    },
+    {
+        name: P2PToolType.FILE_LIST,
+        description: '列出远程节点目录文件',
+        parameters: ['targetDid', 'path']
+    }
+];
+console.log('[P2P Tools] 工具模块已加载');