@adversity/coding-tool-x 2.2.0

package/src/server/api/agents.js

@@ -0,0 +1,248 @@
+/**
+ * Agents API 路由
+ *
+ * 管理 Claude Code 自定义代理
+ */
+
+const express = require('express');
+const { AgentsService } = require('../services/agents-service');
+
+const router = express.Router();
+const agentsService = new AgentsService();
+
+/**
+ * 获取代理列表
+ * GET /api/agents
+ * Query: projectPath - 项目路径（可选，用于获取项目级代理）
+ */
+router.get('/', (req, res) => {
+  try {
+    const { projectPath } = req.query;
+    const result = agentsService.listAgents(projectPath || null);
+
+    res.json({
+      success: true,
+      ...result
+    });
+  } catch (err) {
+    console.error('[Agents API] List agents error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    });
+  }
+});
+
+/**
+ * 获取代理统计
+ * GET /api/agents/stats
+ */
+router.get('/stats', (req, res) => {
+  try {
+    const { projectPath } = req.query;
+    const stats = agentsService.getStats(projectPath || null);
+
+    res.json({
+      success: true,
+      ...stats
+    });
+  } catch (err) {
+    console.error('[Agents API] Get stats error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    });
+  }
+});
+
+/**
+ * 获取单个代理详情
+ * GET /api/agents/:scope/:fileName
+ */
+router.get('/:scope/:fileName', (req, res) => {
+  try {
+    const { scope, fileName } = req.params;
+    const { projectPath } = req.query;
+
+    if (!['user', 'project'].includes(scope)) {
+      return res.status(400).json({
+        success: false,
+        message: '无效的 scope，必须是 user 或 project'
+      });
+    }
+
+    if (scope === 'project' && !projectPath) {
+      return res.status(400).json({
+        success: false,
+        message: '获取项目级代理需要提供 projectPath'
+      });
+    }
+
+    const agent = agentsService.getAgent(fileName, scope, projectPath || null);
+
+    if (!agent) {
+      return res.status(404).json({
+        success: false,
+        message: `代理 "${fileName}" 不存在`
+      });
+    }
+
+    res.json({
+      success: true,
+      agent
+    });
+  } catch (err) {
+    console.error('[Agents API] Get agent error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    });
+  }
+});
+
+/**
+ * 创建代理
+ * POST /api/agents
+ * Body: { fileName, scope, projectPath?, name, description, tools?, model?, permissionMode?, skills?, systemPrompt? }
+ */
+router.post('/', (req, res) => {
+  try {
+    const { fileName, scope, projectPath, name, description, tools, model, permissionMode, skills, systemPrompt } = req.body;
+
+    if (!fileName) {
+      return res.status(400).json({
+        success: false,
+        message: '代理文件名不能为空'
+      });
+    }
+
+    if (!['user', 'project'].includes(scope)) {
+      return res.status(400).json({
+        success: false,
+        message: '无效的 scope，必须是 user 或 project'
+      });
+    }
+
+    if (scope === 'project' && !projectPath) {
+      return res.status(400).json({
+        success: false,
+        message: '创建项目级代理需要提供 projectPath'
+      });
+    }
+
+    const agent = agentsService.createAgent({
+      fileName,
+      scope,
+      projectPath: projectPath || null,
+      name: name || fileName,
+      description: description || '',
+      tools: tools || '',
+      model: model || '',
+      permissionMode: permissionMode || '',
+      skills: skills || '',
+      systemPrompt: systemPrompt || ''
+    });
+
+    res.json({
+      success: true,
+      agent,
+      message: '代理创建成功'
+    });
+  } catch (err) {
+    console.error('[Agents API] Create agent error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    });
+  }
+});
+
+/**
+ * 更新代理
+ * PUT /api/agents/:scope/:fileName
+ */
+router.put('/:scope/:fileName', (req, res) => {
+  try {
+    const { scope, fileName } = req.params;
+    const { projectPath, name, description, tools, model, permissionMode, skills, systemPrompt } = req.body;
+
+    if (!['user', 'project'].includes(scope)) {
+      return res.status(400).json({
+        success: false,
+        message: '无效的 scope，必须是 user 或 project'
+      });
+    }
+
+    if (scope === 'project' && !projectPath) {
+      return res.status(400).json({
+        success: false,
+        message: '更新项目级代理需要提供 projectPath'
+      });
+    }
+
+    const agent = agentsService.updateAgent({
+      fileName,
+      scope,
+      projectPath: projectPath || null,
+      name: name || fileName,
+      description: description || '',
+      tools: tools || '',
+      model: model || '',
+      permissionMode: permissionMode || '',
+      skills: skills || '',
+      systemPrompt: systemPrompt || ''
+    });
+
+    res.json({
+      success: true,
+      agent,
+      message: '代理更新成功'
+    });
+  } catch (err) {
+    console.error('[Agents API] Update agent error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    });
+  }
+});
+
+/**
+ * 删除代理
+ * DELETE /api/agents/:scope/:fileName
+ */
+router.delete('/:scope/:fileName', (req, res) => {
+  try {
+    const { scope, fileName } = req.params;
+    const { projectPath } = req.query;
+
+    if (!['user', 'project'].includes(scope)) {
+      return res.status(400).json({
+        success: false,
+        message: '无效的 scope，必须是 user 或 project'
+      });
+    }
+
+    if (scope === 'project' && !projectPath) {
+      return res.status(400).json({
+        success: false,
+        message: '删除项目级代理需要提供 projectPath'
+      });
+    }
+
+    const result = agentsService.deleteAgent(fileName, scope, projectPath || null);
+
+    res.json({
+      success: result.success,
+      message: result.message
+    });
+  } catch (err) {
+    console.error('[Agents API] Delete agent error:', err);
+    res.status(500).json({
+      success: false,
+      message: err.message
+    });
+  }
+});
+
+module.exports = router;

package/src/server/api/aliases.js

@@ -0,0 +1,36 @@
+const express = require('express');
+const router = express.Router();
+const { setAlias, deleteAlias } = require('../services/alias');
+
+module.exports = () => {
+  // POST /api/aliases - Set alias for a session
+  router.post('/', (req, res) => {
+    try {
+      const { sessionId, alias } = req.body;
+
+      if (!sessionId || !alias) {
+        return res.status(400).json({ error: 'sessionId and alias are required' });
+      }
+
+      const aliases = setAlias(sessionId, alias);
+      res.json({ success: true, aliases });
+    } catch (error) {
+      console.error('Error setting alias:', error);
+      res.status(500).json({ error: error.message });
+    }
+  });
+
+  // DELETE /api/aliases/:sessionId - Delete alias
+  router.delete('/:sessionId', (req, res) => {
+    try {
+      const { sessionId } = req.params;
+      const aliases = deleteAlias(sessionId);
+      res.json({ success: true, aliases });
+    } catch (error) {
+      console.error('Error deleting alias:', error);
+      res.status(500).json({ error: error.message });
+    }
+  });
+
+  return router;
+};

package/src/server/api/channels.js

@@ -0,0 +1,258 @@
+const express = require('express');
+const router = express.Router();
+const {
+  getAllChannels,
+  applyChannelToSettings,
+  createChannel,
+  updateChannel,
+  deleteChannel,
+  getCurrentSettings,
+  getBestChannelForRestore
+} = require('../services/channels');
+const { getSchedulerState } = require('../services/channel-scheduler');
+const { getChannelHealthStatus, getAllChannelHealthStatus, resetChannelHealth } = require('../services/channel-health');
+const { testChannelSpeed, testMultipleChannels, getLatencyLevel } = require('../services/speed-test');
+const { broadcastLog, broadcastProxyState, broadcastSchedulerState } = require('../websocket-server');
+
+// GET /api/channels - Get all channels with health status
+router.get('/', (req, res) => {
+  try {
+    const channels = getAllChannels();
+    // 为每个渠道附加健康状态
+    const channelsWithHealth = channels.map(ch => ({
+      ...ch,
+      health: getChannelHealthStatus(ch.id)
+    }));
+    res.json({ channels: channelsWithHealth });
+  } catch (error) {
+    console.error('Error fetching channels:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+router.get('/pool/status', (req, res) => {
+  try {
+    const source = req.query.source || 'claude';
+    const scheduler = getSchedulerState(source);
+    res.json({ source, scheduler });
+  } catch (error) {
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// GET /api/channels/current - Get current settings
+router.get('/current', (req, res) => {
+  try {
+    const settings = getCurrentSettings();
+    const channels = getAllChannels();
+    let currentChannel = null;
+
+    if (settings) {
+      currentChannel = channels.find(ch =>
+        ch.baseUrl === settings.baseUrl && ch.apiKey === settings.apiKey
+      );
+    }
+
+    res.json({ channel: currentChannel, settings });
+  } catch (error) {
+    console.error('Error fetching current settings:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// POST /api/channels - Create new channel
+router.post('/', (req, res) => {
+  try {
+    const { name, baseUrl, apiKey, websiteUrl, enabled, weight, maxConcurrency } = req.body;
+
+    if (!name || !baseUrl || !apiKey) {
+      return res.status(400).json({ error: 'Missing required fields' });
+    }
+
+    const channel = createChannel(name, baseUrl, apiKey, websiteUrl, {
+      enabled,
+      weight,
+      maxConcurrency
+    });
+    res.json({ channel });
+    broadcastSchedulerState('claude', getSchedulerState('claude'));
+  } catch (error) {
+    console.error('Error creating channel:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// GET /api/channels/best-for-restore - Get best channel for restore (must be before /:id)
+router.get('/best-for-restore', (req, res) => {
+  try {
+    const channel = getBestChannelForRestore();
+    res.json({ channel });
+  } catch (error) {
+    console.error('Error getting best channel for restore:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// PUT /api/channels/:id - Update channel
+router.put('/:id', (req, res) => {
+  try {
+    const { id } = req.params;
+    const updates = req.body;
+
+    const channel = updateChannel(id, updates);
+    res.json({ channel });
+    broadcastSchedulerState('claude', getSchedulerState('claude'));
+  } catch (error) {
+    console.error('Error updating channel:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// DELETE /api/channels/:id - Delete channel
+router.delete('/:id', (req, res) => {
+  try {
+    const { id } = req.params;
+    const result = deleteChannel(id);
+    res.json(result);
+    broadcastSchedulerState('claude', getSchedulerState('claude'));
+  } catch (error) {
+    console.error('Error deleting channel:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+router.post('/:id/apply-to-settings', async (req, res) => {
+  try {
+    const { id } = req.params;
+    const channel = applyChannelToSettings(id);
+
+    // Check if proxy is running
+    const { getProxyStatus } = require('../proxy-server');
+    const proxyStatus = getProxyStatus();
+
+    broadcastLog({
+      type: 'action',
+      action: 'apply_settings',
+      message: `已将 (${channel.name}) 渠道写入配置文件中`,
+      channelName: channel.name,
+      timestamp: Date.now(),
+      source: 'claude'
+    });
+
+    // Stop proxy if running
+    if (proxyStatus && proxyStatus.running) {
+      console.log(`Proxy is running, stopping to apply channel settings: ${channel.name}`);
+
+      // Stop proxy and restore backup
+      const { stopProxyServer } = require('../proxy-server');
+      await stopProxyServer({ clearStartTime: false });
+
+      console.log(`✅ 已停���动态切换，默认使用当前渠道`);
+      broadcastLog({
+        type: 'action',
+        action: 'stop_proxy',
+        message: `已停止动态切换，默认使用当前渠道`,
+        timestamp: Date.now(),
+        source: 'claude'
+      });
+
+      // 广播代理状态更新，通知前端代理已停止
+      const { broadcastProxyState } = require('../websocket-server');
+      broadcastProxyState('claude', {
+        running: false,
+        port: null,
+        runtime: null,
+        startTime: null
+      }, null, getAllChannels());
+    }
+
+    res.json({
+      message: `已将 (${channel.name}) 渠道写入配置文件中`,
+      channel
+    });
+  } catch (error) {
+    console.error('Error applying channel to settings:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// POST /api/channels/:id/reset-health - Reset channel health status
+router.post('/:id/reset-health', (req, res) => {
+  try {
+    const { id } = req.params;
+    resetChannelHealth(id, 'claude');
+    res.json({
+      success: true,
+      message: '渠道健康状态已重置',
+      health: getChannelHealthStatus(id)
+    });
+  } catch (error) {
+    console.error('Error resetting channel health:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// POST /api/channels/:id/speed-test - Test single channel speed
+router.post('/:id/speed-test', async (req, res) => {
+  try {
+    const { id } = req.params;
+    const { timeout = 10000 } = req.body;
+    const channels = getAllChannels();
+    const channel = channels.find(ch => ch.id === id);
+
+    if (!channel) {
+      return res.status(404).json({ error: '渠道不存在' });
+    }
+
+    // Claude 渠道使用 'claude' 类型
+    const result = await testChannelSpeed(channel, timeout, 'claude');
+    result.level = getLatencyLevel(result.latency);
+
+    res.json(result);
+  } catch (error) {
+    console.error('Error testing channel speed:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// POST /api/channels/speed-test-all - Test all channels speed
+router.post('/speed-test-all', async (req, res) => {
+  try {
+    const { timeout = 10000 } = req.body;
+    const channels = getAllChannels();
+
+    if (channels.length === 0) {
+      return res.json({ results: [], message: '没有可测试的渠道' });
+    }
+
+    // Claude 渠道使用 'claude' 类型
+    const results = await testMultipleChannels(channels, timeout, 'claude');
+    // 添加延迟等级
+    results.forEach(r => {
+      r.level = getLatencyLevel(r.latency);
+    });
+
+    res.json({
+      results,
+      summary: {
+        total: results.length,
+        success: results.filter(r => r.success).length,
+        failed: results.filter(r => !r.success).length,
+        avgLatency: calculateAvgLatency(results)
+      }
+    });
+  } catch (error) {
+    console.error('Error testing all channels speed:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// 计算平均延迟
+function calculateAvgLatency(results) {
+  const successResults = results.filter(r => r.success && r.latency);
+  if (successResults.length === 0) return null;
+  const sum = successResults.reduce((acc, r) => acc + r.latency, 0);
+  return Math.round(sum / successResults.length);
+}
+
+module.exports = router;