@voko/lite 0.3.1

package/src/server/agent-manager.js

@@ -0,0 +1,580 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { discoverHermes } = require('./hermes-discovery');
+const { getHermesProfilePath } = require('../core/hermes-paths');
+
+class AgentManager {
+  constructor(databaseAPI) {
+    this.databaseAPI = databaseAPI || null;
+    this.openclawConfigPath = path.join(os.homedir(), '.openclaw', 'openclaw.json');
+    this.channelConfigPath = path.join(__dirname, '..', '..', 'config', 'channel_config.json');
+  }
+
+  // 读取 openclaw.json 配置
+  _readOpenclawConfig() {
+    try {
+      const content = fs.readFileSync(this.openclawConfigPath, 'utf-8');
+      return JSON.parse(content);
+    } catch (err) {
+      console.error('[AgentManager] 读取 openclaw.json 失败:', err.message);
+      return { agents: { list: [] } };
+    }
+  }
+
+  // 写入 openclaw.json 配置
+  _writeOpenclawConfig(config) {
+    try {
+      fs.writeFileSync(this.openclawConfigPath, JSON.stringify(config, null, 2), 'utf-8');
+      return true;
+    } catch (err) {
+      console.error('[AgentManager] 写入 openclaw.json 失败:', err.message);
+      return false;
+    }
+  }
+
+  // 读取 OpenClaw 默认模型
+  _getDefaultModel() {
+    try {
+      const config = this._readOpenclawConfig();
+      // agents.defaults.model.primary (OpenClaw 常用结构)
+      if (config.agents?.defaults?.model?.primary) return config.agents.defaults.model.primary;
+      // auth.profiles 中的模型
+      const profiles = config.auth?.profiles || {};
+      for (const key of Object.keys(profiles)) {
+        const profile = profiles[key];
+        if (profile.model?.primary) return profile.model.primary;
+        if (profile.model) return profile.model;
+      }
+      // 尝试 agents.defaultModel
+      if (config.agents?.defaultModel) return config.agents.defaultModel;
+      // 尝试顶层 gateway.model
+      if (config.gateway?.model) return config.gateway.model;
+      return null;
+    } catch (e) {
+      return null;
+    }
+  }
+
+  getDefaultModel() {
+    return this._getDefaultModel();
+  }
+
+  // 读取 channel_config（优先 DB，首次运行从 JSON 迁移）
+  _readChannelConfig() {
+    try {
+      if (this.databaseAPI) {
+        const fromDb = this.databaseAPI.getConfigFromDb();
+        if (fromDb) return fromDb;
+      }
+      // fallback: 从 JSON 文件读取（首次迁移）
+      const content = fs.readFileSync(this.channelConfigPath, 'utf-8');
+      return JSON.parse(content);
+    } catch (err) {
+      console.error('[AgentManager] 读取 channel_config 失败:', err.message);
+      return {};
+    }
+  }
+
+  // 写入 channel_config（仅 DB，不落明文文件）
+  _writeChannelConfig(config) {
+    try {
+      if (this.databaseAPI) this.databaseAPI.saveConfigToDb(config);
+      return true;
+    } catch (err) {
+      console.error('[AgentManager] 写入 channel_config 失败:', err.message);
+      return false;
+    }
+  }
+
+  // 展开 ~ 为用户家目录
+  _expandPath(p) {
+    if (p && p.startsWith('~')) {
+      return path.join(os.homedir(), p.slice(1));
+    }
+    return p;
+  }
+
+  // 获取 routingRules
+  getRoutingRules() {
+    const config = this._readChannelConfig();
+    return config.routingRules || [];
+  }
+
+  // 获取当前 voko 对接的 agent
+  getVokoAgent() {
+    const rules = this.getRoutingRules();
+    const defaultRule = rules.find(r => r.wsChannel === 'default');
+    return defaultRule ? defaultRule.agentId : null;
+  }
+
+  // 设置 voko 对接的 agent
+  setVokoAgent(agentId) {
+    const config = this._readChannelConfig();
+    if (!config.routingRules) {
+      config.routingRules = [];
+    }
+    const idx = config.routingRules.findIndex(r => r.wsChannel === 'default');
+    if (idx >= 0) {
+      config.routingRules[idx] = { wsChannel: 'default', sessionPrefix: `agent:${agentId}:`, agentId };
+    } else {
+      config.routingRules.push({ wsChannel: 'default', sessionPrefix: `agent:${agentId}:`, agentId });
+    }
+    return this._writeChannelConfig(config);
+  }
+
+  // 列出所有 agent
+  listAgents() {
+    const config = this._readOpenclawConfig();
+    const openclawAgents = config?.agents?.list || [];
+    const vokoAgentId = this.getVokoAgent();
+
+    // 发现 Hermes profiles
+    let hermesAgents = [];
+    try {
+      hermesAgents = discoverHermes();
+    } catch (err) {
+      console.error('[AgentManager] Hermes 发现失败:', err.message);
+    }
+
+    // 读取 channel_config 中的 Hermes 端口配置
+    const channelCfg = this._readChannelConfig();
+    const hermesProfiles = channelCfg?.hermes_config?.profiles || {};
+
+    const result = {
+      agents: [
+        // OpenClaw agents
+        ...openclawAgents.map(a => ({
+          id: a.id,
+          name: a.name || a.id,
+          workspace: a.workspace,
+          agentDir: a.agentDir,
+          model: a.model,
+          skills: a.skills || [],
+          tools: a.tools || { allow: [], deny: [] },
+          isVoko: a.id === vokoAgentId,
+          backend: 'openclaw'
+        })),
+        // Hermes agents
+        ...hermesAgents.map(p => ({
+          id: p.name,
+          name: p.name,
+          model: p.model,
+          isDefault: p.isDefault,
+          backend: 'hermes',
+          port: hermesProfiles[p.name]?.port || null
+        }))
+      ],
+      openclawFound: !!config?.agents,
+      hermesFound: hermesAgents.length > 0
+    };
+
+    // 合并 SQLite agents 表中已发布的 agent（去重）
+    try {
+      if (this.databaseAPI) {
+        const dbRows = this.databaseAPI.query(
+          "SELECT agent_id, agent_name, backend_type, publish_status FROM agents"
+        );
+        const existingIds = new Set(result.agents.map(a => a.id));
+        for (const r of dbRows) {
+          if (!existingIds.has(r.agent_id)) {
+            result.agents.push({
+              id: r.agent_id,
+              name: r.agent_name || r.agent_id,
+              backend: r.backend_type || 'others',
+              workspace: null,
+              model: null,
+              skills: [],
+              tools: { allow: [], deny: [] },
+              isVoko: false
+            });
+          }
+        }
+      }
+    } catch (e) {
+      console.error('[AgentManager] 读取 DB agents 失败:', e.message);
+    }
+
+    return result;
+  }
+
+  // 读取单个 agent 详情
+  getAgentDetail(agentId) {
+    // 先查 OpenClaw
+    const config = this._readOpenclawConfig();
+    const agents = config.agents?.list || [];
+    const agent = agents.find(a => a.id === agentId);
+    if (agent) {
+      let description = '';
+      if (agent.workspace) {
+        const workspacePath = this._expandPath(agent.workspace);
+        const agentsMdPath = path.join(workspacePath, 'AGENTS.md');
+        if (fs.existsSync(agentsMdPath)) {
+          const content = fs.readFileSync(agentsMdPath, 'utf-8');
+          const lines = content.split('\n');
+          for (const line of lines) {
+            const trimmed = line.trim();
+            if (trimmed && !trimmed.startsWith('#')) {
+              description = trimmed;
+              break;
+            }
+          }
+        }
+      }
+      return {
+        id: agent.id,
+        name: agent.name || agent.id,
+        workspace: agent.workspace,
+        model: agent.model,
+        skills: agent.skills || [],
+        tools: agent.tools || { allow: [], deny: [] },
+        description,
+        isVoko: agent.id === this.getVokoAgent(),
+        backend: 'openclaw'
+      };
+    }
+
+    // 再查 Hermes
+    try {
+      const hermesProfiles = discoverHermes();
+      const profile = hermesProfiles.find(p => p.name === agentId);
+      if (profile) {
+        const profilePath = getHermesProfilePath(profile.name);
+        // 解析 SOUL.md 获取权限信息
+        let hermesTools = { allow: [], deny: [] };
+        let hermesSkills = [];
+        const soulPath = path.join(profilePath, 'SOUL.md');
+        try {
+          if (fs.existsSync(soulPath)) {
+            const soulContent = fs.readFileSync(soulPath, 'utf-8');
+            const permMatch = soulContent.match(/##\s*(权限|Permission|Permissions)[\s\S]*?(?=##|$)/);
+            if (permMatch) {
+              const permSection = permMatch[0];
+              for (const line of permSection.split('\n')) {
+                const trimmed = line.trim();
+                if (trimmed.includes('✅')) {
+                  const text = trimmed.replace(/^[-*\s]*✅\s*/, '');
+                  // 取括号前的内容，然后按 /、, 分割
+                  const beforeBracket = text.replace(/[（(【][\s\S]*$/, '');
+                  const parts = beforeBracket.split(/[/、,]/);
+                  for (const part of parts) {
+                    const name = part.trim();
+                    if (name && name.length > 1) {
+                      // 只取第一个词
+                      const firstWord = name.split(/\s+/)[0];
+                      // 判断：如果第一个词后紧跟着 "skill" → 技能，否则→工具
+                      const restAfter = name.substring(firstWord.length).trim();
+                      const isSkillItem = /^skill/i.test(restAfter);
+                      if (isSkillItem) {
+                        if (firstWord && !hermesSkills.includes(firstWord)) hermesSkills.push(firstWord);
+                      } else if (firstWord.length > 1) {
+                        if (!hermesTools.allow.includes(firstWord)) hermesTools.allow.push(firstWord);
+                      }
+                    }
+                  }
+                }
+                if (trimmed.includes('❌')) {
+                  const text = trimmed.replace(/^[-*\s]*❌\s*/, '');
+                  const beforeBracket = text.replace(/[（(【][\s\S]*$/, '');
+                  const parts = beforeBracket.split(/[/、,]/);
+                  for (const part of parts) {
+                    let name = part.trim();
+                    // 去掉后面多余的中文描述
+                    name = name.replace(/\s+(skill|toolset).*$/i, '').trim();
+                    // 只取第一个词（中文描述前）
+                    name = name.split(/\s+/)[0];
+                    if (name && name.length > 1 && !hermesTools.deny.includes(name)) {
+                      hermesTools.deny.push(name);
+                    }
+                  }
+                }
+              }
+            }
+          }
+        } catch (e) {
+          console.error('[AgentManager] 解析 SOUL.md 权限失败:', e.message);
+        }
+        return {
+          id: profile.name,
+          name: profile.name,
+          model: profile.model,
+          description: `Hermes Agent - ${profile.name}${profile.isDefault ? ' (默认)' : ''}`,
+          backend: 'hermes',
+          profilePath,
+          skills: hermesSkills,
+          tools: hermesTools
+        };
+      }
+    } catch (err) {
+      console.error('[AgentManager] Hermes 详情查询失败:', err.message);
+    }
+
+    // 回退：从 SQLite agents 表查询（MCP 注册的 agent）
+    try {
+      if (this.databaseAPI) {
+        const rows = this.databaseAPI.query(
+          "SELECT agent_id, agent_name, description, short_description, category, category_label, backend_type FROM agents WHERE agent_id = '" + agentId.replace(/'/g, "''") + "' LIMIT 1"
+        );
+        if (rows && rows.length > 0) {
+          const r = rows[0];
+          return {
+            id: r.agent_id,
+            name: r.agent_name || r.agent_id,
+            description: r.description || r.short_description || '',
+            backend: r.backend_type || 'others',
+            workspace: null,
+            model: null,
+            skills: [],
+            tools: { allow: [], deny: [] },
+            isVoko: false
+          };
+        }
+      }
+    } catch (e) {
+      console.error('[AgentManager] DB agent 详情查询失败:', e.message);
+    }
+
+    return null;
+  }
+
+  // 获取 agent workspace 下的核心文件列表
+  getAgentFiles(agentId) {
+    const config = this._readOpenclawConfig();
+    const agents = config.agents?.list || [];
+    const agent = agents.find(a => a.id === agentId);
+    if (!agent || !agent.workspace) {
+      return [];
+    }
+
+    const workspacePath = this._expandPath(agent.workspace);
+    const coreFiles = ['AGENTS.md', 'SOUL.md', 'IDENTITY.md', 'TOOLS.md', 'USER.md', 'knowledge.md'];
+
+    try {
+      if (!fs.existsSync(workspacePath)) {
+        return [];
+      }
+
+      const files = [];
+      for (const file of coreFiles) {
+        const filePath = path.join(workspacePath, file);
+        if (fs.existsSync(filePath)) {
+          const stat = fs.statSync(filePath);
+          files.push({
+            name: file,
+            path: filePath,
+            size: stat.size,
+            modified: stat.mtime
+          });
+        }
+      }
+
+      // 尝试读取 skills 目录
+      const skillsDir = path.join(workspacePath, 'skills');
+      if (fs.existsSync(skillsDir)) {
+        const skillDirs = fs.readdirSync(skillsDir).filter(d => {
+          const subPath = path.join(skillsDir, d);
+          return fs.statSync(subPath).isDirectory();
+        });
+        for (const skillDir of skillDirs) {
+          const skillFile = path.join(skillsDir, skillDir, 'SKILL.md');
+          if (fs.existsSync(skillFile)) {
+            const stat = fs.statSync(skillFile);
+            files.push({
+              name: `skills/${skillDir}/SKILL.md`,
+              path: skillFile,
+              size: stat.size,
+              modified: stat.mtime
+            });
+          }
+        }
+      }
+
+      return files;
+    } catch (err) {
+      console.error('[AgentManager] 获取 agent 文件列表失败:', err.message);
+      return [];
+    }
+  }
+
+  // 读取单个文件内容
+  readFile(agentId, filename) {
+    const config = this._readOpenclawConfig();
+    const agents = config.agents?.list || [];
+    const agent = agents.find(a => a.id === agentId);
+    if (!agent || !agent.workspace) {
+      throw new Error('Agent not found');
+    }
+
+    const workspacePath = this._expandPath(agent.workspace);
+    const filePath = path.join(workspacePath, filename);
+
+    // 安全检查：不允许路径遍历
+    if (!filePath.startsWith(workspacePath)) {
+      throw new Error('Invalid path');
+    }
+
+    try {
+      return fs.readFileSync(filePath, 'utf-8');
+    } catch (err) {
+      console.error('[AgentManager] 读取文件失败:', err.message);
+      throw err;
+    }
+  }
+
+  // 写入单个文件
+  writeFile(agentId, filename, content) {
+    const config = this._readOpenclawConfig();
+    const agents = config.agents?.list || [];
+    const agent = agents.find(a => a.id === agentId);
+    if (!agent || !agent.workspace) {
+      throw new Error('Agent not found');
+    }
+
+    const workspacePath = this._expandPath(agent.workspace);
+    const filePath = path.join(workspacePath, filename);
+
+    // 安全检查：不允许路径遍历
+    if (!filePath.startsWith(workspacePath)) {
+      throw new Error('Invalid path');
+    }
+
+    try {
+      fs.writeFileSync(filePath, content, 'utf-8');
+      return true;
+    } catch (err) {
+      console.error('[AgentManager] 写入文件失败:', err.message);
+      throw err;
+    }
+  }
+
+  // 创建新 agent
+  createAgent(agentConfig) {
+    const { id, name, desc, workspace, model } = agentConfig;
+    if (!id || !workspace) {
+      throw new Error('id 和 workspace 是必填项');
+    }
+
+    const config = this._readOpenclawConfig();
+    if (!config.agents) {
+      config.agents = { list: [] };
+    }
+
+    // 检查是否已存在
+    if (config.agents.list.find(a => a.id === id)) {
+      throw new Error('Agent ID 已存在');
+    }
+
+    // 创建 workspace 目录
+    const workspacePath = this._expandPath(workspace);
+    fs.mkdirSync(workspacePath, { recursive: true });
+
+    // 构建 AGENTS.md 内容
+    let agentsContent = `# ${name || id} Agent`;
+    if (desc) {
+      agentsContent += `\n\n> ${desc}`;
+    }
+    agentsContent += `
+
+你是一个有用的 AI 助手。
+
+## 核心能力
+- ...
+
+## 限制
+- ...
+`;
+
+    // 创建默认文件
+    const defaultFiles = {
+      'AGENTS.md': agentsContent,
+      'SOUL.md': `# ${name || id} - 灵魂设定
+
+## 角色定义
+...
+
+## 沟通风格
+...
+`,
+      'IDENTITY.md': `# ${name || id}
+
+## 基本信息
+- 名称: ${name || id}
+- 角色: AI 助手
+
+## 个性
+...
+`,
+      'TOOLS.md': `# 工具使用规范
+
+## 可用工具
+...
+`,
+      'USER.md': `# 用户画像
+
+## 目标用户
+...
+`
+    };
+
+    for (const [filename, fileContent] of Object.entries(defaultFiles)) {
+      const filePath = path.join(workspacePath, filename);
+      fs.writeFileSync(filePath, fileContent, 'utf-8');
+    }
+
+    // 添加到 openclaw.json
+    config.agents.list.push({
+      id,
+      name: name || id,
+      workspace,
+      model: model || null,
+      skills: []
+    });
+
+    this._writeOpenclawConfig(config);
+    return { id, name: name || id, workspace };
+  }
+
+  // 删除 agent
+  deleteAgent(agentId) {
+    const config = this._readOpenclawConfig();
+    const idx = config.agents?.list?.findIndex(a => a.id === agentId);
+    if (idx === undefined || idx < 0) {
+      throw new Error('Agent not found');
+    }
+
+    const agent = config.agents.list[idx];
+    config.agents.list.splice(idx, 1);
+    this._writeOpenclawConfig(config);
+
+    // 从 routingRules 移除
+    const channelConfig = this._readChannelConfig();
+    if (channelConfig.routingRules) {
+      channelConfig.routingRules = channelConfig.routingRules.filter(r => r.agentId !== agentId);
+      this._writeChannelConfig(channelConfig);
+    }
+
+    return true;
+  }
+
+  // 引入人工介入接口
+  injectOwnerCli(agentId) {
+    const config = this._readOpenclawConfig();
+    const agents = config.agents?.list || [];
+    const agent = agents.find(a => a.id === agentId);
+    if (!agent || !agent.workspace) {
+      throw new Error('Agent not found');
+    }
+    const workspacePath = this._expandPath(agent.workspace);
+    const scriptsDir = path.join(workspacePath, 'scripts');
+    const srcFile = path.join(__dirname, '..', '..', 'scripts', 'owner-intervention-cli.js');
+    const dstFile = path.join(scriptsDir, 'owner-intervention-cli.js');
+    if (!fs.existsSync(srcFile)) {
+      throw new Error('源文件不存在: scripts/owner-intervention-cli.js');
+    }
+    fs.mkdirSync(scriptsDir, { recursive: true });
+    fs.copyFileSync(srcFile, dstFile);
+    return true;
+  }
+}
+
+module.exports = AgentManager;