@adversity/coding-tool-x 2.2.0

package/src/server/services/agents-service.js

@@ -0,0 +1,354 @@
+/**
+ * Agents 服务
+ *
+ * 管理 Claude Code 自定义代理的 CRUD 操作
+ * 代理目录:
+ * - 用户级: ~/.claude/agents/
+ * - 项目级: .claude/agents/
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+// 代理目录路径
+const USER_AGENTS_DIR = path.join(os.homedir(), '.claude', 'agents');
+
+/**
+ * 确保目录存在
+ */
+function ensureDir(dirPath) {
+  if (!fs.existsSync(dirPath)) {
+    fs.mkdirSync(dirPath, { recursive: true });
+  }
+}
+
+/**
+ * 解析 YAML frontmatter
+ */
+function parseFrontmatter(content) {
+  const result = {
+    frontmatter: {},
+    body: content
+  };
+
+  // 移除 BOM
+  content = content.trim().replace(/^\uFEFF/, '');
+
+  // 解析 YAML frontmatter
+  const match = content.match(/^---\s*\n([\s\S]*?)\n---\s*\n?([\s\S]*)$/);
+  if (!match) {
+    return result;
+  }
+
+  const frontmatterText = match[1];
+  result.body = match[2].trim();
+
+  // 简单解析 YAML（支持基本字段）
+  const lines = frontmatterText.split('\n');
+  for (const line of lines) {
+    const colonIndex = line.indexOf(':');
+    if (colonIndex === -1) continue;
+
+    const key = line.slice(0, colonIndex).trim();
+    let value = line.slice(colonIndex + 1).trim();
+
+    // 去除引号
+    if ((value.startsWith('"') && value.endsWith('"')) ||
+        (value.startsWith("'") && value.endsWith("'"))) {
+      value = value.slice(1, -1);
+    }
+
+    result.frontmatter[key] = value;
+  }
+
+  return result;
+}
+
+/**
+ * 生成 frontmatter 字符串
+ */
+function generateFrontmatter(data) {
+  const lines = ['---'];
+
+  // 必需字段
+  if (data.name) {
+    lines.push(`name: ${data.name}`);
+  }
+  if (data.description) {
+    lines.push(`description: "${data.description}"`);
+  }
+
+  // 可选字段
+  if (data.tools) {
+    lines.push(`tools: ${data.tools}`);
+  }
+  if (data.model) {
+    lines.push(`model: ${data.model}`);
+  }
+  if (data.permissionMode) {
+    lines.push(`permissionMode: ${data.permissionMode}`);
+  }
+  if (data.skills) {
+    lines.push(`skills: ${data.skills}`);
+  }
+
+  lines.push('---');
+  return lines.join('\n');
+}
+
+/**
+ * 递归扫描目录获取代理文件
+ */
+function scanAgentsDir(dir, basePath, scope) {
+  const agents = [];
+
+  if (!fs.existsSync(dir)) {
+    return agents;
+  }
+
+  try {
+    const entries = fs.readdirSync(dir, { withFileTypes: true });
+
+    for (const entry of entries) {
+      const fullPath = path.join(dir, entry.name);
+
+      if (entry.isDirectory() && !entry.name.startsWith('.')) {
+        // 递归扫描子目录
+        const subAgents = scanAgentsDir(fullPath, basePath, scope);
+        agents.push(...subAgents);
+      } else if (entry.isFile() && entry.name.endsWith('.md')) {
+        // 解析代理文件
+        try {
+          const content = fs.readFileSync(fullPath, 'utf-8');
+          const { frontmatter, body } = parseFrontmatter(content);
+
+          // 计算相对路径
+          const relativePath = path.relative(basePath, fullPath);
+          const fileName = entry.name.replace(/\.md$/, '');
+
+          agents.push({
+            name: frontmatter.name || fileName,
+            fileName,
+            scope,
+            path: relativePath,
+            fullPath,
+            description: frontmatter.description || '',
+            tools: frontmatter.tools || '',
+            model: frontmatter.model || '',
+            permissionMode: frontmatter.permissionMode || '',
+            skills: frontmatter.skills || '',
+            systemPrompt: body,
+            fullContent: content,
+            updatedAt: fs.statSync(fullPath).mtime.getTime()
+          });
+        } catch (err) {
+          console.warn(`[AgentsService] Failed to parse ${fullPath}:`, err.message);
+        }
+      }
+    }
+  } catch (err) {
+    console.error(`[AgentsService] Failed to scan ${dir}:`, err.message);
+  }
+
+  return agents;
+}
+
+/**
+ * Agents 服务类
+ */
+class AgentsService {
+  constructor() {
+    this.userAgentsDir = USER_AGENTS_DIR;
+    ensureDir(this.userAgentsDir);
+  }
+
+  /**
+   * 获取所有代理列表
+   * @param {string} projectPath - 项目路径（可选，用于获取项目级代理）
+   */
+  listAgents(projectPath = null) {
+    const agents = [];
+
+    // 获取用户级代理
+    const userAgents = scanAgentsDir(this.userAgentsDir, this.userAgentsDir, 'user');
+    agents.push(...userAgents);
+
+    // 获取项目级代理（如果提供了项目路径）
+    if (projectPath) {
+      const projectAgentsDir = path.join(projectPath, '.claude', 'agents');
+      const projectAgents = scanAgentsDir(projectAgentsDir, projectAgentsDir, 'project');
+      agents.push(...projectAgents);
+    }
+
+    // 按名称排序
+    agents.sort((a, b) => a.name.toLowerCase().localeCompare(b.name.toLowerCase()));
+
+    return {
+      agents,
+      total: agents.length,
+      userCount: userAgents.length,
+      projectCount: agents.length - userAgents.length
+    };
+  }
+
+  /**
+   * 获取单个代理详情
+   */
+  getAgent(fileName, scope, projectPath = null) {
+    const baseDir = scope === 'user'
+      ? this.userAgentsDir
+      : path.join(projectPath, '.claude', 'agents');
+
+    const filePath = path.join(baseDir, `${fileName}.md`);
+
+    if (!fs.existsSync(filePath)) {
+      return null;
+    }
+
+    const content = fs.readFileSync(filePath, 'utf-8');
+    const { frontmatter, body } = parseFrontmatter(content);
+
+    return {
+      name: frontmatter.name || fileName,
+      fileName,
+      scope,
+      path: `${fileName}.md`,
+      fullPath: filePath,
+      description: frontmatter.description || '',
+      tools: frontmatter.tools || '',
+      model: frontmatter.model || '',
+      permissionMode: frontmatter.permissionMode || '',
+      skills: frontmatter.skills || '',
+      systemPrompt: body,
+      fullContent: content,
+      updatedAt: fs.statSync(filePath).mtime.getTime()
+    };
+  }
+
+  /**
+   * 创建代理
+   */
+  createAgent({ fileName, scope, projectPath, name, description, tools, model, permissionMode, skills, systemPrompt }) {
+    if (!fileName || !fileName.trim()) {
+      throw new Error('代理文件名不能为空');
+    }
+
+    // 验证文件名：只允许字母、数字、横杠、下划线
+    if (!/^[a-zA-Z0-9_-]+$/.test(fileName)) {
+      throw new Error('代理文件名只能包含字母、数字、横杠和下划线');
+    }
+
+    if (!name || !name.trim()) {
+      throw new Error('代理名称不能为空');
+    }
+
+    if (!description || !description.trim()) {
+      throw new Error('代理描述不能为空');
+    }
+
+    const baseDir = scope === 'user'
+      ? this.userAgentsDir
+      : path.join(projectPath, '.claude', 'agents');
+
+    ensureDir(baseDir);
+
+    const filePath = path.join(baseDir, `${fileName}.md`);
+
+    // 检查是否已存在
+    if (fs.existsSync(filePath)) {
+      throw new Error(`代理 "${fileName}" 已存在`);
+    }
+
+    // 生成文件内容
+    const frontmatterData = { name, description };
+    if (tools) frontmatterData.tools = tools;
+    if (model) frontmatterData.model = model;
+    if (permissionMode) frontmatterData.permissionMode = permissionMode;
+    if (skills) frontmatterData.skills = skills;
+
+    const content = generateFrontmatter(frontmatterData) + '\n\n' + (systemPrompt || '');
+
+    fs.writeFileSync(filePath, content, 'utf-8');
+
+    return this.getAgent(fileName, scope, projectPath);
+  }
+
+  /**
+   * 更新代理
+   */
+  updateAgent({ fileName, scope, projectPath, name, description, tools, model, permissionMode, skills, systemPrompt }) {
+    const baseDir = scope === 'user'
+      ? this.userAgentsDir
+      : path.join(projectPath, '.claude', 'agents');
+
+    const filePath = path.join(baseDir, `${fileName}.md`);
+
+    if (!fs.existsSync(filePath)) {
+      throw new Error(`代理 "${fileName}" 不存在`);
+    }
+
+    // 生成文件内容
+    const frontmatterData = {
+      name: name || fileName,
+      description: description || ''
+    };
+    if (tools) frontmatterData.tools = tools;
+    if (model) frontmatterData.model = model;
+    if (permissionMode) frontmatterData.permissionMode = permissionMode;
+    if (skills) frontmatterData.skills = skills;
+
+    const content = generateFrontmatter(frontmatterData) + '\n\n' + (systemPrompt || '');
+
+    fs.writeFileSync(filePath, content, 'utf-8');
+
+    return this.getAgent(fileName, scope, projectPath);
+  }
+
+  /**
+   * 删除代理
+   */
+  deleteAgent(fileName, scope, projectPath = null) {
+    const baseDir = scope === 'user'
+      ? this.userAgentsDir
+      : path.join(projectPath, '.claude', 'agents');
+
+    const filePath = path.join(baseDir, `${fileName}.md`);
+
+    if (!fs.existsSync(filePath)) {
+      return { success: false, message: '代理不存在' };
+    }
+
+    fs.unlinkSync(filePath);
+
+    return { success: true, message: '代理已删除' };
+  }
+
+  /**
+   * 获取统计信息
+   */
+  getStats(projectPath = null) {
+    const { agents, userCount, projectCount } = this.listAgents(projectPath);
+
+    // 按模型分组
+    const models = {};
+    for (const agent of agents) {
+      const m = agent.model || 'default';
+      if (!models[m]) {
+        models[m] = 0;
+      }
+      models[m]++;
+    }
+
+    return {
+      total: agents.length,
+      userCount,
+      projectCount,
+      models
+    };
+  }
+}
+
+module.exports = {
+  AgentsService
+};

package/src/server/services/alias.js

@@ -0,0 +1,71 @@
+const fs = require('fs');
+const path = require('path');
+const { PATHS } = require('../../config/paths');
+
+const ALIAS_DIR = PATHS.base;
+const ALIAS_FILE = PATHS.aliases;
+
+// Ensure alias directory exists
+function ensureAliasDir() {
+  if (!fs.existsSync(ALIAS_DIR)) {
+    fs.mkdirSync(ALIAS_DIR, { recursive: true });
+  }
+}
+
+// Load all aliases
+function loadAliases() {
+  ensureAliasDir();
+
+  if (!fs.existsSync(ALIAS_FILE)) {
+    return {};
+  }
+
+  try {
+    const content = fs.readFileSync(ALIAS_FILE, 'utf8');
+    return JSON.parse(content);
+  } catch (error) {
+    console.error('Error loading aliases:', error);
+    return {};
+  }
+}
+
+// Save aliases
+function saveAliases(aliases) {
+  ensureAliasDir();
+
+  try {
+    fs.writeFileSync(ALIAS_FILE, JSON.stringify(aliases, null, 2), 'utf8');
+  } catch (error) {
+    console.error('Error saving aliases:', error);
+    throw error;
+  }
+}
+
+// Set alias for a session
+function setAlias(sessionId, alias) {
+  const aliases = loadAliases();
+  aliases[sessionId] = alias;
+  saveAliases(aliases);
+  return aliases;
+}
+
+// Delete alias
+function deleteAlias(sessionId) {
+  const aliases = loadAliases();
+  delete aliases[sessionId];
+  saveAliases(aliases);
+  return aliases;
+}
+
+// Get alias for a session
+function getAlias(sessionId) {
+  const aliases = loadAliases();
+  return aliases[sessionId] || null;
+}
+
+module.exports = {
+  loadAliases,
+  setAlias,
+  deleteAlias,
+  getAlias
+};

package/src/server/services/channel-health.js

@@ -0,0 +1,234 @@
+// 渠道健康检查和智能切换模块
+
+const healthConfig = {
+  // 故障检测
+  failureThreshold: 3,           // 连续失败3次触发冻结
+
+  // 冻结时间配置
+  initialFreezeTime: 60 * 1000,  // 初始冻结1分钟
+  maxFreezeTime: 30 * 60 * 1000, // 最大冻结30分钟
+  freezeMultiplier: 2,           // 冻结时间倍增
+
+  // 健康检测
+  healthCheckWindow: 5,          // 健康检测需要连续5次成功
+};
+
+// 渠道健康状态
+const channelHealth = new Map(); // `${source}:${channelId}` → health info
+
+// 冻结回调（用于通知调度器解绑会话）
+let onChannelFrozenCallback = null;
+
+/**
+ * 设置渠道冻结时的回调
+ */
+function setOnChannelFrozen(callback) {
+  onChannelFrozenCallback = callback;
+}
+
+/**
+ * 初始化渠道健康信息
+ */
+function makeKey(source, channelId) {
+  return `${source || 'claude'}:${channelId}`;
+}
+
+function initChannelHealth(channelId, source = 'claude') {
+  const key = makeKey(source, channelId);
+  if (!channelHealth.has(key)) {
+    channelHealth.set(key, {
+      status: 'healthy',           // healthy, frozen, checking
+      consecutiveFailures: 0,      // 连续失败次数
+      consecutiveSuccesses: 0,     // 连续成功次数
+      totalFailures: 0,           // 总失败次数
+      totalSuccesses: 0,          // 总成功次数
+      freezeUntil: 0,             // 冻结到期时间
+      nextFreezeTime: healthConfig.initialFreezeTime,
+      lastCheckTime: null,        // 最后检查时间
+      source
+    });
+  }
+  return channelHealth.get(key);
+}
+
+/**
+ * 记录成功请求
+ */
+function recordSuccess(channelId, source = 'claude') {
+  const health = initChannelHealth(channelId, source);
+  const now = Date.now();
+
+  health.totalSuccesses++;
+  health.consecutiveSuccesses++;
+  health.consecutiveFailures = 0;
+  health.lastCheckTime = now;
+
+  // 如果在检测中状态，检查是否可以恢复
+  if (health.status === 'checking') {
+    if (health.consecutiveSuccesses >= healthConfig.healthCheckWindow) {
+      // 恢复健康状态
+      health.status = 'healthy';
+      health.nextFreezeTime = healthConfig.initialFreezeTime; // 重置冻结时间
+      console.log(`[ChannelHealth] Channel ${channelId} recovered and marked as healthy`);
+    }
+  }
+}
+
+/**
+ * 记录失败请求
+ */
+function recordFailure(channelId, source = 'claude', error) {
+  const health = initChannelHealth(channelId, source);
+  const now = Date.now();
+
+  health.totalFailures++;
+  health.consecutiveFailures++;
+  health.consecutiveSuccesses = 0;
+  health.lastCheckTime = now;
+
+  // 如果当前是健康状态或检测中状态，检查是否需要冻结
+  if (health.status === 'healthy' || health.status === 'checking') {
+    if (health.consecutiveFailures >= healthConfig.failureThreshold) {
+      // 触发冻结
+      const previousStatus = health.status;
+      health.status = 'frozen';
+      health.freezeUntil = now + health.nextFreezeTime;
+
+      const freezeMinutes = Math.round(health.nextFreezeTime / 60000);
+      console.warn(`[ChannelHealth] Channel ${channelId} frozen due to ${health.consecutiveFailures} consecutive failures (was ${previousStatus}). Frozen for ${freezeMinutes} minutes`);
+
+      // 更新下次冻结时间（翻倍，不超过最大值）
+      health.nextFreezeTime = Math.min(
+        health.nextFreezeTime * healthConfig.freezeMultiplier,
+        healthConfig.maxFreezeTime
+      );
+
+      // 触发冻结回调（通知调度器解绑会话）
+      if (onChannelFrozenCallback) {
+        onChannelFrozenCallback(source || 'claude', channelId);
+      }
+    }
+  }
+}
+
+/**
+ * 检查渠道是否可用
+ */
+function isChannelAvailable(channelId, source = 'claude') {
+  const key = makeKey(source, channelId);
+  const health = channelHealth.get(key);
+  if (!health) return true;
+
+  const now = Date.now();
+
+  switch (health.status) {
+    case 'healthy':
+      return true;
+
+    case 'frozen':
+      // 检查冻结时间是否到期
+      if (now >= health.freezeUntil) {
+        // 进入检测状态
+        health.status = 'checking';
+        health.consecutiveSuccesses = 0;
+        console.log(`[ChannelHealth] Channel ${channelId} freeze expired, entering checking mode`);
+        return true; // 允许一个请求用于健康检测
+      }
+      return false;
+
+    case 'checking':
+      // 在检测中的渠道可用，等待成功记录
+      return true;
+
+    default:
+      return true;
+  }
+}
+
+/**
+ * 从渠道列表中过滤出可用的渠道
+ */
+function getAvailableChannels(channels, source = 'claude') {
+  return channels.filter(channel => isChannelAvailable(channel.id, source));
+}
+
+/**
+ * 获取渠道健康状态（用于前端显示）
+ */
+function getChannelHealthStatus(channelId, source = 'claude') {
+  const key = makeKey(source, channelId);
+  const health = channelHealth.get(key);
+  if (!health) {
+    return {
+      status: 'healthy',
+      statusText: '健康',
+      statusColor: '#18a058',
+      consecutiveFailures: 0,
+      consecutiveSuccesses: 0,
+      totalFailures: 0,
+      totalSuccesses: 0,
+      freezeUntil: null,
+      freezeRemaining: 0,
+    };
+  }
+
+  const now = Date.now();
+  const freezeRemaining = Math.max(0, health.freezeUntil - now);
+
+  const statusMap = {
+    'healthy': { text: '健康', color: '#18a058' },
+    'frozen': { text: '冻结', color: '#d03050' },
+    'checking': { text: '检测中', color: '#f0a020' }
+  };
+
+  return {
+    status: health.status,
+    statusText: statusMap[health.status]?.text || '未知',
+    statusColor: statusMap[health.status]?.color || '#909399',
+    consecutiveFailures: health.consecutiveFailures,
+    consecutiveSuccesses: health.consecutiveSuccesses,
+    totalFailures: health.totalFailures,
+    totalSuccesses: health.totalSuccesses,
+    freezeUntil: health.freezeUntil,
+    freezeRemaining: Math.ceil(freezeRemaining / 1000), // 剩余秒数
+  };
+}
+
+/**
+ * 获取所有渠道的健康状态
+ */
+function getAllChannelHealthStatus(source = 'claude') {
+  const result = {};
+  for (const [key] of channelHealth) {
+    const [keySource, channelId] = key.split(':');
+    if (keySource === (source || 'claude')) {
+      result[channelId] = getChannelHealthStatus(channelId, keySource);
+    }
+  }
+  return result;
+}
+
+/**
+ * 手动重置渠道健康状态（用于测试或管理员操作）
+ */
+function resetChannelHealth(channelId, source = 'claude') {
+  const health = initChannelHealth(channelId, source);
+  health.status = 'healthy';
+  health.consecutiveFailures = 0;
+  health.consecutiveSuccesses = 0;
+  health.freezeUntil = 0;
+  health.nextFreezeTime = healthConfig.initialFreezeTime;
+  console.log(`[ChannelHealth] Channel ${channelId} health status reset`);
+}
+
+module.exports = {
+  recordSuccess,
+  recordFailure,
+  isChannelAvailable,
+  getAvailableChannels,
+  getChannelHealthStatus,
+  getAllChannelHealthStatus,
+  resetChannelHealth,
+  setOnChannelFrozen,
+  healthConfig,
+};