protocol-proxy 2.6.0 → 2.8.0

package/server.js

@@ -249,6 +249,89 @@ async function init() {
     return proxyManager.startProxy(proxyConfig);
   }
 
+  // ==================== API Key 健康检查 ====================
+
+  const keyHealth = new Map(); // providerId -> { status, lastCheck, keys: [{index, ok, message}] }
+  let healthCheckRunning = false;
+
+  async function checkAllProviderKeys() {
+    if (healthCheckRunning) return;
+    healthCheckRunning = true;
+    try {
+      const providers = configStore.getProviders();
+      logger.log(`[Health] 开始检查 ${providers.length} 个供应商的 API Key...`);
+      for (const provider of providers) {
+        await checkProviderKeys(provider);
+      }
+      logger.log('[Health] API Key 健康检查完成');
+    } finally {
+      healthCheckRunning = false;
+    }
+  }
+
+  async function checkProviderKeys(provider) {
+    const keys = (provider.apiKeys || []).filter(k => k.enabled !== false);
+    if (keys.length === 0) {
+      keyHealth.set(provider.id, { status: 'unknown', lastCheck: Date.now(), keys: [] });
+      return;
+    }
+
+    const protocol = provider.protocol || 'openai';
+    const base = provider.url.replace(/\/$/, '');
+    const hasV1Suffix = base.endsWith('/v1');
+    const isAzure = protocol === 'openai' && !!provider.azureDeployment;
+
+    const results = await Promise.all(keys.map(async (k, i) => {
+      try {
+        let testUrl, fetchOpts;
+        if (protocol === 'openai') {
+          if (isAzure) {
+            const ver = provider.azureApiVersion || '2024-02-01';
+            testUrl = `${base}/openai/deployments/${provider.azureDeployment}/models?api-version=${ver}`;
+            fetchOpts = { headers: { 'api-key': k.key } };
+          } else {
+            testUrl = hasV1Suffix ? `${base}/models` : `${base}/v1/models`;
+            fetchOpts = { headers: { 'Authorization': `Bearer ${k.key}` } };
+          }
+        } else if (protocol === 'anthropic') {
+          const testModel = (provider.models && provider.models[0]) || 'claude-3-haiku-20240307';
+          testUrl = hasV1Suffix ? `${base}/messages` : `${base}/v1/messages`;
+          fetchOpts = {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json', 'x-api-key': k.key, 'anthropic-version': '2023-06-01' },
+            body: JSON.stringify({ model: testModel, max_tokens: 1, messages: [{ role: 'user', content: 'hi' }] }),
+          };
+        } else if (protocol === 'gemini') {
+          testUrl = `${base}/v1beta/models?key=${k.key}`;
+          fetchOpts = {};
+        } else {
+          return { index: i, ok: false, message: '不支持的协议' };
+        }
+        const res = await fetch(testUrl, { ...fetchOpts, signal: AbortSignal.timeout(15000) });
+        if (!res.ok) {
+          const hint = res.status === 401 || res.status === 403 ? 'Key 无效或无权限' : `HTTP ${res.status}`;
+          return { index: i, ok: false, message: hint };
+        }
+        return { index: i, ok: true };
+      } catch (err) {
+        return { index: i, ok: false, message: err.name === 'TimeoutError' ? '连接超时' : err.message };
+      }
+    }));
+
+    const allOk = results.every(r => r.ok);
+    const anyOk = results.some(r => r.ok);
+    keyHealth.set(provider.id, {
+      status: allOk ? 'healthy' : anyOk ? 'partial' : 'unhealthy',
+      lastCheck: Date.now(),
+      keys: results,
+    });
+  }
+
+  // 启动后延迟 5 秒执行首次检查
+  setTimeout(() => checkAllProviderKeys(), 5000);
+  // 每 24 小时检查一次
+  setInterval(() => checkAllProviderKeys(), 24 * 60 * 60 * 1000);
+
   // ==================== 供应商 API ====================
 
   app.get('/api/providers', (req, res) => {
@@ -551,6 +634,7 @@ async function init() {
 
   // 创建代理
   app.post('/api/proxies', async (req, res) => {
+    configStore.saveSnapshot('create-proxy');
     const { name, port, requireAuth, authToken, providerId, defaultModel, routingStrategy, providerPool, providerWeight } = req.body;
 
     if (!name || !port || !providerId) {
@@ -592,6 +676,7 @@ async function init() {
 
   // 更新代理
   app.put('/api/proxies/:id', async (req, res) => {
+    configStore.saveSnapshot('update-proxy');
     const existing = configStore.getProxyById(req.params.id);
     if (!existing) return res.status(404).json({ error: 'Proxy not found' });
 
@@ -640,6 +725,7 @@ async function init() {
 
   // 删除代理
   app.delete('/api/proxies/:id', async (req, res) => {
+    configStore.saveSnapshot('delete-proxy');
     const existing = configStore.getProxyById(req.params.id);
     if (!existing) return res.status(404).json({ error: 'Proxy not found' });
 
@@ -666,6 +752,36 @@ async function init() {
     res.json({ success: true, running: false });
   });
 
+  // 批量启动所有代理
+  app.post('/api/proxies/start-all', async (req, res) => {
+    const proxies = configStore.getProxies();
+    const results = [];
+    for (const proxy of proxies) {
+      if (proxyManager.isRunning(proxy.id)) {
+        results.push({ id: proxy.id, name: proxy.name, skipped: true });
+        continue;
+      }
+      try {
+        await startProxyWithProvider(proxy);
+        results.push({ id: proxy.id, name: proxy.name, success: true });
+      } catch (err) {
+        results.push({ id: proxy.id, name: proxy.name, success: false, error: err.message });
+      }
+    }
+    res.json({ results });
+  });
+
+  // 批量停止所有代理
+  app.post('/api/proxies/stop-all', async (req, res) => {
+    const running = proxyManager.getRunningPorts();
+    const results = [];
+    for (const r of running) {
+      await proxyManager.stopProxy(r.id);
+      results.push({ id: r.id, name: r.name, success: true });
+    }
+    res.json({ results });
+  });
+
   // 获取运行状态
   app.get('/api/status', (req, res) => {
     res.json({
@@ -687,6 +803,37 @@ async function init() {
     });
   });
 
+  // API Key 健康状态
+  app.get('/api/key-health', (req, res) => {
+    const result = {};
+    for (const [providerId, health] of keyHealth) {
+      result[providerId] = health;
+    }
+    res.json(result);
+  });
+
+  // 手动触发健康检查
+  app.post('/api/key-health/check', async (req, res) => {
+    await checkAllProviderKeys();
+    res.json({ success: true });
+  });
+
+  // 设置
+  app.get('/api/settings', (req, res) => {
+    res.json(configStore.getSettings());
+  });
+
+  app.put('/api/settings', (req, res) => {
+    const settings = req.body;
+    if (!settings || typeof settings !== 'object') {
+      return res.status(400).json({ error: '需要 settings 对象' });
+    }
+    for (const [key, value] of Object.entries(settings)) {
+      configStore.setSetting(key, value);
+    }
+    res.json(configStore.getSettings());
+  });
+
   // Token 用量统计
   app.get('/api/stats', (req, res) => {
     const { range, startDate, endDate, proxyId } = req.query;
@@ -704,6 +851,22 @@ async function init() {
     res.json({ ...stats, proxies });
   });
 
+  // 日志查看
+  app.get('/api/logs', (req, res) => {
+    const lines = Math.min(parseInt(req.query.lines) || 200, 2000);
+    try {
+      if (!fs.existsSync(logger.LOG_FILE)) {
+        return res.json({ lines: [] });
+      }
+      const content = fs.readFileSync(logger.LOG_FILE, 'utf8');
+      const allLines = content.split('\n').filter(l => l.trim());
+      const tail = allLines.slice(-lines);
+      res.json({ lines: tail, total: allLines.length });
+    } catch (err) {
+      res.json({ lines: [], error: err.message });
+    }
+  });
+
   // ==================== 配置导入/导出 ====================
 
   app.get('/api/config/export', (req, res) => {
@@ -733,6 +896,8 @@ async function init() {
       return res.status(400).json({ error: '需要 config 和 mode（overwrite/merge）' });
     }
 
+    configStore.saveSnapshot('import-' + mode);
+
     // 校验结构
     if (!Array.isArray(config.providers) || !Array.isArray(config.proxies)) {
       return res.status(400).json({ error: '配置格式错误：需要 providers 和 proxies 数组' });
@@ -837,6 +1002,21 @@ async function init() {
     });
   });
 
+  // ==================== 配置版本历史 ====================
+
+  app.get('/api/config/history', (req, res) => {
+    const snapshots = configStore.getSnapshots();
+    res.json({ snapshots });
+  });
+
+  app.post('/api/config/rollback', async (req, res) => {
+    const { file } = req.body;
+    if (!file) return res.status(400).json({ error: '需要指定快照文件' });
+    const result = configStore.restoreSnapshot(file);
+    if (result.error) return res.status(400).json({ error: result.error });
+    res.json({ success: true });
+  });
+
   // 前端首页
   app.get('/', (req, res) => {
     res.sendFile(path.join(__dirname, 'public', 'index.html'));