mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-11 18:09:55 +08:00
619f81cce4
新增 `/config` 路由，提供一个可视化的配置编辑页面 (`config_editor.html`)。 用户现在可以通过网页界面管理： - API 密钥（包括批量添加和重置确认） - API 基础配置 (允许的令牌, 认证令牌, 基础URL, 最大失败次数, 超时) - 模型相关配置 (测试模型, 图像/搜索/过滤模型列表, 代码执行/搜索链接/思考过程开关) - 图像生成配置 (付费密钥, 模型, 上传提供商及相关密钥/URL) - 流式输出优化器配置 (开关, 延迟, 阈值, 分块大小) 同时更新了 `/keys` 页面 (`keys_status.html`)： - 页面主标题更改为 "Gemini Balance"。 - 添加了顶部导航选项卡，方便在 "配置编辑" (`/config`) 和 "密钥管理" (`/keys`) 之间切换。
556 lines
19 KiB
JavaScript
556 lines
19 KiB
JavaScript
document.addEventListener('DOMContentLoaded', function() {
|
||
// 初始化配置
|
||
initConfig();
|
||
|
||
// 标签切换
|
||
const tabButtons = document.querySelectorAll('.tab-btn');
|
||
tabButtons.forEach(button => {
|
||
button.addEventListener('click', function(e) {
|
||
// 防止事件冒泡
|
||
e.stopPropagation();
|
||
const tabId = this.getAttribute('data-tab');
|
||
switchTab(tabId);
|
||
});
|
||
});
|
||
|
||
// 上传提供商切换
|
||
const uploadProviderSelect = document.getElementById('UPLOAD_PROVIDER');
|
||
if (uploadProviderSelect) {
|
||
uploadProviderSelect.addEventListener('change', function() {
|
||
toggleProviderConfig(this.value);
|
||
});
|
||
}
|
||
|
||
// 切换按钮事件
|
||
const toggleSwitches = document.querySelectorAll('.toggle-switch');
|
||
toggleSwitches.forEach(toggleSwitch => {
|
||
toggleSwitch.addEventListener('click', function(e) {
|
||
// 防止事件冒泡
|
||
e.stopPropagation();
|
||
const checkbox = this.querySelector('input[type="checkbox"]');
|
||
if (checkbox) {
|
||
checkbox.checked = !checkbox.checked;
|
||
}
|
||
});
|
||
});
|
||
|
||
// 保存按钮
|
||
const saveBtn = document.getElementById('saveBtn');
|
||
if (saveBtn) {
|
||
saveBtn.addEventListener('click', saveConfig);
|
||
}
|
||
|
||
// 重置按钮
|
||
const resetBtn = document.getElementById('resetBtn');
|
||
if (resetBtn) {
|
||
resetBtn.addEventListener('click', resetConfig);
|
||
}
|
||
|
||
// 滚动按钮
|
||
window.addEventListener('scroll', toggleScrollButtons);
|
||
|
||
// --- 新增：API Key 模态框和搜索相关 ---
|
||
const apiKeyModal = document.getElementById('apiKeyModal');
|
||
const addApiKeyBtn = document.getElementById('addApiKeyBtn');
|
||
const closeApiKeyModalBtn = document.getElementById('closeApiKeyModalBtn');
|
||
const cancelAddApiKeyBtn = document.getElementById('cancelAddApiKeyBtn');
|
||
const confirmAddApiKeyBtn = document.getElementById('confirmAddApiKeyBtn');
|
||
const apiKeyBulkInput = document.getElementById('apiKeyBulkInput');
|
||
const apiKeySearchInput = document.getElementById('apiKeySearchInput');
|
||
|
||
// --- 新增：重置确认模态框相关 ---
|
||
const resetConfirmModal = document.getElementById('resetConfirmModal');
|
||
const closeResetModalBtn = document.getElementById('closeResetModalBtn');
|
||
const cancelResetBtn = document.getElementById('cancelResetBtn');
|
||
const confirmResetBtn = document.getElementById('confirmResetBtn');
|
||
// --- 结束：新增 ---
|
||
|
||
|
||
// 打开模态框
|
||
if (addApiKeyBtn) {
|
||
addApiKeyBtn.addEventListener('click', () => {
|
||
if (apiKeyModal) {
|
||
apiKeyModal.classList.add('show');
|
||
}
|
||
if (apiKeyBulkInput) apiKeyBulkInput.value = ''; // 清空输入框
|
||
});
|
||
}
|
||
|
||
// 关闭模态框 (X 按钮)
|
||
if (closeApiKeyModalBtn) {
|
||
closeApiKeyModalBtn.addEventListener('click', () => {
|
||
if (apiKeyModal) {
|
||
apiKeyModal.classList.remove('show');
|
||
}
|
||
});
|
||
}
|
||
|
||
// 关闭模态框 (取消按钮)
|
||
if (cancelAddApiKeyBtn) {
|
||
cancelAddApiKeyBtn.addEventListener('click', () => {
|
||
if (apiKeyModal) {
|
||
apiKeyModal.classList.remove('show');
|
||
}
|
||
});
|
||
}
|
||
|
||
// 点击模态框外部关闭 (处理两个模态框)
|
||
window.addEventListener('click', (event) => {
|
||
if (event.target == apiKeyModal) {
|
||
apiKeyModal.classList.remove('show');
|
||
}
|
||
if (event.target == resetConfirmModal) { // 新增对重置模态框的处理
|
||
resetConfirmModal.classList.remove('show');
|
||
}
|
||
});
|
||
|
||
// 确认添加 API Key
|
||
if (confirmAddApiKeyBtn) {
|
||
confirmAddApiKeyBtn.addEventListener('click', handleBulkAddApiKeys);
|
||
}
|
||
|
||
// API Key 搜索 (稍后实现具体逻辑)
|
||
if (apiKeySearchInput) {
|
||
apiKeySearchInput.addEventListener('input', handleApiKeySearch);
|
||
}
|
||
// --- 结束：API Key 相关 ---
|
||
|
||
// --- 新增：重置确认模态框事件监听 (移到 DOMContentLoaded 内部) ---
|
||
if (closeResetModalBtn) {
|
||
closeResetModalBtn.addEventListener('click', () => {
|
||
if (resetConfirmModal) {
|
||
resetConfirmModal.classList.remove('show');
|
||
}
|
||
});
|
||
}
|
||
if (cancelResetBtn) {
|
||
cancelResetBtn.addEventListener('click', () => {
|
||
if (resetConfirmModal) {
|
||
resetConfirmModal.classList.remove('show');
|
||
}
|
||
});
|
||
}
|
||
if (confirmResetBtn) {
|
||
// 调用之前定义的 executeReset 函数
|
||
confirmResetBtn.addEventListener('click', () => {
|
||
if (resetConfirmModal) {
|
||
resetConfirmModal.classList.remove('show'); // 关闭模态框
|
||
}
|
||
executeReset(); // 执行重置逻辑
|
||
});
|
||
}
|
||
// --- 结束：重置相关 ---
|
||
|
||
}); // <-- DOMContentLoaded 结束括号
|
||
|
||
// 初始化配置
|
||
async function initConfig() {
|
||
try {
|
||
showNotification('正在加载配置...', 'info');
|
||
const response = await fetch('/api/config');
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`HTTP error! status: ${response.status}`);
|
||
}
|
||
|
||
const config = await response.json();
|
||
|
||
// 确保数组字段有默认值
|
||
if (!config.API_KEYS || !Array.isArray(config.API_KEYS) || config.API_KEYS.length === 0) {
|
||
config.API_KEYS = ['请在此处输入 API 密钥'];
|
||
}
|
||
|
||
if (!config.ALLOWED_TOKENS || !Array.isArray(config.ALLOWED_TOKENS) || config.ALLOWED_TOKENS.length === 0) {
|
||
config.ALLOWED_TOKENS = [''];
|
||
}
|
||
|
||
if (!config.IMAGE_MODELS || !Array.isArray(config.IMAGE_MODELS) || config.IMAGE_MODELS.length === 0) {
|
||
config.IMAGE_MODELS = ['gemini-1.5-pro-latest'];
|
||
}
|
||
|
||
if (!config.SEARCH_MODELS || !Array.isArray(config.SEARCH_MODELS) || config.SEARCH_MODELS.length === 0) {
|
||
config.SEARCH_MODELS = ['gemini-1.5-flash-latest'];
|
||
}
|
||
|
||
if (!config.FILTERED_MODELS || !Array.isArray(config.FILTERED_MODELS) || config.FILTERED_MODELS.length === 0) {
|
||
config.FILTERED_MODELS = ['gemini-1.0-pro-latest'];
|
||
}
|
||
|
||
populateForm(config);
|
||
|
||
// 确保上传提供商有默认值
|
||
const uploadProvider = document.getElementById('UPLOAD_PROVIDER');
|
||
if (uploadProvider && !uploadProvider.value) {
|
||
uploadProvider.value = 'smms'; // 设置默认值为 smms
|
||
toggleProviderConfig('smms');
|
||
}
|
||
|
||
showNotification('配置加载成功', 'success');
|
||
} catch (error) {
|
||
console.error('加载配置失败:', error);
|
||
showNotification('加载配置失败: ' + error.message, 'error');
|
||
|
||
// 加载失败时，使用默认配置
|
||
const defaultConfig = {
|
||
API_KEYS: [''],
|
||
ALLOWED_TOKENS: [''],
|
||
IMAGE_MODELS: ['gemini-1.5-pro-latest'],
|
||
SEARCH_MODELS: ['gemini-1.5-flash-latest'],
|
||
FILTERED_MODELS: ['gemini-1.0-pro-latest'],
|
||
UPLOAD_PROVIDER: 'smms'
|
||
};
|
||
|
||
populateForm(defaultConfig);
|
||
toggleProviderConfig('smms');
|
||
}
|
||
}
|
||
|
||
// 填充表单
|
||
function populateForm(config) {
|
||
for (const [key, value] of Object.entries(config)) {
|
||
// 首先检查是否是数组类型
|
||
if (Array.isArray(value)) {
|
||
const container = document.getElementById(`${key}_container`);
|
||
if (container) {
|
||
// 清除现有项
|
||
const existingItems = container.querySelectorAll('.array-item');
|
||
existingItems.forEach(item => item.remove());
|
||
// 添加数组项
|
||
value.forEach(item => {
|
||
// 确保只添加非空字符串项（如果需要）
|
||
// if (item && typeof item === 'string' && item.trim() !== '') {
|
||
addArrayItemWithValue(key, item);
|
||
// }
|
||
});
|
||
}
|
||
// 处理完数组后，跳过本次循环的剩余部分
|
||
continue;
|
||
}
|
||
|
||
// 如果不是数组，再尝试查找对应的单个元素
|
||
const element = document.getElementById(key);
|
||
if (element) {
|
||
if (typeof value === 'boolean') {
|
||
element.checked = value;
|
||
} else {
|
||
// 处理其他类型 (确保 value 不是 null 或 undefined)
|
||
element.value = value ?? ''; // 使用空字符串作为默认值
|
||
}
|
||
}
|
||
// 如果既不是数组，也找不到对应 ID 的元素，则忽略该配置项
|
||
}
|
||
|
||
// 初始化上传提供商配置 (保持不变)
|
||
const uploadProvider = document.getElementById('UPLOAD_PROVIDER');
|
||
if (uploadProvider) {
|
||
toggleProviderConfig(uploadProvider.value);
|
||
}
|
||
}
|
||
|
||
// --- 新增：处理批量添加 API Key 的逻辑 ---
|
||
function handleBulkAddApiKeys() {
|
||
const apiKeyBulkInput = document.getElementById('apiKeyBulkInput');
|
||
const apiKeyContainer = document.getElementById('API_KEYS_container');
|
||
const apiKeyModal = document.getElementById('apiKeyModal');
|
||
|
||
if (!apiKeyBulkInput || !apiKeyContainer || !apiKeyModal) return;
|
||
|
||
const bulkText = apiKeyBulkInput.value;
|
||
const keyRegex = /AIzaSy\S{33}/g; // 全局匹配
|
||
const extractedKeys = bulkText.match(keyRegex) || [];
|
||
|
||
// 获取当前已有的 keys
|
||
const currentKeyInputs = apiKeyContainer.querySelectorAll('.array-input');
|
||
const currentKeys = Array.from(currentKeyInputs).map(input => input.value).filter(key => key.trim() !== '');
|
||
|
||
// 合并并去重
|
||
const combinedKeys = new Set([...currentKeys, ...extractedKeys]);
|
||
const uniqueKeys = Array.from(combinedKeys);
|
||
|
||
// 清空现有列表显示
|
||
const existingItems = apiKeyContainer.querySelectorAll('.array-item');
|
||
existingItems.forEach(item => item.remove());
|
||
|
||
// 重新填充列表
|
||
uniqueKeys.forEach(key => {
|
||
addArrayItemWithValue('API_KEYS', key);
|
||
});
|
||
|
||
// 关闭模态框
|
||
apiKeyModal.classList.remove('show');
|
||
showNotification(`添加/更新了 ${uniqueKeys.length} 个唯一密钥`, 'success');
|
||
}
|
||
|
||
// --- 新增：处理 API Key 搜索的逻辑 ---
|
||
function handleApiKeySearch() {
|
||
const apiKeySearchInput = document.getElementById('apiKeySearchInput');
|
||
const apiKeyContainer = document.getElementById('API_KEYS_container');
|
||
|
||
if (!apiKeySearchInput || !apiKeyContainer) return;
|
||
|
||
const searchTerm = apiKeySearchInput.value.toLowerCase();
|
||
const keyItems = apiKeyContainer.querySelectorAll('.array-item');
|
||
|
||
keyItems.forEach(item => {
|
||
const input = item.querySelector('.array-input');
|
||
if (input) {
|
||
const key = input.value.toLowerCase();
|
||
if (key.includes(searchTerm)) {
|
||
item.style.display = 'flex'; // 或者 'block'，取决于你的布局
|
||
} else {
|
||
item.style.display = 'none';
|
||
}
|
||
}
|
||
});
|
||
}
|
||
|
||
// 切换标签
|
||
function switchTab(tabId) {
|
||
// 更新标签按钮状态
|
||
const tabButtons = document.querySelectorAll('.tab-btn');
|
||
tabButtons.forEach(button => {
|
||
if (button.getAttribute('data-tab') === tabId) {
|
||
button.classList.add('active');
|
||
} else {
|
||
button.classList.remove('active');
|
||
}
|
||
});
|
||
|
||
// 更新内容区域
|
||
const sections = document.querySelectorAll('.config-section');
|
||
sections.forEach(section => {
|
||
if (section.id === `${tabId}-section`) {
|
||
section.classList.add('active');
|
||
} else {
|
||
section.classList.remove('active');
|
||
}
|
||
});
|
||
}
|
||
|
||
// 切换上传提供商配置
|
||
function toggleProviderConfig(provider) {
|
||
const providerConfigs = document.querySelectorAll('.provider-config');
|
||
providerConfigs.forEach(config => {
|
||
if (config.getAttribute('data-provider') === provider) {
|
||
config.classList.add('active');
|
||
} else {
|
||
config.classList.remove('active');
|
||
}
|
||
});
|
||
}
|
||
|
||
|
||
// 添加数组项
|
||
function addArrayItem(key) {
|
||
const container = document.getElementById(`${key}_container`);
|
||
if (!container) return;
|
||
|
||
addArrayItemWithValue(key, '');
|
||
}
|
||
|
||
// 添加带值的数组项
|
||
function addArrayItemWithValue(key, value) {
|
||
const container = document.getElementById(`${key}_container`);
|
||
if (!container) return;
|
||
|
||
const arrayItem = document.createElement('div');
|
||
arrayItem.className = 'array-item';
|
||
|
||
const input = document.createElement('input');
|
||
input.type = 'text';
|
||
input.name = `${key}[]`;
|
||
input.value = value;
|
||
input.className = 'array-input';
|
||
|
||
const removeBtn = document.createElement('button');
|
||
removeBtn.type = 'button';
|
||
removeBtn.className = 'remove-btn';
|
||
removeBtn.innerHTML = '<i class="fas fa-times"></i>';
|
||
removeBtn.addEventListener('click', function() {
|
||
arrayItem.remove();
|
||
});
|
||
|
||
arrayItem.appendChild(input);
|
||
arrayItem.appendChild(removeBtn);
|
||
|
||
// 插入到添加按钮之前
|
||
const controls = container.querySelector('.array-controls');
|
||
container.insertBefore(arrayItem, controls);
|
||
}
|
||
|
||
// 收集表单数据
|
||
function collectFormData() {
|
||
const formData = {};
|
||
|
||
// 处理普通输入
|
||
const inputs = document.querySelectorAll('input[type="text"], input[type="number"], select');
|
||
inputs.forEach(input => {
|
||
if (!input.name.includes('[]')) {
|
||
if (input.type === 'number') {
|
||
formData[input.name] = parseFloat(input.value);
|
||
} else {
|
||
formData[input.name] = input.value;
|
||
}
|
||
}
|
||
});
|
||
|
||
// 处理复选框
|
||
const checkboxes = document.querySelectorAll('input[type="checkbox"]');
|
||
checkboxes.forEach(checkbox => {
|
||
formData[checkbox.name] = checkbox.checked;
|
||
});
|
||
|
||
// 处理数组
|
||
const arrayContainers = document.querySelectorAll('.array-container');
|
||
arrayContainers.forEach(container => {
|
||
const key = container.id.replace('_container', '');
|
||
const arrayInputs = container.querySelectorAll('.array-input');
|
||
formData[key] = Array.from(arrayInputs).map(input => input.value).filter(value => value.trim() !== '');
|
||
});
|
||
|
||
return formData;
|
||
}
|
||
|
||
// 保存配置
|
||
async function saveConfig() {
|
||
try {
|
||
const formData = collectFormData();
|
||
|
||
showNotification('正在保存配置...', 'info');
|
||
|
||
const response = await fetch('/api/config', {
|
||
method: 'PUT',
|
||
headers: {
|
||
'Content-Type': 'application/json'
|
||
},
|
||
body: JSON.stringify(formData)
|
||
});
|
||
|
||
if (!response.ok) {
|
||
const errorData = await response.json();
|
||
throw new Error(errorData.detail || `HTTP error! status: ${response.status}`);
|
||
}
|
||
|
||
const result = await response.json();
|
||
|
||
// 显示保存状态
|
||
const saveStatus = document.getElementById('saveStatus');
|
||
saveStatus.classList.add('show');
|
||
|
||
setTimeout(() => {
|
||
saveStatus.classList.remove('show');
|
||
}, 3000);
|
||
|
||
showNotification('配置保存成功', 'success');
|
||
} catch (error) {
|
||
console.error('保存配置失败:', error);
|
||
|
||
// 显示错误状态
|
||
const saveStatus = document.getElementById('saveStatus');
|
||
saveStatus.classList.add('show', 'error');
|
||
saveStatus.querySelector('.status-icon i').className = 'fas fa-times-circle';
|
||
saveStatus.querySelector('.status-text').textContent = '配置保存失败';
|
||
|
||
setTimeout(() => {
|
||
saveStatus.classList.remove('show', 'error');
|
||
}, 3000);
|
||
|
||
showNotification('保存配置失败: ' + error.message, 'error');
|
||
}
|
||
}
|
||
|
||
// 重置配置 (现在只负责打开模态框)
|
||
function resetConfig(event) {
|
||
// 阻止事件冒泡和默认行为
|
||
if (event) {
|
||
event.preventDefault();
|
||
event.stopPropagation();
|
||
}
|
||
|
||
console.log('resetConfig called. Event target:', event ? event.target.id : 'No event');
|
||
|
||
// 确保只有当事件来自重置按钮时才显示模态框
|
||
if (!event || event.target.id === 'resetBtn' || event.currentTarget.id === 'resetBtn') {
|
||
const resetConfirmModal = document.getElementById('resetConfirmModal');
|
||
if (resetConfirmModal) {
|
||
resetConfirmModal.classList.add('show');
|
||
} else {
|
||
// Fallback if modal doesn't exist for some reason
|
||
console.error("Reset confirmation modal not found! Falling back to default confirm.");
|
||
// Fallback to original confirm behavior
|
||
if (!confirm('确定要重置所有配置吗？这将恢复到默认值。')) {
|
||
return;
|
||
}
|
||
// If confirmed, proceed with the reset logic directly (less ideal)
|
||
executeReset();
|
||
}
|
||
}
|
||
}
|
||
|
||
// --- 新增：将实际重置逻辑提取到一个单独的函数 ---
|
||
async function executeReset() {
|
||
try {
|
||
showNotification('正在重置配置...', 'info');
|
||
const response = await fetch('/api/config/reset', { method: 'POST' });
|
||
if (!response.ok) {
|
||
throw new Error(`HTTP error! status: ${response.status}`);
|
||
}
|
||
const config = await response.json();
|
||
populateForm(config);
|
||
showNotification('配置已重置为默认值', 'success');
|
||
} catch (error) {
|
||
console.error('重置配置失败:', error);
|
||
showNotification('重置配置失败: ' + error.message, 'error');
|
||
}
|
||
}
|
||
|
||
// 显示通知
|
||
function showNotification(message, type = 'info') {
|
||
const notification = document.getElementById('notification');
|
||
notification.textContent = message;
|
||
notification.className = 'notification show';
|
||
|
||
if (type) {
|
||
notification.classList.add(type);
|
||
}
|
||
|
||
setTimeout(() => {
|
||
notification.classList.remove('show');
|
||
}, 3000);
|
||
}
|
||
|
||
// 刷新页面
|
||
function refreshPage(button) {
|
||
button.classList.add('loading');
|
||
location.reload();
|
||
}
|
||
|
||
// 滚动到顶部
|
||
function scrollToTop() {
|
||
const container = document.querySelector('.container');
|
||
container.scrollTo({
|
||
top: 0,
|
||
behavior: 'smooth'
|
||
});
|
||
}
|
||
|
||
// 滚动到底部
|
||
function scrollToBottom() {
|
||
const container = document.querySelector('.container');
|
||
container.scrollTo({
|
||
top: container.scrollHeight,
|
||
behavior: 'smooth'
|
||
});
|
||
}
|
||
|
||
// 切换滚动按钮显示
|
||
function toggleScrollButtons() {
|
||
const scrollButtons = document.querySelector('.scroll-buttons');
|
||
|
||
if (window.scrollY > 200) {
|
||
scrollButtons.style.display = 'flex';
|
||
} else {
|
||
scrollButtons.style.display = 'none';
|
||
}
|
||
}
|