\n \n \n \n \n \n \n \n \n \n
\n\n \n \n
\n \n JimiHub
\n\n \n \n \n
\n \n 加载中...\n
\n \n \n 错误:\n \n
\n \n \n 成功:\n \n
\n \n \n \n
\n\n \n \n
\n\n \n Gemini API Keys
\n \n\n
\n \n
\n 正在运行所有测试
\n \n\n
\n \n 进度\n 0 / 0\n
\n \n \n
\n \n 准备测试中...\n
\n \n \n
\n \n 加载密钥中...
\n\n \n \n \n
\n \n \n
\n Vertex AI Configuration
\n\n \n\n
\n\n \n \n \n
\n 当前 Vertex 配置
\n\n \n
\n \n
\n 加载配置中...
\nWorker API Keys
\n\n \n
\n \n 加载密钥中...
\n\n
\n \n 安全设置： 默认启用。禁用时，模型允许生成 NSFW 内容。
\n\n
\n Managed Models
\n \n\n \n
\n \n 加载模型中...
\n\n
\n\n \n \n
\n \n
\n \n \n 设置类别配额
\n \n\n
\n\n \n \n
\n \n
\n \n \n 设置配额
\n \n\n
\n\n \n\n\n"
},
{
"path": "public/admin/script.js",
"content": "document.addEventListener('DOMContentLoaded', () => {\n // --- UI Elements ---\n const authCheckingUI = document.getElementById('auth-checking');\n const unauthorizedUI = document.getElementById('unauthorized');\n const mainContentUI = document.getElementById('main-content');\n\n // --- Global State & Elements ---\n const loadingIndicator = document.getElementById('loading-indicator');\n const errorMessageDiv = document.getElementById('error-message');\n const errorTextSpan = document.getElementById('error-text');\n const successMessageDiv = document.getElementById('success-message');\n const successTextSpan = document.getElementById('success-text');\n const geminiKeysListDiv = document.getElementById('gemini-keys-list');\n const addGeminiKeyForm = document.getElementById('add-gemini-key-form');\n const workerKeysListDiv = document.getElementById('worker-keys-list');\n const addWorkerKeyForm = document.getElementById('add-worker-key-form');\n const generateWorkerKeyBtn = document.getElementById('generate-worker-key');\n\n // Tab elements\n const geminiTab = document.getElementById('gemini-tab');\n const vertexTab = document.getElementById('vertex-tab');\n const geminiContent = document.getElementById('gemini-content');\n const vertexContent = document.getElementById('vertex-content');\n\n // Vertex configuration elements\n const vertexConfigForm = document.getElementById('vertex-config-form');\n const vertexConfigDisplay = document.getElementById('vertex-config-display');\n const vertexConfigInfo = document.getElementById('vertex-config-info');\n const vertexStatus = document.getElementById('vertex-status');\n const testVertexConfigBtn = document.getElementById('test-vertex-config');\n const clearVertexConfigBtn = document.getElementById('clear-vertex-config');\n const workerKeyValueInput = document.getElementById('worker-key-value');\n const modelsListDiv = document.getElementById('models-list');\n const addModelForm = document.getElementById('add-model-form');\n const modelCategorySelect = document.getElementById('model-category');\n const customQuotaDiv = document.getElementById('custom-quota-div');\n const modelQuotaInput = document.getElementById('model-quota');\n const modelIdInput = document.getElementById('model-id');\n const setCategoryQuotasBtn = document.getElementById('set-category-quotas-btn');\n const categoryQuotasModal = document.getElementById('category-quotas-modal');\n const closeCategoryQuotasModalBtn = document.getElementById('close-category-quotas-modal');\n const cancelCategoryQuotasBtn = document.getElementById('cancel-category-quotas');\n const categoryQuotasForm = document.getElementById('category-quotas-form');\n const proQuotaInput = document.getElementById('pro-quota');\n const flashQuotaInput = document.getElementById('flash-quota');\n const categoryQuotasErrorDiv = document.getElementById('category-quotas-error');\n const geminiKeyErrorContainer = document.getElementById('gemini-key-error-container'); // Container for error messages in modal\n // Individual Quota Elements\n const individualQuotaModal = document.getElementById('individual-quota-modal');\n const closeIndividualQuotaModalBtn = document.getElementById('close-individual-quota-modal');\n const cancelIndividualQuotaBtn = document.getElementById('cancel-individual-quota');\n const individualQuotaForm = document.getElementById('individual-quota-form');\n const individualQuotaModelIdInput = document.getElementById('individual-quota-model-id');\n const individualQuotaValueInput = document.getElementById('individual-quota-value');\n const individualQuotaErrorDiv = document.getElementById('individual-quota-error');\n const logoutButton = document.getElementById('logout-button');\n const darkModeToggle = document.getElementById('dark-mode-toggle');\n const sunIcon = document.getElementById('sun-icon');\n const moonIcon = document.getElementById('moon-icon');\n // Run All Test Elements\n const runAllTestBtn = document.getElementById('run-all-test-btn');\n const ignoreAllErrorsBtn = document.getElementById('ignore-all-errors-btn');\n const cleanErrorKeysBtn = document.getElementById('clean-error-keys-btn');\n const geminiKeysActionsDiv = document.getElementById('gemini-keys-actions');\n const testProgressArea = document.getElementById('test-progress-area');\n const cancelAllTestBtn = document.getElementById('cancel-all-test-btn');\n const testProgressBar = document.getElementById('test-progress-bar');\n const testProgressText = document.getElementById('test-progress-text');\n const testStatusText = document.getElementById('test-status-text');\n\n // --- Global Cache ---\n let cachedModels = [];\n let cachedGeminiModels = []; // Add cache for available Gemini models\n let cachedCategoryQuotas = { proQuota: 0, flashQuota: 0 };\n\n // --- Global Test State ---\n let isRunningAllTests = false;\n let testCancelRequested = false;\n let currentTestBatch = [];\n let operationInProgress = false; // Prevent concurrent database operations\n // No need for a separate errorKeyIds cache, as errorStatus is now part of the key data\n\n // --- Run All Test Functions ---\n async function runAllGeminiKeysTest() {\n // Prevent concurrent operations\n if (operationInProgress) {\n showError(t('operation_in_progress') || 'Another operation is in progress. Please wait.');\n return;\n }\n\n try {\n operationInProgress = true; // Lock operations\n isRunningAllTests = true;\n testCancelRequested = false;\n\n // Show progress area\n testProgressArea.classList.remove('hidden');\n runAllTestBtn.disabled = true;\n cancelAllTestBtn.disabled = false;\n\n // Get all Gemini keys\n const keys = await apiFetch('/gemini-keys');\n if (!keys || keys.length === 0) {\n showError(t('no_gemini_keys_found'));\n return;\n }\n\n const totalKeys = keys.length;\n let completedTests = 0;\n const testModel = 'gemini-2.0-flash'; // Fixed model for testing\n\n // Update initial progress\n updateTestProgress(completedTests, totalKeys, t('preparing_tests'));\n\n // Process keys in batches to balance performance and server load\n const batchSize = 5; // Optimal batch size for testing\n for (let i = 0; i < keys.length; i += batchSize) {\n if (testCancelRequested) {\n break;\n }\n\n const batch = keys.slice(i, i + batchSize);\n currentTestBatch = batch;\n\n updateTestProgress(completedTests, totalKeys, t('testing_batch', Math.floor(i / batchSize) + 1));\n\n // Run tests for current batch concurrently\n const batchPromises = batch.map(key => testSingleKey(key.id, testModel));\n const batchResults = await Promise.allSettled(batchPromises);\n\n // Update progress\n completedTests += batch.length;\n updateTestProgress(completedTests, totalKeys, t('completed_tests', completedTests, totalKeys));\n\n // Increased delay between batches to reduce server load\n if (i + batchSize < keys.length && !testCancelRequested) {\n await new Promise(resolve => setTimeout(resolve, 1000)); // Increased from 500ms to 1000ms\n }\n }\n\n // Final status\n if (testCancelRequested) {\n updateTestProgress(completedTests, totalKeys, t('tests_cancelled'));\n showError(t('test_run_cancelled'));\n } else {\n updateTestProgress(completedTests, totalKeys, t('all_tests_completed'));\n showSuccess(t('completed_testing', totalKeys, testModel));\n\n // Auto-hide progress area after 3 seconds\n setTimeout(() => {\n testProgressArea.classList.add('hidden');\n }, 3000);\n }\n\n // Reload keys to show updated status\n await loadGeminiKeys();\n\n } catch (error) {\n console.error('Error running all tests:', error);\n showError(t('failed_to_run_tests', error.message));\n updateTestProgress(0, 0, t('test_run_failed'));\n } finally {\n operationInProgress = false; // Release lock\n isRunningAllTests = false;\n runAllTestBtn.disabled = false;\n cancelAllTestBtn.disabled = true;\n currentTestBatch = [];\n }\n }\n\n async function testSingleKey(keyId, modelId) {\n try {\n // Use direct fetch to get detailed error information\n const response = await fetch('/api/admin/test-gemini-key', {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n },\n credentials: 'include',\n body: JSON.stringify({ keyId, modelId })\n });\n\n // Parse response regardless of status code\n let result = null;\n const contentType = response.headers.get(\"content-type\");\n if (contentType && contentType.indexOf(\"application/json\") !== -1) {\n result = await response.json();\n } else {\n const textContent = await response.text();\n result = {\n success: false,\n status: response.status,\n content: textContent || 'No response content'\n };\n }\n\n return {\n keyId,\n success: result?.success || false,\n status: result?.status || response.status,\n error: result?.success ? null : (result?.content || 'Test failed')\n };\n } catch (error) {\n return {\n keyId,\n success: false,\n status: 'error',\n error: error.message || 'Network error'\n };\n }\n }\n\n function updateTestProgress(completed, total, statusMessage) {\n const percentage = total > 0 ? Math.round((completed / total) * 100) : 0;\n\n testProgressBar.style.width = `${percentage}%`;\n testProgressText.textContent = `${completed} / ${total}`;\n testStatusText.textContent = statusMessage;\n }\n\n // --- Utility Functions ---\n function showLoading() {\n loadingIndicator.classList.remove('hidden');\n }\n\n function hideLoading() {\n loadingIndicator.classList.add('hidden');\n }\n\n // Function to enable/disable add model form based on Gemini keys availability\n function updateAddModelFormState(hasGeminiKeys) {\n const addModelFormElements = addModelForm.querySelectorAll('input, select, button');\n addModelFormElements.forEach(element => {\n element.disabled = !hasGeminiKeys;\n });\n\n // Add visual indication when disabled\n if (hasGeminiKeys) {\n addModelForm.classList.remove('opacity-50', 'pointer-events-none');\n } else {\n addModelForm.classList.add('opacity-50', 'pointer-events-none');\n }\n }\n\n function showError(message, element = errorTextSpan, container = errorMessageDiv) {\n element.textContent = message;\n container.classList.remove('hidden');\n // Auto-hide after 5 seconds\n setTimeout(() => {\n hideError(container);\n }, 5000);\n }\n\nfunction hideError(container = errorMessageDiv) {\n container.classList.add('hidden');\n const textSpan = container.querySelector('span#error-text'); \n if (textSpan) textSpan.textContent = ''; // Only clear the message span\n}\n\n // Function to show success message and auto-hide\n function showSuccess(message, element = successTextSpan, container = successMessageDiv) {\n element.textContent = message;\n container.classList.remove('hidden');\n // Auto-hide after 3 seconds\n setTimeout(() => {\n hideSuccess(container);\n }, 3000);\n }\n\n // Function to hide success message\n function hideSuccess(container = successMessageDiv) {\n container.classList.add('hidden');\n const textSpan = container.querySelector('span');\n if (textSpan) textSpan.textContent = '';\n }\n\n // Generic API fetch function (using cookie auth now)\n // 新增 suppressGlobalError 参数，允许调用方控制是否全局报错\n async function apiFetch(endpoint, options = {}, suppressGlobalError = false) {\n showLoading();\n hideError();\n hideError(categoryQuotasErrorDiv);\n hideSuccess(); // Hide success message on new request\n\n // No need for Authorization header, rely on HttpOnly cookie\n const defaultHeaders = {\n 'Content-Type': 'application/json',\n };\n\n try {\n const response = await fetch(`/api/admin${endpoint}`, {\n credentials: 'include', \n ...options,\n headers: {\n ...defaultHeaders,\n ...(options.headers || {}),\n },\n });\n\n // Check for auth errors (401 Unauthorized, 403 Forbidden)\n if (response.status === 401 || response.status === 403) {\n console.log(\"Authentication required or session expired. Redirecting to login.\");\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n return null;\n }\n \n // Check for redirects that might indicate auth issues (302, 307, etc.)\n if (response.redirected) {\n const redirectUrl = new URL(response.url);\n // Check if redirected to login page or similar auth pages\n if (redirectUrl.pathname.includes('login') || \n !redirectUrl.pathname.includes('/api/admin')) {\n console.log(\"Detected redirect to login page. Session likely expired.\");\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n return null;\n }\n }\n \n // Additional check for 3xx status codes\n if (response.status >= 300 && response.status < 400) {\n console.log(`Redirect status detected: ${response.status}. Handling potential auth issue.`);\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n return null;\n }\n\n let data = null;\n const contentType = response.headers.get(\"content-type\");\n if (contentType && contentType.indexOf(\"application/json\") !== -1) {\n try {\n data = await response.json();\n } catch (e) {\n if (response.ok) {\n console.warn(\"Received OK response but failed to parse JSON body.\");\n return { success: true };\n } else {\n const errorText = await response.text();\n throw new Error(`HTTP error! status: ${response.status} ${response.statusText} - ${errorText}`);\n }\n }\n } else if (!response.ok) {\n const errorText = await response.text();\n throw new Error(`HTTP error! status: ${response.status} ${response.statusText} - ${errorText}`);\n } else {\n console.log(`Received non-JSON response with status ${response.status}`);\n return { success: true };\n }\n\n // 409 Conflict\n if (response.status === 409) {\n throw new Error(data?.error || 'Existing API key');\n }\n\n if (!response.ok) {\n throw new Error(data?.error || `HTTP error! status: ${response.status}`);\n }\n\n return data;\n } catch (error) {\n console.error('API Fetch Error:', error);\n if (suppressGlobalError) {\n throw error;\n }\n if (endpoint === '/category-quotas') {\n showError(error.message || 'An unknown error occurred.', categoryQuotasErrorDiv, categoryQuotasErrorDiv);\n } else {\n showError(error.message || 'An unknown error occurred.');\n }\n return null;\n } finally {\n hideLoading();\n }\n }\n\n // --- Rendering Functions ---\n\n // Helper to format quota display (Infinity becomes ∞)\n function formatQuota(quota) {\n return (quota === undefined || quota === null || quota === Infinity) ? '∞' : quota;\n }\n\n // Helper to calculate remaining percentage for progress bar\n function calculateRemainingPercentage(count, quota) {\n if (quota === undefined || quota === null || quota === Infinity || quota <= 0) {\n return 100;\n }\n const percentage = Math.max(0, 100 - (count / quota * 100));\n return percentage;\n }\n\n // Helper to get progress bar color based on percentage\n function getProgressColor(percentage) {\n if (percentage < 25) return 'bg-red-500';\n if (percentage < 50) return 'bg-yellow-500';\n return 'bg-green-500';\n }\n\n\nasync function renderGeminiKeys(keys) {\n geminiKeysListDiv.innerHTML = ''; // Clear previous list\n if (!keys || keys.length === 0) {\n geminiKeysListDiv.innerHTML = '\n
\n \n
\n \n
\n\n \n 系统设置
\n \nNo Gemini keys configured.
';\n // Hide action buttons when no keys\n geminiKeysActionsDiv.classList.add('hidden');\n // Disable add model form when no Gemini keys\n updateAddModelFormState(false);\n return;\n }\n\n // Check if there are any error keys\n const hasErrorKeys = keys.some(key => key.errorStatus === 400 || key.errorStatus === 401 || key.errorStatus === 403);\n\n // Show/hide error-related buttons based on error keys existence\n if (hasErrorKeys) {\n ignoreAllErrorsBtn.classList.remove('hidden');\n cleanErrorKeysBtn.classList.remove('hidden');\n } else {\n ignoreAllErrorsBtn.classList.add('hidden');\n cleanErrorKeysBtn.classList.add('hidden');\n }\n\n // Show action buttons when keys exist\n geminiKeysActionsDiv.classList.remove('hidden');\n\n // Enable add model form when Gemini keys exist\n updateAddModelFormState(true);\n\n // Ensure models and category quotas are cached (should be loaded in initialLoad)\n if (cachedModels.length === 0) {\n console.warn(\"Models cache is empty during renderGeminiKeys. Load may be incomplete.\");\n }\n\n // Calculate statistics\n const totalKeys = keys.length;\n const totalUsage = keys.reduce((sum, key) => sum + (parseInt(key.usage) || 0), 0);\n \n // Create main container\n const keysContainer = document.createElement('div');\n keysContainer.className = 'keys-container';\n geminiKeysListDiv.appendChild(keysContainer);\n \n // Create statistics bar that's always visible\n const statsBar = document.createElement('div');\n // Removed justify-between, added relative for positioning context and select-none to prevent text selection\n statsBar.className = 'stats-bar relative flex items-center justify-center p-3 rounded-md mb-4 cursor-pointer transition-colors select-none';\n statsBar.innerHTML = `\n \n
\n \n `;\n keysContainer.appendChild(statsBar);\n \n // Create collapsible grid container with fixed height for responsive design\n const keysGrid = document.createElement('div');\n // Mobile: 1 column, show 6 items (6 rows)\n // Tablet: 2 columns, show 6 items (3 rows)\n // Desktop: 3 columns, show 9 items (3 rows)\n keysGrid.className = 'keys-grid grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-4 transition-all duration-300 overflow-y-auto border border-gray-200 rounded-lg p-2';\n\n // Function to calculate and apply dynamic height\n const updateGridHeight = () => {\n // Each card is 80px height + 16px gap between cards + 8px padding\n const cardHeight = 80;\n const gapSize = 16;\n const containerPadding = 8;\n\n // Calculate number of rows based on screen size and total keys\n let columnsCount, maxRows, actualRows;\n\n // Determine columns and max rows based on screen size\n if (window.innerWidth >= 1024) { // lg breakpoint\n columnsCount = 3;\n maxRows = 3; // Show max 3 rows on desktop\n } else if (window.innerWidth >= 768) { // md breakpoint\n columnsCount = 2;\n maxRows = 3; // Show max 3 rows on tablet\n } else {\n columnsCount = 1;\n maxRows = 6; // Show max 6 rows on mobile\n }\n\n // Calculate actual rows needed\n actualRows = Math.ceil(totalKeys / columnsCount);\n\n // Use the smaller of actual rows needed or max rows allowed\n const displayRows = Math.min(actualRows, maxRows);\n\n // Calculate dynamic height: rows * cardHeight + (rows-1) * gap + padding\n const dynamicHeight = displayRows * cardHeight + (displayRows - 1) * gapSize + containerPadding * 2;\n const maxHeight = maxRows * cardHeight + (maxRows - 1) * gapSize + containerPadding * 2;\n\n // Apply dynamic height with max height constraint\n keysGrid.style.height = `${dynamicHeight}px`;\n keysGrid.style.maxHeight = `${maxHeight}px`;\n };\n\n // Initial height calculation\n updateGridHeight();\n\n // Add resize listener to recalculate height when window size changes\n const resizeHandler = () => updateGridHeight();\n window.addEventListener('resize', resizeHandler);\n\n // Store the resize handler for cleanup (optional)\n keysGrid._resizeHandler = resizeHandler;\n \n // Set initial expanded/collapsed state based on key count\n // With fixed height containers, we can be more generous with initial expansion\n // Mobile: show if <= 6 keys, Desktop: show if <= 9 keys\n const isInitiallyExpanded = totalKeys <= 9;\n if (!isInitiallyExpanded) {\n keysGrid.classList.add('hidden');\n // Hide action buttons when grid is initially collapsed\n geminiKeysActionsDiv.classList.add('hidden');\n }\n \n // Update icon display\n const expandIcon = statsBar.querySelector('.expand-icon'); // Left arrow - collapsed state\n const collapseIcon = statsBar.querySelector('.collapse-icon'); // Down arrow - expanded state\n\n if (isInitiallyExpanded) {\n // Content expanded state (visible): show down arrow\n expandIcon.classList.add('hidden');\n collapseIcon.classList.remove('hidden');\n } else {\n // Content collapsed state (hidden): show left arrow\n expandIcon.classList.remove('hidden');\n collapseIcon.classList.add('hidden');\n }\n \n keysContainer.appendChild(keysGrid);\n\n // Add click event listener to toggle the grid visibility\n statsBar.addEventListener('click', (e) => {\n // 防止文本选择\n e.preventDefault();\n const isCurrentlyHidden = keysGrid.classList.contains('hidden');\n keysGrid.classList.toggle('hidden');\n expandIcon.classList.toggle('hidden');\n collapseIcon.classList.toggle('hidden');\n\n // Toggle action buttons visibility based on grid visibility\n if (isCurrentlyHidden) {\n // Grid is being shown, show action buttons\n geminiKeysActionsDiv.classList.remove('hidden');\n } else {\n // Grid is being hidden, hide action buttons\n geminiKeysActionsDiv.classList.add('hidden');\n }\n });\n\n keys.forEach(key => {\n // Create a simplified card for each key with optimized height\n const cardItem = document.createElement('div');\n cardItem.className = 'card-item p-3 border rounded-md bg-white shadow-sm hover:shadow-md transition-shadow cursor-pointer select-none h-[80px] flex flex-col justify-between';\n cardItem.dataset.keyId = key.id;\n\n // Show warning icon or usage badge\n let rightSideContent = '';\n if (key.errorStatus === 400 || key.errorStatus === 401 || key.errorStatus === 403) {\n rightSideContent = `\n \n `;\n } else {\n rightSideContent = `\n \n API Keys:\n ${totalKeys}\n
\n \n 24hr Usage:\n ${totalUsage}\n
\n \n ${key.usage}\n
\n `;\n }\n\n // Optimized card content with better spacing and typography\n cardItem.innerHTML = `\n \n
\n `;\n\n\n keysGrid.appendChild(cardItem);\n\n // Create a hidden detailed information modal\n const detailModal = document.createElement('div');\n detailModal.className = 'fixed inset-0 bg-black bg-opacity-50 flex justify-center items-center z-50 hidden';\n detailModal.dataset.modalFor = key.id;\n\n // --- Start Modal HTML ---\n let modalHTML = `\n `;\n // --- End Modal HTML ---\n\n detailModal.innerHTML = modalHTML;\n document.body.appendChild(detailModal);\n\n\n // Add click event to the card to display the detailed information modal\n cardItem.addEventListener('click', (e) => {\n // 防止文本选择\n e.preventDefault();\n detailModal.classList.remove('hidden');\n });\n\n // Add event to the close button\n const closeBtn = detailModal.querySelector('.close-modal');\n closeBtn.addEventListener('click', () => {\n detailModal.classList.add('hidden');\n });\n\n // Close by clicking outside the modal\n detailModal.addEventListener('click', (e) => {\n if (e.target === detailModal) {\n detailModal.classList.add('hidden');\n }\n });\n });\n\n // Note: Event listeners for .test-gemini-key and .run-test-btn are now handled\n // by global event delegation to prevent duplicate listeners and DOM reference issues\n }\n\n function renderWorkerKeys(keys) {\n workerKeysListDiv.innerHTML = ''; // Clear previous list\n if (!keys || keys.length === 0) {\n workerKeysListDiv.innerHTML = '\n
\n ${key.name || key.id}
\nID: ${key.id}
\n${key.keyPreview}
\n\n ${rightSideContent}\n
\n No Worker keys configured.
';\n return;\n }\n\n keys.forEach(key => {\n const isSafetyEnabled = key.safetyEnabled !== undefined ? key.safetyEnabled : true;\n\n const item = document.createElement('div');\n item.className = 'p-3 border rounded-md';\n item.innerHTML = `\n\n
\n \n
\n \n ${key.key}
\n${key.description || t('no_description')} (${t('created')}: ${new Date(key.createdAt).toLocaleDateString()})
\n\n
\n `;\n\n workerKeysListDiv.appendChild(item);\n });\n\n // Add styles for toggle switch\n const style = document.createElement('style');\n style.textContent = `\n .toggle-checkbox:checked {\n /* Adjusted translation to keep handle within bounds */\n transform: translateX(1rem); /* Was 100% */\n border-color: #68D391;\n }\n .toggle-checkbox:checked + .toggle-label {\n background-color: #68D391;\n }\n .toggle-label {\n transition: background-color 0.2s ease-in-out;\n }\n `;\n document.head.appendChild(style);\n\n // Add event listeners for safety toggles\n document.querySelectorAll('.safety-toggle').forEach(toggle => {\n toggle.addEventListener('change', function() {\n const key = this.dataset.key;\n const isEnabled = this.checked;\n const statusText = this.parentElement.nextElementSibling;\n statusText.textContent = isEnabled ? t('enabled') : t('disabled');\n statusText.className = `text-xs font-medium ${isEnabled ? 'text-green-600' : 'text-red-600'}`;\n saveSafetySettingsToServer(key, isEnabled);\n\n console.log(`Safety settings for key ${key} set to ${isEnabled ? 'enabled' : 'disabled'}`);\n });\n });\n }\n\n function renderModels(models) {\n modelsListDiv.innerHTML = ''; // Clear previous list\n if (!models || models.length === 0) {\n modelsListDiv.innerHTML = '\n \n
\n \n \n \n
\n \n ${isSafetyEnabled ? t('enabled') : t('disabled')}\n \n No models configured.
';\n return;\n }\n models.forEach(model => {\n const item = document.createElement('div');\n item.className = 'p-3 border rounded-md flex items-center justify-between';\n \n let quotaDisplay = model.category;\n if (model.category === 'Custom') {\n quotaDisplay += ` (${t('quota')}: ${model.dailyQuota === undefined ? t('unlimited') : model.dailyQuota})`;\n } else if (model.individualQuota) {\n // Show individual quota if it exists for Pro/Flash models\n quotaDisplay += ` (${t('individual_quota')}: ${model.individualQuota})`;\n }\n\n let actionsHtml = '';\n // Only show Set Individual Quota button for Pro and Flash models\n if (model.category === 'Pro' || model.category === 'Flash') {\n actionsHtml = `\n \n `;\n }\n actionsHtml += ``;\n\n item.innerHTML = `\n\n
\n ${model.id}
\n${quotaDisplay}
\n\n ${actionsHtml}\n
\n `;\n modelsListDiv.appendChild(item);\n });\n\n // Add event listeners for individual quota buttons\n document.querySelectorAll('.set-individual-quota').forEach(btn => {\n btn.addEventListener('click', (e) => {\n const modelId = e.target.dataset.id;\n const category = e.target.dataset.category;\n const currentQuota = parseInt(e.target.dataset.quota, 10);\n \n // Set the form values\n individualQuotaModelIdInput.value = modelId;\n individualQuotaValueInput.value = currentQuota || 0;\n \n // Show the modal\n hideError(individualQuotaErrorDiv);\n individualQuotaModal.classList.remove('hidden');\n });\n });\n }\n\n // --- Data Loading Functions ---\n async function loadGeminiKeys() {\n const keys = await apiFetch('/gemini-keys');\n if (keys) {\n renderGeminiKeys(keys);\n } else {\n geminiKeysListDiv.innerHTML = 'Failed to load Gemini keys.
';\n // Disable add model form when failed to load keys\n updateAddModelFormState(false);\n }\n }\n\n async function loadWorkerKeys() {\n const keys = await apiFetch('/worker-keys');\n if (keys) {\n renderWorkerKeys(keys);\n } else {\n workerKeysListDiv.innerHTML = 'Failed to load Worker keys.
';\n }\n }\n\n async function loadModels() {\n const models = await apiFetch('/models');\n if (models) {\n cachedModels = models;\n renderModels(models);\n } else {\n modelsListDiv.innerHTML = 'Failed to load models.
';\n }\n }\n\n // New function to load category quotas\n async function loadCategoryQuotas() {\n const quotas = await apiFetch('/category-quotas');\n if (quotas) {\n cachedCategoryQuotas = quotas;\n } else {\n showError(\"Failed to load category quotas.\");\n }\n return quotas;\n }\n\n // New function to load available Gemini models\n async function loadGeminiAvailableModels(forceRefresh = false) {\n // Only proceed if we have Gemini keys\n const geminiKeysList = document.querySelectorAll('#gemini-keys-list .card-item');\n if (geminiKeysList.length === 0) {\n console.log(\"No Gemini keys available, skipping model list fetch\");\n // Clear cached models if no keys available\n cachedGeminiModels = [];\n return;\n }\n\n // Skip if we already have cached models and not forcing refresh\n if (!forceRefresh && cachedGeminiModels.length > 0) {\n console.log(\"Using cached Gemini models\");\n updateModelIdDropdown(cachedGeminiModels);\n return;\n }\n\n try {\n console.log(\"Fetching available Gemini models from server...\");\n const models = await apiFetch('/gemini-models');\n if (models && Array.isArray(models)) {\n cachedGeminiModels = models;\n\n // Update the model-id input field to include dropdown\n updateModelIdDropdown(models);\n\n console.log(`Loaded ${models.length} available Gemini models`);\n } else {\n console.warn(\"No models returned from server\");\n cachedGeminiModels = [];\n }\n } catch (error) {\n console.error(\"Failed to load Gemini models:\", error);\n cachedGeminiModels = [];\n }\n }\n\n // Update the model-id input to include dropdown functionality\n function updateModelIdDropdown(models) {\n if (!modelIdInput) return;\n \n // Create custom dropdown menu\n const createCustomDropdown = () => {\n // Remove old dropdown menu (if it exists)\n const existingDropdown = document.getElementById('custom-model-dropdown');\n if (existingDropdown) {\n existingDropdown.remove();\n }\n \n // Create new dropdown menu container\n const dropdownContainer = document.createElement('div');\n dropdownContainer.id = 'custom-model-dropdown';\n dropdownContainer.className = 'absolute z-10 mt-1 w-full bg-white shadow-lg max-h-60 rounded-md py-1 text-base overflow-auto focus:outline-none sm:text-sm hidden';\n dropdownContainer.style.maxHeight = '200px';\n dropdownContainer.style.overflowY = 'auto';\n dropdownContainer.style.border = '1px solid #d1d5db';\n \n // Add model options to the dropdown menu\n models.forEach(model => {\n const option = document.createElement('div');\n option.className = 'cursor-pointer select-none relative py-2 pl-3 pr-9 hover:bg-gray-100';\n option.textContent = model.id;\n option.dataset.value = model.id;\n \n option.addEventListener('click', () => {\n modelIdInput.value = model.id;\n dropdownContainer.classList.add('hidden');\n \n // Automatically select category based on model name\n const modelValue = model.id.toLowerCase();\n if (modelValue.includes('pro')) {\n modelCategorySelect.value = 'Pro';\n customQuotaDiv.classList.add('hidden');\n modelQuotaInput.required = false;\n } else if (modelValue.includes('flash')) {\n modelCategorySelect.value = 'Flash';\n customQuotaDiv.classList.add('hidden');\n modelQuotaInput.required = false;\n }\n \n // Trigger input event so other listeners can respond\n modelIdInput.dispatchEvent(new Event('input'));\n });\n \n dropdownContainer.appendChild(option);\n });\n \n // Add the dropdown menu to the input element's parent\n modelIdInput.parentNode.appendChild(dropdownContainer);\n \n return dropdownContainer;\n };\n \n // Create dropdown menu\n const dropdown = createCustomDropdown();\n console.log(`Created custom dropdown with ${models.length} model options`);\n \n // Add input event to automatically select category based on model name and filter dropdown options\n modelIdInput.addEventListener('input', function() {\n const modelValue = this.value.toLowerCase();\n \n // Filter dropdown options based on input value\n const options = dropdown.querySelectorAll('div[data-value]');\n let hasVisibleOptions = false;\n \n options.forEach(option => {\n const optionValue = option.dataset.value.toLowerCase();\n if (optionValue.includes(modelValue)) {\n option.style.display = 'block';\n hasVisibleOptions = true;\n } else {\n option.style.display = 'none';\n }\n });\n \n // If there are matching options, show the dropdown menu\n if (hasVisibleOptions && modelValue) {\n dropdown.classList.remove('hidden');\n } else {\n dropdown.classList.add('hidden');\n }\n \n // Automatically select category based on input value\n if (modelValue.includes('pro')) {\n modelCategorySelect.value = 'Pro';\n customQuotaDiv.classList.add('hidden');\n modelQuotaInput.required = false;\n } else if (modelValue.includes('flash')) {\n modelCategorySelect.value = 'Flash';\n customQuotaDiv.classList.add('hidden');\n modelQuotaInput.required = false;\n }\n });\n \n // Add click event to show the dropdown menu and refresh models if needed\n modelIdInput.addEventListener('click', async function() {\n // Check if we need to refresh the model list\n const geminiKeysList = document.querySelectorAll('#gemini-keys-list .card-item');\n if (geminiKeysList.length > 0 && (cachedGeminiModels.length === 0 || models.length === 0)) {\n console.log(\"Refreshing Gemini models list on input click...\");\n await loadGeminiAvailableModels();\n return; // loadGeminiAvailableModels will recreate the dropdown with updated models\n }\n\n if (models.length > 0) {\n // Show all options\n const options = dropdown.querySelectorAll('div[data-value]');\n options.forEach(option => {\n option.style.display = 'block';\n });\n\n dropdown.classList.remove('hidden');\n }\n });\n \n // Hide dropdown menu when clicking elsewhere on the page\n document.addEventListener('click', function(e) {\n if (e.target !== modelIdInput && !dropdown.contains(e.target)) {\n dropdown.classList.add('hidden');\n }\n });\n \n // Remove datalist attribute from input if it exists\n modelIdInput.removeAttribute('list');\n }\n\n\n // --- Event Handlers ---\n\n // Add Gemini Key with support for batch input\n addGeminiKeyForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n\n // Prevent concurrent operations\n if (operationInProgress) {\n showError(t('operation_in_progress') || 'Another operation is in progress. Please wait.');\n return;\n }\n\n const formData = new FormData(addGeminiKeyForm);\n const data = Object.fromEntries(formData.entries());\n const geminiKeyInput = data.key ? data.key.trim() : '';\n\n // Check if input is empty\n if (!geminiKeyInput) {\n showError(\"API Key Value is required.\");\n return;\n }\n\n // Split input, supporting comma-separated keys (both English and Chinese commas) and line-separated keys\n const geminiKeys = geminiKeyInput\n .split(/[,，\\n\\r]+/) // Split by English comma, Chinese comma, newline, or carriage return\n .map(key => key.trim())\n .filter(key => key !== '');\n \n // Check if there are any keys to process\n if (geminiKeys.length === 0) {\n showError(\"No valid API Keys found.\");\n return;\n }\n\n // Gemini API Key format validation regex\n const geminiKeyRegex = /^AIzaSy[A-Za-z0-9_-]{33}$/;\n \n // Check format and remove duplicates\n const validKeys = [];\n const invalidKeys = [];\n const seenKeys = new Set();\n \n for (const key of geminiKeys) {\n // Skip duplicates\n if (seenKeys.has(key)) {\n continue;\n }\n \n seenKeys.add(key);\n \n // Validate format\n if (!geminiKeyRegex.test(key)) {\n invalidKeys.push(key);\n } else {\n validKeys.push(key);\n }\n }\n \n // If no valid keys, exit\n if (validKeys.length === 0) {\n showError(\"No valid API Keys found. Please check the format.\");\n return;\n }\n \n // Show warnings about invalid keys but continue with valid ones\n if (invalidKeys.length > 0) {\n const maskedInvalidKeys = invalidKeys.map(key => {\n if (key.length > 10) {\n return `${key.substring(0, 6)}...${key.substring(key.length - 4)}`;\n }\n return key;\n });\n showError(`Invalid API key format detected: ${maskedInvalidKeys.join(', ')}`);\n }\n \n operationInProgress = true; // Lock operations\n showLoading();\n let successCount = 0;\n let failureCount = 0;\n\n try {\n if (validKeys.length === 1) {\n // Single key - use original API with name support\n let keyData = { key: validKeys[0] };\n if (data.name) {\n keyData.name = data.name.trim();\n }\n\n const result = await apiFetch('/gemini-keys', {\n method: 'POST',\n body: JSON.stringify(keyData),\n });\n\n if (result && result.success) {\n successCount = 1;\n failureCount = 0;\n } else {\n successCount = 0;\n failureCount = 1;\n }\n } else if (validKeys.length <= 50) {\n // Medium batch - use single batch API call\n const result = await apiFetch('/gemini-keys/batch', {\n method: 'POST',\n body: JSON.stringify({ keys: validKeys }),\n });\n\n if (result && result.success) {\n successCount = result.successCount || 0;\n failureCount = result.failureCount || 0;\n\n // Log detailed results for debugging\n if (result.results && result.results.length > 0) {\n const failures = result.results.filter(r => !r.success);\n if (failures.length > 0) {\n console.warn('Some keys failed to add:', failures);\n }\n }\n } else {\n successCount = 0;\n failureCount = validKeys.length;\n }\n } else {\n // Large batch - split into chunks and process with limited concurrency\n const chunkSize = 20; // Process 20 keys per chunk\n const maxConcurrency = 3; // Maximum 3 concurrent requests\n\n // Split keys into chunks\n const chunks = [];\n for (let i = 0; i < validKeys.length; i += chunkSize) {\n chunks.push(validKeys.slice(i, i + chunkSize));\n }\n\n // Process chunks with limited concurrency and progress feedback\n for (let i = 0; i < chunks.length; i += maxConcurrency) {\n const currentChunks = chunks.slice(i, i + maxConcurrency);\n\n // Show progress\n const processedChunks = Math.floor(i / maxConcurrency) + 1;\n const totalChunks = Math.ceil(chunks.length / maxConcurrency);\n console.log(`Processing batch ${processedChunks}/${totalChunks} (${validKeys.length} total keys)`);\n\n // Process current batch of chunks concurrently\n const chunkPromises = currentChunks.map((chunk, chunkIndex) =>\n apiFetch('/gemini-keys/batch', {\n method: 'POST',\n body: JSON.stringify({ keys: chunk }),\n }).then(result => {\n console.log(`Chunk ${i + chunkIndex + 1} completed: ${result?.successCount || 0} success, ${result?.failureCount || 0} failed`);\n return result;\n }).catch(error => {\n console.error(`Error in chunk ${i + chunkIndex + 1} processing:`, error);\n return { success: false, successCount: 0, failureCount: chunk.length };\n })\n );\n\n const chunkResults = await Promise.all(chunkPromises);\n\n // Aggregate results\n chunkResults.forEach(result => {\n if (result && result.success) {\n successCount += result.successCount || 0;\n failureCount += result.failureCount || 0;\n } else {\n // If the entire chunk failed, count all keys as failures\n const chunkIndex = chunkResults.indexOf(result);\n const chunkSize = currentChunks[chunkIndex]?.length || 0;\n failureCount += chunkSize;\n }\n });\n\n // Small delay between batches to avoid overwhelming the server\n if (i + maxConcurrency < chunks.length) {\n await new Promise(resolve => setTimeout(resolve, 200));\n }\n }\n }\n } catch (error) {\n console.error('Error during batch add:', error);\n successCount = 0;\n failureCount = validKeys.length;\n } finally {\n operationInProgress = false; // Release lock\n hideLoading();\n }\n\n // Reset form and reload keys\n addGeminiKeyForm.reset();\n await loadGeminiKeys();\n\n // If keys were successfully added, refresh the available models list\n if (successCount > 0) {\n await loadGeminiAvailableModels(true); // Force refresh after adding new keys\n }\n\n // Show appropriate message based on results\n if (successCount > 0) {\n showSuccess(`Successfully added ${successCount} Gemini ${successCount === 1 ? 'key' : 'keys'}.`);\n } else {\n showError(`Failed to add any keys.`);\n }\n });\n\n // Global event delegation for Gemini key actions\n document.addEventListener('click', async (e) => {\n // Handle test gemini key button clicks\n if (e.target.classList.contains('test-gemini-key')) {\n const keyId = e.target.dataset.id;\n const testSection = document.querySelector(`.test-model-section[data-key-id=\"${keyId}\"]`);\n\n // Check if testSection exists (防止DOM重新渲染后元素不存在的错误)\n if (!testSection) {\n console.warn('Test section not found for keyId:', keyId);\n return;\n }\n\n // Toggle display status\n if (testSection.classList.contains('hidden')) {\n // Hide all other test areas\n document.querySelectorAll('.test-model-section').forEach(section => {\n section.classList.add('hidden');\n section.querySelector('.test-result')?.classList.add('hidden');\n });\n\n // Show current test area\n testSection.classList.remove('hidden');\n } else {\n testSection.classList.add('hidden');\n }\n return;\n }\n\n // Handle run test button clicks\n if (e.target.classList.contains('run-test-btn') && !e.target.id) { // Exclude the main \"run all test\" button\n const testSection = e.target.closest('.test-model-section');\n\n // Check if testSection exists (防止DOM重新渲染后元素不存在的错误)\n if (!testSection) {\n console.warn('Test section not found, possibly due to DOM re-rendering');\n return;\n }\n\n const keyId = testSection.dataset.keyId;\n const modelSelect = testSection.querySelector('.model-select');\n const resultDiv = testSection.querySelector('.test-result');\n const resultPre = resultDiv?.querySelector('pre');\n\n // Additional safety checks\n if (!keyId || !modelSelect || !resultDiv || !resultPre) {\n console.warn('Required elements not found in test section');\n return;\n }\n\n const modelId = modelSelect.value;\n\n if (!modelId) {\n showError(t('please_select_model'));\n return;\n }\n\n // Show result area and set \"Loading\" text\n resultDiv.classList.remove('hidden');\n resultPre.textContent = t('testing');\n\n // Send test request directly to handle both success and error responses\n let result = null;\n try {\n // Use direct fetch instead of apiFetch to get raw response\n const response = await fetch('/api/admin/test-gemini-key', {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n },\n credentials: 'include',\n body: JSON.stringify({ keyId, modelId })\n });\n\n // Parse response regardless of status code\n const contentType = response.headers.get(\"content-type\");\n if (contentType && contentType.indexOf(\"application/json\") !== -1) {\n result = await response.json();\n } else {\n const textContent = await response.text();\n result = {\n success: false,\n status: response.status,\n content: textContent || 'No response content'\n };\n }\n\n if (result) {\n const formattedContent = typeof result.content === 'object'\n ? JSON.stringify(result.content, null, 2)\n : result.content;\n\n if (result.success) {\n resultPre.textContent = `${t('test_passed')}\\n${t('status')}: ${result.status}\\n\\n${t('response')}:\\n${formattedContent}`;\n resultPre.className = 'text-xs bg-green-50 text-green-800 p-2 rounded overflow-x-auto';\n } else {\n resultPre.textContent = `${t('test_failed')}\\n${t('status')}: ${result.status}\\n\\n${t('response')}:\\n${formattedContent}`;\n resultPre.className = 'text-xs bg-red-50 text-red-800 p-2 rounded overflow-x-auto';\n }\n } else {\n resultPre.textContent = t('test_failed_no_response');\n resultPre.className = 'text-xs bg-red-50 text-red-800 p-2 rounded overflow-x-auto';\n }\n } catch (error) {\n // 只在测试区域显示网络错误\n resultPre.textContent = t('test_failed_network', error.message || t('unknown_error'));\n resultPre.className = 'text-xs bg-red-50 text-red-800 p-2 rounded overflow-x-auto';\n }\n return;\n }\n\n if (e.target.classList.contains('delete-gemini-key')) {\n const keyId = e.target.dataset.id;\n if (confirm(t('delete_confirm_gemini', keyId))) {\n const modal = e.target.closest('.fixed.inset-0');\n if (modal) {\n modal.classList.add('hidden');\n }\n\n const result = await apiFetch(`/gemini-keys/${encodeURIComponent(keyId)}`, {\n method: 'DELETE',\n });\n if (result && result.success) {\n await loadGeminiKeys(); // Wait for the list to reload\n showSuccess(`Gemini key ${keyId} deleted successfully!`);\n }\n }\n }\n\n // --- New: Clear Gemini Key Error ---\n if (e.target.classList.contains('clear-gemini-key-error')) {\n const keyId = e.target.dataset.id;\n const button = e.target;\n const modalErrorContainer = document.getElementById(`gemini-key-error-container-${keyId}`);\n const modalErrorSpan = modalErrorContainer?.querySelector('span');\n\n if (confirm(`Are you sure you want to clear the error status for key: ${keyId}?`)) {\n const result = await apiFetch('/clear-key-error', {\n method: 'POST',\n body: JSON.stringify({ keyId }),\n });\n\n if (result && result.success) {\n // Get the corresponding card and data\n const cardItem = document.querySelector(`.card-item[data-key-id=\"${keyId}\"]`);\n \n // Find the current key's data to get the usage value\n const keyData = result.updatedKey || { usage: 0 }; // Use the updated key data from the API response if available, otherwise default to 0\n \n // Replace the warning icon container with the Total display\n const warningContainer = cardItem?.querySelector('.warning-icon-container');\n if (warningContainer) {\n const totalHTML = `\n\n ${keyData.usage || '0'}\n
\n `;\n warningContainer.outerHTML = totalHTML;\n }\n \n // Remove error status text from modal\n const errorStatusP = button.closest('.modal-content').querySelector('p.text-red-600');\n if (errorStatusP) {\n errorStatusP.remove();\n }\n \n // Remove the button itself\n button.remove();\n showSuccess(`Error status cleared for key ${keyId}.`);\n } else {\n // Show error within the modal\n if (modalErrorContainer && modalErrorSpan) {\n modalErrorSpan.textContent = result?.error || 'Failed to clear error status.';\n modalErrorContainer.classList.remove('hidden');\n setTimeout(() => modalErrorContainer.classList.add('hidden'), 5000);\n } else {\n showError(result?.error || 'Failed to clear error status.'); // Fallback to global error\n }\n }\n }\n }\n // --- End Clear Gemini Key Error ---\n });\n\n // Add Worker Key with validation\n addWorkerKeyForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n const formData = new FormData(addWorkerKeyForm);\n const data = Object.fromEntries(formData.entries());\n \n // Validate worker key format - allow alphanumeric, hyphens, and underscores\n const workerKeyValue = data.key?.trim();\n if (!workerKeyValue) {\n showError('Worker key is required.');\n return;\n }\n \n const validKeyRegex = /^[a-zA-Z0-9_\\-]+$/;\n if (!validKeyRegex.test(workerKeyValue)) {\n showError('Worker key can only contain letters, numbers, underscores (_), and hyphens (-).');\n return;\n }\n \n const result = await apiFetch('/worker-keys', {\n method: 'POST',\n body: JSON.stringify(data),\n });\n if (result && result.success) {\n addWorkerKeyForm.reset();\n await loadWorkerKeys(); // Wait for the list to reload\n showSuccess('Worker key added successfully!');\n }\n });\n\n // Delete Worker Key (no changes needed)\n workerKeysListDiv.addEventListener('click', async (e) => {\n if (e.target.classList.contains('delete-worker-key')) {\n const key = e.target.dataset.key;\n\n // Use key in the path for deletion, matching backend expectation\n if (confirm(t('delete_confirm_worker', key))) {\n const result = await apiFetch(`/worker-keys/${encodeURIComponent(key)}`, {\n method: 'DELETE',\n });\n if (result && result.success) {\n await loadWorkerKeys(); // Wait for the list to reload\n showSuccess(`Worker key ${key} deleted successfully!`);\n }\n }\n }\n });\n\n // Save safety settings (no changes needed)\n async function saveSafetySettingsToServer(key, isEnabled) {\n try {\n const result = await apiFetch('/worker-keys/safety-settings', {\n method: 'POST',\n body: JSON.stringify({\n key: key,\n safetyEnabled: isEnabled\n }),\n });\n if (!result || !result.success) {\n console.error('Failed to save safety settings to server');\n showError('Failed to sync safety settings with server. Changes may not persist across browsers.');\n }\n } catch (error) {\n console.error('Error saving safety settings to server:', error);\n showError('Failed to sync safety settings with server. Changes may not persist across browsers.');\n }\n }\n\n // Generate Random Worker Key with valid format\n generateWorkerKeyBtn.addEventListener('click', () => {\n const validChars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';\n let randomKey = 'sk-';\n \n for (let i = 0; i < 20; i++) {\n const randomIndex = Math.floor(Math.random() * validChars.length);\n randomKey += validChars[randomIndex];\n }\n \n workerKeyValueInput.value = randomKey;\n });\n\n // --- Model Form Logic ---\n // Show/hide Custom Quota input based on category selection\n modelCategorySelect.addEventListener('change', (e) => {\n if (e.target.value === 'Custom') {\n customQuotaDiv.classList.remove('hidden');\n modelQuotaInput.required = true;\n } else {\n customQuotaDiv.classList.add('hidden');\n modelQuotaInput.required = false;\n modelQuotaInput.value = '';\n }\n });\n\n // Add/Update Model - Modified Submit Handler\n addModelForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n\n // Check if there are any Gemini API keys before allowing model addition\n const geminiKeysList = document.querySelectorAll('#gemini-keys-list .card-item');\n if (geminiKeysList.length === 0) {\n showError('无法添加模型：请先添加至少一个 Gemini API 密钥。');\n return;\n }\n\n const formData = new FormData(addModelForm);\n const data = {\n id: formData.get('id').trim(),\n category: formData.get('category')\n };\n\n // Only include dailyQuota if category is 'Custom' and input is visible/filled\n if (data.category === 'Custom') {\n const quotaInput = formData.get('dailyQuota')?.trim().toLowerCase();\n if (quotaInput === undefined || quotaInput === null || quotaInput === '') {\n showError(\"Daily Quota is required for Custom models. Enter a positive number, 'none', or '0'.\");\n return; // Stop submission\n }\n\n if (quotaInput === 'none' || quotaInput === '0') {\n } else {\n const quotaValue = parseInt(quotaInput, 10);\n if (isNaN(quotaValue) || quotaValue <= 0 || quotaInput !== quotaValue.toString()) {\n showError(\"Daily Quota for Custom models must be a positive whole number, 'none', or '0'.\");\n return;\n }\n data.dailyQuota = quotaValue;\n }\n }\n\n const result = await apiFetch('/models', {\n method: 'POST',\n body: JSON.stringify(data),\n });\n\n if (result && result.success) {\n addModelForm.reset();\n customQuotaDiv.classList.add('hidden');\n modelQuotaInput.required = false;\n await loadModels(); // Wait for models to reload\n await loadGeminiKeys(); // Wait for gemini keys to reload (as model changes affect them)\n showSuccess(`Model ${data.id} added/updated successfully!`);\n }\n });\n\n // Delete Model (no changes needed)\n modelsListDiv.addEventListener('click', async (e) => {\n if (e.target.classList.contains('delete-model')) {\n const modelId = e.target.dataset.id;\n\n // Use model ID in the path for deletion, matching backend expectation\n if (confirm(t('delete_confirm_model', modelId))) {\n const result = await apiFetch(`/models/${encodeURIComponent(modelId)}`, {\n method: 'DELETE',\n });\n if (result && result.success) {\n await loadModels(); // Wait for models to reload\n await loadGeminiKeys(); // Wait for gemini keys to reload\n showSuccess(`Model ${modelId} deleted successfully!`);\n }\n }\n }\n });\n\n // --- Category Quotas Modal Logic ---\n setCategoryQuotasBtn.addEventListener('click', async () => {\n hideError(categoryQuotasErrorDiv);\n const currentQuotas = await loadCategoryQuotas();\n if (currentQuotas) {\n proQuotaInput.value = currentQuotas.proQuota ?? 50;\n flashQuotaInput.value = currentQuotas.flashQuota ?? 1500;\n \n // Set placeholders to show default values\n proQuotaInput.placeholder = \"Default: 50\";\n flashQuotaInput.placeholder = \"Default: 1500\";\n \n categoryQuotasModal.classList.remove('hidden');\n } else {\n showError(\"Could not load current category quotas.\", categoryQuotasErrorDiv, categoryQuotasErrorDiv);\n }\n });\n\n closeCategoryQuotasModalBtn.addEventListener('click', () => {\n categoryQuotasModal.classList.add('hidden');\n });\n\n cancelCategoryQuotasBtn.addEventListener('click', () => {\n categoryQuotasModal.classList.add('hidden');\n });\n\n categoryQuotasModal.addEventListener('click', (e) => {\n if (e.target === categoryQuotasModal) {\n categoryQuotasModal.classList.add('hidden');\n }\n });\n\n categoryQuotasForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n hideError(categoryQuotasErrorDiv);\n\n const proQuota = parseInt(proQuotaInput.value, 10);\n const flashQuota = parseInt(flashQuotaInput.value, 10);\n\n if (isNaN(proQuota) || proQuota < 0 || isNaN(flashQuota) || flashQuota < 0) {\n showError(\"Quotas must be non-negative numbers.\", categoryQuotasErrorDiv, categoryQuotasErrorDiv);\n return;\n }\n\n const result = await apiFetch('/category-quotas', {\n method: 'POST',\n body: JSON.stringify({ proQuota, flashQuota }),\n });\n\n if (result && result.success) {\n cachedCategoryQuotas = { proQuota, flashQuota };\n categoryQuotasModal.classList.add('hidden');\n await loadGeminiKeys(); // Wait for gemini keys to reload\n showSuccess('Category quotas saved successfully!');\n } else {\n // Error already shown by apiFetch\n showError(result?.error || \"Failed to save category quotas.\", categoryQuotasErrorDiv, categoryQuotasErrorDiv);\n }\n });\n\n // --- Individual Quota Modal Logic ---\n closeIndividualQuotaModalBtn.addEventListener('click', () => {\n individualQuotaModal.classList.add('hidden');\n });\n\n cancelIndividualQuotaBtn.addEventListener('click', () => {\n individualQuotaModal.classList.add('hidden');\n });\n\n // --- Run All Test Logic ---\n runAllTestBtn.addEventListener('click', async () => {\n if (isRunningAllTests) {\n return; // Prevent multiple concurrent tests\n }\n\n await runAllGeminiKeysTest();\n });\n\n cancelAllTestBtn.addEventListener('click', () => {\n testCancelRequested = true;\n testStatusText.textContent = t('cancelling_tests');\n cancelAllTestBtn.disabled = true;\n });\n\n // Ignore All Errors Logic\n ignoreAllErrorsBtn.addEventListener('click', async () => {\n // Prevent concurrent operations\n if (operationInProgress) {\n showError(t('operation_in_progress') || 'Another operation is in progress. Please wait.');\n return;\n }\n\n if (!confirm(t('ignore_all_errors_confirm'))) {\n return;\n }\n\n try {\n operationInProgress = true; // Lock operations\n showLoading();\n const result = await apiFetch('/clear-all-errors', {\n method: 'POST',\n });\n\n if (result && result.success) {\n if (result.clearedCount === 0) {\n showSuccess(t('no_error_keys_found'));\n } else {\n showSuccess(t('error_keys_ignored', result.clearedCount));\n }\n await loadGeminiKeys(); // Reload the keys list\n }\n } catch (error) {\n console.error('Error ignoring error keys:', error);\n showError(t('failed_to_ignore_error_keys', error.message));\n } finally {\n operationInProgress = false; // Release lock\n hideLoading();\n }\n });\n\n // Clean Error Keys Logic\n cleanErrorKeysBtn.addEventListener('click', async () => {\n // Prevent concurrent operations\n if (operationInProgress) {\n showError(t('operation_in_progress') || 'Another operation is in progress. Please wait.');\n return;\n }\n\n if (!confirm(t('clean_error_keys_confirm'))) {\n return;\n }\n\n try {\n operationInProgress = true; // Lock operations\n showLoading();\n const result = await apiFetch('/error-keys', {\n method: 'DELETE',\n });\n\n if (result && result.success) {\n if (result.deletedCount === 0) {\n showSuccess(t('no_error_keys_found'));\n } else {\n showSuccess(t('error_keys_cleaned', result.deletedCount));\n }\n await loadGeminiKeys(); // Reload the keys list\n }\n } catch (error) {\n console.error('Error cleaning error keys:', error);\n showError(t('failed_to_clean_error_keys', error.message));\n } finally {\n operationInProgress = false; // Release lock\n hideLoading();\n }\n });\n\n individualQuotaModal.addEventListener('click', (e) => {\n if (e.target === individualQuotaModal) {\n individualQuotaModal.classList.add('hidden');\n }\n });\n\n individualQuotaForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n hideError(individualQuotaErrorDiv);\n\n const modelId = individualQuotaModelIdInput.value;\n const individualQuota = parseInt(individualQuotaValueInput.value, 10);\n\n if (isNaN(individualQuota) || individualQuota < 0) {\n showError(\"Individual quota must be a non-negative number.\", individualQuotaErrorDiv, individualQuotaErrorDiv);\n return;\n }\n\n // Find the existing model to update\n const modelToUpdate = cachedModels.find(m => m.id === modelId);\n if (!modelToUpdate) {\n showError(`Model ${modelId} not found.`, individualQuotaErrorDiv, individualQuotaErrorDiv);\n return;\n }\n\n // Create the payload with existing data plus the new individualQuota\n const payload = {\n id: modelId,\n category: modelToUpdate.category,\n individualQuota: individualQuota > 0 ? individualQuota : undefined // If 0, set to undefined to remove quota\n };\n\n // If it's a Custom model, preserve the dailyQuota\n if (modelToUpdate.category === 'Custom' && modelToUpdate.dailyQuota) {\n payload.dailyQuota = modelToUpdate.dailyQuota;\n }\n\n const result = await apiFetch('/models', {\n method: 'POST',\n body: JSON.stringify(payload),\n });\n\n if (result && result.success) {\n individualQuotaModal.classList.add('hidden');\n await loadModels(); // Reload models to show updated quota\n await loadGeminiKeys(); // Reload keys as they display model usage\n \n if (individualQuota > 0) {\n showSuccess(`Individual quota for ${modelId} set to ${individualQuota}.`);\n } else {\n showSuccess(`Individual quota for ${modelId} removed.`);\n }\n } else {\n showError(result?.error || \"Failed to set individual quota.\", individualQuotaErrorDiv, individualQuotaErrorDiv);\n }\n });\n\n\n // Verify if the user is authorized; redirect directly if not\n async function checkAuth() {\n try {\n if (localStorage.getItem('isLoggedIn') !== 'true') {\n window.location.href = '/login';\n return false;\n }\n\n const response = await fetch('/api/admin/models', { // Use an existing simple GET endpoint\n method: 'GET',\n credentials: 'include'\n });\n \n // Check for redirects that might indicate auth issues\n if (response.redirected) {\n const redirectUrl = new URL(response.url);\n if (redirectUrl.pathname.includes('login') || \n !redirectUrl.pathname.includes('/api/admin')) {\n console.log('Detected redirect to login page. Session likely expired.');\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n return false;\n }\n }\n\n if (!response.ok) {\n if (response.status === 401 || response.status === 403 || \n (response.status >= 300 && response.status < 400)) {\n console.log(`User is not authorized. Auth check failed with status: ${response.status}. Redirecting to login page.`);\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n }\n return false;\n }\n \n localStorage.setItem('isLoggedIn', 'true');\n authCheckingUI.classList.add('hidden');\n unauthorizedUI.classList.add('hidden');\n mainContentUI.classList.remove('hidden');\n return true;\n\n } catch (error) {\n console.error('Authorization check failed:', error);\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n return false;\n }\n }\n\n // --- Initial Load ---\n async function initialLoad() {\n const isAuthorized = await checkAuth();\n if (!isAuthorized) {\n console.log('User is not authorized. Aborting initial load.');\n return;\n }\n\n try {\n const results = await Promise.allSettled([\n loadModels(),\n loadCategoryQuotas(),\n loadWorkerKeys()\n ]);\n\n // Check results for critical failures (models/quotas)\n if (results[0].status === 'rejected') {\n console.error(`Initial load failed for models:`, results[0].reason);\n showError('Failed to load essential model data. Please refresh.');\n return;\n }\n if (results[1].status === 'rejected') {\n console.error(`Initial load failed for category quotas:`, results[1].reason);\n showError('Failed to load category quotas. Display might be incorrect.');\n }\n if (results[2].status === 'rejected') {\n console.error(`Initial load failed for worker keys:`, results[2].reason);\n }\n\n await loadGeminiKeys();\n // After loading Gemini keys, try to load available Gemini models\n await loadGeminiAvailableModels();\n\n // Check for updates\n await checkForUpdates();\n\n } catch (error) {\n console.error('Failed to load data:', error);\n showError('Failed to load data. Please refresh the page or try again later.');\n }\n\n // Add logout button functionality\n if (logoutButton) {\n logoutButton.addEventListener('click', async () => {\n showLoading();\n try {\n localStorage.removeItem('isLoggedIn'); // Clear local login status\n const response = await fetch('/api/logout', { method: 'POST', credentials: 'include' });\n window.location.href = '/login';\n } catch (error) {\n showError('Error during logout.');\n } finally {\n hideLoading();\n }\n });\n }\n }\n\n function initDarkMode() {\n const savedTheme = localStorage.getItem('theme');\n if (savedTheme === 'dark') {\n document.body.setAttribute('data-theme', 'dark');\n sunIcon.classList.add('hidden');\n moonIcon.classList.remove('hidden');\n } else {\n document.body.setAttribute('data-theme', 'light');\n sunIcon.classList.remove('hidden');\n moonIcon.classList.add('hidden');\n }\n\n darkModeToggle.addEventListener('click', () => {\n const currentTheme = document.body.getAttribute('data-theme');\n\n if (currentTheme === 'light') {\n document.body.setAttribute('data-theme', 'dark');\n localStorage.setItem('theme', 'dark');\n sunIcon.classList.add('hidden');\n moonIcon.classList.remove('hidden');\n } else {\n document.body.setAttribute('data-theme', 'light');\n localStorage.setItem('theme', 'light');\n moonIcon.classList.add('hidden');\n sunIcon.classList.remove('hidden');\n }\n });\n }\n\n function setupAuthRefresh() {\n const authCheckInterval = 5 * 60 * 1000;\n \n setInterval(async () => {\n console.log(\"Performing scheduled auth check...\");\n try {\n const response = await fetch('/api/admin/models', {\n method: 'GET',\n credentials: 'include'\n });\n \n if (response.redirected) {\n const redirectUrl = new URL(response.url);\n if (redirectUrl.pathname.includes('login') || \n !redirectUrl.pathname.includes('/api/admin')) {\n console.log('Session expired during scheduled check. Redirecting to login.');\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n }\n }\n \n if (!response.ok) {\n if (response.status === 401 || response.status === 403 || \n (response.status >= 300 && response.status < 400)) {\n console.log(`Auth check failed with status: ${response.status}. Redirecting to login.`);\n localStorage.removeItem('isLoggedIn');\n window.location.href = '/login';\n }\n }\n } catch (error) {\n console.error('Scheduled auth check failed:', error);\n }\n }, authCheckInterval);\n }\n\n // --- Tab Switching Functions ---\n function switchTab(tabName) {\n // Update tab buttons\n document.querySelectorAll('.api-tab').forEach(tab => {\n tab.classList.remove('active');\n });\n document.getElementById(`${tabName}-tab`).classList.add('active');\n\n // Update tab content\n document.querySelectorAll('.tab-content').forEach(content => {\n content.classList.add('hidden');\n });\n document.getElementById(`${tabName}-content`).classList.remove('hidden');\n\n // Handle Managed Models container visibility\n const modelsListElement = document.getElementById('models-list');\n const managedModelsSection = modelsListElement ? modelsListElement.closest('section') : null;\n if (managedModelsSection) {\n if (tabName === 'vertex') {\n // Hide Managed Models container when Vertex tab is active\n managedModelsSection.classList.add('hidden');\n } else {\n // Show Managed Models container when Gemini tab is active\n managedModelsSection.classList.remove('hidden');\n }\n }\n }\n\n // --- Vertex Configuration Functions ---\n async function loadVertexConfig() {\n try {\n const config = await apiFetch('/vertex-config');\n if (config) {\n renderVertexConfig(config);\n } else {\n renderVertexConfig(null);\n }\n } catch (error) {\n console.error('Error loading Vertex config:', error);\n renderVertexConfig(null);\n }\n }\n\n function renderVertexConfig(config) {\n if (!config || (!config.expressApiKey && !config.vertexJson)) {\n vertexConfigInfo.innerHTML = '';\n vertexStatus.innerHTML = '';\n\n // Clear form\n document.getElementById('express-api-key').value = '';\n document.getElementById('vertex-json').value = '';\n document.querySelector('input[name=\"auth_mode\"][value=\"service_account\"]').checked = true;\n toggleAuthMode();\n\n // Apply translations\n if (window.i18n) {\n window.i18n.applyTranslations();\n }\n return;\n }\n\n // Update status\n vertexStatus.innerHTML = '';\n\n // Update info display\n if (config.expressApiKey) {\n vertexConfigInfo.innerHTML = `\n :
\n: ${config.expressApiKey.substring(0, 10)}...${config.expressApiKey.substring(config.expressApiKey.length - 4)}
\n `;\n\n // Don't populate form fields when displaying existing config\n // This ensures the form is clean for new input\n document.querySelector('input[name=\"auth_mode\"][value=\"express\"]').checked = true;\n document.getElementById('express-api-key').value = '';\n document.getElementById('vertex-json').value = '';\n } else if (config.vertexJson) {\n try {\n const jsonData = JSON.parse(config.vertexJson);\n vertexConfigInfo.innerHTML = `\n:
\n: ${jsonData.project_id || 'N/A'}
\n: ${jsonData.client_email || 'N/A'}
\n `;\n\n // Don't populate form fields when displaying existing config\n document.querySelector('input[name=\"auth_mode\"][value=\"service_account\"]').checked = true;\n document.getElementById('express-api-key').value = '';\n document.getElementById('vertex-json').value = '';\n } catch (e) {\n vertexConfigInfo.innerHTML = '';\n }\n }\n\n toggleAuthMode();\n\n // Apply translations\n if (window.i18n) {\n window.i18n.applyTranslations();\n }\n }\n\n function toggleAuthMode() {\n const authMode = document.querySelector('input[name=\"auth_mode\"]:checked').value;\n const expressSection = document.getElementById('express-api-key-section');\n const serviceAccountSection = document.getElementById('service-account-section');\n\n if (authMode === 'service_account') {\n serviceAccountSection.classList.remove('hidden');\n expressSection.classList.add('hidden');\n } else {\n expressSection.classList.remove('hidden');\n serviceAccountSection.classList.add('hidden');\n }\n }\n\n async function saveVertexConfig(configData) {\n try {\n const result = await apiFetch('/vertex-config', {\n method: 'POST',\n body: JSON.stringify(configData),\n });\n\n if (result && result.success) {\n showSuccess('Vertex 配置保存成功！');\n\n // Clear the form after successful save\n document.getElementById('express-api-key').value = '';\n document.getElementById('vertex-json').value = '';\n\n await loadVertexConfig(); // Reload to show updated config\n return true;\n } else {\n showError('保存 Vertex 配置失败');\n return false;\n }\n } catch (error) {\n console.error('Error saving Vertex config:', error);\n showError('保存 Vertex 配置时发生错误');\n return false;\n }\n }\n\n async function testVertexConfig() {\n try {\n showLoading();\n const result = await apiFetch('/vertex-config/test', {\n method: 'POST',\n });\n\n hideLoading();\n\n if (result && result.success) {\n showSuccess('Vertex 配置测试成功！');\n } else {\n showError(result?.error || '测试 Vertex 配置失败');\n }\n } catch (error) {\n hideLoading();\n console.error('Error testing Vertex config:', error);\n showError('测试 Vertex 配置时发生错误');\n }\n }\n\n async function clearVertexConfig() {\n if (!confirm('确定要清除 Vertex 配置吗？')) {\n return;\n }\n\n try {\n const result = await apiFetch('/vertex-config', {\n method: 'DELETE',\n });\n\n if (result && result.success) {\n showSuccess('Vertex 配置已清除');\n await loadVertexConfig(); // Reload to show cleared config\n } else {\n showError('清除 Vertex 配置失败');\n }\n } catch (error) {\n console.error('Error clearing Vertex config:', error);\n showError('清除 Vertex 配置时发生错误');\n }\n }\n\n // --- Event Listeners ---\n\n // Tab switching\n geminiTab.addEventListener('click', () => switchTab('gemini'));\n vertexTab.addEventListener('click', () => switchTab('vertex'));\n\n // Authentication mode toggle\n document.querySelectorAll('input[name=\"auth_mode\"]').forEach(radio => {\n radio.addEventListener('change', toggleAuthMode);\n });\n\n // Vertex configuration form\n vertexConfigForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n const formData = new FormData(vertexConfigForm);\n const authMode = formData.get('auth_mode');\n\n let configData = {};\n\n if (authMode === 'express') {\n const expressApiKey = formData.get('express_api_key');\n if (!expressApiKey || !expressApiKey.trim()) {\n showError('请输入 Express API Key');\n return;\n }\n configData.expressApiKey = expressApiKey.trim();\n } else {\n const vertexJson = formData.get('vertex_json');\n if (!vertexJson || !vertexJson.trim()) {\n showError('请输入 Service Account JSON');\n return;\n }\n\n // Validate JSON\n try {\n JSON.parse(vertexJson);\n configData.vertexJson = vertexJson.trim();\n } catch (e) {\n showError('无效的 JSON 格式');\n return;\n }\n }\n\n // Check if configuration already exists\n try {\n const existingConfig = await apiFetch('/vertex-config');\n if (existingConfig && (existingConfig.expressApiKey || existingConfig.vertexJson)) {\n // Configuration exists, ask for confirmation\n const confirmMessage = window.i18n ?\n window.i18n.translate('vertex_config_overwrite_confirm') :\n '检测到已存在 Vertex 配置，是否要覆盖当前配置？';\n\n if (!confirm(confirmMessage)) {\n return; // User cancelled\n }\n }\n } catch (error) {\n console.warn('Failed to check existing Vertex config:', error);\n // Continue with save even if check fails\n }\n\n await saveVertexConfig(configData);\n });\n\n // Test and clear buttons\n testVertexConfigBtn.addEventListener('click', testVertexConfig);\n clearVertexConfigBtn.addEventListener('click', clearVertexConfig);\n\n // Settings modal functionality\n setupSettingsModal();\n\n initialLoad();\n initDarkMode();\n setupAuthRefresh();\n\n // Load Vertex config after initial load\n loadVertexConfig();\n\n // --- Settings Modal Functions ---\n function setupSettingsModal() {\n const settingsButton = document.getElementById('settings-button');\n const settingsModal = document.getElementById('settings-modal');\n const closeModalButton = document.getElementById('close-settings-modal');\n const cancelButton = document.getElementById('cancel-settings');\n const settingsForm = document.getElementById('settings-form');\n const keepaliveToggle = document.getElementById('keepalive-toggle');\n const maxRetryInput = document.getElementById('max-retry-input');\n\n // Open modal\n settingsButton.addEventListener('click', () => {\n loadSystemSettings();\n settingsModal.classList.remove('hidden');\n });\n\n // Close modal\n function closeModal() {\n settingsModal.classList.add('hidden');\n }\n\n closeModalButton.addEventListener('click', closeModal);\n cancelButton.addEventListener('click', closeModal);\n\n // Close modal when clicking outside\n settingsModal.addEventListener('click', (e) => {\n if (e.target === settingsModal) {\n closeModal();\n }\n });\n\n // Handle form submission\n settingsForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n await saveSystemSettings();\n });\n }\n\n async function loadSystemSettings() {\n try {\n const settings = await apiFetch('/system-settings');\n console.log('Loaded settings:', settings); // Debug log\n\n // Set KEEPALIVE toggle\n const keepaliveToggle = document.getElementById('keepalive-toggle');\n keepaliveToggle.checked = settings.keepalive === '1' || settings.keepalive === 1 || settings.keepalive === true;\n\n // Set MAX_RETRY input\n const maxRetryInput = document.getElementById('max-retry-input');\n maxRetryInput.value = settings.maxRetry || 3;\n\n // Set Web Search toggle\n const webSearchToggle = document.getElementById('web-search-toggle');\n webSearchToggle.checked = settings.webSearch === '1' || settings.webSearch === 1 || settings.webSearch === true;\n\n // Set Auto Test toggle\n const autoTestToggle = document.getElementById('auto-test-toggle');\n autoTestToggle.checked = settings.autoTest === '1' || settings.autoTest === 1 || settings.autoTest === true;\n\n } catch (error) {\n console.error('Error loading system settings:', error);\n // Set default values\n document.getElementById('keepalive-toggle').checked = false;\n document.getElementById('max-retry-input').value = 3;\n document.getElementById('web-search-toggle').checked = false;\n document.getElementById('auto-test-toggle').checked = false;\n }\n }\n\n async function saveSystemSettings() {\n try {\n const keepaliveToggle = document.getElementById('keepalive-toggle');\n const maxRetryInput = document.getElementById('max-retry-input');\n const webSearchToggle = document.getElementById('web-search-toggle');\n const autoTestToggle = document.getElementById('auto-test-toggle');\n\n const settings = {\n keepalive: keepaliveToggle.checked ? '1' : '0',\n maxRetry: parseInt(maxRetryInput.value) || 3,\n webSearch: webSearchToggle.checked ? '1' : '0',\n autoTest: autoTestToggle.checked ? '1' : '0'\n };\n\n const result = await apiFetch('/system-settings', {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json'\n },\n body: JSON.stringify(settings)\n });\n\n if (result.success) {\n showSuccess('系统设置已保存');\n document.getElementById('settings-modal').classList.add('hidden');\n } else {\n showError('保存系统设置失败');\n }\n } catch (error) {\n console.error('Error saving system settings:', error);\n showError('保存系统设置时发生错误');\n }\n }\n});\n\n // --- Update Check Functions ---\n /**\n * Compares two semantic version strings.\n * @param {string} v1 - The first version string.\n * @param {string} v2 - The second version string.\n * @returns {number} 1 if v1 > v2, -1 if v1 < v2, 0 if v1 === v2.\n */\n function compareVersions(v1, v2) {\n const parts1 = v1.split('.').map(Number);\n const parts2 = v2.split('.').map(Number);\n const len = Math.max(parts1.length, parts2.length);\n\n for (let i = 0; i < len; i++) {\n const p1 = parts1[i] || 0;\n const p2 = parts2[i] || 0;\n if (p1 > p2) return 1;\n if (p1 < p2) return -1;\n }\n return 0;\n }\n\n async function checkForUpdates() {\n try {\n // 1. Fetch local version\n const localVersionResponse = await fetch('/admin/version.txt?t=' + new Date().getTime());\n if (!localVersionResponse.ok) {\n console.warn('Could not fetch local version.txt');\n return;\n }\n const localVersion = (await localVersionResponse.text()).trim();\n\n // 2. Fetch latest release from GitHub\n const githubApiResponse = await fetch('https://api.github.com/repos/dreamhartley/gemini-proxy-panel/releases/latest');\n if (!githubApiResponse.ok) {\n console.warn('Could not fetch latest release from GitHub.');\n // Show version display even if GitHub check fails\n showVersionDisplay(localVersion);\n return;\n }\n const latestRelease = await githubApiResponse.json();\n const latestVersion = latestRelease.tag_name.replace('v', '').trim();\n\n // 3. Compare versions and show appropriate notifier\n const updateNotifier = document.getElementById('update-notifier');\n if (!updateNotifier) return;\n\n if (compareVersions(latestVersion, localVersion) > 0) {\n // Show update available notification (red)\n updateNotifier.classList.remove('hidden', 'version-display');\n updateNotifier.textContent = 'New';\n updateNotifier.setAttribute('data-tooltip', t('update_available'));\n updateNotifier.removeAttribute('title');\n } else {\n // Show current version (blue)\n showVersionDisplay(localVersion);\n }\n } catch (error) {\n console.error('Error checking for updates:', error);\n // Try to show version display even if update check fails\n try {\n const localVersionResponse = await fetch('/admin/version.txt?t=' + new Date().getTime());\n if (localVersionResponse.ok) {\n const localVersion = (await localVersionResponse.text()).trim();\n showVersionDisplay(localVersion);\n }\n } catch (versionError) {\n console.error('Error fetching local version:', versionError);\n }\n }\n }\n\n function showVersionDisplay(version) {\n const updateNotifier = document.getElementById('update-notifier');\n if (updateNotifier) {\n updateNotifier.classList.remove('hidden');\n updateNotifier.classList.add('version-display');\n updateNotifier.textContent = `v${version}`;\n updateNotifier.setAttribute('data-tooltip', t('current_is_latest'));\n updateNotifier.removeAttribute('title');\n }\n }\n\n// --- Debugging Commands ---\nwindow.show = function(what) {\n if (what === 'update') {\n const updateNotifier = document.getElementById('update-notifier');\n if (updateNotifier) {\n updateNotifier.classList.remove('hidden', 'version-display');\n updateNotifier.textContent = 'New';\n updateNotifier.setAttribute('data-tooltip', t('update_available'));\n updateNotifier.removeAttribute('title');\n console.log(\"Debug: Forcibly showing update notifier.\");\n return \"Update notifier shown.\";\n } else {\n const msg = \"Debug Error: #update-notifier element not found.\";\n console.error(msg);\n return msg;\n }\n } else if (what === 'version') {\n const updateNotifier = document.getElementById('update-notifier');\n if (updateNotifier) {\n showVersionDisplay('1.2.0');\n console.log(\"Debug: Forcibly showing version display.\");\n return \"Version display shown.\";\n } else {\n const msg = \"Debug Error: #update-notifier element not found.\";\n console.error(msg);\n return msg;\n }\n }\n return `Unknown command: ${what}`;\n};\n" }, { "path": "public/admin/style.css", "content": "/* Optional: Add custom CSS rules here if needed */\n\n/* Light mode warm background colors */\nbody[data-theme=\"light\"] {\n background-color: #f5f3f0 !important; /* Warm light khaki background */\n}\n\nbody[data-theme=\"light\"] .bg-gray-100 {\n background-color: #f5f3f0 !important; /* Warm light khaki background */\n}\n\nbody[data-theme=\"light\"] .bg-white {\n background-color: #faf9f7 !important; /* Slightly warmer white for cards */\n}\n\nbody[data-theme=\"light\"] section {\n background-color: #faf9f7 !important; /* Slightly warmer white for sections */\n}\n\n/* Light mode border and shadow adjustments for warm theme */\nbody[data-theme=\"light\"] .border,\nbody[data-theme=\"light\"] .border-gray-200,\nbody[data-theme=\"light\"] .border-gray-300 {\n border-color: #e6ddd4 !important; /* Warmer border color */\n}\n\nbody[data-theme=\"light\"] .shadow,\nbody[data-theme=\"light\"] .shadow-md,\nbody[data-theme=\"light\"] .shadow-lg {\n box-shadow: 0 1px 3px 0 rgba(139, 116, 88, 0.1), 0 1px 2px 0 rgba(139, 116, 88, 0.06) !important; /* Warmer shadow */\n}\n\n/* Adjust modal and card backgrounds for warm theme */\nbody[data-theme=\"light\"] .modal-content,\nbody[data-theme=\"light\"] .card-item {\n background-color: #faf9f7 !important;\n border-color: #e6ddd4 !important;\n}\n\n/* Light mode input field adjustments for warm theme */\nbody[data-theme=\"light\"] input[type=\"text\"],\nbody[data-theme=\"light\"] input[type=\"password\"],\nbody[data-theme=\"light\"] input[type=\"number\"],\nbody[data-theme=\"light\"] textarea,\nbody[data-theme=\"light\"] select {\n background-color: #f8f6f3 !important; /* Warm input background */\n border-color: #e6ddd4 !important; /* Warmer border */\n}\n\nbody[data-theme=\"light\"] input[type=\"text\"]:focus,\nbody[data-theme=\"light\"] input[type=\"password\"]:focus,\nbody[data-theme=\"light\"] input[type=\"number\"]:focus,\nbody[data-theme=\"light\"] textarea:focus,\nbody[data-theme=\"light\"] select:focus {\n background-color: #faf9f7 !important; /* Slightly lighter when focused */\n border-color: #3b82f6 !important; /* Keep blue focus border */\n box-shadow: 0 0 0 1px #3b82f6 !important;\n}\n\n/* Light mode button adjustments for warm theme */\nbody[data-theme=\"light\"] .bg-gray-100 {\n background-color: #f0ede8 !important; /* Warmer gray-100 */\n}\n\nbody[data-theme=\"light\"] .bg-gray-200 {\n background-color: #e8e3dc !important; /* Warmer gray-200 */\n}\n\nbody[data-theme=\"light\"] .hover\\:bg-gray-200:hover {\n background-color: #e0d9d0 !important; /* Warmer hover state */\n}\n\nbody[data-theme=\"light\"] .hover\\:bg-gray-300:hover {\n background-color: #d6cfc4 !important; /* Warmer hover state */\n}\n\n/* Light mode additional button and hover state adjustments */\nbody[data-theme=\"light\"] .bg-white {\n background-color: #faf9f7 !important; /* Warmer white for buttons */\n}\n\nbody[data-theme=\"light\"] .hover\\:bg-gray-50:hover {\n background-color: #f5f2ed !important; /* Warmer hover state */\n}\n\nbody[data-theme=\"light\"] .hover\\:bg-red-50:hover {\n background-color: #fef7f7 !important; /* Slightly warmer red hover */\n}\n\n/* Statistics bar styles */\n.stats-bar {\n transition: box-shadow 0.2s ease; /* Removed background-color transition */\n border: 1px solid #dbeafe; /* Light blue border */\n box-shadow: 0 1px 2px 0 rgba(0, 0, 0, 0.05); /* Subtle shadow */\n}\n\n.stats-bar:hover {\n box-shadow: 0 2px 4px 0 rgba(0, 0, 0, 0.08); /* Slightly larger shadow on hover */\n}\n\n/* Key cards container transition - smoother animation */\n.keys-grid {\n overflow: hidden;\n transition: max-height 0.5s cubic-bezier(0.4, 0, 0.2, 1), opacity 0.3s ease-in-out;\n max-height: 1000px; /* Set a reasonable max-height for transition */\n}\n\n/* Add specific style for hidden state for smoother transition */\n.keys-grid.hidden {\n max-height: 0;\n opacity: 0;\n margin-top: 0; /* Avoid margin when hidden */\n /* Ensure padding doesn't interfere when height is 0 */\n padding-top: 0;\n padding-bottom: 0;\n /* Add delay to opacity transition to wait for height change */\n transition: max-height 0.5s cubic-bezier(0.4, 0, 0.2, 1), opacity 0.2s ease-in-out 0.1s;\n}\n\n\n/* Toggle icon transitions with rotation */\n.toggle-icon svg {\n transition: transform 0.3s cubic-bezier(0.4, 0, 0.2, 1); /* Smoother cubic-bezier */\n}\n\n/* Rotate expand icon when grid is collapsed (default state when hidden) */\n.stats-bar .expand-icon {\n transform: rotate(0deg);\n}\n.stats-bar .collapse-icon {\n transform: rotate(180deg);\n}\n/* No need for specific rules when hidden/shown, script handles swapping icons */\n\n\n/* Dark mode adjustments for statistics bar */\nbody[data-theme=\"dark\"] .stats-bar {\n /* Removed background-color for dark mode */\n color: #d1d5db !important; /* Adjusted dark text */\n border-color: #4b5563 !important; /* Adjusted dark border */\n box-shadow: 0 1px 2px 0 rgba(0, 0, 0, 0.1); /* Slightly darker shadow */\n}\n\nbody[data-theme=\"dark\"] .stats-bar:hover {\n box-shadow: 0 2px 4px 0 rgba(0, 0, 0, 0.15); /* Slightly darker hover shadow */\n}\n\nbody[data-theme=\"dark\"] .toggle-icon svg {\n color: #9ca3af !important; /* Adjusted dark icon color */\n}\n\n/* Style for the individual key cards (optional improvements) */\nbody[data-theme=\"dark\"] .card-item {\n background-color: #374151; /* Dark background for cards */\n border-color: #4b5563;\n color: #d1d5db;\n}\nbody[data-theme=\"dark\"] .card-item h3 {\n color: #f3f4f6; /* Lighter heading in dark mode */\n}\nbody[data-theme=\"dark\"] .card-item p {\n color: #9ca3af; /* Lighter secondary text */\n}\nbody[data-theme=\"dark\"] .card-item .bg-blue-100 {\n background-color: #2563eb; /* Darker blue badge bg */\n color: #eff6ff; /* Light text for badge */\n}\n\n/* Tab styles */\n.api-tab {\n border-bottom-color: transparent;\n color: #6b7280;\n transition: all 0.2s ease;\n}\n\n.api-tab:hover {\n color: #374151;\n border-bottom-color: #d1d5db;\n}\n\n.api-tab.active {\n color: #3b82f6;\n border-bottom-color: #3b82f6;\n}\n\n/* Tab content */\n.tab-content {\n transition: opacity 0.2s ease;\n}\n\n.tab-content.hidden {\n display: none;\n}\n\n/* Dark mode tab styles */\nbody[data-theme=\"dark\"] .api-tab {\n color: #9ca3af;\n}\n\nbody[data-theme=\"dark\"] .api-tab:hover {\n color: #d1d5db;\n border-bottom-color: #4b5563;\n}\n\nbody[data-theme=\"dark\"] .api-tab.active {\n color: #60a5fa;\n border-bottom-color: #60a5fa;\n}\n\n/* Vertex configuration display styles */\n#vertex-config-display {\n transition: background-color 0.2s ease;\n}\n\nbody[data-theme=\"dark\"] #vertex-config-display {\n background-color: #374151;\n border-color: #4b5563;\n color: #d1d5db;\n}\n\n/* Authentication mode radio buttons */\n.form-radio {\n color: #3b82f6;\n}\n\nbody[data-theme=\"dark\"] .form-radio {\n background-color: #374151;\n border-color: #4b5563;\n color: #60a5fa;\n}\n\nbody[data-theme=\"dark\"] .form-radio:checked {\n background-color: #3b82f6;\n border-color: #3b82f6;\n}\n\n/* Status badges in dark mode */\n\n\n/* Button styles in dark mode */\nbody[data-theme=\"dark\"] .border-red-300 {\n border-color: #dc2626 !important;\n}\n\nbody[data-theme=\"dark\"] .text-red-700 {\n color: #fca5a5 !important;\n}\n\nbody[data-theme=\"dark\"] .hover\\:bg-red-50:hover {\n background-color: #7f1d1d !important;\n}\n\n/* Form elements in dark mode */\nbody[data-theme=\"dark\"] input[type=\"text\"],\nbody[data-theme=\"dark\"] textarea,\nbody[data-theme=\"dark\"] select {\n background-color: #374151;\n border-color: #4b5563;\n color: #d1d5db;\n}\n\nbody[data-theme=\"dark\"] input[type=\"text\"]:focus,\nbody[data-theme=\"dark\"] textarea:focus,\nbody[data-theme=\"dark\"] select:focus {\n border-color: #60a5fa;\n box-shadow: 0 0 0 1px #60a5fa;\n}\n\nbody[data-theme=\"dark\"] input[type=\"text\"]::placeholder,\nbody[data-theme=\"dark\"] textarea::placeholder {\n color: #9ca3af;\n}\n\n/* Dark mode base styles */\nbody[data-theme=\"dark\"] {\n background-color: #1a202c !important;\n color: #e2e8f0 !important;\n}\n\nbody[data-theme=\"dark\"] .container,\nbody[data-theme=\"dark\"] #main-content {\n background-color: #1a202c !important;\n}\n\nbody[data-theme=\"dark\"] .bg-white,\nbody[data-theme=\"dark\"] section,\nbody[data-theme=\"dark\"] .modal-content,\nbody[data-theme=\"dark\"] .card-item {\n background-color: #2d3748 !important;\n}\n\nbody[data-theme=\"dark\"] .bg-gray-100 {\n background-color: #1a202c !important;\n}\n\nbody[data-theme=\"dark\"] .text-gray-700,\nbody[data-theme=\"dark\"] .text-gray-800,\nbody[data-theme=\"dark\"] .text-gray-900,\nbody[data-theme=\"dark\"] .font-medium,\nbody[data-theme=\"dark\"] h1,\nbody[data-theme=\"dark\"] h2,\nbody[data-theme=\"dark\"] h3 {\n color: #e2e8f0 !important;\n}\n\nbody[data-theme=\"dark\"] .text-gray-500,\nbody[data-theme=\"dark\"] .text-gray-600 {\n color: #a0aec0 !important;\n}\n\nbody[data-theme=\"dark\"] .border,\nbody[data-theme=\"dark\"] input,\nbody[data-theme=\"dark\"] select,\nbody[data-theme=\"dark\"] textarea {\n border-color: #4a5568 !important;\n}\n\nbody[data-theme=\"dark\"] input,\nbody[data-theme=\"dark\"] select,\nbody[data-theme=\"dark\"] textarea {\n background-color: #2d3748 !important;\n color: #e2e8f0 !important;\n}\n\nbody[data-theme=\"dark\"] .bg-gray-200,\nbody[data-theme=\"dark\"] .bg-gray-300 {\n background-color: #4a5568 !important;\n}\n\nbody[data-theme=\"dark\"] .hover\\:bg-gray-300:hover {\n background-color: #2d3748 !important;\n}\n\nbody[data-theme=\"dark\"] .bg-indigo-600 {\n background-color: #4c51bf !important;\n}\n\nbody[data-theme=\"dark\"] .hover\\:bg-indigo-700:hover {\n background-color: #434190 !important;\n}\n\nbody[data-theme=\"dark\"] .text-indigo-600 {\n color: #7f9cf5 !important;\n}\n\nbody[data-theme=\"dark\"] .hover\\:text-indigo-800:hover {\n color: #6b46c1 !important;\n}\n\nbody[data-theme=\"dark\"] .bg-blue-50,\nbody[data-theme=\"dark\"] .bg-blue-100 {\n background-color: #2c5282 !important;\n}\n\nbody[data-theme=\"dark\"] .text-blue-800 {\n color: #bee3f8 !important;\n}\n\nbody[data-theme=\"dark\"] .text-xs.px-2.py-1.bg-blue-100.text-blue-800.rounded-full {\n background-color: #3182ce !important;\n color: #ffffff !important;\n}\n\nbody[data-theme=\"dark\"] .p-4.border.rounded-md.bg-white {\n background-color: #2d3748 !important;\n}\n\nbody[data-theme=\"dark\"] .text-gray-900 {\n color: #e2e8f0 !important;\n}\n\nbody[data-theme=\"dark\"] .focus\\:border-indigo-500:focus,\nbody[data-theme=\"dark\"] .focus\\:ring-indigo-500:focus {\n border-color: #7f9cf5 !important;\n box-shadow: 0 0 0 1px #7f9cf5 !important;\n}\n\n/* Legacy toggle switches (for other parts of the app) */\nbody[data-theme=\"dark\"] .toggle-label {\n background-color: #4a5568 !important;\n}\n\nbody[data-theme=\"dark\"] .toggle-checkbox:checked + .toggle-label {\n background-color: #68D391 !important;\n}\n\nbody[data-theme=\"dark\"] .toggle-checkbox {\n border-color: #4a5568 !important;\n}\n\nbody[data-theme=\"dark\"] .toggle-checkbox:checked {\n border-color: #68D391 !important;\n}\n\nbody[data-theme=\"dark\"] label[for^=\"safety-toggle\"] {\n color: #e2e8f0 !important;\n}\n\nbody[data-theme=\"dark\"] .text-green-600 {\n color: #68D391 !important;\n}\n\nbody[data-theme=\"dark\"] .text-red-600 {\n color: #F56565 !important;\n}\n\n/* Settings Modal Toggle Switch */\n.toggle-switch {\n position: relative;\n display: inline-block;\n width: 56px;\n height: 32px;\n align-self: center;\n flex-shrink: 0;\n}\n\n.toggle-switch input {\n opacity: 0;\n width: 0;\n height: 0;\n}\n\n.toggle-slider {\n position: absolute;\n cursor: pointer;\n top: 0;\n left: 0;\n right: 0;\n bottom: 0;\n background-color: #ccc;\n transition: all .3s ease;\n border-radius: 32px;\n}\n\n.toggle-slider:before {\n position: absolute;\n content: \"\";\n height: 24px;\n width: 24px;\n left: 4px;\n top: 4px;\n background-color: white;\n transition: all .3s ease;\n border-radius: 50%;\n box-shadow: 0 2px 4px rgba(0,0,0,0.2);\n}\n\ninput:checked + .toggle-slider {\n background-color: #10B981;\n}\n\ninput:checked + .toggle-slider:before {\n transform: translateX(24px);\n}\n\n/* Dark mode for settings modal */\nbody[data-theme=\"dark\"] #settings-modal .bg-white {\n background-color: #1f2937 !important;\n color: #e5e7eb !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal .text-gray-900 {\n color: #e5e7eb !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal .text-gray-700 {\n color: #d1d5db !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal .text-gray-500 {\n color: #9ca3af !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal .border-gray-300 {\n border-color: #4b5563 !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal input[type=\"number\"] {\n background-color: #374151 !important;\n color: #e5e7eb !important;\n border-color: #4b5563 !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal .bg-gray-100 {\n background-color: #374151 !important;\n color: #e5e7eb !important;\n}\n\nbody[data-theme=\"dark\"] #settings-modal .bg-gray-100:hover {\n background-color: #4b5563 !important;\n}\n\n/* Dark mode for toggle switch */\nbody[data-theme=\"dark\"] .toggle-slider {\n background-color: #4b5563 !important;\n}\n\nbody[data-theme=\"dark\"] input:checked + .toggle-slider {\n background-color: #10B981 !important;\n}\n\n/* Custom dropdown dark mode styles */\nbody[data-theme=\"dark\"] #custom-model-dropdown {\n background-color: #2d3748 !important;\n border-color: #4a5568 !important;\n}\n\nbody[data-theme=\"dark\"] #custom-model-dropdown div {\n color: #e2e8f0 !important;\n}\n\nbody[data-theme=\"dark\"] #custom-model-dropdown div:hover {\n background-color: #4a5568 !important;\n}\n\n/* Dark mode button adjustments for Managed Models */\nbody[data-theme=\"dark\"] .set-individual-quota {\n color: #90cdf4 !important; /* Lighter blue */\n}\nbody[data-theme=\"dark\"] .set-individual-quota:hover {\n color: #bee3f8 !important; /* Even lighter blue on hover */\n}\nbody[data-theme=\"dark\"] .delete-model {\n color: #fbb6ce !important; /* Lighter red/pink */\n}\nbody[data-theme=\"dark\"] .delete-model:hover {\n color: #fecaca !important; /* Even lighter red/pink on hover */\n}\n\n/* Dark mode adjustments for text-based delete buttons */\nbody[data-theme=\"dark\"] .delete-worker-key,\nbody[data-theme=\"dark\"] .delete-gemini-key { /* .delete-model already covered above */\n color: #fbb6ce !important; /* Lighter red/pink */\n}\nbody[data-theme=\"dark\"] .delete-worker-key:hover,\nbody[data-theme=\"dark\"] .delete-gemini-key:hover { /* .delete-model already covered above */\n color: #fecaca !important; /* Even lighter red/pink on hover */\n}\n\n/* Dark mode adjustments for text-based action buttons */\nbody[data-theme=\"dark\"] .test-gemini-key { /* .set-individual-quota already covered above */\n color: #90cdf4 !important; /* Lighter blue */\n}\nbody[data-theme=\"dark\"] .test-gemini-key:hover { /* .set-individual-quota already covered above */\n color: #bee3f8 !important; /* Even lighter blue on hover */\n}\n\n/* Adjust Generate Worker Key button (text-based) */\nbody[data-theme=\"dark\"] #generate-worker-key {\n color: #a3bffa !important; /* Lighter indigo */\n}\nbody[data-theme=\"dark\"] #generate-worker-key:hover {\n color: #c3dafe !important; /* Even lighter indigo */\n}\n\n/* Adjust Set Category Quotas button (background) */\nbody[data-theme=\"dark\"] #set-category-quotas-btn {\n background-color: #4299e1 !important; /* Brighter blue */\n color: #ffffff !important;\n}\nbody[data-theme=\"dark\"] #set-category-quotas-btn:hover {\n background-color: #63b3ed !important; /* Lighter blue on hover */\n}\n\n/* Adjust Run Test button (background) */\nbody[data-theme=\"dark\"] .run-test-btn {\n background-color: #48bb78 !important; /* Brighter green */\n color: #ffffff !important;\n}\nbody[data-theme=\"dark\"] .run-test-btn:hover {\n background-color: #68d391 !important; /* Lighter green on hover */\n}\n\n/* Adjust Modal Cancel buttons (background) */\nbody[data-theme=\"dark\"] #cancel-category-quotas,\nbody[data-theme=\"dark\"] #cancel-individual-quota {\n background-color: #4a5568 !important; /* Darker gray background */\n color: #e2e8f0 !important; /* Light text */\n border-color: #718096 !important; /* Gray border */\n}\nbody[data-theme=\"dark\"] #cancel-category-quotas:hover,\nbody[data-theme=\"dark\"] #cancel-individual-quota:hover {\n background-color: #718096 !important; /* Lighter gray background on hover */\n}\n\n/* Dark mode adjustments for Ignore Error button */\nbody[data-theme=\"dark\"] .clear-gemini-key-error {\n color: #faf089 !important; /* Lighter yellow (Tailwind yellow-300) */\n}\nbody[data-theme=\"dark\"] .clear-gemini-key-error:hover {\n color: #f6e05e !important;\n}\n\n/* Dark mode adjustments for Vertex configuration buttons */\nbody[data-theme=\"dark\"] #clear-vertex-config {\n color: #fbb6ce !important; /* Lighter red/pink */\n border-color: #f56565 !important; /* Red border */\n}\nbody[data-theme=\"dark\"] #clear-vertex-config:hover {\n background-color: #742a2a !important; /* Dark red background on hover */\n color: #fecaca !important; /* Even lighter red/pink on hover */\n}\n\n/* Dark mode adjustments for radio buttons */\nbody[data-theme=\"dark\"] input[type=\"radio\"] {\n background-color: #2d3748 !important;\n border-color: #4a5568 !important;\n}\n\nbody[data-theme=\"dark\"] input[type=\"radio\"]:checked {\n background-color: #3182ce !important;\n border-color: #3182ce !important;\n}\n\nbody[data-theme=\"dark\"] input[type=\"radio\"]:focus {\n box-shadow: 0 0 0 2px #3182ce !important;\n}\n\n/* Dark mode adjustments for status badges */\nbody[data-theme=\"dark\"] .bg-green-100.text-green-800 {\n background-color: #065f46 !important;\n color: #d1fae5 !important;\n}\n\nbody[data-theme=\"dark\"] .bg-gray-100.text-gray-800 {\n background-color: #374151 !important;\n color: #d1d5db !important;\n}\n\n/* Update Notifier Styles */\n#update-notifier {\n display: inline-flex;\n align-items: center;\n justify-content: center;\n min-width: 32px;\n height: 18px;\n background: linear-gradient(135deg, #ff4444, #ff6666);\n border-radius: 9px;\n margin-left: 12px;\n vertical-align: text-top;\n cursor: pointer;\n transition: all 0.3s ease-in-out;\n font-size: 10px;\n font-weight: bold;\n color: white;\n text-shadow: 0 1px 2px rgba(0, 0, 0, 0.3);\n box-shadow: 0 2px 4px rgba(255, 68, 68, 0.3);\n position: relative;\n transform: translateY(-2px);\n padding: 0 6px;\n}\n\n/* Version Display Styles (when no update available) */\n#update-notifier.version-display {\n background: linear-gradient(135deg, #3b82f6, #60a5fa) !important;\n box-shadow: 0 2px 4px rgba(59, 130, 246, 0.3) !important;\n min-width: auto;\n padding: 0 8px;\n}\n\n#update-notifier.version-display:hover {\n transform: translateY(-2px) scale(1.05);\n box-shadow: 0 4px 8px rgba(59, 130, 246, 0.4);\n}\n\n#update-notifier:not(.version-display):hover {\n transform: translateY(-2px) scale(1.1);\n box-shadow: 0 4px 8px rgba(255, 68, 68, 0.4);\n}\n\n#update-notifier.hidden {\n display: none;\n}\n\n/* Update Notifier Tooltip */\n#update-notifier::after {\n content: attr(data-tooltip);\n position: absolute;\n top: 100%;\n left: 50%;\n transform: translateX(-50%);\n background-color: #333;\n color: white;\n padding: 8px 12px;\n border-radius: 6px;\n font-size: 12px;\n font-weight: normal;\n white-space: nowrap;\n opacity: 0;\n visibility: hidden;\n transition: all 0.3s ease-in-out;\n z-index: 1000;\n margin-top: 5px;\n text-shadow: none;\n box-shadow: 0 2px 8px rgba(0, 0, 0, 0.2);\n}\n\n#update-notifier:hover::after {\n opacity: 1;\n visibility: visible;\n}\n" }, { "path": "public/admin/version.txt", "content": "1.3.4" }, { "path": "public/i18n.js", "content": "// 国际化配置和管理脚本\nclass I18n {\n constructor() {\n this.currentLanguage = 'zh'; // 默认中文\n this.translations = {\n zh: {\n // 登录页面\n 'password': '密码',\n 'enter_admin_password': '请输入管理员密码',\n 'login': '登录',\n \n // 通用\n 'loading': '加载中...',\n 'error': '错误',\n 'success': '成功',\n 'cancel': '取消',\n 'save': '保存',\n 'delete': '删除',\n 'edit': '编辑',\n 'add': '添加',\n 'test': '测试',\n 'generate': '生成',\n 'optional': '可选',\n 'required': '必填',\n \n // 认证相关\n 'verifying_identity': '正在验证身份...',\n 'auth_check_timeout': '如果页面长时间无响应，请',\n 'return_to_login': '返回登录页面',\n 'unauthorized_access': '未授权访问',\n 'need_login_message': '您需要登录才能访问管理页面。',\n 'go_to_login': '前往登录',\n 'logout': '退出登录',\n \n // Gemini API Keys 部分\n 'add_new_gemini_key': '添加新的 Gemini 密钥',\n 'name_optional': '名称（可选）',\n 'name_placeholder': '例如：个人密钥',\n 'name_help': '用于识别的友好名称。如果未提供，将使用自动生成的ID。',\n 'api_key_value': 'API 密钥值',\n 'enter_gemini_api_key': '请输入 Gemini API 密钥',\n 'gemini_key_batch_help': '支持批量添加：使用逗号分隔多个密钥，或每行一个密钥。',\n 'add_gemini_key': '添加 Gemini 密钥',\n 'run_all_test': '运行所有测试',\n 'ignore_all_errors': '忽略所有报错',\n 'clean_error_keys': '清理报错密钥',\n 'loading_keys': '加载密钥中...',\n\n // Vertex AI 配置部分\n 'current_vertex_config': '当前 Vertex 配置',\n 'loading_vertex_config': '加载配置中...',\n 'vertex_config_title': 'Vertex AI 配置',\n 'auth_mode': '认证模式',\n 'express_mode': '快捷模式 (API Key)',\n 'service_account_mode': '服务账号 (JSON)',\n 'express_api_key': 'Express API Key',\n 'enter_express_api_key': '请输入 Vertex AI Express API Key',\n 'express_api_key_help': '用于 Vertex AI Express Mode 的 API Key',\n 'service_account_json': 'Service Account JSON',\n 'enter_vertex_json': '请输入 Vertex AI Service Account JSON',\n 'vertex_json_help': '完整的 Google Cloud Service Account JSON 配置',\n 'save_vertex_config': '保存 Vertex 配置',\n 'test_vertex_config': '测试配置',\n 'clear_vertex_config': '清除配置',\n 'no_vertex_config': '未配置 Vertex AI',\n 'project_id': '项目 ID',\n 'client_email': '客户端邮箱',\n 'api_key': 'API Key',\n 'invalid_json': '无效的 JSON 配置',\n 'vertex_config_overwrite_confirm': '检测到已存在 Vertex 配置，是否要覆盖当前配置？',\n 'clean_error_keys_confirm': '确定要删除所有带错误标记的 Gemini 密钥吗？此操作不可撤销。',\n 'ignore_all_errors_confirm': '确定要清除所有带错误标记的 Gemini 密钥的错误状态吗？密钥将保留但错误标记会被移除。',\n 'no_error_keys_found': '没有找到带错误标记的密钥。',\n 'error_keys_cleaned': '成功清理了 {0} 个报错密钥。',\n 'error_keys_ignored': '成功忽略了 {0} 个报错密钥的错误状态。',\n 'failed_to_clean_error_keys': '清理报错密钥失败：{0}',\n 'failed_to_ignore_error_keys': '忽略报错密钥失败：{0}',\n \n // Worker API Keys 部分\n 'add_new_worker_key': '添加新的 Worker 密钥',\n 'generate_or_enter_key': '生成或输入强密钥',\n 'generate_random_key': '生成随机密钥',\n 'description_optional': '描述（可选）',\n 'description_placeholder': '例如：客户端应用 A',\n 'add_worker_key': '添加 Worker 密钥',\n 'safety_settings_help': '安全设置：默认启用。禁用时，模型允许生成 NSFW 内容。',\n \n // Models 部分\n 'add_model': '添加模型',\n 'model_id': '模型 ID',\n 'model_id_placeholder': '例如：gemini-1.5-flash-latest',\n 'model_id_help': '选择或输入模型 ID',\n 'category': '类别',\n 'daily_quota_custom': '每日配额（自定义）',\n 'quota_placeholder': '例如：1500，或 \\'none\\'/\\'0\\' 表示无限制',\n 'quota_help': '仅适用于\"自定义\"类别。设置此特定模型的每日最大请求数。输入 \\'none\\' 或 \\'0\\' 表示无限制。',\n 'set_category_quotas': '设置类别配额',\n 'loading_models': '加载模型中...',\n \n // 配额设置模态框\n 'set_category_quotas_title': '设置类别配额',\n 'pro_models_daily_quota': 'Pro 模型每日配额',\n 'flash_models_daily_quota': 'Flash 模型每日配额',\n 'save_quotas': '保存配额',\n \n // 独立配额模态框\n 'set_quota': '设置配额',\n 'individual_daily_quota': '独立配额',\n 'individual_quota_placeholder': '默认：0（无独立配额）',\n 'individual_quota_help': '输入 0 表示无独立配额。独立配额将覆盖类别配额。',\n 'save_quota': '保存配额',\n \n // 测试进度\n 'running_all_tests': '正在运行所有测试',\n 'progress': '进度',\n 'preparing_tests': '准备测试中...',\n \n // 按钮和操作\n 'set_individual_quota': '设置独立配额',\n 'ignore_error': '忽略错误',\n 'clear_error': '清除错误',\n\n // 弹窗和动态内容\n 'id': 'ID',\n 'key_preview': '密钥预览',\n 'total_usage_today': '今日总使用量',\n 'date': '日期',\n 'error_status': '错误状态',\n 'category_usage': '类别使用情况',\n 'pro_models': 'Pro 模型',\n 'flash_models': 'Flash 模型',\n 'test_api_key': '测试 API 密钥',\n 'select_a_model': '选择一个模型...',\n 'run_test': '运行测试',\n 'testing': '测试中...',\n 'test_passed': '测试通过！',\n 'test_failed': '测试失败。',\n 'status': '状态',\n 'response': '响应',\n 'no_description': '无描述',\n 'created': '创建于',\n 'safety_settings': '安全设置',\n 'enabled': '已启用',\n 'disabled': '已禁用',\n 'unlimited': '无限制',\n 'individual_quota': '独立配额',\n 'quota': '配额',\n 'set_quota_btn': '设置配额',\n 'delete_confirm_gemini': '您确定要删除 Gemini 密钥 ID：{0} 吗？',\n 'delete_confirm_worker': '您确定要删除 Worker 密钥：{0} 吗？',\n 'delete_confirm_model': '您确定要删除模型：{0} 吗？',\n 'please_select_model': '请选择一个模型进行测试',\n 'tests_cancelled': '测试已被用户取消',\n 'all_tests_completed': '所有测试已完成！',\n 'completed_testing': '已完成测试 {0} 个 Gemini 密钥，使用模型 {1}。',\n 'test_run_failed': '测试运行失败',\n 'cancelling_tests': '正在取消测试...',\n 'testing_batch': '正在测试批次 {0}...',\n 'completed_tests': '已完成 {0} / {1} 测试',\n 'no_gemini_keys_found': '未找到要测试的 Gemini 密钥。',\n 'network_error': '网络错误',\n 'test_failed_no_response': '测试失败：服务器无响应',\n 'test_failed_network': '测试失败：网络错误 - {0}',\n 'unknown_error': '未知错误',\n 'test_run_cancelled': '测试运行已取消。',\n 'failed_to_run_tests': '运行所有测试失败：{0}',\n\n // 系统设置\n 'system_settings': '系统设置',\n 'keepalive_setting': 'KEEPALIVE 模式',\n 'keepalive_description': '启用后将使用保持连接模式处理请求',\n 'max_retry_setting': '最大重试次数',\n 'max_retry_description': 'API请求失败时的最大重试次数（默认：3）',\n 'web_search_setting': '联网搜索',\n 'web_search_description': '启用后将在模型列表中显示带-search后缀的联网搜索模型',\n 'auto_test_setting': '自动批量测试',\n 'auto_test_description': '启用后将在每天北京时间4点自动进行批量测试',\n 'update_available': '有可用的新版本',\n 'current_is_latest': '当前为最新版本'\n },\n en: {\n // 登录页面\n 'password': 'Password',\n 'enter_admin_password': 'Enter admin password',\n 'login': 'Login',\n \n // 通用\n 'loading': 'Loading...',\n 'error': 'Error',\n 'success': 'Success',\n 'cancel': 'Cancel',\n 'save': 'Save',\n 'delete': 'Delete',\n 'edit': 'Edit',\n 'add': 'Add',\n 'test': 'Test',\n 'generate': 'Generate',\n 'optional': 'Optional',\n 'required': 'Required',\n \n // 认证相关\n 'verifying_identity': 'Verifying identity...',\n 'auth_check_timeout': 'If the page doesn\\'t respond for a long time, please',\n 'return_to_login': 'return to the login page',\n 'unauthorized_access': 'Unauthorized Access',\n 'need_login_message': 'You need to log in to access the admin page.',\n 'go_to_login': 'Go to Login',\n 'logout': 'Logout',\n \n // Gemini API Keys 部分\n 'add_new_gemini_key': 'Add New Gemini Key',\n 'name_optional': 'Name (Optional)',\n 'name_placeholder': 'e.g., Personal Key',\n 'name_help': 'A friendly name for identification. If not provided, an auto-generated ID will be used.',\n 'api_key_value': 'API Key Value',\n 'enter_gemini_api_key': 'Enter Gemini API Key',\n 'gemini_key_batch_help': 'Batch addition supported: Use commas to separate multiple keys, or one key per line.',\n 'add_gemini_key': 'Add Gemini Key',\n 'run_all_test': 'Run All Test',\n 'ignore_all_errors': 'Ignore All Errors',\n 'clean_error_keys': 'Clean Error Keys',\n 'loading_keys': 'Loading keys...',\n 'clean_error_keys_confirm': 'Are you sure you want to delete all Gemini keys with error status? This action cannot be undone.',\n 'ignore_all_errors_confirm': 'Are you sure you want to clear error status for all Gemini keys with errors? Keys will be kept but error marks will be removed.',\n 'no_error_keys_found': 'No keys with error status found.',\n 'error_keys_cleaned': 'Successfully cleaned {0} error keys.',\n 'error_keys_ignored': 'Successfully ignored error status for {0} keys.',\n 'failed_to_clean_error_keys': 'Failed to clean error keys: {0}',\n 'failed_to_ignore_error_keys': 'Failed to ignore error keys: {0}',\n\n // Vertex AI Configuration\n 'current_vertex_config': 'Current Vertex Configuration',\n 'loading_vertex_config': 'Loading configuration...',\n 'vertex_config_title': 'Vertex AI Configuration',\n 'auth_mode': 'Authentication Mode',\n 'express_mode': 'Express Mode (API Key)',\n 'service_account_mode': 'Service Account (JSON)',\n 'express_api_key': 'Express API Key',\n 'enter_express_api_key': 'Enter Vertex AI Express API Key',\n 'express_api_key_help': 'API Key for Vertex AI Express Mode',\n 'service_account_json': 'Service Account JSON',\n 'enter_vertex_json': 'Enter Vertex AI Service Account JSON',\n 'vertex_json_help': 'Complete Google Cloud Service Account JSON configuration',\n 'save_vertex_config': 'Save Vertex Configuration',\n 'test_vertex_config': 'Test Configuration',\n 'clear_vertex_config': 'Clear Configuration',\n 'no_vertex_config': 'Vertex AI not configured',\n 'project_id': 'Project ID',\n 'client_email': 'Client Email',\n 'api_key': 'API Key',\n 'invalid_json': 'Invalid JSON configuration',\n 'vertex_config_overwrite_confirm': 'Existing Vertex configuration detected. Do you want to overwrite the current configuration?',\n\n // Worker API Keys 部分\n 'add_new_worker_key': 'Add New Worker Key',\n 'generate_or_enter_key': 'Generate or enter a strong key',\n 'generate_random_key': 'Generate Random Key',\n 'description_optional': 'Description (Optional)',\n 'description_placeholder': 'e.g., Client App A',\n 'add_worker_key': 'Add Worker Key',\n 'safety_settings_help': 'Safety Settings: Enabled by default. When disabled, the model is allowed to generate NSFW content.',\n \n // Models 部分\n 'add_model': 'Add Model',\n 'model_id': 'Model ID',\n 'model_id_placeholder': 'e.g., gemini-1.5-flash-latest',\n 'model_id_help': 'Select or enter model ID',\n 'category': 'Category',\n 'daily_quota_custom': 'Daily Quota (Custom)',\n 'quota_placeholder': 'e.g., 1500, or \\'none\\'/\\'0\\' for unlimited',\n 'quota_help': 'Only for \\'Custom\\' category. Sets max daily requests for this specific model. Enter \\'none\\' or \\'0\\' for unlimited.',\n 'set_category_quotas': 'Set Category Quotas',\n 'loading_models': 'Loading models...',\n \n // 配额设置模态框\n 'set_category_quotas_title': 'Set Category Quotas',\n 'pro_models_daily_quota': 'Pro Models Daily Quota',\n 'flash_models_daily_quota': 'Flash Models Daily Quota',\n 'save_quotas': 'Save Quotas',\n \n // 独立配额模态框\n 'set_quota': 'Set Quota',\n 'individual_daily_quota': 'Individual Daily Quota',\n 'individual_quota_placeholder': 'Default: 0 (No individual quota)',\n 'individual_quota_help': 'Enter 0 for no individual quota. Individual quota is applied in addition to category quota.',\n 'save_quota': 'Save Quota',\n \n // 测试进度\n 'running_all_tests': 'Running All Tests',\n 'progress': 'Progress',\n 'preparing_tests': 'Preparing tests...',\n \n // 按钮和操作\n 'set_individual_quota': 'Set Individual Quota',\n 'ignore_error': 'Ignore Error',\n 'clear_error': 'Clear Error',\n\n // 弹窗和动态内容\n 'id': 'ID',\n 'key_preview': 'Key Preview',\n 'total_usage_today': 'Total Usage Today',\n 'date': 'Date',\n 'error_status': 'Error Status',\n 'category_usage': 'Category Usage',\n 'pro_models': 'Pro Models',\n 'flash_models': 'Flash Models',\n 'test_api_key': 'Test API Key',\n 'select_a_model': 'Select a model...',\n 'run_test': 'Run Test',\n 'testing': 'Testing...',\n 'test_passed': 'Test Passed!',\n 'test_failed': 'Test Failed.',\n 'status': 'Status',\n 'response': 'Response',\n 'no_description': 'No description',\n 'created': 'Created',\n 'safety_settings': 'Safety Settings',\n 'enabled': 'Enabled',\n 'disabled': 'Disabled',\n 'unlimited': 'Unlimited',\n 'individual_quota': 'Individual Quota',\n 'quota': 'Quota',\n 'set_quota_btn': 'Set Quota',\n 'delete_confirm_gemini': 'Are you sure you want to delete Gemini key with ID: {0}?',\n 'delete_confirm_worker': 'Are you sure you want to delete Worker key: {0}?',\n 'delete_confirm_model': 'Are you sure you want to delete model: {0}?',\n 'please_select_model': 'Please select a model to test',\n 'tests_cancelled': 'Tests cancelled by user',\n 'all_tests_completed': 'All tests completed!',\n 'completed_testing': 'Completed testing {0} Gemini keys with model {1}.',\n 'test_run_failed': 'Test run failed',\n 'cancelling_tests': 'Cancelling tests...',\n 'testing_batch': 'Testing batch {0}...',\n 'completed_tests': 'Completed {0} of {1} tests',\n 'no_gemini_keys_found': 'No Gemini keys found to test.',\n 'network_error': 'Network error',\n 'test_failed_no_response': 'Test failed: No response from server',\n 'test_failed_network': 'Test failed: Network error - {0}',\n 'unknown_error': 'Unknown error',\n 'test_run_cancelled': 'Test run was cancelled.',\n 'failed_to_run_tests': 'Failed to run all tests: {0}',\n\n // System Settings\n 'system_settings': 'System Settings',\n 'keepalive_setting': 'KEEPALIVE Mode',\n 'keepalive_description': 'Enable keep-alive mode for request processing',\n 'max_retry_setting': 'Max Retry Count',\n 'max_retry_description': 'Maximum retry attempts for failed API requests (default: 3)',\n 'web_search_setting': 'Web Search',\n 'web_search_description': 'Enable to show models with -search suffix for web search functionality',\n 'auto_test_setting': 'Auto Batch Test',\n 'auto_test_description': 'Enable to automatically run batch tests daily at 4 AM Beijing time',\n 'update_available': 'A new version is available',\n 'current_is_latest': 'Current is latest version'\n }\n };\n \n this.init();\n }\n \n init() {\n // 检测浏览器语言\n this.detectLanguage();\n\n // 应用翻译\n this.applyTranslations();\n\n // 监听语言变化\n this.setupLanguageChangeListener();\n }\n \n detectLanguage() {\n // 检测浏览器语言，不再使用本地存储\n const browserLanguage = navigator.language || navigator.userLanguage;\n if (browserLanguage.startsWith('zh')) {\n this.currentLanguage = 'zh';\n } else {\n this.currentLanguage = 'en';\n }\n }\n \n\n \n translate(key, ...args) {\n let translation = this.translations[this.currentLanguage][key] || key;\n\n // 支持参数替换，例如 translate('message', 'arg1', 'arg2')\n if (args.length > 0) {\n args.forEach((arg, index) => {\n translation = translation.replace(`{${index}}`, arg);\n });\n }\n\n return translation;\n }\n \n applyTranslations() {\n // 翻译所有带有 data-i18n 属性的元素\n document.querySelectorAll('[data-i18n]').forEach(element => {\n const key = element.getAttribute('data-i18n');\n const translation = this.translate(key);\n \n if (element.tagName === 'INPUT' && (element.type === 'text' || element.type === 'password')) {\n element.placeholder = translation;\n } else {\n element.textContent = translation;\n }\n });\n \n // 翻译所有带有 data-i18n-placeholder 属性的元素\n document.querySelectorAll('[data-i18n-placeholder]').forEach(element => {\n const key = element.getAttribute('data-i18n-placeholder');\n element.placeholder = this.translate(key);\n });\n \n // 翻译所有带有 data-i18n-title 属性的元素\n document.querySelectorAll('[data-i18n-title]').forEach(element => {\n const key = element.getAttribute('data-i18n-title');\n element.title = this.translate(key);\n });\n }\n \n setupLanguageChangeListener() {\n // 监听DOM变化，自动翻译新添加的元素\n const observer = new MutationObserver((mutations) => {\n mutations.forEach((mutation) => {\n mutation.addedNodes.forEach((node) => {\n if (node.nodeType === Node.ELEMENT_NODE) {\n // 翻译新添加的元素\n if (node.hasAttribute && node.hasAttribute('data-i18n')) {\n const key = node.getAttribute('data-i18n');\n const translation = this.translate(key);\n \n if (node.tagName === 'INPUT' && (node.type === 'text' || node.type === 'password')) {\n node.placeholder = translation;\n } else {\n node.textContent = translation;\n }\n }\n \n // 翻译新添加元素的子元素\n if (node.querySelectorAll) {\n node.querySelectorAll('[data-i18n]').forEach(element => {\n const key = element.getAttribute('data-i18n');\n const translation = this.translate(key);\n \n if (element.tagName === 'INPUT' && (element.type === 'text' || element.type === 'password')) {\n element.placeholder = translation;\n } else {\n element.textContent = translation;\n }\n });\n \n node.querySelectorAll('[data-i18n-placeholder]').forEach(element => {\n const key = element.getAttribute('data-i18n-placeholder');\n element.placeholder = this.translate(key);\n });\n \n node.querySelectorAll('[data-i18n-title]').forEach(element => {\n const key = element.getAttribute('data-i18n-title');\n element.title = this.translate(key);\n });\n }\n }\n });\n });\n });\n \n observer.observe(document.body, {\n childList: true,\n subtree: true\n });\n }\n}\n\n// 页面加载完成后初始化国际化\ndocument.addEventListener('DOMContentLoaded', () => {\n window.i18n = new I18n();\n});\n\n// 全局翻译函数，方便在其他脚本中使用\nwindow.t = function(key, ...args) {\n if (window.i18n) {\n return window.i18n.translate(key, ...args);\n }\n return key;\n};\n" }, { "path": "public/login.html", "content": "\n\n\n \n \n\n
\n\n \n\n\n"
},
{
"path": "public/login.script.js",
"content": "document.addEventListener('DOMContentLoaded', () => {\n const loginForm = document.getElementById('login-form');\n const passwordInput = document.getElementById('password');\n const loginButton = document.getElementById('login-button');\n const loadingSpinner = document.getElementById('loading-spinner');\n const errorMessageDiv = document.getElementById('error-message');\n const errorTextSpan = document.getElementById('error-text');\n \n function showLoading() {\n loginButton.disabled = true;\n loadingSpinner.classList.remove('hidden');\n }\n \n function hideLoading() {\n loginButton.disabled = false;\n loadingSpinner.classList.add('hidden');\n }\n \n function showError(message) {\n errorTextSpan.textContent = message;\n errorMessageDiv.classList.remove('hidden');\n }\n \n function hideError() {\n errorMessageDiv.classList.add('hidden');\n errorTextSpan.textContent = '';\n }\n \n loginForm.addEventListener('submit', async (e) => {\n e.preventDefault();\n hideError();\n showLoading();\n const password = passwordInput.value;\n \n try {\n const response = await fetch('/api/login', {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n },\n body: JSON.stringify({ password }),\n });\n \n const data = await response.json();\n \n if (response.ok && data.success) {\n localStorage.setItem('isLoggedIn', 'true');\n window.location.href = '/admin';\n } else {\n showError(data.error || 'Login failed. Please check your password.');\n passwordInput.focus();\n }\n } catch (error) {\n console.error('Login Error:', error);\n showError('An error occurred during login. Please try again.');\n } finally {\n hideLoading();\n }\n });\n});"
},
{
"path": "src/db/index.js",
"content": "const sqlite3 = require('sqlite3').verbose();\nconst path = require('path');\nconst fs = require('fs');\nconst GitHubSync = require('../utils/githubSync');\n\n// Construct the database path\nlet dataDir;\n\n// Use /home/user/data directory on Hugging Face Space\nif (process.env.HUGGING_FACE === '1') {\n dataDir = '/home/user/data';\n console.log(`Using Hugging Face persistent data directory: ${dataDir}`);\n} else {\n dataDir = path.resolve(__dirname, '..', '..', 'data');\n}\n\nif (!fs.existsSync(dataDir)) {\n console.log(`Creating data directory: ${dataDir}`);\n try {\n fs.mkdirSync(dataDir, { recursive: true });\n } catch (err) {\n console.error(`Error creating data directory: ${err.message}`);\n console.error('Will attempt to use ./data as fallback');\n dataDir = path.resolve(__dirname, '..', '..', 'data');\n if (!fs.existsSync(dataDir)) {\n fs.mkdirSync(dataDir, { recursive: true });\n }\n }\n}\n\nconst dbPath = path.resolve(dataDir, 'database.db');\nconsole.log(`Database path: ${dbPath}`); // Log the path for debugging\n\n// Initialize GitHub sync if configured\nconst githubProject = process.env.GITHUB_PROJECT;\nconst githubToken = process.env.GITHUB_PROJECT_PAT;\nconst githubEncryptKey = process.env.GITHUB_ENCRYPT_KEY;\nlet githubSync = null;\n\nif (githubProject && githubToken) {\n console.log(`GitHub sync configured for repository: ${githubProject}`);\n githubSync = new GitHubSync(githubProject, githubToken, dbPath, githubEncryptKey);\n \n if (githubEncryptKey && githubEncryptKey.length >= 32) {\n console.log('GitHub data encryption enabled, using AES-256-CBC algorithm');\n } else if (githubEncryptKey) {\n console.warn('GitHub encryption key length is insufficient, requires at least 32 characters, data will be stored unencrypted');\n } else {\n console.log('GitHub data encryption not enabled, data will be stored unencrypted');\n }\n}\n\n// Function to validate if a file is a valid SQLite database\nfunction validateDatabaseFile(filePath) {\n try {\n if (!fs.existsSync(filePath)) {\n return { valid: false, reason: 'File does not exist' };\n }\n\n const buffer = fs.readFileSync(filePath, { encoding: null });\n if (buffer.length < 16) {\n return { valid: false, reason: 'File too small to be a valid SQLite database' };\n }\n\n // Check SQLite file header\n const sqliteHeader = Buffer.from(\"SQLite format 3\\0\");\n const fileHeader = buffer.subarray(0, 16);\n\n if (Buffer.compare(fileHeader, sqliteHeader) === 0) {\n return { valid: true, reason: 'Valid SQLite database' };\n } else {\n return { valid: false, reason: 'Invalid SQLite header' };\n }\n } catch (error) {\n return { valid: false, reason: `Error reading file: ${error.message}` };\n }\n}\n\n// Initialize database with proper GitHub sync handling\nasync function initializeDatabase() {\n // Try to download database from GitHub BEFORE opening the database connection\n if (githubSync) {\n try {\n console.log('Attempting to download database from GitHub before opening connection...');\n const downloadSuccess = await githubSync.downloadDatabase();\n\n // Validate the downloaded database file\n if (downloadSuccess && fs.existsSync(dbPath)) {\n const validation = validateDatabaseFile(dbPath);\n if (!validation.valid) {\n console.warn(`Downloaded database file is invalid: ${validation.reason}`);\n console.log('Removing invalid database file and creating a new one...');\n try {\n fs.unlinkSync(dbPath);\n } catch (unlinkErr) {\n console.error('Failed to remove invalid database file:', unlinkErr.message);\n }\n } else {\n console.log('Downloaded database file validation passed');\n }\n }\n } catch (err) {\n console.error('Failed to download database from GitHub:', err.message);\n console.log('Continuing with local database...');\n }\n }\n\n return new Promise((resolve, reject) => {\n // Initialize the database connection after GitHub sync is complete\n // The OPEN_READWRITE | OPEN_CREATE flag ensures the file is created if it doesn't exist.\n const db = new sqlite3.Database(dbPath, sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE, async (err) => {\n if (err) {\n console.error('Error opening database:', err.message);\n reject(err); // Reject to stop the application if DB connection fails\n } else {\n console.log('Connected to the SQLite database.');\n\n // Initialize database schema\n try {\n await new Promise((schemaResolve, schemaReject) => {\n // Pass the current database instance to the schema initialization function\n initializeDatabaseSchemaInternal.call({ db: db }, (schemaErr) => {\n if (schemaErr) schemaReject(schemaErr);\n else schemaResolve();\n });\n });\n\n resolve(db);\n } catch (schemaErr) {\n console.error('Failed to initialize database schema:', schemaErr.message);\n reject(schemaErr);\n }\n }\n });\n });\n}\n\n// Start the database initialization\nlet db;\ninitializeDatabase()\n .then(async (database) => {\n db = database;\n console.log('Database initialization completed successfully.');\n\n // Initialize Vertex service after database is ready and exported\n try {\n const vertexService = require('../services/vertexProxyService');\n console.log('Initializing Vertex AI service after database setup...');\n await vertexService.initializeVertexCredentials();\n } catch (err) {\n console.error('Failed to initialize Vertex service:', err.message);\n }\n\n // Initialize scheduler service after database is ready\n try {\n const schedulerService = require('../services/schedulerService');\n console.log('Initializing Scheduler Service after database setup...');\n await schedulerService.initialize();\n console.log('Scheduler Service: Initialized successfully');\n } catch (err) {\n console.error('Scheduler Service: Failed to initialize:', err.message);\n }\n })\n .catch((err) => {\n console.error('Fatal error during database initialization:', err.message);\n process.exit(1);\n });\n\n// Function to trigger GitHub sync (now always delayed)\nasync function syncToGitHub() {\n if (!githubSync) {\n return false;\n }\n\n try {\n // Schedule the delayed sync\n await githubSync.scheduleSync();\n return true; // Indicate scheduling was successful\n } catch (err) {\n console.error('Failed to schedule GitHub sync:', err.message);\n return false;\n }\n}\n\n// SQL statements to create tables (if they don't exist)\nconst createTablesSQL = `\n CREATE TABLE IF NOT EXISTS gemini_keys (\n id TEXT PRIMARY KEY,\n api_key TEXT NOT NULL UNIQUE,\n name TEXT,\n usage_date TEXT,\n model_usage TEXT DEFAULT '{}', -- Store as JSON string\n category_usage TEXT DEFAULT '{}', -- Store as JSON string\n error_status INTEGER, -- 401, 403, or NULL\n consecutive_429_counts TEXT DEFAULT '{}', -- Store as JSON string\n created_at TEXT DEFAULT CURRENT_TIMESTAMP\n );\n\n CREATE TABLE IF NOT EXISTS worker_keys (\n api_key TEXT PRIMARY KEY,\n description TEXT,\n safety_enabled INTEGER DEFAULT 1, -- 1 for true, 0 for false\n created_at TEXT DEFAULT CURRENT_TIMESTAMP\n );\n\n CREATE TABLE IF NOT EXISTS models_config (\n model_id TEXT PRIMARY KEY,\n category TEXT NOT NULL CHECK(category IN ('Pro', 'Flash', 'Custom')),\n daily_quota INTEGER, -- NULL means unlimited\n individual_quota INTEGER -- NULL means no individual limit\n );\n\n CREATE TABLE IF NOT EXISTS settings (\n key TEXT PRIMARY KEY,\n value TEXT -- Can store JSON strings or simple values\n );\n\n -- Initialize default category quotas if not present\n INSERT OR IGNORE INTO settings (key, value) VALUES\n ('category_quotas', '{\"proQuota\": 50, \"flashQuota\": 1500}');\n\n -- Initialize gemini_key_list if not present (as an empty JSON array)\n INSERT OR IGNORE INTO settings (key, value) VALUES\n ('gemini_key_list', '[]');\n\n -- Initialize gemini_key_index if not present\n INSERT OR IGNORE INTO settings (key, value) VALUES\n ('gemini_key_index', '0');\n\n -- Add other default settings as needed, e.g., last used key ID\n INSERT OR IGNORE INTO settings (key, value) VALUES\n ('last_used_gemini_key_id', '');\n`;\n\n// Function to initialize the database schema\nfunction initializeDatabaseSchemaInternal(callback) {\n // Use the database instance passed via 'this' context or fall back to global db\n const currentDb = this?.db || db;\n if (!currentDb) {\n const error = new Error('Database instance not available for schema initialization');\n console.error(error.message);\n if (callback) callback(error);\n return;\n }\n\n currentDb.exec(createTablesSQL, (err) => {\n if (err) {\n console.error('Error creating database tables:', err.message);\n if (callback) callback(err);\n } else {\n console.log('Database tables checked/created successfully.');\n // You might seed initial data here if necessary\n if (callback) callback(null);\n }\n });\n}\n\n// Function to safely close the database connection\nfunction closeDatabase() {\n if (db) {\n db.close((err) => {\n if (err) {\n console.error('Error closing database:', err.message);\n } else {\n console.log('Database connection closed.');\n }\n });\n }\n}\n\n// Gracefully close the database on application exit\nprocess.on('SIGINT', () => {\n closeDatabase();\n process.exit(0);\n});\n\nprocess.on('SIGTERM', () => {\n closeDatabase();\n process.exit(0);\n});\n\n// Export the database connection instance and sync functions\nmodule.exports = {\n get db() { return db; }, // Use getter to ensure db is available when accessed\n syncToGitHub\n};\n"
},
{
"path": "src/index.js",
"content": "// Load environment variables from .env file FIRST\nrequire('dotenv').config();\n\nconst express = require('express');\nconst path = require('path');\nconst cors = require('cors');\nconst cookieParser = require('cookie-parser');\n\n// Import the database module (this will also trigger initialization)\nconst dbModule = require('./db');\n\n// Import Vertex service but don't initialize yet - will be done after DB is ready\nconst vertexService = require('./services/vertexProxyService');\n\n// Note: schedulerService is imported lazily in routes/adminApi.js to avoid database initialization issues\n\n// Import route handlers\nconst authRoutes = require('./routes/auth');\nconst adminApiRoutes = require('./routes/adminApi');\nconst apiV1Routes = require('./routes/apiV1');\n\n// Import services and utils (ensure proxyPool is imported to trigger its initialization)\nrequire('./services/geminiProxyService'); // Still need to import this for other initializations if any\nconst proxyPool = require('./utils/proxyPool');\n\n// Import middleware\nconst requireAdminAuth = require('./middleware/adminAuth');\n\nconst app = express();\nconst port = process.env.PORT || 3000; // Default to 3000 if PORT not set\n\n// --- Middleware ---\n\n// Enable CORS for all origins (adjust for production if needed)\napp.use(cors({\n origin: '*', // Allow all origins for now\n credentials: true, // Allow cookies for authenticated requests (like admin UI)\n methods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'],\n allowedHeaders: ['Content-Type', 'Authorization', 'x-requested-with'],\n maxAge: 86400 // Cache preflight requests for 1 day\n}));\n\n// Handle OPTIONS preflight requests globally (alternative to handling in each route)\napp.options('*', cors());\n\n// Parse JSON request bodies\napp.use(express.json({ limit: '100mb' }));\n\n// Parse URL-encoded request bodies\napp.use(express.urlencoded({ extended: true, limit: '100mb' }));\n\n// Parse cookies\napp.use(cookieParser());\n\n// Serve static files from the 'public' directory\n// __dirname now refers to the src directory, need to go up one level\napp.use(express.static(path.join(__dirname, '..', 'public')));\n\n// --- Basic Routes ---\n\n// Root route: Redirects to /admin/index.html if logged in, otherwise requireAdminAuth redirects to /login.html\napp.get('/', (req, res) => {\n res.redirect('/login.html');\n});\n\n// Redirect /login to the static HTML file\napp.get('/login', (req, res) => {\n res.redirect('/login.html');\n});\n\n// Admin route: Protect the route and serve the static file\napp.get('/admin', requireAdminAuth, (req, res) => {\n res.redirect('/admin/'); // Redirect to the directory path\n});\n\napp.use('/admin', requireAdminAuth, express.static(path.join(__dirname, '..', 'public', 'admin')));\n\n// --- API Routes ---\napp.use('/api', authRoutes); \napp.use('/api/admin', requireAdminAuth, adminApiRoutes); \napp.use('/v1', apiV1Routes); \n\n// --- Global Error Handler ---\napp.use((err, req, res, next) => {\n console.error('Unhandled Error:', err.stack || err);\n res.status(err.status || 500).json({\n error: {\n message: err.message || 'Internal Server Error',\n type: err.type || 'unhandled_error',\n ...(process.env.NODE_ENV === 'development' && { stack: err.stack })\n }\n });\n});\n\n// --- Start Server ---\napp.listen(port, '0.0.0.0', async () => {\n console.log(`JimiHub (Node.js version) listening on port ${port} (all interfaces)`);\n\n // Log Proxy Pool Status\n const proxyStatus = proxyPool.getProxyPoolStatus(); // Get status from proxyPool module\n if (proxyStatus.enabled) {\n console.log(`Proxy Pool: Enabled (Loaded ${proxyStatus.count} SOCKS5 proxies)`);\n } else if (proxyStatus.count > 0 && !proxyStatus.agentLoaded) {\n console.log(`Proxy Pool: Configured (${proxyStatus.count} proxies) but DISABLED (missing 'socks-proxy-agent' dependency)`);\n } else {\n console.log(`Proxy Pool: Disabled (PROXY environment variable not set or contains no valid SOCKS5 proxies)`);\n }\n \n // Log Vertex AI Status using the check function\n if (vertexService.isVertexEnabled()) {\n // Check if we're using Express Mode\n if (process.env.EXPRESS_API_KEY) {\n console.log(`Vertex AI: Enabled with Express Mode (API Key authentication, additional [v] prefixed models available)`);\n } else {\n console.log(`Vertex AI: Enabled (Service Account credentials, additional [v] prefixed models available)`);\n }\n } else {\n console.log(`Vertex AI: Disabled (VERTEX variable and EXPRESS_API_KEY not found or invalid in .env file)`);\n }\n \n // Check if running in Hugging Face Space\n if (process.env.HUGGING_FACE === '1' && process.env.SPACE_HOST) {\n const adminUrl = `https://${process.env.SPACE_HOST}/admin`;\n const endpointUrl = `https://${process.env.SPACE_HOST}/v1`;\n console.log(`Hugging Face Space Admin UI: ${adminUrl}`);\n console.log(`Hugging Face Space Endpoint: ${endpointUrl}`);\n } else {\n // Fallback for local or other environments\n const adminUrl = `http://localhost:${port}/admin`;\n const endpointUrl = `http://localhost:${port}/v1`;\n console.log(`Admin UI available at: ${adminUrl} (or the server's public address)`);\n console.log(`API Endpoint available at: ${endpointUrl} (or the server's public address)`);\n }\n});\n"
},
{
"path": "src/middleware/adminAuth.js",
"content": "const { verifySessionCookie } = require('../utils/session');\n\n/**\n * Express middleware to protect routes requiring admin authentication.\n * Verifies the session cookie. If invalid or missing, redirects to /login.html.\n * @param {import('express').Request} req\n * @param {import('express').Response} res\n * @param {import('express').NextFunction} next\n */\nasync function requireAdminAuth(req, res, next) {\n try {\n const isAuthenticated = await verifySessionCookie(req);\n\n if (!isAuthenticated) {\n console.log('AdminAuth Middleware: Session invalid or expired. Redirecting to login.');\n // Redirect to the login page if not authenticated\n // Check if the original request was for an API endpoint\n if (req.originalUrl.startsWith('/api/admin')) {\n // For API requests, send a 401 Unauthorized status instead of redirecting\n return res.status(401).json({ error: 'Unauthorized. Please log in again.' });\n } else {\n // For page requests (like /admin), redirect to the login page\n return res.redirect('/login.html');\n }\n }\n\n // If authenticated, proceed to the next middleware or route handler\n // console.log('AdminAuth Middleware: Session valid.'); // Optional: Log success\n next();\n } catch (error) {\n console.error('Error in admin authentication middleware:', error);\n // Pass the error to the global error handler\n next(error);\n }\n}\n\nmodule.exports = requireAdminAuth;\n"
},
{
"path": "src/middleware/workerAuth.js",
"content": "const dbModule = require('../db'); // Import the database module\n\n/**\n * Express middleware to validate the Worker API Key provided in the Authorization header.\n * Checks against the `worker_keys` table in the database.\n * @param {import('express').Request} req\n * @param {import('express').Response} res\n * @param {import('express').NextFunction} next\n */\nasync function requireWorkerAuth(req, res, next) {\n const authHeader = req.headers.authorization;\n const workerApiKey = authHeader?.startsWith('Bearer ') ? authHeader.substring(7) : null;\n\n if (!workerApiKey) {\n return res.status(401).json({ error: 'Missing API key. Provide it in the Authorization header as \"Bearer YOUR_KEY\".' });\n }\n\n try {\n // Get database instance\n const db = dbModule.db;\n if (!db) {\n console.error('Database not available for worker key validation');\n return res.status(500).json({ error: 'Database not available' });\n }\n\n // Query the database to see if the key exists\n const sql = `SELECT api_key FROM worker_keys WHERE api_key = ?`;\n db.get(sql, [workerApiKey], (err, row) => {\n if (err) {\n console.error('Database error during worker key validation:', err);\n // Pass error to the global error handler\n return next(err);\n }\n\n if (!row) {\n // Key not found in the database\n console.warn(`Worker key validation failed: Key \"${workerApiKey.slice(0, 5)}...\" not found.`);\n return res.status(401).json({ error: 'Invalid API key.' });\n }\n\n // Key is valid, attach it to the request object for potential use later\n // (e.g., determining safety settings)\n req.workerApiKey = workerApiKey;\n\n // Proceed to the next middleware or route handler\n next();\n });\n } catch (error) {\n console.error('Unexpected error during worker key validation:', error);\n next(error); // Pass to global error handler\n }\n}\n\nmodule.exports = requireWorkerAuth;\n"
},
{
"path": "src/routes/adminApi.js",
"content": "const express = require('express');\nconst requireAdminAuth = require('../middleware/adminAuth');\nconst configService = require('../services/configService');\nconst geminiKeyService = require('../services/geminiKeyService');\nconst vertexProxyService = require('../services/vertexProxyService');\nconst batchTestService = require('../services/batchTestService');\n// Note: schedulerService is imported lazily when needed to avoid database initialization issues\nconst fetch = require('node-fetch');\nconst dbModule = require('../db');\nconst proxyPool = require('../utils/proxyPool'); // Import the proxy pool module\nconst router = express.Router();\n\n// Apply admin authentication middleware to all /api/admin routes\nrouter.use(requireAdminAuth);\n\n// --- Helper for parsing request body (already exists in helpers.js, but useful here) ---\n// Ensure express.json() middleware is applied in server.js\nfunction parseBody(req) {\n if (!req.body) {\n throw new Error(\"Request body not parsed. Ensure express.json() middleware is used.\");\n }\n return req.body;\n}\n\n// --- Gemini Key Management --- (/api/admin/gemini-keys)\nrouter.route('/gemini-keys')\n .get(async (req, res, next) => {\n try {\n const keys = await geminiKeyService.getAllGeminiKeysWithUsage();\n res.json(keys);\n } catch (error) {\n next(error);\n }\n })\n .post(async (req, res, next) => {\n try {\n const { key, name } = parseBody(req);\n if (!key || typeof key !== 'string') {\n return res.status(400).json({ error: 'Request body must include a valid API key (string)' });\n }\n const result = await geminiKeyService.addGeminiKey(key, name);\n res.status(201).json({ success: true, ...result });\n } catch (error) {\n if (error.message.includes('duplicate API key')) {\n return res.status(409).json({ error: 'Cannot add duplicate API key' });\n }\n next(error);\n }\n });\n\n// --- Batch Add Gemini Keys --- (/api/admin/gemini-keys/batch)\nrouter.post('/gemini-keys/batch', async (req, res, next) => {\n try {\n const { keys } = parseBody(req);\n if (!Array.isArray(keys) || keys.length === 0) {\n return res.status(400).json({ error: 'Request body must include a valid array of API keys' });\n }\n\n // Validate that all items are strings\n const invalidKeys = keys.filter(key => !key || typeof key !== 'string');\n if (invalidKeys.length > 0) {\n return res.status(400).json({ error: 'All API keys must be valid strings' });\n }\n\n const result = await geminiKeyService.addMultipleGeminiKeys(keys);\n res.status(201).json({\n success: true,\n ...result\n });\n } catch (error) {\n next(error);\n }\n});\n\nrouter.delete('/gemini-keys/:id', async (req, res, next) => {\n try {\n const keyId = req.params.id;\n if (!keyId) {\n return res.status(400).json({ error: 'Missing key ID in path' });\n }\n await geminiKeyService.deleteGeminiKey(keyId);\n res.json({ success: true, id: keyId });\n } catch (error) {\n if (error.message.includes('not found')) {\n return res.status(404).json({ error: error.message });\n }\n next(error);\n }\n});\n\n// Base Gemini API URL\nconst BASE_GEMINI_URL = process.env.GEMINI_BASE_URL || 'https://generativelanguage.googleapis.com';\n\n// Helper function to check if a 400 error should be marked for key error\nfunction shouldMark400Error(responseBody) {\n try {\n // Only mark 400 errors if the message indicates invalid API key\n if (responseBody && responseBody.error) {\n const errorMessage = responseBody.error.message;\n\n // Check for the specific \"API key not valid\" error\n if (errorMessage && errorMessage.includes('API key not valid. Please pass a valid API key.')) {\n return true;\n }\n }\n return false;\n } catch (e) {\n // If we can't parse the error, don't mark it\n return false;\n }\n}\n\n// --- Test Gemini Key --- (/api/admin/test-gemini-key)\nrouter.post('/test-gemini-key', async (req, res, next) => {\n try {\n const { keyId, modelId } = parseBody(req);\n if (!keyId || !modelId) {\n return res.status(400).json({ error: 'Request body must include keyId and modelId' });\n }\n\n // Fetch the actual key from the database\n const keyInfo = await configService.getDb('SELECT api_key FROM gemini_keys WHERE id = ?', [keyId]);\n if (!keyInfo || !keyInfo.api_key) {\n return res.status(404).json({ error: `API Key with ID '${keyId}' not found or invalid.` });\n }\n const apiKey = keyInfo.api_key;\n\n // Fetch model category for potential usage increment\n const modelsConfig = await configService.getModelsConfig();\n let modelCategory = modelsConfig[modelId]?.category;\n\n // If model is not configured, infer category from model name\n if (!modelCategory) {\n if (modelId.includes('flash')) {\n modelCategory = 'Flash';\n } else if (modelId.includes('pro')) {\n modelCategory = 'Pro';\n } else {\n // Default to Flash for unknown models (most common case)\n modelCategory = 'Flash';\n }\n console.log(`Model ${modelId} not configured, inferred category: ${modelCategory}`);\n }\n\n const testGeminiRequestBody = { contents: [{ role: \"user\", parts: [{ text: \"Hi\" }] }] };\n const geminiUrl = `${BASE_GEMINI_URL}/v1beta/models/${modelId}:generateContent`;\n\n let testResponseStatus = 500;\n let testResponseBody = null;\n let isSuccess = false;\n\n try {\n // Get proxy agent\n const agent = proxyPool.getNextProxyAgent();\n const fetchOptions = {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n 'x-goog-api-key': apiKey\n },\n body: JSON.stringify(testGeminiRequestBody)\n };\n if (agent) {\n fetchOptions.agent = agent;\n console.log(`Admin API (Test Key): Sending request via proxy ${agent.proxy.href}`);\n } else {\n console.log(`Admin API (Test Key): Sending request directly.`);\n }\n\n const response = await fetch(geminiUrl, fetchOptions);\n testResponseStatus = response.status;\n testResponseBody = await response.json(); // Attempt to parse JSON\n isSuccess = response.ok;\n\nif (isSuccess) {\n // Increment usage and sync to GitHub\n await geminiKeyService.incrementKeyUsage(keyId, modelId, modelCategory);\n\n // Clear error status if the key was previously marked with an error\n // This allows previously failed keys to be restored when they work again during batch testing\n try {\n const wasCleared = await geminiKeyService.clearKeyError(keyId);\n if (wasCleared) {\n console.log(`Restored key ${keyId} - cleared previous error status during testing.`);\n }\n } catch (clearError) {\n // Log but don't fail the test if clearing error status fails\n console.warn(`Failed to clear error status for key ${keyId}:`, clearError);\n }\n } else {\n // Record 400/401/403 errors (invalid API key, unauthorized, forbidden)\n // But only mark 400 errors if they indicate invalid API key\n if (testResponseStatus === 401 || testResponseStatus === 403) {\n await geminiKeyService.recordKeyError(keyId, testResponseStatus);\n } else if (testResponseStatus === 400) {\n // Check if this is an invalid API key 400 error that should be marked\n if (shouldMark400Error(testResponseBody)) {\n await geminiKeyService.recordKeyError(keyId, testResponseStatus);\n } else {\n console.log(`Skipping error marking for key ${keyId} - 400 error not related to invalid API key.`);\n }\n }\n }\n\n } catch (fetchError) {\n console.error(`Error testing Gemini API key ${keyId}:`, fetchError);\n testResponseBody = { error: `Fetch error: ${fetchError.message}` };\n isSuccess = false;\n // Don't assume network error means key is bad, could be temporary\n }\n\n res.status(isSuccess ? 200 : testResponseStatus).json({\n success: isSuccess,\n status: testResponseStatus,\n content: testResponseBody\n });\n\n } catch (error) {\n // Errors from fetching keyInfo etc.\n next(error);\n }\n});\n\n// --- Get Available Gemini Models --- (/api/admin/gemini-models)\nrouter.get('/gemini-models', async (req, res, next) => {\n try {\n // Helper function to fetch models with a specific key\n const fetchModelsWithKey = async (key) => {\n const geminiUrl = `${BASE_GEMINI_URL}/v1beta/models`;\n\n // Get proxy agent\n const agent = proxyPool.getNextProxyAgent();\n const fetchOptions = {\n method: 'GET',\n headers: {\n 'Content-Type': 'application/json',\n 'x-goog-api-key': key.key\n }\n };\n if (agent) {\n fetchOptions.agent = agent;\n console.log(`Admin API (Get Models): Sending request via proxy ${agent.proxy.href}`);\n } else {\n console.log(`Admin API (Get Models): Sending request directly.`);\n }\n\n const response = await fetch(geminiUrl, fetchOptions);\n return { response, keyId: key.id };\n };\n\n // Try to get models with up to 3 different keys\n let lastError = null;\n for (let attempt = 1; attempt <= 3; attempt++) {\n // Find *any* valid key to make the models list request, without updating the rotation index\n // This prevents writing to the database and GitHub sync on page refreshes\n const availableKey = await geminiKeyService.getNextAvailableGeminiKey(null, false);\n if (!availableKey) {\n console.warn(`Attempt ${attempt}: No available Gemini key found to fetch models list.`);\n break;\n }\n\n console.log(`Attempt ${attempt}: Fetching models with key ${availableKey.id}`);\n\n try {\n const { response, keyId } = await fetchModelsWithKey(availableKey);\n\n if (response.ok) {\n // Success! Process and return the models\n const data = await response.json();\n const processedModels = (data.models || [])\n .filter(model => model.name?.startsWith('models/')) // Ensure correct format\n .map((model) => ({\n id: model.name.substring(7), // Extract ID\n name: model.displayName || model.name.substring(7), // Prefer displayName\n description: model.description,\n // Add other potentially useful fields: supportedGenerationMethods, version, etc.\n }));\n\n console.log(`Successfully fetched ${processedModels.length} models with key ${keyId}`);\n return res.json(processedModels);\n } else {\n // Handle error response\n const errorBody = await response.text();\n console.error(`Attempt ${attempt}: Error fetching Gemini models list (key ${keyId}): ${response.status} ${response.statusText}`, errorBody);\n\n // Mark key as invalid if it's a persistent error (401/403/400)\n if (response.status === 401 || response.status === 403) {\n console.log(`Marking key ${keyId} as invalid due to ${response.status} error during model list fetch`);\n await geminiKeyService.recordKeyError(keyId, response.status);\n } else if (response.status === 400) {\n // Check if this is an invalid API key 400 error that should be marked\n try {\n const errorBodyJson = JSON.parse(errorBody);\n if (shouldMark400Error(errorBodyJson)) {\n console.log(`Marking key ${keyId} as invalid due to 400 error during model list fetch`);\n await geminiKeyService.recordKeyError(keyId, response.status);\n } else {\n console.log(`Skipping error marking for key ${keyId} during model list fetch - 400 error not related to invalid API key.`);\n }\n } catch (parseError) {\n // If we can't parse the error body, don't mark it as error\n console.log(`Skipping error marking for key ${keyId} during model list fetch - unparseable 400 response`);\n }\n }\n\n lastError = { status: response.status, body: errorBody };\n // Continue to next attempt with a different key\n }\n } catch (fetchError) {\n console.error(`Attempt ${attempt}: Network error fetching models with key ${availableKey.id}:`, fetchError);\n lastError = fetchError;\n // Continue to next attempt\n }\n }\n\n // If we get here, all attempts failed\n console.warn(\"All attempts to fetch Gemini models failed. Returning empty list.\");\n return res.json([]); // Return empty list if all attempts fail\n\n } catch (error) {\n console.error('Error handling /api/admin/gemini-models:', error);\n next(error);\n }\n});\n\n\n// --- Error Key Management ---\nrouter.get('/error-keys', async (req, res, next) => {\n try {\n const errorKeys = await geminiKeyService.getErrorKeys();\n res.json(errorKeys);\n } catch (error) {\n next(error);\n }\n});\n\nrouter.post('/clear-key-error', async (req, res, next) => {\n try {\n const { keyId } = parseBody(req);\n if (!keyId || typeof keyId !== 'string') {\n return res.status(400).json({ error: 'Request body must include a valid keyId (string)' });\n }\n await geminiKeyService.clearKeyError(keyId);\n res.json({ success: true, id: keyId });\n } catch (error) {\n if (error.message.includes('not found')) {\n return res.status(404).json({ error: error.message });\n }\n next(error);\n }\n});\n\nrouter.delete('/error-keys', async (req, res, next) => {\n try {\n const result = await geminiKeyService.deleteAllErrorKeys();\n res.json({\n success: true,\n deletedCount: result.deletedCount,\n deletedKeys: result.deletedKeys\n });\n } catch (error) {\n next(error);\n }\n});\n\nrouter.post('/clear-all-errors', async (req, res, next) => {\n try {\n const result = await geminiKeyService.clearAllErrorKeys();\n res.json({\n success: true,\n clearedCount: result.clearedCount,\n clearedKeys: result.clearedKeys\n });\n } catch (error) {\n next(error);\n }\n});\n\n\n// --- Worker Key Management --- (/api/admin/worker-keys)\nrouter.route('/worker-keys')\n .get(async (req, res, next) => {\n try {\n const keys = await configService.getAllWorkerKeys();\n res.json(keys);\n } catch (error) {\n next(error);\n }\n })\n .post(async (req, res, next) => {\n try {\n const { key, description } = parseBody(req);\n if (!key || typeof key !== 'string' || key.trim() === '') {\n return res.status(400).json({ error: 'Request body must include a valid non-empty string: key' });\n }\n await configService.addWorkerKey(key.trim(), description);\n res.status(201).json({ success: true, key: key.trim() });\n } catch (error) {\n if (error.message.includes('already exists')) {\n return res.status(409).json({ error: error.message });\n }\n next(error);\n }\n });\n\nrouter.delete('/worker-keys/:key', async (req, res, next) => { // Use key in path param\n try {\n const keyToDelete = decodeURIComponent(req.params.key); // Decode URL component\n if (!keyToDelete) {\n return res.status(400).json({ error: 'Missing worker key in path' });\n }\n await configService.deleteWorkerKey(keyToDelete);\n res.json({ success: true, key: keyToDelete });\n } catch (error) {\n if (error.message.includes('not found')) {\n return res.status(404).json({ error: error.message });\n }\n next(error);\n }\n});\n\nrouter.post('/worker-keys/safety-settings', async (req, res, next) => { // Specific path for safety\n try {\n const { key, safetyEnabled } = parseBody(req);\n if (!key || typeof key !== 'string' || typeof safetyEnabled !== 'boolean') {\n return res.status(400).json({ error: 'Request body must include key (string) and safetyEnabled (boolean)' });\n }\n await configService.updateWorkerKeySafety(key, safetyEnabled);\n res.json({ success: true, key: key, safetyEnabled: safetyEnabled });\n } catch (error) {\n if (error.message.includes('not found')) {\n return res.status(404).json({ error: error.message });\n }\n next(error);\n }\n});\n\n\n// --- Model Configuration Management --- (/api/admin/models)\nrouter.route('/models')\n .get(async (req, res, next) => {\n try {\n const config = await configService.getModelsConfig();\n // Convert to array format expected by UI\n const modelList = Object.entries(config).map(([id, data]) => ({ id, ...data }));\n res.json(modelList);\n } catch (error) {\n next(error);\n }\n })\n .post(async (req, res, next) => { // Add or Update\n try {\n const { id, category, dailyQuota, individualQuota } = parseBody(req);\n if (!id || !category || !['Pro', 'Flash', 'Custom'].includes(category)) {\n return res.status(400).json({ error: 'Request body must include valid id and category (Pro, Flash, or Custom)' });\n }\n // Basic validation for quotas (more in service layer)\n const dailyQuotaNum = (dailyQuota === null || dailyQuota === undefined || dailyQuota === '') ? null : Number(dailyQuota);\n const individualQuotaNum = (individualQuota === null || individualQuota === undefined || individualQuota === '') ? null : Number(individualQuota);\n\n if ((dailyQuotaNum !== null && isNaN(dailyQuotaNum)) || (individualQuotaNum !== null && isNaN(individualQuotaNum))) {\n return res.status(400).json({ error: 'Quotas must be numbers or null/empty.' });\n }\n\n await configService.setModelConfig(id, category, dailyQuotaNum, individualQuotaNum);\n res.status(200).json({ success: true, id, category, dailyQuota: dailyQuotaNum, individualQuota: individualQuotaNum }); // Use 200 for add/update simplicity\n } catch (error) {\n if (error.message.includes('must be a non-negative integer')) {\n return res.status(400).json({ error: error.message });\n }\n next(error);\n }\n });\n\nrouter.delete('/models/:id', async (req, res, next) => { // Use ID in path\n try {\n const modelIdToDelete = decodeURIComponent(req.params.id);\n if (!modelIdToDelete) {\n return res.status(400).json({ error: 'Missing model ID in path' });\n }\n await configService.deleteModelConfig(modelIdToDelete);\n res.json({ success: true, id: modelIdToDelete });\n } catch (error) {\n if (error.message.includes('not found')) {\n return res.status(404).json({ error: error.message });\n }\n next(error);\n }\n});\n\n\n// --- Category Quota Management --- (/api/admin/category-quotas)\nrouter.route('/category-quotas')\n .get(async (req, res, next) => {\n try {\n const quotas = await configService.getCategoryQuotas();\n res.json(quotas);\n } catch (error) {\n next(error);\n }\n })\n .post(async (req, res, next) => {\n try {\n const { proQuota, flashQuota } = parseBody(req);\n // Service layer handles detailed validation\n await configService.setCategoryQuotas(proQuota, flashQuota);\n res.json({ success: true, proQuota, flashQuota });\n } catch (error) {\n if (error.message.includes('must be non-negative numbers')) {\n return res.status(400).json({ error: error.message });\n }\n next(error);\n }\n });\n\n// --- Vertex Configuration Management --- (/api/admin/vertex-config)\nrouter.route('/vertex-config')\n .get(async (req, res, next) => {\n try {\n const config = await configService.getSetting('vertex_config', null);\n res.json(config);\n } catch (error) {\n next(error);\n }\n })\n .post(async (req, res, next) => {\n try {\n const { expressApiKey, vertexJson } = parseBody(req);\n\n // Validate that at least one authentication method is provided\n if (!expressApiKey && !vertexJson) {\n return res.status(400).json({ error: 'Either Express API Key or Vertex JSON must be provided' });\n }\n\n // Validate that only one authentication method is provided\n if (expressApiKey && vertexJson) {\n return res.status(400).json({ error: 'Only one authentication method can be configured at a time' });\n }\n\n let configData = {};\n\n if (expressApiKey) {\n // Validate Express API Key format (basic validation)\n if (typeof expressApiKey !== 'string' || expressApiKey.trim().length === 0) {\n return res.status(400).json({ error: 'Express API Key must be a non-empty string' });\n }\n configData.expressApiKey = expressApiKey.trim();\n }\n\n if (vertexJson) {\n // Validate JSON format\n try {\n const jsonData = JSON.parse(vertexJson);\n\n // Basic validation of required fields\n const requiredKeys = [\"type\", \"project_id\", \"private_key_id\", \"private_key\", \"client_email\", \"client_id\"];\n const missingKeys = requiredKeys.filter(key => !(key in jsonData));\n\n if (missingKeys.length > 0) {\n return res.status(400).json({\n error: `Invalid Service Account JSON. Missing required keys: ${missingKeys.join(', ')}`\n });\n }\n\n if (jsonData.type !== \"service_account\") {\n return res.status(400).json({\n error: \"Invalid Service Account JSON. 'type' must be 'service_account'\"\n });\n }\n\n configData.vertexJson = vertexJson.trim();\n } catch (e) {\n return res.status(400).json({ error: 'Invalid JSON format for Vertex configuration' });\n }\n }\n\n // Save configuration to database\n await configService.setSetting('vertex_config', configData);\n\n // Reinitialize Vertex service with new configuration\n await vertexProxyService.reinitializeWithDatabaseConfig();\n\n res.json({ success: true, message: 'Vertex configuration saved successfully' });\n } catch (error) {\n next(error);\n }\n })\n .delete(async (req, res, next) => {\n try {\n // Clear the configuration\n await configService.setSetting('vertex_config', null);\n\n // Reinitialize Vertex service to clear configuration\n await vertexProxyService.reinitializeWithDatabaseConfig();\n\n res.json({ success: true, message: 'Vertex configuration cleared successfully' });\n } catch (error) {\n next(error);\n }\n });\n\n// Test Vertex Configuration\nrouter.post('/vertex-config/test', async (req, res, next) => {\n try {\n // Get current configuration\n const config = await configService.getSetting('vertex_config', null);\n\n if (!config || (!config.expressApiKey && !config.vertexJson)) {\n return res.status(400).json({ error: 'No Vertex configuration found. Please configure Vertex first.' });\n }\n\n // Test the configuration by checking if Vertex is enabled\n const isEnabled = vertexProxyService.isVertexEnabled();\n const supportedModels = vertexProxyService.getVertexSupportedModels();\n\n if (!isEnabled || supportedModels.length === 0) {\n return res.status(400).json({ error: 'Vertex configuration test failed. Service is not properly initialized.' });\n }\n\n res.json({\n success: true,\n message: 'Vertex configuration test successful',\n supportedModels: supportedModels.length,\n authMode: config.expressApiKey ? 'Express Mode' : 'Service Account'\n });\n } catch (error) {\n next(error);\n }\n});\n\n// --- System Settings Management --- (/api/admin/system-settings)\nrouter.route('/system-settings')\n .get(async (req, res, next) => {\n try {\n // Get settings from database\n const keepalive = await configService.getSetting('keepalive', '0');\n const maxRetry = await configService.getSetting('max_retry', '3');\n const webSearch = await configService.getSetting('web_search', '0');\n const autoTest = await configService.getSetting('auto_test', '0');\n\n // Ensure consistent data types\n res.json({\n keepalive: String(keepalive), // Ensure it's a string\n maxRetry: parseInt(maxRetry) || 3,\n webSearch: String(webSearch),\n autoTest: String(autoTest)\n });\n } catch (error) {\n next(error);\n }\n })\n .post(async (req, res, next) => {\n try {\n const { keepalive, maxRetry, webSearch, autoTest } = parseBody(req);\n\n // Validate inputs\n if (keepalive !== '0' && keepalive !== '1') {\n return res.status(400).json({ error: 'KEEPALIVE must be \"0\" or \"1\"' });\n }\n\n const maxRetryNum = parseInt(maxRetry);\n if (isNaN(maxRetryNum) || maxRetryNum < 0 || maxRetryNum > 10) {\n return res.status(400).json({ error: 'MAX_RETRY must be a number between 0 and 10' });\n }\n\n if (webSearch !== '0' && webSearch !== '1') {\n return res.status(400).json({ error: 'WEB_SEARCH must be \"0\" or \"1\"' });\n }\n\n if (autoTest !== '0' && autoTest !== '1') {\n return res.status(400).json({ error: 'AUTO_TEST must be \"0\" or \"1\"' });\n }\n\n // Save to database (skip sync for first three, sync on the last one)\n await configService.setSetting('keepalive', keepalive, true); // Skip sync\n await configService.setSetting('max_retry', maxRetryNum.toString(), true); // Skip sync\n await configService.setSetting('web_search', webSearch, true); // Skip sync\n await configService.setSetting('auto_test', autoTest); // Trigger sync on last setting\n\n // Update scheduler service when auto_test setting changes\n try {\n const schedulerService = require('../services/schedulerService');\n await schedulerService.updateBatchTestSchedule();\n console.log('Scheduler updated after auto_test setting change');\n } catch (schedulerError) {\n console.error('Failed to update scheduler:', schedulerError);\n // Don't fail the request if scheduler update fails\n }\n\n res.json({\n success: true,\n keepalive: keepalive,\n maxRetry: maxRetryNum,\n webSearch: webSearch,\n autoTest: autoTest\n });\n } catch (error) {\n next(error);\n }\n });\n\n// --- Batch Test Management --- (/api/admin/batch-test)\nrouter.post('/batch-test/run', async (req, res, next) => {\n try {\n console.log('Manual batch test triggered via API');\n const result = await batchTestService.runBatchTest();\n\n res.json({\n success: true,\n message: 'Batch test completed',\n ...result\n });\n } catch (error) {\n console.error('Error running manual batch test:', error);\n next(error);\n }\n});\n\nrouter.get('/batch-test/status', async (req, res, next) => {\n try {\n const schedulerService = require('../services/schedulerService');\n const schedulerStatus = schedulerService.getStatus();\n const autoTestEnabled = await configService.getSetting('auto_test', '0');\n\n res.json({\n success: true,\n autoTestEnabled: autoTestEnabled === '1',\n schedulerStatus: schedulerStatus\n });\n } catch (error) {\n console.error('Error getting batch test status:', error);\n next(error);\n }\n});\n\n\nmodule.exports = router;\n"
},
{
"path": "src/routes/apiV1.js",
"content": "// src/routes/apiV1.js\n\nconst express = require('express');\nconst { Readable, Transform } = require('stream'); // For handling streams and transforming\nconst requireWorkerAuth = require('../middleware/workerAuth');\nconst geminiProxyService = require('../services/geminiProxyService');\nconst configService = require('../services/configService'); // For /v1/models\nconst transformUtils = require('../utils/transform');\n\n// Import vertexProxyService, which now includes manual loading logic\nconst vertexProxyService = require('../services/vertexProxyService');\n\nconst router = express.Router();\n\n// Apply worker authentication middleware to all /v1 routes\nrouter.use(requireWorkerAuth);\n\n// --- /v1/models ---\nrouter.get('/models', async (req, res, next) => {\n try {\n const modelsConfig = await configService.getModelsConfig();\n let modelsData = Object.keys(modelsConfig).map(modelId => ({\n id: modelId,\n object: \"model\",\n created: Math.floor(Date.now() / 1000), // Placeholder timestamp\n owned_by: \"google\", // Assuming all configured models are Google's\n // Add other relevant properties if available/needed\n }));\n\n // Check if web search is enabled\n const webSearchEnabled = String(await configService.getSetting('web_search', '0')) === '1';\n\n // Add search versions for gemini-2.0+ series models only if web search is enabled\n let searchModels = [];\n if (webSearchEnabled) {\n searchModels = Object.keys(modelsConfig)\n .filter(modelId =>\n // Match gemini-2.0, gemini-2.5, gemini-3.0, etc. series models\n /^gemini-[2-9]\\.\\d/.test(modelId) &&\n // Exclude models that are already search versions\n !modelId.endsWith('-search')\n )\n .map(modelId => ({\n id: `${modelId}-search`,\n object: \"model\",\n created: Math.floor(Date.now() / 1000),\n owned_by: \"google\",\n }));\n }\n \n // Add non-thinking versions for gemini-2.5-flash-preview models\n const nonThinkingModels = Object.keys(modelsConfig)\n .filter(modelId => \n // Currently only gemini-2.5-flash-preview supports thinkingBudget\n modelId.includes('gemini-2.5-flash-preview') && \n // Exclude models that are already non-thinking versions\n !modelId.endsWith(':non-thinking')\n )\n .map(modelId => ({\n id: `${modelId}:non-thinking`,\n object: \"model\",\n created: Math.floor(Date.now() / 1000),\n owned_by: \"google\",\n }));\n \n // Merge regular, search and non-thinking model lists\n modelsData = [...modelsData, ...searchModels, ...nonThinkingModels];\n\n // If Vertex feature is enabled (via manual loading), add Vertex AI supported models\n if (vertexProxyService.isVertexEnabled()) {\n const vertexModels = vertexProxyService.getVertexSupportedModels().map(modelId => ({\n id: modelId, // Model ID including [v] prefix\n object: \"model\",\n created: Math.floor(Date.now() / 1000),\n owned_by: \"google\",\n }));\n \n // Add Vertex models to the list\n modelsData = [...modelsData, ...vertexModels];\n }\n\n res.json({ object: \"list\", data: modelsData });\n } catch (error) {\n console.error(\"Error handling /v1/models:\", error);\n next(error); // Pass to global error handler\n }\n});\n\n// --- /v1/chat/completions ---\nrouter.post('/chat/completions', async (req, res, next) => {\n const openAIRequestBody = req.body;\n const workerApiKey = req.workerApiKey; // Attached by requireWorkerAuth middleware\n const stream = openAIRequestBody?.stream ?? false;\n const requestedModelId = openAIRequestBody?.model; // Keep track for transformations\n \n try {\n // --- Model Validation Step ---\n // Get all available models to validate against the request\n const modelsConfig = await configService.getModelsConfig();\n let enabledModels = Object.keys(modelsConfig);\n\n // Add search versions if web search is enabled\n const webSearchEnabled = String(await configService.getSetting('web_search', '0')) === '1';\n if (webSearchEnabled) {\n const searchModels = Object.keys(modelsConfig)\n .filter(modelId => /^gemini-[2-9]\\.\\d/.test(modelId) && !modelId.endsWith('-search'))\n .map(modelId => `${modelId}-search`);\n enabledModels = [...enabledModels, ...searchModels];\n }\n \n // Add non-thinking versions\n const nonThinkingModels = Object.keys(modelsConfig)\n .filter(modelId => modelId.includes('gemini-2.5-flash-preview') && !modelId.endsWith(':non-thinking'))\n .map(modelId => `${modelId}:non-thinking`);\n enabledModels = [...enabledModels, ...nonThinkingModels];\n\n // Add Vertex models if the feature is enabled\n if (vertexProxyService.isVertexEnabled()) {\n const vertexModels = vertexProxyService.getVertexSupportedModels();\n enabledModels = [...enabledModels, ...vertexModels];\n }\n\n // Validate that the requested model is in the enabled list\n if (!requestedModelId || !enabledModels.includes(requestedModelId)) {\n return res.status(400).json({\n error: {\n message: `Model not found or not enabled: ${requestedModelId}. Please check the /v1/models endpoint for available models.`,\n type: 'invalid_request_error',\n param: 'model'\n }\n });\n }\n // --- End Model Validation ---\n \n // Check if this is a non-thinking model request\n const isNonThinking = requestedModelId?.endsWith(':non-thinking');\n // Remove the suffix for actual model lookup, but keep original for response\n const actualModelId = isNonThinking ? requestedModelId.replace(':non-thinking', '') : requestedModelId;\n \n // Set thinkingBudget to 0 for non-thinking models\n const thinkingBudget = isNonThinking ? 0 : undefined;\n \n // If model was modified, update the request body with the actual model ID\n if (isNonThinking) {\n openAIRequestBody.model = actualModelId;\n }\n\n let result;\n\n // KEEPALIVE mode setup - prepare heartbeat callback if needed\n let keepAliveCallback = null;\n const keepAliveEnabled = String(await configService.getSetting('keepalive', '0')) === '1';\n const isSafetyEnabled = await configService.getWorkerKeySafetySetting(workerApiKey);\n const useKeepAlive = keepAliveEnabled && stream && !isSafetyEnabled;\n\n // Debug logging for KEEPALIVE mode\n console.log(`KEEPALIVE Debug - keepAliveEnabled: ${keepAliveEnabled}, stream: ${stream}, isSafetyEnabled: ${isSafetyEnabled}, useKeepAlive: ${useKeepAlive}`);\n\n if (useKeepAlive) {\n // Set up KEEPALIVE heartbeat management\n const { Readable } = require('stream');\n const keepAliveSseStream = new Readable({ read() {} });\n let keepAliveTimerId = null;\n let isConnectionClosed = false;\n\n // Function to safely clean up resources\n const cleanup = () => {\n if (keepAliveTimerId) {\n clearInterval(keepAliveTimerId);\n keepAliveTimerId = null;\n }\n isConnectionClosed = true;\n };\n\n // Monitor client connection status\n res.on('close', () => {\n console.log('KEEPALIVE: Client connection closed');\n cleanup();\n });\n\n res.on('error', (err) => {\n console.error('KEEPALIVE: Client connection error:', err);\n cleanup();\n });\n\n // Handle stream errors\n keepAliveSseStream.on('error', (err) => {\n console.error('KEEPALIVE: Stream error:', err);\n cleanup();\n });\n\n keepAliveSseStream.on('end', () => {\n console.log('KEEPALIVE: Stream ended');\n cleanup();\n });\n\n keepAliveSseStream.on('finish', () => {\n console.log('KEEPALIVE: Stream finished');\n cleanup();\n });\n\n const sendKeepAliveSseChunk = () => {\n // Check multiple connection states\n if (isConnectionClosed || res.writableEnded || res.destroyed || !res.writable) {\n cleanup();\n return;\n }\n\n try {\n const keepAliveSseData = {\n id: \"keepalive\",\n object: \"chat.completion.chunk\",\n created: Math.floor(Date.now() / 1000),\n model: requestedModelId,\n choices: [{ index: 0, delta: {}, finish_reason: null }]\n };\n keepAliveSseStream.push(`data: ${JSON.stringify(keepAliveSseData)}\\n\\n`);\n } catch (err) {\n console.error('KEEPALIVE: Error sending heartbeat:', err);\n cleanup();\n }\n };\n\n // Set streaming headers for KEEPALIVE mode\n res.setHeader('Content-Type', 'text/event-stream; charset=utf-8');\n res.setHeader('Cache-Control', 'no-cache');\n res.setHeader('Connection', 'keep-alive');\n res.setHeader('X-Proxied-By', 'gemini-proxy-panel-node');\n\n // Pipe stream to response after setting up error handlers\n const pipeStream = keepAliveSseStream.pipe(res);\n\n // Handle pipe errors\n pipeStream.on('error', (err) => {\n console.error('KEEPALIVE: Pipe error:', err);\n cleanup();\n });\n\n // Create callback object for geminiProxyService\n keepAliveCallback = {\n startHeartbeat: () => {\n console.log('KEEPALIVE: Starting heartbeat (3 second intervals)');\n keepAliveTimerId = setInterval(sendKeepAliveSseChunk, 3000); // 3 second intervals\n sendKeepAliveSseChunk(); // Send first one immediately\n },\n stopHeartbeat: () => {\n console.log('KEEPALIVE: Stopping heartbeat');\n cleanup();\n },\n sendFinalResponse: (responseData) => {\n try {\n // Double-check connection status\n if (res.writableEnded || res.destroyed || !res.writable) {\n console.warn(\"KEEPALIVE: Response stream ended before data could be sent.\");\n return;\n }\n\n const openAIResponse = JSON.parse(transformUtils.transformGeminiResponseToOpenAI(\n responseData,\n requestedModelId\n ));\n const content = openAIResponse.choices[0].message.content || \"\";\n const completeChunk = {\n id: `chatcmpl-${Date.now()}-${Math.random().toString(36).substring(2, 8)}`,\n object: \"chat.completion.chunk\",\n created: Math.floor(Date.now() / 1000),\n model: requestedModelId,\n choices: [{\n index: 0,\n delta: { role: \"assistant\", content: content },\n finish_reason: openAIResponse.choices[0].finish_reason || \"stop\"\n }]\n };\n\n keepAliveSseStream.push(`data: ${JSON.stringify(completeChunk)}\\n\\n`);\n keepAliveSseStream.push('data: [DONE]\\n\\n');\n keepAliveSseStream.push(null); // End the stream\n } catch (error) {\n console.error(\"Error processing KEEPALIVE final response:\", error);\n const errorPayload = {\n error: {\n message: error.message || 'Failed to process KEEPALIVE response',\n type: error.type || 'keepalive_proxy_error',\n code: error.code,\n status: error.status\n }\n };\n keepAliveSseStream.push(`data: ${JSON.stringify(errorPayload)}\\n\\n`);\n keepAliveSseStream.push('data: [DONE]\\n\\n');\n keepAliveSseStream.push(null);\n }\n },\n sendError: (errorData) => {\n try {\n // Double-check connection status\n if (res.writableEnded || res.destroyed || !res.writable) {\n console.warn(\"KEEPALIVE: Response stream ended before error could be sent.\");\n return;\n }\n\n const errorPayload = {\n error: {\n message: errorData.message || 'Upstream API error',\n type: errorData.type || 'upstream_error',\n code: errorData.code\n }\n };\n\n keepAliveSseStream.push(`data: ${JSON.stringify(errorPayload)}\\n\\n`);\n keepAliveSseStream.push('data: [DONE]\\n\\n');\n keepAliveSseStream.push(null); // End the stream\n } catch (error) {\n console.error(\"Error sending KEEPALIVE error response:\", error);\n // Try to end the stream gracefully\n try {\n keepAliveSseStream.push(null);\n } catch (e) {\n console.error(\"Failed to end stream after error:\", e);\n }\n }\n }\n };\n }\n\n // Check if it's a Vertex model (with [v] prefix) and confirm Vertex feature is enabled\n if (requestedModelId && requestedModelId.startsWith('[v]') && vertexProxyService.isVertexEnabled()) {\n // Use Vertex proxy service to handle the request\n console.log(`Using Vertex AI to process model: ${requestedModelId}`);\n result = await vertexProxyService.proxyVertexChatCompletions(\n openAIRequestBody,\n workerApiKey,\n stream,\n keepAliveCallback\n );\n } else {\n // Use Gemini proxy service to handle the request with optional thinkingBudget\n result = await geminiProxyService.proxyChatCompletions(\n openAIRequestBody,\n workerApiKey,\n stream,\n thinkingBudget,\n keepAliveCallback\n );\n }\n\n // Check if the service returned an error\n if (result.error) {\n // In KEEPALIVE mode, send error through the heartbeat stream\n if (useKeepAlive && keepAliveCallback) {\n console.log('KEEPALIVE: Sending error response through heartbeat stream');\n try {\n // Stop heartbeat first\n keepAliveCallback.stopHeartbeat();\n\n // Send error through the stream\n const errorPayload = {\n error: {\n message: result.error.message || 'Upstream API error',\n type: result.error.type || 'upstream_error',\n code: result.error.code,\n status: result.status || 500\n }\n };\n\n // Use the existing stream to send error\n const { Readable } = require('stream');\n const errorStream = new Readable({ read() {} });\n errorStream.pipe(res);\n errorStream.push(`data: ${JSON.stringify(errorPayload)}\\n\\n`);\n errorStream.push('data: [DONE]\\n\\n');\n errorStream.push(null);\n return;\n } catch (streamError) {\n console.error('KEEPALIVE: Failed to send error through stream:', streamError);\n // Fallback: if stream fails, we can't do much more since headers are already sent\n return;\n }\n } else {\n // Normal mode: set headers and send JSON error\n res.setHeader('Content-Type', 'application/json');\n return res.status(result.status || 500).json({ error: result.error });\n }\n }\n\n // Destructure the successful result\n const { response: geminiResponse, selectedKeyId, modelCategory } = result;\n\n // --- Handle Response ---\n\n // Check if this is a KEEPALIVE special response first\n if (result.isKeepAlive) {\n console.log(`KEEPALIVE mode activated for model ${requestedModelId} - response will be handled asynchronously`);\n // In the new KEEPALIVE mode, the response is handled completely asynchronously\n // The heartbeat is already started and the response will be sent when ready\n // We just return here as everything is handled in the background\n return; // Exit early for KEEPALIVE mode\n }\n\n // Set common headers (only for non-KEEPALIVE mode)\n res.setHeader('X-Proxied-By', 'gemini-proxy-panel-node');\n res.setHeader('X-Selected-Key-ID', selectedKeyId); // Send back which key was used (optional)\n\n if (stream) {\n // --- Streaming Response ---\n res.setHeader('Content-Type', 'text/event-stream; charset=utf-8');\n res.setHeader('Cache-Control', 'no-cache');\n res.setHeader('Connection', 'keep-alive');\n // Apply CORS headers if not already handled globally by middleware\n // res.setHeader('Access-Control-Allow-Origin', '*'); // Example if needed\n\n\n // Check in advance if it's keepalive mode, if so, no need to check the body stream\n if (!result.isKeepAlive) {\n if (!geminiResponse.body || typeof geminiResponse.body.pipe !== 'function') {\n console.error('Gemini response body is not a readable stream for streaming request.');\n // Send a valid SSE error event before closing\n const errorPayload = JSON.stringify({ error: { message: 'Upstream response body is not readable.', type: 'proxy_error' } });\n res.write(`data: ${errorPayload}\\n\\n`);\n res.write('data: [DONE]\\n\\n');\n return res.end();\n }\n }\n\n const decoder = new TextDecoder();\n let buffer = '';\n let lineBuffer = '';\n let jsonCollector = '';\n let isCollectingJson = false;\n let openBraces = 0;\n let closeBraces = 0;\n\n // Implement stream processing transformer for both Gemini and Vertex streams\n const streamTransformer = new Transform({\n transform(chunk, encoding, callback) {\n try {\n const chunkStr = decoder.decode(chunk, { stream: true });\n buffer += chunkStr;\n\n // Process based on the source (Gemini or Vertex)\n if (selectedKeyId === 'vertex-ai') {\n // Vertex stream response is a series of continuous JSON objects without newline separation\n // Use a method similar to Gemini to process JSON objects\n let startPos = -1;\n let endPos = -1;\n let bracketDepth = 0;\n let inString = false;\n let escapeNext = false;\n let flushed = false;\n \n // Scan the entire buffer to find complete JSON objects\n for (let i = 0; i < buffer.length; i++) {\n const char = buffer[i];\n \n // Handle characters inside strings\n if (inString) {\n if (escapeNext) {\n escapeNext = false;\n } else if (char === '\\\\') {\n escapeNext = true;\n } else if (char === '\"') {\n inString = false;\n }\n continue;\n }\n \n // Handle characters outside strings\n if (char === '{') {\n if (bracketDepth === 0) {\n startPos = i; // Record the starting position of a new JSON object\n }\n bracketDepth++;\n } else if (char === '}') {\n bracketDepth--;\n if (bracketDepth === 0 && startPos !== -1) {\n endPos = i;\n \n // Extract and process the complete JSON object\n const jsonStr = buffer.substring(startPos, endPos + 1);\n try {\n // Check if it's the 'done' marker from vertexProxyService's flush\n // We only need to parse if we suspect it might be the done object.\n // Otherwise, jsonStr is already the stringified chunk we want.\n if (jsonStr.includes('\"done\":true')) { // Quick check\n try {\n const jsonObj = JSON.parse(jsonStr);\n if (jsonObj.done) {\n // This is the '{\"done\":true}' from vertexProxyService's flush.\n // The main flush of apiV1's transformer will send 'data: [DONE]\\n\\n'. So, ignore this one.\n } else {\n // It wasn't the done object, but was parsable. Send it.\n this.push(`data: ${jsonStr}\\n\\n`);\n if (typeof res.flush === 'function') res.flush();\n }\n } catch (e) {\n // Parsing failed, but it might still be a valid (non-done) chunk.\n // This case should ideally not happen if vertexProxyService sends valid JSONs.\n console.error(\"Error parsing potential Vertex JSON object:\", e, \"Original string:\", jsonStr);\n this.push(`data: ${jsonStr}\\n\\n`); // Send as is if parsing fails but wasn't 'done'\n if (typeof res.flush === 'function') res.flush();\n }\n } else {\n // Not the 'done' marker, so jsonStr is a data chunk.\n this.push(`data: ${jsonStr}\\n\\n`);\n if (typeof res.flush === 'function') res.flush();\n }\n } catch (e) {\n // This outer catch handles errors from buffer.substring or other unexpected issues\n console.error(\"Error processing Vertex JSON chunk:\", e, \"Original string:\", jsonStr);\n }\n \n // Continue searching for the next object\n startPos = -1;\n \n // Truncate the processed part\n if (i + 1 < buffer.length) {\n buffer = buffer.substring(endPos + 1);\n i = -1; // Reset index to scan the remaining buffer from the beginning\n } else {\n buffer = '';\n break; // Exit loop if buffer is exhausted\n }\n }\n } else if (char === '\"') {\n inString = true;\n }\n }\n } else {\n // Original Gemini stream processing (find raw Gemini JSON chunks)\n let startPos = -1;\n let endPos = -1;\n let bracketDepth = 0;\n let inString = false;\n let escapeNext = false;\n \n // Scan the entire buffer to find complete JSON objects\n for (let i = 0; i < buffer.length; i++) {\n const char = buffer[i];\n \n // Handle characters within strings\n if (inString) {\n if (escapeNext) {\n escapeNext = false;\n } else if (char === '\\\\') {\n escapeNext = true;\n } else if (char === '\"') {\n inString = false;\n }\n continue;\n }\n \n // Handle characters outside strings\n if (char === '{') {\n if (bracketDepth === 0) {\n startPos = i; // Record the starting position of a new JSON object\n }\n bracketDepth++;\n } else if (char === '}') {\n bracketDepth--;\n if (bracketDepth === 0 && startPos !== -1) {\n endPos = i;\n \n // Extract and process the complete JSON object\n const jsonStr = buffer.substring(startPos, endPos + 1);\n try {\n const jsonObj = JSON.parse(jsonStr);\n // Immediately process and send this object\n processGeminiObject(jsonObj, this);\n } catch (e) {\n console.error(\"Error parsing JSON object:\", e);\n }\n \n // Continue searching for the next object\n startPos = -1;\n }\n } else if (char === '\"') {\n inString = true;\n } else if (char === '[' && !inString && startPos === -1) {\n // Ignore the start marker of JSON arrays, as we process each object individually\n continue;\n } else if (char === ']' && !inString && bracketDepth === 0) {\n // Ignore the end marker of JSON arrays\n continue;\n } else if (char === ',') {\n // If there's a comma after an object, continue processing the next object\n continue;\n }\n }\n \n // Keep the unprocessed part for Gemini stream\n if (startPos !== -1 && endPos !== -1 && endPos > startPos) {\n buffer = buffer.substring(endPos + 1);\n } else if (startPos !== -1) {\n buffer = buffer.substring(startPos);\n } else {\n buffer = '';\n }\n } // End of else (Gemini stream processing)\n \n callback();\n } catch (e) {\n console.error(\"Error in stream transform:\", e);\n callback(e);\n }\n },\n \n flush(callback) {\n try {\n // Handling the remaining buffer\n if (buffer.trim()) {\n if (selectedKeyId === 'vertex-ai') {\n if (buffer.trim()) {\n let startPos = -1;\n let endPos = -1;\n let bracketDepth = 0;\n let inString = false;\n let escapeNext = false;\n \n for (let i = 0; i < buffer.length; i++) {\n const char = buffer[i];\n \n if (inString) {\n if (escapeNext) {\n escapeNext = false;\n } else if (char === '\\\\') {\n escapeNext = true;\n } else if (char === '\"') {\n inString = false;\n }\n continue;\n }\n \n if (char === '{') {\n if (bracketDepth === 0) {\n startPos = i;\n }\n bracketDepth++;\n } else if (char === '}') {\n bracketDepth--;\n if (bracketDepth === 0 && startPos !== -1) {\n endPos = i;\n \n try {\n const jsonStr = buffer.substring(startPos, endPos + 1);\n const jsonObj = JSON.parse(jsonStr);\n if (!jsonObj.done) { // Avoid duplicate DONE\n this.push(`data: ${JSON.stringify(jsonObj)}\\n\\n`);\n }\n } catch (e) {\n console.debug(\"Could not parse Vertex buffer JSON:\", e);\n }\n \n // Update the buffer and reset the index\n if (endPos + 1 < buffer.length) {\n buffer = buffer.substring(endPos + 1);\n i = -1; // Reset index\n } else {\n buffer = '';\n break;\n }\n }\n } else if (char === '\"') {\n inString = true;\n }\n }\n }\n } else {\n // Try parsing remaining Gemini JSON object\n try {\n const jsonObj = JSON.parse(buffer);\n processGeminiObject(jsonObj, this); // Use existing Gemini processing\n } catch (e) {\n console.debug(\"Could not parse final Gemini buffer:\", buffer, e);\n }\n }\n }\n \n // Always send the final [DONE] event\n // console.log(\"Stream transformer flushing, sending [DONE].\"); // Removed log\n this.push('data: [DONE]\\n\\n');\n callback();\n } catch (e) {\n console.error(\"Error in stream flush:\", e); // Keep error log in English\n callback(e);\n }\n }\n });\n \n // Process a single Gemini API response object and convert it to OpenAI format\n function processGeminiObject(geminiObj, stream) {\n if (!geminiObj) return;\n \n // If it's a valid Gemini response object (contains candidates)\n if (geminiObj.candidates && geminiObj.candidates.length > 0) {\n // Convert and send directly\n const openaiChunkStr = transformUtils.transformGeminiStreamChunk(geminiObj, requestedModelId);\n if (openaiChunkStr) {\n stream.push(openaiChunkStr);\n }\n } else if (Array.isArray(geminiObj)) {\n // If it's an array, process each element\n for (const item of geminiObj) {\n processGeminiObject(item, stream);\n }\n } else if (geminiObj.text) {\n // Single text fragment, construct Gemini format\n const mockGeminiChunk = {\n candidates: [{\n content: {\n parts: [{ text: geminiObj.text }],\n role: \"model\"\n }\n }]\n };\n \n const openaiChunkStr = transformUtils.transformGeminiStreamChunk(mockGeminiChunk, requestedModelId);\n if (openaiChunkStr) {\n stream.push(openaiChunkStr);\n }\n }\n // May need to handle other response types...\n }\n\n // Standard (non-KEEPALIVE) Gemini and Vertex streams\n if (!geminiResponse || !geminiResponse.body || typeof geminiResponse.body.pipe !== 'function') {\n console.error('Upstream response body is not a readable stream for standard streaming request.');\n const errorPayload = JSON.stringify({ error: { message: 'Upstream response body is not readable.', type: 'proxy_error' } });\n res.write(`data: ${errorPayload}\\n\\n`); // Use res.write for SSE\n res.write('data: [DONE]\\n\\n');\n return res.end();\n }\n\n console.log(`Piping ${selectedKeyId === 'vertex-ai' ? 'Vertex' : 'Gemini'} stream through transformer.`);\n geminiResponse.body.pipe(streamTransformer).pipe(res);\n\n geminiResponse.body.on('error', (err) => {\n console.error(`Error reading stream from upstream (${selectedKeyId}):`, err);\n if (!res.headersSent) {\n // If headers not sent, we can still send a JSON error\n res.status(500).json({ error: { message: 'Error reading stream from upstream API.' } });\n } else if (!res.writableEnded) {\n // If headers sent but stream not ended, try to send an SSE error then end\n const sseError = JSON.stringify({ error: { message: 'Upstream stream error', type: 'upstream_error'} });\n res.write(`data: ${sseError}\\n\\n`);\n res.write('data: [DONE]\\n\\n');\n res.end();\n }\n // If res.writableEnded is true, nothing more we can do.\n });\n\n streamTransformer.on('error', (err) => {\n console.error('Error in stream transformer:', err);\n if (!res.headersSent) {\n res.status(500).json({ error: { message: 'Error processing stream data.' } });\n } else if (!res.writableEnded) {\n const sseError = JSON.stringify({ error: { message: 'Stream processing error', type: 'transform_error'} });\n res.write(`data: ${sseError}\\n\\n`);\n res.write('data: [DONE]\\n\\n');\n res.end();\n }\n });\n\n console.log(`Streaming response initiated for key ${selectedKeyId}`);\n\n\n } else {\n // --- Non-Streaming Response ---\n res.setHeader('Content-Type', 'application/json; charset=utf-8');\n\n try {\n if (selectedKeyId === 'vertex-ai') {\n // Vertex service already transformed the response to OpenAI format\n const openaiJson = await geminiResponse.json(); // Get the pre-transformed JSON\n res.status(geminiResponse.status || 200).json(openaiJson); // Send it directly\n console.log(`Non-stream Vertex request completed, status: ${geminiResponse.status || 200}`);\n } else {\n // Original Gemini service response handling\n const geminiJson = await geminiResponse.json(); // Parse the raw upstream Gemini JSON\n const openaiJsonString = transformUtils.transformGeminiResponseToOpenAI(geminiJson, requestedModelId); // Transform it\n // Use Gemini's original status code if available and OK, otherwise default to 200\n res.status(geminiResponse.ok ? geminiResponse.status : 200).send(openaiJsonString);\n console.log(`Non-stream Gemini request completed for key ${selectedKeyId}, status: ${geminiResponse.status}`);\n }\n } catch (jsonError) {\n console.error(\"Error parsing Gemini non-stream JSON response:\", jsonError);\n // Check if response text might give clues\n try {\n const errorText = await geminiResponse.text(); // Need to re-read or clone earlier\n console.error(\"Gemini non-stream response text:\", errorText);\n } catch(e){}\n next(new Error(\"Failed to parse upstream API response.\")); // Pass to global error handler\n }\n }\n\n } catch (error) {\n console.error(\"Error in /v1/chat/completions handler:\", error);\n next(error); // Pass error to the global Express error handler\n }\n});\n\nmodule.exports = router;\n"
},
{
"path": "src/routes/auth.js",
"content": "const express = require('express');\nconst { generateSessionToken, setSessionCookie, clearSessionCookie } = require('../utils/session');\nconst { readRequestBody } = require('../utils/helpers'); // Although body-parser is used, keep for consistency\n\nconst router = express.Router();\n\nconst ADMIN_PASSWORD = process.env.ADMIN_PASSWORD;\n\nif (!ADMIN_PASSWORD) {\n console.error(\"FATAL: ADMIN_PASSWORD environment variable is not set. Admin login disabled.\");\n}\n\n// --- Login Route ---\n// Path: /api/login (mounted under /api in server.js)\nrouter.post('/login', async (req, res, next) => {\n if (!ADMIN_PASSWORD) {\n return res.status(500).json({ error: 'Server configuration error: Admin password not set.' });\n }\n\n try {\n // express.json() middleware populates req.body\n const body = req.body;\n if (!body || typeof body.password !== 'string') {\n return res.status(400).json({ error: 'Password is required.' });\n }\n\n if (body.password === ADMIN_PASSWORD) {\n // Password matches, generate and set session token\n const token = await generateSessionToken();\n if (!token) {\n // Error already logged in generateSessionToken\n return res.status(500).json({ error: 'Failed to generate session token.' });\n }\n\n setSessionCookie(res, token); // Set the cookie on the response\n console.log('Admin login successful.');\n return res.status(200).json({ success: true });\n\n } else {\n // Invalid password\n console.warn('Admin login failed: Invalid password.');\n return res.status(401).json({ error: 'Invalid password.' });\n }\n } catch (error) {\n console.error(\"Error during login:\", error);\n // Pass error to the global error handler\n next(error);\n }\n});\n\n// --- Logout Route ---\n// Path: /api/logout (mounted under /api in server.js)\nrouter.post('/logout', (req, res) => {\n try {\n clearSessionCookie(res); // Clear the cookie\n console.log('Admin logout successful.');\n // Send a simple success response. Client should handle redirect.\n res.status(200).json({ success: true });\n } catch (error) {\n console.error(\"Error during logout:\", error);\n // Pass error to the global error handler\n // Note: synchronous errors might not be caught by Express error handler unless passed explicitly\n next(error); // Ensure error is passed if any occurs unexpectedly\n }\n});\n\nmodule.exports = router;\n"
},
{
"path": "src/services/batchTestService.js",
"content": "const fetch = require('node-fetch');\nconst configService = require('./configService');\nconst geminiKeyService = require('./geminiKeyService');\nconst proxyPool = require('../utils/proxyPool');\n\n// Base Gemini API URL\nconst BASE_GEMINI_URL = process.env.GEMINI_BASE_URL || 'https://generativelanguage.googleapis.com';\n\n// Helper function to check if a 400 error should be marked for key error\nfunction shouldMark400Error(responseBody) {\n try {\n // Only mark 400 errors if the message indicates invalid API key\n if (responseBody && responseBody.error) {\n const errorMessage = responseBody.error.message;\n\n // Check for the specific \"API key not valid\" error\n if (errorMessage && errorMessage.includes('API key not valid. Please pass a valid API key.')) {\n return true;\n }\n }\n return false;\n } catch (e) {\n // If we can't parse the error, don't mark it\n return false;\n }\n}\n\n/**\n * Tests a single Gemini API key\n * @param {string} keyId - The key ID to test\n * @param {string} modelId - The model ID to test with\n * @returns {Promise<{keyId: string, success: boolean, status: number|string, error?: string}>}\n */\nasync function testSingleKey(keyId, modelId) {\n try {\n // Fetch the actual key from the database\n const keyInfo = await configService.getDb('SELECT api_key FROM gemini_keys WHERE id = ?', [keyId]);\n if (!keyInfo || !keyInfo.api_key) {\n return {\n keyId,\n success: false,\n status: 'not_found',\n error: `API Key with ID '${keyId}' not found or invalid.`\n };\n }\n const apiKey = keyInfo.api_key;\n\n // Fetch model category for potential usage increment\n const modelsConfig = await configService.getModelsConfig();\n let modelCategory = modelsConfig[modelId]?.category;\n\n // If model is not configured, infer category from model name\n if (!modelCategory) {\n if (modelId.includes('flash')) {\n modelCategory = 'Flash';\n } else if (modelId.includes('pro')) {\n modelCategory = 'Pro';\n } else {\n // Default to Flash for unknown models (most common case)\n modelCategory = 'Flash';\n }\n }\n\n const testGeminiRequestBody = { contents: [{ role: \"user\", parts: [{ text: \"Hi\" }] }] };\n const geminiUrl = `${BASE_GEMINI_URL}/v1beta/models/${modelId}:generateContent`;\n\n let testResponseStatus = 500;\n let testResponseBody = null;\n let isSuccess = false;\n\n try {\n // Get proxy agent\n const agent = proxyPool.getNextProxyAgent();\n const fetchOptions = {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n 'x-goog-api-key': apiKey\n },\n body: JSON.stringify(testGeminiRequestBody)\n };\n if (agent) {\n fetchOptions.agent = agent;\n console.log(`Batch Test (Key ${keyId}): Sending request via proxy ${agent.proxy.href}`);\n } else {\n console.log(`Batch Test (Key ${keyId}): Sending request directly.`);\n }\n\n const response = await fetch(geminiUrl, fetchOptions);\n testResponseStatus = response.status;\n testResponseBody = await response.json(); // Attempt to parse JSON\n isSuccess = response.ok;\n\n if (isSuccess) {\n // Increment usage and sync to GitHub\n await geminiKeyService.incrementKeyUsage(keyId, modelId, modelCategory);\n\n // Clear error status if the key was previously marked with an error\n try {\n const wasCleared = await geminiKeyService.clearKeyError(keyId);\n if (wasCleared) {\n console.log(`Batch Test: Restored key ${keyId} - cleared previous error status.`);\n }\n } catch (clearError) {\n // Log but don't fail the test if clearing error status fails\n console.warn(`Batch Test: Failed to clear error status for key ${keyId}:`, clearError);\n }\n } else {\n // Record 400/401/403 errors (invalid API key, unauthorized, forbidden)\n // But only mark 400 errors if they indicate invalid API key\n if (testResponseStatus === 401 || testResponseStatus === 403) {\n await geminiKeyService.recordKeyError(keyId, testResponseStatus);\n } else if (testResponseStatus === 400) {\n // Check if this is an invalid API key 400 error that should be marked\n if (shouldMark400Error(testResponseBody)) {\n await geminiKeyService.recordKeyError(keyId, testResponseStatus);\n } else {\n console.log(`Batch Test: Skipping error marking for key ${keyId} - 400 error not related to invalid API key.`);\n }\n }\n }\n\n } catch (fetchError) {\n console.error(`Batch Test: Error testing Gemini API key ${keyId}:`, fetchError);\n testResponseBody = { error: `Fetch error: ${fetchError.message}` };\n isSuccess = false;\n testResponseStatus = 'network_error';\n // Don't assume network error means key is bad, could be temporary\n }\n\n return {\n keyId,\n success: isSuccess,\n status: testResponseStatus,\n error: isSuccess ? null : (testResponseBody?.error?.message || testResponseBody?.error || 'Test failed')\n };\n\n } catch (error) {\n console.error(`Batch Test: Error processing key ${keyId}:`, error);\n return {\n keyId,\n success: false,\n status: 'processing_error',\n error: error.message || 'Processing error'\n };\n }\n}\n\n/**\n * Runs batch test on all Gemini keys\n * @returns {Promise<{totalKeys: number, successCount: number, failureCount: number, results: Array}>}\n */\nasync function runBatchTest() {\n console.log('Starting automated batch test...');\n \n try {\n // Get all Gemini keys\n const keys = await geminiKeyService.getAllGeminiKeysWithUsage();\n if (!keys || keys.length === 0) {\n console.log('Batch Test: No Gemini keys found to test.');\n return {\n totalKeys: 0,\n successCount: 0,\n failureCount: 0,\n results: []\n };\n }\n\n const totalKeys = keys.length;\n const testModel = 'gemini-2.0-flash'; // Fixed model for testing\n const results = [];\n let successCount = 0;\n let failureCount = 0;\n\n console.log(`Batch Test: Testing ${totalKeys} keys with model ${testModel}`);\n\n // Process keys in batches to balance performance and server load\n const batchSize = 5; // Optimal batch size for testing\n for (let i = 0; i < keys.length; i += batchSize) {\n const batch = keys.slice(i, i + batchSize);\n \n console.log(`Batch Test: Processing batch ${Math.floor(i / batchSize) + 1} (${batch.length} keys)`);\n\n // Run tests for current batch concurrently\n const batchPromises = batch.map(key => testSingleKey(key.id, testModel));\n const batchResults = await Promise.allSettled(batchPromises);\n\n // Process results\n batchResults.forEach((result, index) => {\n if (result.status === 'fulfilled') {\n const testResult = result.value;\n results.push(testResult);\n \n if (testResult.success) {\n successCount++;\n console.log(`Batch Test: Key ${testResult.keyId} - SUCCESS`);\n } else {\n failureCount++;\n console.log(`Batch Test: Key ${testResult.keyId} - FAILED (${testResult.status}): ${testResult.error}`);\n }\n } else {\n const keyId = batch[index].id;\n failureCount++;\n results.push({\n keyId,\n success: false,\n status: 'promise_rejected',\n error: result.reason?.message || 'Promise rejected'\n });\n console.log(`Batch Test: Key ${keyId} - PROMISE REJECTED: ${result.reason?.message}`);\n }\n });\n\n // Delay between batches to reduce server load\n if (i + batchSize < keys.length) {\n console.log('Batch Test: Waiting 1 second before next batch...');\n await new Promise(resolve => setTimeout(resolve, 1000));\n }\n }\n\n const summary = {\n totalKeys,\n successCount,\n failureCount,\n results\n };\n\n console.log(`Batch Test completed: ${successCount} successful, ${failureCount} failed out of ${totalKeys} total keys.`);\n return summary;\n\n } catch (error) {\n console.error('Batch Test: Error during batch test execution:', error);\n throw error;\n }\n}\n\nmodule.exports = {\n testSingleKey,\n runBatchTest\n};\n"
},
{
"path": "src/services/configService.js",
"content": "const dbModule = require('../db');\n\n// --- Helper Functions for DB Interaction ---\n\n/**\n * Helper function to get database instance\n * @returns {object} Database instance\n */\nconst getDbInstance = () => {\n const db = dbModule.db;\n if (!db) {\n throw new Error('Database not initialized');\n }\n return db;\n};\n\n/**\n * Helper function to run a single SQL query with parameters.\n * Returns a Promise.\n * @param {string} sql The SQL query string.\n * @param {Array} params Query parameters.\n * @returns {PromiseJimiHub
\n\n \n\n \n
\n\n \n