coding-tool-x 3.2.0

package/src/server/api/gemini-channels.js

@@ -0,0 +1,366 @@
+const express = require('express');
+const router = express.Router();
+const {
+  getChannels,
+  createChannel,
+  updateChannel,
+  deleteChannel,
+  getEnabledChannels,
+  saveChannelOrder
+} = require('../services/gemini-channels');
+const { getSchedulerState } = require('../services/channel-scheduler');
+const { getChannelHealthStatus, resetChannelHealth } = require('../services/channel-health');
+const { broadcastSchedulerState } = require('../websocket-server');
+const { isGeminiInstalled } = require('../services/gemini-config');
+const {
+  testChannelSpeed,
+  getLatencyLevel,
+  sanitizeBatchConcurrency,
+  runWithConcurrencyLimit
+} = require('../services/speed-test');
+const { clearGeminiRedirectCache } = require('../gemini-proxy-server');
+const { getDefaultSpeedTestModelByToolType } = require('../../config/model-metadata');
+const GEMINI_GATEWAY_SOURCE_TYPE = 'gemini';
+
+function getDefaultGeminiModel() {
+  return getDefaultSpeedTestModelByToolType('gemini');
+}
+
+module.exports = (config) => {
+  /**
+   * GET /api/gemini/channels
+   * 获取所有 Gemini 渠道（包含健康状态）
+   */
+  router.get('/', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.json({
+          channels: [],
+          error: 'Gemini CLI not installed'
+        });
+      }
+
+      const data = getChannels();
+      // 为每个渠道添加健康状态
+      const channelsWithHealth = (data.channels || []).map(ch => ({
+        ...ch,
+        health: getChannelHealthStatus(ch.id, 'gemini')
+      }));
+      res.json({ channels: channelsWithHealth });
+    } catch (err) {
+      console.error('[Gemini Channels API] Failed to get channels:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * GET /api/gemini/channels/:id/models
+   * 获取渠道可用模型列表
+   */
+  router.get('/:id/models', async (req, res) => {
+    try {
+      const { id } = req.params;
+      const channels = getChannels().channels || [];
+      const channel = channels.find(ch => ch.id === id);
+
+      if (!channel) {
+        return res.status(404).json({ error: '渠道不存在' });
+      }
+
+      const gatewaySourceType = GEMINI_GATEWAY_SOURCE_TYPE;
+      const models = [getDefaultGeminiModel()];
+      const result = {
+        models,
+        supported: models.length > 0,
+        cached: false,
+        fallbackUsed: false,
+        lastChecked: new Date().toISOString(),
+        error: models.length > 0 ? null : '未配置默认模型列表',
+        errorHint: models.length > 0 ? null : '请在设置中配置 Gemini 默认模型'
+      };
+
+      res.json({
+        channelId: id,
+        gatewaySourceType,
+        models: result.models,
+        supported: result.supported,
+        cached: result.cached,
+        fallbackUsed: result.fallbackUsed,
+        fetchedAt: result.lastChecked || new Date().toISOString(),
+        error: result.error,
+        errorHint: result.errorHint
+      });
+    } catch (error) {
+      console.error('[Gemini Channels API] Error fetching models:', error);
+      res.status(500).json({
+        error: '获取模型列表失败',
+        channelId: req.params.id
+      });
+    }
+  });
+
+  /**
+   * POST /api/gemini/channels
+   * 创建新渠道
+   * Body: { name, baseUrl, apiKey, model, websiteUrl }
+   */
+  router.post('/', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const {
+        name,
+        baseUrl,
+        apiKey,
+        model,
+        websiteUrl,
+        enabled,
+        weight,
+        maxConcurrency,
+        modelRedirects,
+        speedTestModel,
+        presetId,
+        gatewaySourceType
+      } = req.body;
+
+      if (!name || !baseUrl) {
+        return res.status(400).json({ error: 'Missing required fields: name, baseUrl' });
+      }
+
+      if (!apiKey) {
+        return res.status(400).json({ error: 'Missing required fields: apiKey' });
+      }
+
+      const channel = createChannel(name, baseUrl, apiKey, model || getDefaultGeminiModel(), {
+        websiteUrl,
+        enabled,
+        weight,
+        maxConcurrency,
+        modelRedirects: modelRedirects || [],
+        speedTestModel: speedTestModel || null,
+        presetId: presetId || null,
+        gatewaySourceType
+      });
+      res.json(channel);
+      broadcastSchedulerState('gemini', getSchedulerState('gemini'));
+    } catch (err) {
+      console.error('[Gemini Channels API] Failed to create channel:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * PUT /api/gemini/channels/:channelId
+   * 更新渠道
+   */
+  router.put('/:channelId', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { channelId } = req.params;
+      const updates = req.body;
+
+      const channel = updateChannel(channelId, updates);
+      // 清除该渠道的模型重定向日志缓存，使下次请求时重新打印
+      clearGeminiRedirectCache(channelId);
+      res.json(channel);
+      broadcastSchedulerState('gemini', getSchedulerState('gemini'));
+    } catch (err) {
+      console.error('[Gemini Channels API] Failed to update channel:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * DELETE /api/gemini/channels/:channelId
+   * 删除渠道
+   */
+  router.delete('/:channelId', async (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { channelId } = req.params;
+      const result = await deleteChannel(channelId);
+      res.json(result);
+      broadcastSchedulerState('gemini', getSchedulerState('gemini'));
+    } catch (err) {
+      console.error('[Gemini Channels API] Failed to delete channel:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * POST /api/gemini/channels/order
+   * 保存渠道顺序
+   */
+  router.post('/order', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { order } = req.body;
+
+      if (!Array.isArray(order)) {
+        return res.status(400).json({ error: 'order must be an array' });
+      }
+
+      saveChannelOrder(order);
+      res.json({ success: true });
+    } catch (err) {
+      console.error('[Gemini Channels API] Failed to save channel order:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * GET /api/gemini/channels/enabled
+   * 获取所有启用的渠道（供调度器使用）
+   */
+  router.get('/enabled', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.json({ channels: [] });
+      }
+
+      const channels = getEnabledChannels();
+      res.json({ channels });
+    } catch (err) {
+      console.error('[Gemini Channels API] Failed to get enabled channels:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * POST /api/gemini/channels/:channelId/speed-test
+   * 测试单个渠道速度
+   */
+  router.post('/:channelId/speed-test', async (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { channelId } = req.params;
+      const { timeout = 10000 } = req.body;
+      const data = getChannels();
+      const channel = data.channels.find(ch => ch.id === channelId);
+
+      if (!channel) {
+        return res.status(404).json({ error: '渠道不存在' });
+      }
+
+      const speedTestType = GEMINI_GATEWAY_SOURCE_TYPE;
+      const result = await testChannelSpeed(channel, timeout, speedTestType);
+      result.level = getLatencyLevel(result.latency);
+      result.gatewaySourceType = speedTestType;
+
+      res.json(result);
+    } catch (error) {
+      console.error('[Gemini Channels API] Error testing channel speed:', error);
+      res.status(500).json({ error: error.message });
+    }
+  });
+
+  /**
+   * POST /api/gemini/channels/speed-test-all
+   * 测试所有渠道速度
+   */
+  router.post('/speed-test-all', async (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.json({ results: [], message: 'Gemini CLI not installed' });
+      }
+
+      const { timeout = 10000, concurrency } = req.body || {};
+      const data = getChannels();
+      const channels = data.channels || [];
+      const safeConcurrency = sanitizeBatchConcurrency(concurrency);
+
+      if (channels.length === 0) {
+        return res.json({ results: [], message: '没有可测试的渠道' });
+      }
+
+      const results = await runWithConcurrencyLimit(
+        channels,
+        safeConcurrency,
+        async channel => {
+          const speedTestType = GEMINI_GATEWAY_SOURCE_TYPE;
+          const result = await testChannelSpeed(channel, timeout, speedTestType);
+          result.level = getLatencyLevel(result.latency);
+          result.gatewaySourceType = speedTestType;
+          return result;
+        }
+      );
+
+      // 成功在前，成功结果按延迟升序
+      results.sort((a, b) => {
+        if (a.success && !b.success) return -1;
+        if (!a.success && b.success) return 1;
+        if (a.success && b.success) {
+          const aLatency = (a.latency === null || a.latency === undefined) ? Infinity : a.latency;
+          const bLatency = (b.latency === null || b.latency === undefined) ? Infinity : b.latency;
+          return aLatency - bLatency;
+        }
+        return 0;
+      });
+
+      res.json({
+        results,
+        summary: {
+          total: results.length,
+          success: results.filter(r => r.success).length,
+          failed: results.filter(r => !r.success).length,
+          avgLatency: calculateAvgLatency(results),
+          concurrency: safeConcurrency
+        }
+      });
+    } catch (error) {
+      console.error('[Gemini Channels API] Error testing all channels speed:', error);
+      res.status(500).json({ error: error.message });
+    }
+  });
+
+  /**
+   * POST /api/gemini/channels/:channelId/reset-health
+   * 重置渠道健康状态
+   */
+  router.post('/:channelId/reset-health', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { channelId } = req.params;
+      resetChannelHealth(channelId, 'gemini');
+      broadcastSchedulerState('gemini', getSchedulerState('gemini'));
+
+      res.json({
+        success: true,
+        message: '渠道健康状态已重置',
+        health: getChannelHealthStatus(channelId, 'gemini')
+      });
+    } catch (error) {
+      console.error('[Gemini Channels API] Error resetting channel health:', error);
+      res.status(500).json({ error: error.message });
+    }
+  });
+
+  return router;
+};
+
+// 计算平均延迟
+function calculateAvgLatency(results) {
+  const successResults = results.filter(
+    r => r.success && r.latency !== null && r.latency !== undefined
+  );
+  if (successResults.length === 0) return null;
+  const sum = successResults.reduce((acc, r) => acc + r.latency, 0);
+  return Math.round(sum / successResults.length);
+}

package/src/server/api/gemini-projects.js

@@ -0,0 +1,91 @@
+const express = require('express');
+const router = express.Router();
+const { getProjects, saveProjectOrder, deleteProject } = require('../services/gemini-sessions');
+const { isGeminiInstalled } = require('../services/gemini-config');
+
+module.exports = (config) => {
+  /**
+   * GET /api/gemini/projects
+   * 获取所有 Gemini 项目列表
+   */
+  router.get('/', (req, res) => {
+    try {
+      // 检查 Gemini 是否安装
+      if (!isGeminiInstalled()) {
+        return res.json({
+          projects: [],
+          currentProject: null,
+          error: 'Gemini CLI not installed or not found'
+        });
+      }
+
+      const projects = getProjects();
+
+      res.json({
+        projects,
+        currentProject: projects[0] ? projects[0].name : null
+      });
+    } catch (err) {
+      console.error('[Gemini API] Failed to get projects:', err);
+
+      if (err.code === 'ENOENT') {
+        return res.status(404).json({
+          error: 'Gemini tmp directory not found',
+          projects: []
+        });
+      }
+
+      res.status(500).json({
+        error: err.message,
+        projects: []
+      });
+    }
+  });
+
+  /**
+   * POST /api/gemini/projects/order
+   * 保存项目排序
+   */
+  router.post('/order', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { order } = req.body;
+
+      if (!Array.isArray(order)) {
+        return res.status(400).json({ error: 'order must be an array' });
+      }
+
+      saveProjectOrder(order);
+
+      res.json({ success: true });
+    } catch (err) {
+      console.error('[Gemini API] Failed to save project order:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  /**
+   * DELETE /api/gemini/projects/:projectHash
+   * 删除项目（删除项目下所有会话）
+   */
+  router.delete('/:projectHash', (req, res) => {
+    try {
+      if (!isGeminiInstalled()) {
+        return res.status(404).json({ error: 'Gemini CLI not installed' });
+      }
+
+      const { projectHash } = req.params;
+      const result = deleteProject(projectHash);
+
+      res.json(result);
+    } catch (err) {
+      console.error('[Gemini API] Failed to delete project:', err);
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  return router;
+};

package/src/server/api/gemini-proxy.js

@@ -0,0 +1,173 @@
+const express = require('express');
+const router = express.Router();
+const {
+  startGeminiProxyServer,
+  stopGeminiProxyServer,
+  getGeminiProxyStatus
+} = require('../gemini-proxy-server');
+const {
+  setProxyConfig,
+  restoreSettings,
+  isProxyConfig,
+  getCurrentProxyPort,
+  configExists,
+  hasBackup
+} = require('../services/gemini-settings-manager');
+const { getChannels, getEnabledChannels } = require('../services/gemini-channels');
+const { PATHS, ensureStorageDirMigrated } = require('../../config/paths');
+const fs = require('fs');
+const path = require('path');
+
+function sanitizeChannel(channel) {
+  if (!channel) return null;
+  const { apiKey, ...rest } = channel;
+  return rest;
+}
+
+// 保存激活渠道ID
+function saveActiveChannelId(channelId) {
+  ensureStorageDirMigrated();
+  const filePath = PATHS.activeChannel.gemini;
+  const dir = path.dirname(filePath);
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+  fs.writeFileSync(filePath, JSON.stringify({ activeChannelId: channelId }, null, 2), 'utf8');
+}
+
+function removeActiveChannelFile() {
+  ensureStorageDirMigrated();
+  const filePath = PATHS.activeChannel.gemini;
+  if (fs.existsSync(filePath)) {
+    fs.unlinkSync(filePath);
+    console.log('[Gemini Proxy] Removed gemini-active-channel.json');
+  }
+}
+
+// 获取代理状态
+router.get('/status', (req, res) => {
+  try {
+    const proxyStatus = getGeminiProxyStatus();
+    const configStatus = {
+      isProxyConfig: isProxyConfig(),
+      configExists: configExists(),
+      hasBackup: hasBackup(),
+      currentProxyPort: getCurrentProxyPort()
+    };
+    const { channels } = getChannels();
+    const enabledChannels = channels.filter(ch => ch.enabled !== false);
+    const activeChannel = enabledChannels[0]; // 多渠道模式：第一个启用的渠道
+
+    res.json({
+      proxy: proxyStatus,
+      config: configStatus,
+      activeChannel: sanitizeChannel(activeChannel)
+    });
+  } catch (error) {
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// 启动代理
+router.post('/start', async (req, res) => {
+  try {
+    // 1. 检查 Gemini 配置文件是否存在
+    if (!configExists()) {
+      return res.status(400).json({
+        error: 'Gemini .env not found. Please run Gemini CLI at least once or create ~/.gemini/.env manually.'
+      });
+    }
+
+    // 2. 获取当前启用的渠道（多渠道模式）
+    const enabledChannels = getEnabledChannels();
+    const currentChannel = enabledChannels[0];
+    if (!currentChannel) {
+      return res.status(400).json({
+        error: 'No enabled Gemini channel found. Please create and enable a channel first.'
+      });
+    }
+
+    // 3. 保存当前激活渠道ID（用于代理模式）
+    saveActiveChannelId(currentChannel.id);
+    console.log(`[Gemini Proxy] Saved active channel: ${currentChannel.name} (${currentChannel.id})`);
+
+    // 4. 启动代理服务器
+    const proxyResult = await startGeminiProxyServer();
+
+    if (!proxyResult.success) {
+      return res.status(500).json({ error: 'Failed to start Gemini proxy server' });
+    }
+
+    // 5. 设置代理配置（备份并修改 .env 和 settings.json）
+    setProxyConfig(proxyResult.port);
+
+    const { broadcastProxyState } = require('../websocket-server');
+    const proxyStatus = getGeminiProxyStatus();
+    const { channels: allChannels } = getChannels();
+    const activeChannel = allChannels.filter(ch => ch.enabled !== false)[0];
+    broadcastProxyState('gemini', proxyStatus, activeChannel, allChannels);
+
+    res.json({
+      success: true,
+      port: proxyResult.port,
+      activeChannel: sanitizeChannel(currentChannel),
+      message: `Gemini proxy started on port ${proxyResult.port}, active channel: ${currentChannel.name}`
+    });
+  } catch (error) {
+    console.error('[Gemini Proxy] Error starting proxy:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// 停止代理
+router.post('/stop', async (req, res) => {
+  try {
+    // 1. 获取当前启用的渠道（多渠道模式）
+    const { channels } = getChannels();
+    const enabledChannels = channels.filter(ch => ch.enabled !== false);
+    const activeChannel = enabledChannels[0];
+
+    // 2. 停止代理服务器
+    const proxyResult = await stopGeminiProxyServer();
+
+    // 3. 恢复原始配置
+    const { broadcastProxyState } = require('../websocket-server');
+    if (hasBackup()) {
+      restoreSettings();
+      console.log('[Gemini Proxy] Restored settings from backup');
+
+      // Enforce single-channel mode: apply the active channel and disable all others
+      if (activeChannel) {
+        const { applyChannelToSettings } = require('../services/gemini-channels');
+        applyChannelToSettings(activeChannel.id);
+        console.log(`[Gemini Proxy] Single-channel mode enforced: ${activeChannel.name}`);
+      }
+
+      // 删除 gemini-active-channel.json
+      removeActiveChannelFile();
+
+      const response = {
+        success: true,
+        message: `Gemini proxy stopped, settings restored${activeChannel ? ' (channel: ' + activeChannel.name + ')' : ''}`,
+        port: proxyResult.port,
+        restoredChannel: activeChannel?.name
+      };
+      res.json(response);
+    } else {
+      res.json({
+        success: true,
+        message: 'Gemini proxy stopped (no backup to restore)',
+        port: proxyResult.port
+      });
+    }
+
+    const proxyStatus = getGeminiProxyStatus();
+    const { channels: latestChannels } = getChannels();
+    broadcastProxyState('gemini', proxyStatus, activeChannel, latestChannels);
+  } catch (error) {
+    console.error('[Gemini Proxy] Error stopping proxy:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
+module.exports = router;