protocol-proxy 2.1.6 → 2.3.2

package/public/app.js

@@ -211,6 +211,12 @@ function selectProvider(id) {
   const models = provider?.models || [];
   selectModel(models[0] || '');
   updateModelAddState();
+  // 同步 API Key placeholder
+  document.getElementById('target-key').placeholder = provider?.apiKey ? '已设置（留空则不修改）' : 'sk-...';
+  // 同步 Azure 字段
+  document.getElementById('target-azure-deployment').value = provider?.azureDeployment || '';
+  document.getElementById('target-azure-version').value = provider?.azureApiVersion || '';
+  document.getElementById('azure-fields').style.display = protocol === 'openai' ? '' : 'none';
 }
 
 // ==================== Model 下拉框 ====================
@@ -345,8 +351,254 @@ function updateModelAddState() {
   }
 }
 
+// ==================== 配置导入/导出 ====================
+
+let importData = null;
+
+async function exportConfig() {
+  try {
+    const res = await fetch('/api/config/export');
+    const data = await res.json();
+    const blob = new Blob([JSON.stringify(data, null, 2)], { type: 'application/json' });
+    const url = URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    const date = new Date().toISOString().slice(0, 10).replace(/-/g, '');
+    a.href = url;
+    a.download = `config-backup-${date}.json`;
+    a.click();
+    URL.revokeObjectURL(url);
+    showToast('配置已导出');
+  } catch (err) {
+    showToast('导出失败: ' + err.message, true);
+  }
+}
+
+function handleImportFile(e) {
+  const file = e.target.files[0];
+  if (!file) return;
+  e.target.value = '';
+
+  const reader = new FileReader();
+  reader.onload = () => {
+    try {
+      const data = JSON.parse(reader.result);
+      if (!Array.isArray(data.providers) || !Array.isArray(data.proxies)) {
+        showToast('配置格式错误：需要 providers 和 proxies 数组', true);
+        return;
+      }
+      importData = data;
+      document.getElementById('import-providers-count').textContent = data.providers.length;
+      document.getElementById('import-proxies-count').textContent = data.proxies.length;
+      document.getElementById('import-modal').classList.add('active');
+    } catch (err) {
+      showToast('文件解析失败: ' + err.message, true);
+    }
+  };
+  reader.readAsText(file);
+}
+
+function closeImportModal() {
+  document.getElementById('import-modal').classList.remove('active');
+  importData = null;
+}
+
+async function confirmImport() {
+  if (!importData) return;
+  const mode = document.querySelector('input[name="import-mode"]:checked')?.value || 'merge';
+
+  if (mode === 'overwrite') {
+    const ok = await showConfirm('确认<strong>覆盖</strong>现有配置？此操作不可撤销。');
+    if (!ok) return;
+  }
+
+  try {
+    const res = await fetch('/api/config/import', {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ config: importData, mode }),
+    });
+    const result = await res.json();
+
+    if (!res.ok) {
+      showToast(result.error || '导入失败', true);
+      return;
+    }
+
+    closeImportModal();
+    await Promise.all([loadProxies(), loadProviders()]);
+
+    const added = result.added;
+    let msg = `导入成功（${mode === 'overwrite' ? '覆盖' : '合并'}）`;
+    if (added) msg += `：新增 ${added.providers} 供应商、${added.proxies} 代理`;
+
+    const restart = await showConfirm(`${msg}。<br><br>运行中的代理需要重启才能应用变更，新增的代理需要手动启动。<br><br>是否立即重启所有代理？`);
+    if (restart) {
+      await restartAllProxies();
+    }
+  } catch (err) {
+    showToast('导入失败: ' + err.message, true);
+  }
+}
+
+async function restartAllProxies() {
+  try {
+    // 先停掉所有运行中的代理（不管 ID 是否匹配新配置）
+    const statusRes = await fetch('/api/status');
+    const status = await statusRes.json();
+    const runningIds = (status.running || []).map(r => r.id);
+    for (const id of runningIds) {
+      await fetch(`/api/proxies/${id}/stop`, { method: 'POST' });
+    }
+    // 重新加载配置
+    await loadProxies();
+    // 按新配置启动所有代理
+    for (const p of proxies) {
+      await fetch(`/api/proxies/${p.id}/start`, { method: 'POST' });
+    }
+    await loadProxies();
+    showToast('所有代理已重启');
+  } catch (err) {
+    showToast('重启失败: ' + err.message, true);
+  }
+}
+
 // ==================== 初始化 ====================
 
+// ==================== Token 用量统计 ====================
+
+let statsRange = 'daily';
+let statsProxyId = '';
+
+async function loadStats() {
+  try {
+    const params = new URLSearchParams({ range: statsRange });
+    if (statsProxyId) params.set('proxyId', statsProxyId);
+    const startDate = document.getElementById('stats-start-date')?.value;
+    const endDate = document.getElementById('stats-end-date')?.value;
+    if (startDate) params.set('startDate', startDate);
+    if (endDate) params.set('endDate', endDate);
+    const res = await fetch('/api/stats?' + params);
+    const data = await res.json();
+    renderStatsSummary(data.summary);
+    renderStatsBreakdown(data);
+    renderStatsProxyOptions(data.proxies || []);
+  } catch (err) {
+    console.error('加载统计失败:', err);
+  }
+}
+
+function renderStatsSummary(summary) {
+  document.getElementById('stats-total-tokens').textContent = formatTokens(summary.total);
+  document.getElementById('stats-prompt-tokens').textContent = formatTokens(summary.prompt);
+  document.getElementById('stats-completion-tokens').textContent = formatTokens(summary.completion);
+  document.getElementById('stats-total-requests').textContent = summary.requests.toLocaleString();
+  const badge = document.getElementById('stats-estimated-badge');
+  if (badge) badge.style.display = summary.hasEstimated ? 'inline' : 'none';
+}
+
+function formatTokens(n) {
+  if (!n || n === 0) return '0';
+  if (n >= 1000000) return (n / 1000000).toFixed(1) + 'M';
+  if (n >= 1000) return (n / 1000).toFixed(1) + 'K';
+  return n.toLocaleString();
+}
+
+function renderStatsBreakdown(data) {
+  const container = document.getElementById('stats-breakdown');
+  const { byProvider, byModel, summary } = data;
+
+  if (!byProvider || byProvider.length === 0) {
+    container.innerHTML = '<div class="empty">暂无数据</div>';
+    return;
+  }
+
+  let html = '<table class="stats-table"><thead><tr>';
+  html += '<th>供应商</th><th>模型</th><th style="text-align:right">请求数</th>';
+  html += '<th style="text-align:right">输入 Token</th><th style="text-align:right">输出 Token</th>';
+  html += '<th style="text-align:right">合计</th>';
+  html += '</tr></thead><tbody>';
+
+  for (const item of byModel) {
+    const prefix = item.hasEstimated ? '~' : '';
+    html += '<tr>';
+    html += `<td class="provider-cell">${escapeHtml(item.provider)}</td>`;
+    html += `<td class="model-cell"><code>${escapeHtml(item.model)}</code></td>`;
+    html += `<td class="num">${item.requests.toLocaleString()}</td>`;
+    html += `<td class="num">${prefix ? `<span class="num-estimated" title="估算值">~</span>` : ''}${formatTokens(item.prompt)}</td>`;
+    html += `<td class="num">${prefix ? `<span class="num-estimated" title="估算值">~</span>` : ''}${formatTokens(item.completion)}</td>`;
+    html += `<td class="num">${prefix ? `<span class="num-estimated" title="估算值">~</span>` : ''}${formatTokens(item.total)}</td>`;
+    html += '</tr>';
+  }
+
+  html += '</tbody>';
+  html += '<tfoot><tr>';
+  html += '<td colspan="2">合计</td>';
+  html += `<td class="num">${summary.requests.toLocaleString()}</td>`;
+  html += `<td class="num">${formatTokens(summary.prompt)}</td>`;
+  html += `<td class="num">${formatTokens(summary.completion)}</td>`;
+  html += `<td class="num">${formatTokens(summary.total)}</td>`;
+  html += '</tr></tfoot></table>';
+
+  container.innerHTML = html;
+}
+
+function renderStatsProxyOptions(proxyList) {
+  const container = document.getElementById('stats-proxy-dropdown-options');
+  container.innerHTML = `<div class="model-option${!statsProxyId ? ' selected' : ''}" data-proxy-id="">
+    <span class="model-option-name">全部代理</span>
+  </div>` + proxyList.map(p => `
+    <div class="model-option${p.id === statsProxyId ? ' selected' : ''}" data-proxy-id="${escapeHtml(p.id)}">
+      <span class="model-option-name">${escapeHtml(p.name)}</span>
+      ${p.providerName ? `<span style="color:#64748b;font-size:12px;margin-left:4px">${escapeHtml(p.providerName)}</span>` : ''}
+    </div>
+  `).join('');
+
+  container.querySelectorAll('.model-option').forEach(opt => {
+    opt.addEventListener('click', () => {
+      statsProxyId = opt.dataset.proxyId;
+      document.getElementById('stats-proxy-dropdown-value').textContent =
+        statsProxyId ? (proxyList.find(p => p.id === statsProxyId)?.name || '全部代理') : '全部代理';
+      document.getElementById('stats-proxy-dropdown').classList.remove('open');
+      container.querySelectorAll('.model-option').forEach(o => o.classList.remove('selected'));
+      opt.classList.add('selected');
+      loadStats();
+    });
+  });
+}
+
+function initStatsDropdown() {
+  const trigger = document.getElementById('stats-proxy-dropdown-trigger');
+  const dropdown = document.getElementById('stats-proxy-dropdown');
+
+  trigger.addEventListener('click', (e) => {
+    e.stopPropagation();
+    dropdown.classList.toggle('open');
+  });
+
+  document.addEventListener('click', (e) => {
+    if (!dropdown.contains(e.target)) {
+      dropdown.classList.remove('open');
+    }
+  });
+}
+
+function initStatsRangeBtns() {
+  document.querySelectorAll('.stats-range-btn').forEach(btn => {
+    btn.addEventListener('click', () => {
+      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 = '';
+      loadStats();
+    });
+  });
+  // 日期选择器变化时自动加载
+  document.getElementById('stats-start-date').addEventListener('change', loadStats);
+  document.getElementById('stats-end-date').addEventListener('change', loadStats);
+}
+
 function generateToken() {
   const arr = new Uint8Array(24);
   crypto.getRandomValues(arr);
@@ -385,9 +637,11 @@ function initSimpleDropdown(dropdownId, onChange) {
 }
 
 async function init() {
-  await Promise.all([loadProxies(), loadProviders()]);
+  await Promise.all([loadProxies(), loadProviders(), loadStats()]);
   initProviderDropdown();
   initModelDropdown();
+  initStatsDropdown();
+  initStatsRangeBtns();
   initSimpleDropdown('auth-dropdown', (val) => {
     const enabled = val === 'true';
     document.getElementById('auth-token-group').style.display = enabled ? 'block' : 'none';
@@ -395,7 +649,12 @@ async function init() {
       document.getElementById('proxy-auth-token').value = generateToken();
     }
   });
-  initSimpleDropdown('protocol-dropdown');
+  initSimpleDropdown('protocol-dropdown', (val) => {
+    document.getElementById('azure-fields').style.display = val === 'openai' ? '' : 'none';
+  });
+  // 初始状态：根据当前协议值决定 Azure 字段显示
+  const initProto = document.getElementById('target-protocol').value;
+  document.getElementById('azure-fields').style.display = initProto === 'openai' ? '' : 'none';
 }
 
 // ==================== 代理地址复制 ====================
@@ -541,6 +800,11 @@ function openModal(id = null) {
     selectProvider(p.providerId || '');
     selectModel(p.defaultModel || '');
     document.getElementById('target-key').placeholder = p.hasApiKey ? '已设置（留空则不修改）' : 'sk-...';
+    // Azure 字段从供应商配置读取
+    const provider = providers.find(pr => pr.id === p.providerId);
+    document.getElementById('target-azure-deployment').value = provider?.azureDeployment || '';
+    document.getElementById('target-azure-version').value = provider?.azureApiVersion || '';
+    document.getElementById('azure-fields').style.display = p.protocol === 'openai' ? '' : 'none';
   } else {
     document.getElementById('proxy-id').value = '';
     // 重置认证下拉框
@@ -552,6 +816,9 @@ function openModal(id = null) {
     selectProvider('');
     selectModel('');
     document.getElementById('target-key').placeholder = 'sk-...';
+    document.getElementById('target-azure-deployment').value = '';
+    document.getElementById('target-azure-version').value = '';
+    document.getElementById('azure-fields').style.display = 'none';
   }
 
   updateModelAddState();
@@ -593,6 +860,10 @@ async function handleSubmit(e) {
   const providerUpdates = {};
   if (apiKey) providerUpdates.apiKey = apiKey;
   if (protocol) providerUpdates.protocol = protocol;
+  const azureDeployment = document.getElementById('target-azure-deployment').value.trim();
+  const azureApiVersion = document.getElementById('target-azure-version').value.trim();
+  providerUpdates.azureDeployment = azureDeployment || '';
+  providerUpdates.azureApiVersion = azureApiVersion || '';
   if (Object.keys(providerUpdates).length > 0) {
     try {
       const res = await fetch(`/api/providers/${providerId}`, {

package/public/index.html

@@ -24,10 +24,63 @@
       </div>
     </div>
 
+    <!-- Token 用量统计 -->
+    <section class="card stats-panel">
+      <div class="card-header">
+        <h2>Token 用量统计 <span class="stats-estimated-badge" id="stats-estimated-badge" style="display:none">含估算</span></h2>
+        <div class="stats-controls">
+          <div class="model-dropdown" id="stats-proxy-dropdown">
+            <div class="model-dropdown-trigger stats-filter-trigger" id="stats-proxy-dropdown-trigger">
+              <span id="stats-proxy-dropdown-value">全部代理</span>
+              <span class="model-dropdown-arrow">&#9662;</span>
+            </div>
+            <div class="model-dropdown-menu" id="stats-proxy-dropdown-menu">
+              <div class="model-dropdown-options" id="stats-proxy-dropdown-options"></div>
+            </div>
+          </div>
+          <div class="stats-range-btns">
+            <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>
+            <span class="stats-date-sep">|</span>
+            <input type="date" id="stats-start-date" class="stats-date-input" placeholder="开始日期">
+            <span class="stats-date-sep">~</span>
+            <input type="date" id="stats-end-date" class="stats-date-input" placeholder="结束日期">
+          </div>
+        </div>
+      </div>
+      <div class="stats-summary" id="stats-summary">
+        <div class="stats-summary-item">
+          <span class="stats-summary-value" id="stats-total-tokens">-</span>
+          <span class="stats-summary-label">总 Token</span>
+        </div>
+        <div class="stats-summary-item">
+          <span class="stats-summary-value" id="stats-prompt-tokens">-</span>
+          <span class="stats-summary-label">输入 Token</span>
+        </div>
+        <div class="stats-summary-item">
+          <span class="stats-summary-value" id="stats-completion-tokens">-</span>
+          <span class="stats-summary-label">输出 Token</span>
+        </div>
+        <div class="stats-summary-item">
+          <span class="stats-summary-value" id="stats-total-requests">-</span>
+          <span class="stats-summary-label">请求数</span>
+        </div>
+      </div>
+      <div id="stats-breakdown" class="stats-breakdown">
+        <div class="empty">暂无数据</div>
+      </div>
+    </section>
+
     <section class="card">
       <div class="card-header">
         <h2>代理列表</h2>
-        <button class="btn btn-primary" onclick="openModal()">+ 新建代理</button>
+        <div class="card-header-actions">
+          <button class="btn" onclick="exportConfig()">导出配置</button>
+          <button class="btn" onclick="document.getElementById('import-file').click()">导入配置</button>
+          <input type="file" id="import-file" accept=".json" style="display:none" onchange="handleImportFile(event)">
+          <button class="btn btn-primary" onclick="openModal()">+ 新建代理</button>
+        </div>
       </div>
       <div id="proxy-list" class="proxy-list">
         <div class="empty">加载中...</div>
@@ -111,6 +164,7 @@
                       <div class="model-dropdown-options" id="protocol-dropdown-options">
                         <div class="model-option selected" data-value="openai"><span class="model-option-name">OpenAI</span></div>
                         <div class="model-option" data-value="anthropic"><span class="model-option-name">Anthropic</span></div>
+                        <div class="model-option" data-value="gemini"><span class="model-option-name">Gemini</span></div>
                       </div>
                     </div>
                   </div>
@@ -139,6 +193,16 @@
                   <input type="password" id="target-key" placeholder="sk-...">
                 </div>
               </div>
+              <div class="form-row" id="azure-fields" style="display:none">
+                <div class="form-group">
+                  <label>Azure Deployment <span style="color:#64748b;font-size:0.75rem">（仅 Azure 用户填写）</span></label>
+                  <input type="text" id="target-azure-deployment" placeholder="仅 Azure 用户填写">
+                </div>
+                <div class="form-group">
+                  <label>Azure API Version <span style="color:#64748b;font-size:0.75rem">（仅 Azure 用户填写）</span></label>
+                  <input type="text" id="target-azure-version" placeholder="仅 Azure 用户填写">
+                </div>
+              </div>
             </div>
           </div>
 
@@ -163,6 +227,51 @@
     </div>
   </div>
 
+  <!-- 导入预览弹窗 -->
+  <div class="modal" id="import-modal">
+    <div class="modal-content" style="max-width:500px">
+      <div class="modal-header">
+        <h3>导入配置</h3>
+        <button class="btn-close" onclick="closeImportModal()">&times;</button>
+      </div>
+      <div class="import-preview" id="import-preview">
+        <div class="import-stats">
+          <div class="import-stat">
+            <span class="import-stat-value" id="import-providers-count">0</span>
+            <span class="import-stat-label">供应商</span>
+          </div>
+          <div class="import-stat">
+            <span class="import-stat-value" id="import-proxies-count">0</span>
+            <span class="import-stat-label">代理</span>
+          </div>
+        </div>
+        <div class="import-mode">
+          <label>导入模式</label>
+          <div class="import-mode-options">
+            <label class="import-mode-option">
+              <input type="radio" name="import-mode" value="merge" checked>
+              <span>
+                <strong>合并</strong>
+                <small>按 ID 去重：新增导入项，同 ID 覆盖</small>
+              </span>
+            </label>
+            <label class="import-mode-option">
+              <input type="radio" name="import-mode" value="overwrite">
+              <span>
+                <strong>覆盖</strong>
+                <small>完全替换现有配置</small>
+              </span>
+            </label>
+          </div>
+        </div>
+      </div>
+      <div class="modal-footer">
+        <button class="btn" onclick="closeImportModal()">取消</button>
+        <button class="btn btn-primary" onclick="confirmImport()">确认导入</button>
+      </div>
+    </div>
+  </div>
+
   <script src="app.js"></script>
 </body>
 </html>