protocol-proxy 2.7.0 → 2.8.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "protocol-proxy",
-  "version": "2.7.0",
+  "version": "2.8.0",
   "description": "OpenAI / Anthropic 协议转换透明代理",
   "main": "server.js",
   "bin": {

package/public/app.js

@@ -6,6 +6,8 @@ let importData = null;
 let statsRange = 'daily';
 let statsProxyId = '';
 let providerPoolItems = [];
+let keyHealth = {};
+let statsAutoRefreshTimer = null;
 
 // ==================== 主题切换 ====================
 
@@ -81,6 +83,30 @@ function updateStats() {
     proxies.filter(p => p.running).length;
 }
 
+async function loadKeyHealth() {
+  try {
+    const res = await fetch('/api/key-health');
+    keyHealth = await res.json();
+    refreshHealthUI();
+  } catch (err) {
+    console.error('加载 Key 健康状态失败:', err);
+  }
+}
+
+function refreshHealthUI() {
+  // 更新每个代理卡片上的健康点
+  document.querySelectorAll('.health-dot[data-provider]').forEach(dot => {
+    const h = keyHealth[dot.dataset.provider];
+    dot.className = 'health-dot';
+    if (!h || h.status === 'unknown') { dot.classList.add('health-unknown'); dot.title = '未检测'; }
+    else if (h.status === 'healthy') { dot.classList.add('health-ok'); dot.title = 'Key 正常'; }
+    else if (h.status === 'partial') { dot.classList.add('health-warn'); dot.title = '部分 Key 异常'; }
+    else { dot.classList.add('health-error'); dot.title = 'Key 全部异常'; }
+  });
+  // 更新汇总卡片
+  renderProviderHealthSummary();
+}
+
 function parseProviderPool(value) {
   const text = (value || '').trim();
   if (!text) return [];
@@ -991,9 +1017,18 @@ function initStatsRangeBtns() {
       document.querySelectorAll('.stats-range-btn').forEach(b => b.classList.remove('active'));
       btn.classList.add('active');
       statsRange = btn.dataset.range;
-      // 清空日期选择器，显示全部数据
-      document.getElementById('stats-start-date').value = '';
-      document.getElementById('stats-end-date').value = '';
+      // 清除自动刷新
+      if (statsAutoRefreshTimer) { clearInterval(statsAutoRefreshTimer); statsAutoRefreshTimer = null; }
+      if (statsRange === 'hourly') {
+        // 实时模式：设为今天 + 30 秒自动刷新
+        const today = new Date().toISOString().slice(0, 10);
+        document.getElementById('stats-start-date').value = today;
+        document.getElementById('stats-end-date').value = today;
+        statsAutoRefreshTimer = setInterval(loadStats, 30000);
+      } else {
+        document.getElementById('stats-start-date').value = '';
+        document.getElementById('stats-end-date').value = '';
+      }
       loadStats();
     });
   });
@@ -1054,7 +1089,16 @@ function initProviderPoolDropdown() {
 }
 
 async function init() {
+  // 默认统计范围：当天（HTML 内联脚本已优先设置，此处兜底）
+  const today = new Date().toISOString().slice(0, 10);
+  const sd = document.getElementById('stats-start-date');
+  const ed = document.getElementById('stats-end-date');
+  if (!sd.value) sd.value = today;
+  if (!ed.value) ed.value = today;
   await Promise.all([loadProxies(), loadProviders(), loadStats()]);
+  // 延迟加载 health（等后端启动检测完成），之后每 5 分钟刷新
+  setTimeout(() => loadKeyHealth(), 6000);
+  setInterval(() => loadKeyHealth(), 5 * 60 * 1000);
   renderProxies();
   initProviderDropdown();
   initModelDropdown();
@@ -1327,9 +1371,64 @@ function filterProxies() {
   renderProxies();
 }
 
+function healthDot(providerId) {
+  const h = keyHealth[providerId];
+  const cls = !h || h.status === 'unknown' ? 'health-unknown'
+    : h.status === 'healthy' ? 'health-ok'
+    : h.status === 'partial' ? 'health-warn' : 'health-error';
+  const title = !h || h.status === 'unknown' ? '未检测'
+    : h.status === 'healthy' ? 'Key 正常'
+    : h.status === 'partial' ? '部分 Key 异常' : 'Key 全部异常';
+  return `<span class="health-dot ${cls}" data-provider="${escapeHtml(providerId)}" title="${title}"></span>`;
+}
+
+function renderProviderHealthSummary() {
+  const el = document.getElementById('provider-health-summary');
+  if (!el) return;
+  const allProviders = proxies.map(p => p.providerId).filter(Boolean);
+  const unique = [...new Set(allProviders)];
+  if (unique.length === 0) { el.style.display = 'none'; return; }
+  let healthy = 0, partial = 0, unhealthy = 0, unknown = 0;
+  for (const id of unique) {
+    const h = keyHealth[id];
+    if (!h || h.status === 'unknown') unknown++;
+    else if (h.status === 'healthy') healthy++;
+    else if (h.status === 'partial') partial++;
+    else unhealthy++;
+  }
+  el.style.display = '';
+  el.innerHTML = `
+    <div class="health-stat"><span class="health-dot health-ok"></span><span>正常 ${healthy}</span></div>
+    <div class="health-stat"><span class="health-dot health-warn"></span><span>部分异常 ${partial}</span></div>
+    <div class="health-stat"><span class="health-dot health-error"></span><span>异常 ${unhealthy}</span></div>
+    <div class="health-stat"><span class="health-dot health-unknown"></span><span>未检测 ${unknown}</span></div>
+    <button class="btn btn-sm" onclick="recheckKeys()">重新检测</button>
+  `;
+}
+
+let rechecking = false;
+async function recheckKeys() {
+  if (rechecking) return;
+  rechecking = true;
+  const btn = document.querySelector('.provider-health-summary .btn');
+  if (btn) { btn.disabled = true; btn.textContent = '检测中...'; }
+  showToast('正在检测...');
+  try {
+    await fetch('/api/key-health/check', { method: 'POST' });
+    await loadKeyHealth();
+    showToast('检测完成');
+  } catch (err) {
+    showToast('检测失败: ' + err.message, true);
+  } finally {
+    rechecking = false;
+    if (btn) { btn.disabled = false; btn.textContent = '重新检测'; }
+  }
+}
+
 function renderProxies() {
   const container = document.getElementById('proxy-list');
   const list = getFilteredProxies();
+  renderProviderHealthSummary();
   if (proxies.length === 0) {
     container.innerHTML = '<div class="empty">暂无代理配置，点击右上角创建</div>';
     return;
@@ -1374,6 +1473,7 @@ function renderProxies() {
       </div>
       <div class="proxy-meta">
         <span>端口: <strong>${p.port}</strong></span>
+        <span>供应商: ${healthDot(p.providerId)} ${escapeHtml(p.providerName || '-')}</span>
         <span>认证: ${p.requireAuth ? '已启用' : '未启用'}</span>
       </div>
       <div class="proxy-address">

package/public/index.html

@@ -7,6 +7,16 @@
   <link rel="stylesheet" href="style.css">
 </head>
 <body>
+  <script>
+    // 尽早设置统计日期默认值，避免 init() 时序问题
+    document.addEventListener('DOMContentLoaded', function() {
+      var today = new Date().toISOString().slice(0, 10);
+      var sd = document.getElementById('stats-start-date');
+      var ed = document.getElementById('stats-end-date');
+      if (sd && !sd.value) sd.value = today;
+      if (ed && !ed.value) ed.value = today;
+    });
+  </script>
   <div class="container">
     <button class="theme-toggle" id="theme-toggle" onclick="toggleTheme()" title="切换主题">
       <span id="theme-icon">&#9790;</span>
@@ -43,6 +53,7 @@
             </div>
           </div>
           <div class="stats-range-btns">
+            <button class="btn btn-sm stats-range-btn" data-range="hourly">实时</button>
             <button class="btn btn-sm stats-range-btn active" data-range="daily">每日</button>
             <button class="btn btn-sm stats-range-btn" data-range="monthly">每月</button>
             <button class="btn btn-sm stats-range-btn" data-range="yearly">每年</button>
@@ -97,6 +108,7 @@
           <button class="btn btn-sm" onclick="openLogViewer()">日志</button>
         </div>
       </div>
+      <div class="provider-health-summary" id="provider-health-summary" style="display:none"></div>
       <div id="proxy-list" class="proxy-list">
         <div class="empty">加载中...</div>
       </div>

package/public/style.css

@@ -1871,3 +1871,36 @@ form {
   font-size: 0.78rem;
   white-space: nowrap;
 }
+
+/* Health dots */
+.health-dot {
+  display: inline-block;
+  width: 8px;
+  height: 8px;
+  border-radius: 50%;
+  vertical-align: middle;
+  margin-right: 4px;
+}
+
+.health-ok { background: #22c55e; box-shadow: 0 0 6px rgba(34, 197, 94, 0.4); }
+.health-warn { background: #f59e0b; box-shadow: 0 0 6px rgba(245, 158, 11, 0.4); }
+.health-error { background: #ef4444; box-shadow: 0 0 6px rgba(239, 68, 68, 0.4); }
+.health-unknown { background: #64748b; }
+
+/* Provider health summary */
+.provider-health-summary {
+  display: flex;
+  align-items: center;
+  gap: 18px;
+  padding: 12px 28px;
+  border-bottom: 1px solid var(--border-light);
+  background: var(--bg-surface-alt);
+  font-size: 0.82rem;
+  color: var(--text-muted);
+}
+
+.health-stat {
+  display: flex;
+  align-items: center;
+  gap: 5px;
+}

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) => {
@@ -720,6 +803,21 @@ 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());