import { useEffect, useMemo, useRef, useState, useCallback } from 'react'; import { useTranslation } from 'react-i18next'; import { useInterval } from '@/hooks/useInterval'; import { useHeaderRefresh } from '@/hooks/useHeaderRefresh'; import { Card } from '@/components/ui/Card'; import { Button } from '@/components/ui/Button'; import { LoadingSpinner } from '@/components/ui/LoadingSpinner'; import { Input } from '@/components/ui/Input'; import { AutocompleteInput } from '@/components/ui/AutocompleteInput'; import { Modal } from '@/components/ui/Modal'; import { EmptyState } from '@/components/ui/EmptyState'; import { ToggleSwitch } from '@/components/ui/ToggleSwitch'; import { IconBot, IconCode, IconDownload, IconInfo, IconTrash2, IconX, } from '@/components/ui/icons'; import { useAuthStore, useNotificationStore, useThemeStore } from '@/stores'; import { authFilesApi, usageApi } from '@/services/api'; import { apiClient } from '@/services/api/client'; import type { AuthFileItem, OAuthModelMappingEntry } from '@/types'; import { calculateStatusBarData, collectUsageDetails, normalizeUsageSourceId, type KeyStatBucket, type KeyStats, type UsageDetail, } from '@/utils/usage'; import { formatFileSize } from '@/utils/format'; import { generateId } from '@/utils/helpers'; import styles from './AuthFilesPage.module.scss'; type ThemeColors = { bg: string; text: string; border?: string }; type TypeColorSet = { light: ThemeColors; dark?: ThemeColors }; type ResolvedTheme = 'light' | 'dark'; type AuthFileModelItem = { id: string; display_name?: string; type?: string; owned_by?: string }; // 标签类型颜色配置(对齐重构前 styles.css 的 file-type-badge 颜色) const TYPE_COLORS: Record = { qwen: { light: { bg: '#e8f5e9', text: '#2e7d32' }, dark: { bg: '#1b5e20', text: '#81c784' }, }, gemini: { light: { bg: '#e3f2fd', text: '#1565c0' }, dark: { bg: '#0d47a1', text: '#64b5f6' }, }, 'gemini-cli': { light: { bg: '#e7efff', text: '#1e4fa3' }, dark: { bg: '#1c3f73', text: '#a8c7ff' }, }, aistudio: { light: { bg: '#f0f2f5', text: '#2f343c' }, dark: { bg: '#373c42', text: '#cfd3db' }, }, claude: { light: { bg: '#fce4ec', text: '#c2185b' }, dark: { bg: '#880e4f', text: '#f48fb1' }, }, codex: { light: { bg: '#fff3e0', text: '#ef6c00' }, dark: { bg: '#e65100', text: '#ffb74d' }, }, antigravity: { light: { bg: '#e0f7fa', text: '#006064' }, dark: { bg: '#004d40', text: '#80deea' }, }, iflow: { light: { bg: '#f3e5f5', text: '#7b1fa2' }, dark: { bg: '#4a148c', text: '#ce93d8' }, }, empty: { light: { bg: '#f5f5f5', text: '#616161' }, dark: { bg: '#424242', text: '#bdbdbd' }, }, unknown: { light: { bg: '#f0f0f0', text: '#666666', border: '1px dashed #999999' }, dark: { bg: '#3a3a3a', text: '#aaaaaa', border: '1px dashed #666666' }, }, }; const OAUTH_PROVIDER_PRESETS = [ 'gemini-cli', 'vertex', 'aistudio', 'antigravity', 'claude', 'codex', 'qwen', 'iflow', ]; const OAUTH_PROVIDER_EXCLUDES = new Set(['all', 'unknown', 'empty']); const MIN_CARD_PAGE_SIZE = 3; const MAX_CARD_PAGE_SIZE = 30; const MAX_AUTH_FILE_SIZE = 50 * 1024; const clampCardPageSize = (value: number) => Math.min(MAX_CARD_PAGE_SIZE, Math.max(MIN_CARD_PAGE_SIZE, Math.round(value))); interface ExcludedFormState { provider: string; modelsText: string; } type OAuthModelMappingFormEntry = OAuthModelMappingEntry & { id: string }; interface ModelMappingsFormState { provider: string; mappings: OAuthModelMappingFormEntry[]; } interface PrefixProxyEditorState { fileName: string; loading: boolean; saving: boolean; error: string | null; originalText: string; rawText: string; json: Record | null; prefix: string; proxyUrl: string; } const buildEmptyMappingEntry = (): OAuthModelMappingFormEntry => ({ id: generateId(), name: '', alias: '', fork: false, }); // 标准化 auth_index 值(与 usage.ts 中的 normalizeAuthIndex 保持一致) function normalizeAuthIndexValue(value: unknown): string | null { if (typeof value === 'number' && Number.isFinite(value)) { return value.toString(); } if (typeof value === 'string') { const trimmed = value.trim(); return trimmed ? trimmed : null; } return null; } function isRuntimeOnlyAuthFile(file: AuthFileItem): boolean { const raw = file['runtime_only'] ?? file.runtimeOnly; if (typeof raw === 'boolean') return raw; if (typeof raw === 'string') return raw.trim().toLowerCase() === 'true'; return false; } // 解析认证文件的统计数据 function resolveAuthFileStats(file: AuthFileItem, stats: KeyStats): KeyStatBucket { const defaultStats: KeyStatBucket = { success: 0, failure: 0 }; const rawFileName = file?.name || ''; // 兼容 auth_index 和 authIndex 两种字段名(API 返回的是 auth_index) const rawAuthIndex = file['auth_index'] ?? file.authIndex; const authIndexKey = normalizeAuthIndexValue(rawAuthIndex); // 尝试根据 authIndex 匹配 if (authIndexKey && stats.byAuthIndex?.[authIndexKey]) { return stats.byAuthIndex[authIndexKey]; } // 尝试根据 source (文件名) 匹配 const fileNameId = rawFileName ? normalizeUsageSourceId(rawFileName) : ''; if (fileNameId && stats.bySource?.[fileNameId]) { const fromName = stats.bySource[fileNameId]; if (fromName.success > 0 || fromName.failure > 0) { return fromName; } } // 尝试去掉扩展名后匹配 if (rawFileName) { const nameWithoutExt = rawFileName.replace(/\.[^/.]+$/, ''); if (nameWithoutExt && nameWithoutExt !== rawFileName) { const nameWithoutExtId = normalizeUsageSourceId(nameWithoutExt); const fromNameWithoutExt = nameWithoutExtId ? stats.bySource?.[nameWithoutExtId] : undefined; if ( fromNameWithoutExt && (fromNameWithoutExt.success > 0 || fromNameWithoutExt.failure > 0) ) { return fromNameWithoutExt; } } } return defaultStats; } export function AuthFilesPage() { const { t } = useTranslation(); const { showNotification, showConfirmation } = useNotificationStore(); const connectionStatus = useAuthStore((state) => state.connectionStatus); const resolvedTheme: ResolvedTheme = useThemeStore((state) => state.resolvedTheme); const [files, setFiles] = useState([]); const [loading, setLoading] = useState(true); const [error, setError] = useState(''); const [filter, setFilter] = useState<'all' | string>('all'); const [search, setSearch] = useState(''); const [page, setPage] = useState(1); const [pageSize, setPageSize] = useState(9); const [pageSizeInput, setPageSizeInput] = useState('9'); const [uploading, setUploading] = useState(false); const [deleting, setDeleting] = useState(null); const [deletingAll, setDeletingAll] = useState(false); const [statusUpdating, setStatusUpdating] = useState>({}); const [keyStats, setKeyStats] = useState({ bySource: {}, byAuthIndex: {} }); const [usageDetails, setUsageDetails] = useState([]); // 详情弹窗相关 const [detailModalOpen, setDetailModalOpen] = useState(false); const [selectedFile, setSelectedFile] = useState(null); // 模型列表弹窗相关 const [modelsModalOpen, setModelsModalOpen] = useState(false); const [modelsLoading, setModelsLoading] = useState(false); const [modelsList, setModelsList] = useState([]); const [modelsFileName, setModelsFileName] = useState(''); const [modelsFileType, setModelsFileType] = useState(''); const [modelsError, setModelsError] = useState<'unsupported' | null>(null); const modelsCacheRef = useRef>(new Map()); // OAuth 排除模型相关 const [excluded, setExcluded] = useState>({}); const [excludedError, setExcludedError] = useState<'unsupported' | null>(null); const [excludedModalOpen, setExcludedModalOpen] = useState(false); const [excludedForm, setExcludedForm] = useState({ provider: '', modelsText: '', }); const [savingExcluded, setSavingExcluded] = useState(false); // OAuth 模型映射相关 const [modelMappings, setModelMappings] = useState>({}); const [modelMappingsError, setModelMappingsError] = useState<'unsupported' | null>(null); const [mappingModalOpen, setMappingModalOpen] = useState(false); const [mappingForm, setMappingForm] = useState({ provider: '', mappings: [buildEmptyMappingEntry()], }); const [mappingModelsFileName, setMappingModelsFileName] = useState(''); const [mappingModelsList, setMappingModelsList] = useState([]); const [mappingModelsLoading, setMappingModelsLoading] = useState(false); const [mappingModelsError, setMappingModelsError] = useState<'unsupported' | null>(null); const [savingMappings, setSavingMappings] = useState(false); const [prefixProxyEditor, setPrefixProxyEditor] = useState(null); const fileInputRef = useRef(null); const loadingKeyStatsRef = useRef(false); const excludedUnsupportedRef = useRef(false); const mappingsUnsupportedRef = useRef(false); const normalizeProviderKey = (value: string) => value.trim().toLowerCase(); const disableControls = connectionStatus !== 'connected'; useEffect(() => { setPageSizeInput(String(pageSize)); }, [pageSize]); const modelSourceFileOptions = useMemo(() => { const normalizedProvider = normalizeProviderKey(mappingForm.provider); const matching: string[] = []; const others: string[] = []; const seen = new Set(); files.forEach((file) => { const isRuntimeOnly = isRuntimeOnlyAuthFile(file); const isAistudio = (file.type || '').toLowerCase() === 'aistudio'; const canShowModels = !isRuntimeOnly || isAistudio; if (!canShowModels) return; const fileName = String(file.name || '').trim(); if (!fileName) return; if (seen.has(fileName)) return; seen.add(fileName); if (!normalizedProvider) { matching.push(fileName); return; } const typeKey = normalizeProviderKey(String(file.type || '')); const providerKey = normalizeProviderKey(String(file.provider || '')); const isMatch = typeKey === normalizedProvider || providerKey === normalizedProvider; if (isMatch) { matching.push(fileName); } else { others.push(fileName); } }); matching.sort((a, b) => a.localeCompare(b)); others.sort((a, b) => a.localeCompare(b)); return [...matching, ...others]; }, [files, mappingForm.provider]); useEffect(() => { if (!mappingModalOpen) return; const fileName = mappingModelsFileName.trim(); if (!fileName) { setMappingModelsList([]); setMappingModelsError(null); setMappingModelsLoading(false); return; } const cached = modelsCacheRef.current.get(fileName); if (cached) { setMappingModelsList(cached); setMappingModelsError(null); setMappingModelsLoading(false); return; } let cancelled = false; setMappingModelsLoading(true); setMappingModelsError(null); authFilesApi .getModelsForAuthFile(fileName) .then((models) => { if (cancelled) return; modelsCacheRef.current.set(fileName, models); setMappingModelsList(models); }) .catch((err: unknown) => { if (cancelled) return; const errorMessage = err instanceof Error ? err.message : ''; if ( errorMessage.includes('404') || errorMessage.includes('not found') || errorMessage.includes('Not Found') ) { setMappingModelsList([]); setMappingModelsError('unsupported'); return; } showNotification(`${t('notification.load_failed')}: ${errorMessage}`, 'error'); }) .finally(() => { if (cancelled) return; setMappingModelsLoading(false); }); return () => { cancelled = true; }; }, [mappingModalOpen, mappingModelsFileName, showNotification, t]); const prefixProxyUpdatedText = useMemo(() => { if (!prefixProxyEditor?.json) return prefixProxyEditor?.rawText ?? ''; const next: Record = { ...prefixProxyEditor.json }; if ('prefix' in next || prefixProxyEditor.prefix.trim()) { next.prefix = prefixProxyEditor.prefix; } if ('proxy_url' in next || prefixProxyEditor.proxyUrl.trim()) { next.proxy_url = prefixProxyEditor.proxyUrl; } return JSON.stringify(next); }, [ prefixProxyEditor?.json, prefixProxyEditor?.prefix, prefixProxyEditor?.proxyUrl, prefixProxyEditor?.rawText, ]); const prefixProxyDirty = useMemo(() => { if (!prefixProxyEditor?.json) return false; if (!prefixProxyEditor.originalText) return false; return prefixProxyUpdatedText !== prefixProxyEditor.originalText; }, [prefixProxyEditor?.json, prefixProxyEditor?.originalText, prefixProxyUpdatedText]); const commitPageSizeInput = (rawValue: string) => { const trimmed = rawValue.trim(); if (!trimmed) { setPageSizeInput(String(pageSize)); return; } const value = Number(trimmed); if (!Number.isFinite(value)) { setPageSizeInput(String(pageSize)); return; } const next = clampCardPageSize(value); setPageSize(next); setPageSizeInput(String(next)); setPage(1); }; const handlePageSizeChange = (event: React.ChangeEvent) => { const rawValue = event.currentTarget.value; setPageSizeInput(rawValue); const trimmed = rawValue.trim(); if (!trimmed) return; const parsed = Number(trimmed); if (!Number.isFinite(parsed)) return; const rounded = Math.round(parsed); if (rounded < MIN_CARD_PAGE_SIZE || rounded > MAX_CARD_PAGE_SIZE) return; setPageSize(rounded); setPage(1); }; // 格式化修改时间 const formatModified = (item: AuthFileItem): string => { const raw = item['modtime'] ?? item.modified; if (!raw) return '-'; const asNumber = Number(raw); const date = Number.isFinite(asNumber) && !Number.isNaN(asNumber) ? new Date(asNumber < 1e12 ? asNumber * 1000 : asNumber) : new Date(String(raw)); return Number.isNaN(date.getTime()) ? '-' : date.toLocaleString(); }; // 加载文件列表 const loadFiles = useCallback(async () => { setLoading(true); setError(''); try { const data = await authFilesApi.list(); setFiles(data?.files || []); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : t('notification.refresh_failed'); setError(errorMessage); } finally { setLoading(false); } }, [t]); // 加载 key 统计和 usage 明细(API 层已有60秒超时) const loadKeyStats = useCallback(async () => { // 防止重复请求 if (loadingKeyStatsRef.current) return; loadingKeyStatsRef.current = true; try { const usageResponse = await usageApi.getUsage(); const usageData = usageResponse?.usage ?? usageResponse; const stats = await usageApi.getKeyStats(usageData); setKeyStats(stats); // 收集 usage 明细用于状态栏 const details = collectUsageDetails(usageData); setUsageDetails(details); } catch { // 静默失败 } finally { loadingKeyStatsRef.current = false; } }, []); // 加载 OAuth 排除列表 const loadExcluded = useCallback(async () => { try { const res = await authFilesApi.getOauthExcludedModels(); excludedUnsupportedRef.current = false; setExcluded(res || {}); setExcludedError(null); } catch (err: unknown) { const status = typeof err === 'object' && err !== null && 'status' in err ? (err as { status?: unknown }).status : undefined; if (status === 404) { setExcluded({}); setExcludedError('unsupported'); if (!excludedUnsupportedRef.current) { excludedUnsupportedRef.current = true; showNotification(t('oauth_excluded.upgrade_required'), 'warning'); } return; } // 静默失败 } }, [showNotification, t]); // 加载 OAuth 模型映射 const loadModelMappings = useCallback(async () => { try { const res = await authFilesApi.getOauthModelMappings(); mappingsUnsupportedRef.current = false; setModelMappings(res || {}); setModelMappingsError(null); } catch (err: unknown) { const status = typeof err === 'object' && err !== null && 'status' in err ? (err as { status?: unknown }).status : undefined; if (status === 404) { setModelMappings({}); setModelMappingsError('unsupported'); if (!mappingsUnsupportedRef.current) { mappingsUnsupportedRef.current = true; showNotification(t('oauth_model_mappings.upgrade_required'), 'warning'); } return; } // 静默失败 } }, [showNotification, t]); const handleHeaderRefresh = useCallback(async () => { await Promise.all([loadFiles(), loadKeyStats(), loadExcluded(), loadModelMappings()]); }, [loadFiles, loadKeyStats, loadExcluded, loadModelMappings]); useHeaderRefresh(handleHeaderRefresh); useEffect(() => { loadFiles(); loadKeyStats(); loadExcluded(); loadModelMappings(); }, [loadFiles, loadKeyStats, loadExcluded, loadModelMappings]); // 定时刷新状态数据(每240秒) useInterval(loadKeyStats, 240_000); // 提取所有存在的类型 const existingTypes = useMemo(() => { const types = new Set(['all']); files.forEach((file) => { if (file.type) { types.add(file.type); } }); return Array.from(types); }, [files]); const excludedProviderLookup = useMemo(() => { const lookup = new Map(); Object.keys(excluded).forEach((provider) => { const key = provider.trim().toLowerCase(); if (key && !lookup.has(key)) { lookup.set(key, provider); } }); return lookup; }, [excluded]); const mappingProviderLookup = useMemo(() => { const lookup = new Map(); Object.keys(modelMappings).forEach((provider) => { const key = provider.trim().toLowerCase(); if (key && !lookup.has(key)) { lookup.set(key, provider); } }); return lookup; }, [modelMappings]); const providerOptions = useMemo(() => { const extraProviders = new Set(); Object.keys(excluded).forEach((provider) => { extraProviders.add(provider); }); Object.keys(modelMappings).forEach((provider) => { extraProviders.add(provider); }); files.forEach((file) => { if (typeof file.type === 'string') { extraProviders.add(file.type); } if (typeof file.provider === 'string') { extraProviders.add(file.provider); } }); const normalizedExtras = Array.from(extraProviders) .map((value) => value.trim()) .filter((value) => value && !OAUTH_PROVIDER_EXCLUDES.has(value.toLowerCase())); const baseSet = new Set(OAUTH_PROVIDER_PRESETS.map((value) => value.toLowerCase())); const extraList = normalizedExtras .filter((value) => !baseSet.has(value.toLowerCase())) .sort((a, b) => a.localeCompare(b)); return [...OAUTH_PROVIDER_PRESETS, ...extraList]; }, [excluded, files, modelMappings]); // 过滤和搜索 const filtered = useMemo(() => { return files.filter((item) => { const matchType = filter === 'all' || item.type === filter; const term = search.trim().toLowerCase(); const matchSearch = !term || item.name.toLowerCase().includes(term) || (item.type || '').toString().toLowerCase().includes(term) || (item.provider || '').toString().toLowerCase().includes(term); return matchType && matchSearch; }); }, [files, filter, search]); // 分页计算 const totalPages = Math.max(1, Math.ceil(filtered.length / pageSize)); const currentPage = Math.min(page, totalPages); const start = (currentPage - 1) * pageSize; const pageItems = filtered.slice(start, start + pageSize); // 点击上传 const handleUploadClick = () => { fileInputRef.current?.click(); }; // 处理文件上传(支持多选) const handleFileChange = async (event: React.ChangeEvent) => { const fileList = event.target.files; if (!fileList || fileList.length === 0) return; const filesToUpload = Array.from(fileList); const validFiles: File[] = []; const invalidFiles: string[] = []; const oversizedFiles: string[] = []; filesToUpload.forEach((file) => { if (!file.name.endsWith('.json')) { invalidFiles.push(file.name); return; } if (file.size > MAX_AUTH_FILE_SIZE) { oversizedFiles.push(file.name); return; } validFiles.push(file); }); if (invalidFiles.length > 0) { showNotification(t('auth_files.upload_error_json'), 'error'); } if (oversizedFiles.length > 0) { showNotification( t('auth_files.upload_error_size', { maxSize: formatFileSize(MAX_AUTH_FILE_SIZE) }), 'error' ); } if (validFiles.length === 0) { event.target.value = ''; return; } setUploading(true); let successCount = 0; const failed: { name: string; message: string }[] = []; for (const file of validFiles) { try { await authFilesApi.upload(file); successCount++; } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : 'Unknown error'; failed.push({ name: file.name, message: errorMessage }); } } if (successCount > 0) { const suffix = validFiles.length > 1 ? ` (${successCount}/${validFiles.length})` : ''; showNotification( `${t('auth_files.upload_success')}${suffix}`, failed.length ? 'warning' : 'success' ); await loadFiles(); await loadKeyStats(); } if (failed.length > 0) { const details = failed.map((item) => `${item.name}: ${item.message}`).join('; '); showNotification(`${t('notification.upload_failed')}: ${details}`, 'error'); } setUploading(false); event.target.value = ''; }; // 删除单个文件 const handleDelete = async (name: string) => { showConfirmation({ title: t('auth_files.delete_title', { defaultValue: 'Delete File' }), message: `${t('auth_files.delete_confirm')} "${name}" ?`, variant: 'danger', confirmText: t('common.confirm'), onConfirm: async () => { setDeleting(name); try { await authFilesApi.deleteFile(name); showNotification(t('auth_files.delete_success'), 'success'); setFiles((prev) => prev.filter((item) => item.name !== name)); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('notification.delete_failed')}: ${errorMessage}`, 'error'); } finally { setDeleting(null); } }, }); }; // 删除全部(根据筛选类型) const handleDeleteAll = async () => { const isFiltered = filter !== 'all'; const typeLabel = isFiltered ? getTypeLabel(filter) : t('auth_files.filter_all'); const confirmMessage = isFiltered ? t('auth_files.delete_filtered_confirm', { type: typeLabel }) : t('auth_files.delete_all_confirm'); showConfirmation({ title: t('auth_files.delete_all_title', { defaultValue: 'Delete All Files' }), message: confirmMessage, variant: 'danger', confirmText: t('common.confirm'), onConfirm: async () => { setDeletingAll(true); try { if (!isFiltered) { // 删除全部 await authFilesApi.deleteAll(); showNotification(t('auth_files.delete_all_success'), 'success'); setFiles((prev) => prev.filter((file) => isRuntimeOnlyAuthFile(file))); } else { // 删除筛选类型的文件 const filesToDelete = files.filter((f) => f.type === filter && !isRuntimeOnlyAuthFile(f)); if (filesToDelete.length === 0) { showNotification(t('auth_files.delete_filtered_none', { type: typeLabel }), 'info'); setDeletingAll(false); return; } let success = 0; let failed = 0; const deletedNames: string[] = []; for (const file of filesToDelete) { try { await authFilesApi.deleteFile(file.name); success++; deletedNames.push(file.name); } catch { failed++; } } setFiles((prev) => prev.filter((f) => !deletedNames.includes(f.name))); if (failed === 0) { showNotification( t('auth_files.delete_filtered_success', { count: success, type: typeLabel }), 'success' ); } else { showNotification( t('auth_files.delete_filtered_partial', { success, failed, type: typeLabel }), 'warning' ); } setFilter('all'); } } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('notification.delete_failed')}: ${errorMessage}`, 'error'); } finally { setDeletingAll(false); } }, }); }; // 下载文件 const handleDownload = async (name: string) => { try { const response = await apiClient.getRaw( `/auth-files/download?name=${encodeURIComponent(name)}`, { responseType: 'blob', } ); const blob = new Blob([response.data]); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); a.href = url; a.download = name; a.click(); window.URL.revokeObjectURL(url); showNotification(t('auth_files.download_success'), 'success'); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('notification.download_failed')}: ${errorMessage}`, 'error'); } }; const openPrefixProxyEditor = async (name: string) => { if (disableControls) return; if (prefixProxyEditor?.fileName === name) { setPrefixProxyEditor(null); return; } setPrefixProxyEditor({ fileName: name, loading: true, saving: false, error: null, originalText: '', rawText: '', json: null, prefix: '', proxyUrl: '', }); try { const rawText = await authFilesApi.downloadText(name); const trimmed = rawText.trim(); let parsed: unknown; try { parsed = JSON.parse(trimmed) as unknown; } catch { setPrefixProxyEditor((prev) => { if (!prev || prev.fileName !== name) return prev; return { ...prev, loading: false, error: t('auth_files.prefix_proxy_invalid_json'), rawText: trimmed, originalText: trimmed, }; }); return; } if (!parsed || typeof parsed !== 'object' || Array.isArray(parsed)) { setPrefixProxyEditor((prev) => { if (!prev || prev.fileName !== name) return prev; return { ...prev, loading: false, error: t('auth_files.prefix_proxy_invalid_json'), rawText: trimmed, originalText: trimmed, }; }); return; } const json = parsed as Record; const originalText = JSON.stringify(json); const prefix = typeof json.prefix === 'string' ? json.prefix : ''; const proxyUrl = typeof json.proxy_url === 'string' ? json.proxy_url : ''; setPrefixProxyEditor((prev) => { if (!prev || prev.fileName !== name) return prev; return { ...prev, loading: false, originalText, rawText: originalText, json, prefix, proxyUrl, error: null, }; }); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : t('notification.download_failed'); setPrefixProxyEditor((prev) => { if (!prev || prev.fileName !== name) return prev; return { ...prev, loading: false, error: errorMessage, rawText: '' }; }); showNotification(`${t('notification.download_failed')}: ${errorMessage}`, 'error'); } }; const handlePrefixProxyChange = (field: 'prefix' | 'proxyUrl', value: string) => { setPrefixProxyEditor((prev) => { if (!prev) return prev; if (field === 'prefix') return { ...prev, prefix: value }; return { ...prev, proxyUrl: value }; }); }; const handlePrefixProxySave = async () => { if (!prefixProxyEditor?.json) return; if (!prefixProxyDirty) return; const name = prefixProxyEditor.fileName; const payload = prefixProxyUpdatedText; const fileSize = new Blob([payload]).size; if (fileSize > MAX_AUTH_FILE_SIZE) { showNotification( t('auth_files.upload_error_size', { maxSize: formatFileSize(MAX_AUTH_FILE_SIZE) }), 'error' ); return; } setPrefixProxyEditor((prev) => { if (!prev || prev.fileName !== name) return prev; return { ...prev, saving: true }; }); try { const file = new File([payload], name, { type: 'application/json' }); await authFilesApi.upload(file); showNotification(t('auth_files.prefix_proxy_saved_success', { name }), 'success'); await loadFiles(); await loadKeyStats(); setPrefixProxyEditor(null); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('notification.upload_failed')}: ${errorMessage}`, 'error'); setPrefixProxyEditor((prev) => { if (!prev || prev.fileName !== name) return prev; return { ...prev, saving: false }; }); } }; const handleStatusToggle = async (item: AuthFileItem, enabled: boolean) => { const name = item.name; const nextDisabled = !enabled; const previousDisabled = item.disabled === true; setStatusUpdating((prev) => ({ ...prev, [name]: true })); // Optimistic update for snappy UI. setFiles((prev) => prev.map((f) => (f.name === name ? { ...f, disabled: nextDisabled } : f))); try { const res = await authFilesApi.setStatus(name, nextDisabled); setFiles((prev) => prev.map((f) => (f.name === name ? { ...f, disabled: res.disabled } : f))); showNotification( enabled ? t('auth_files.status_enabled_success', { name }) : t('auth_files.status_disabled_success', { name }), 'success' ); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; setFiles((prev) => prev.map((f) => (f.name === name ? { ...f, disabled: previousDisabled } : f)) ); showNotification(`${t('notification.update_failed')}: ${errorMessage}`, 'error'); } finally { setStatusUpdating((prev) => { if (!prev[name]) return prev; const next = { ...prev }; delete next[name]; return next; }); } }; // 显示详情弹窗 const showDetails = (file: AuthFileItem) => { setSelectedFile(file); setDetailModalOpen(true); }; // 显示模型列表 const showModels = async (item: AuthFileItem) => { setModelsFileName(item.name); setModelsFileType(item.type || ''); setModelsList([]); setModelsError(null); setModelsModalOpen(true); const cached = modelsCacheRef.current.get(item.name); if (cached) { setModelsList(cached); setModelsLoading(false); return; } setModelsLoading(true); try { const models = await authFilesApi.getModelsForAuthFile(item.name); modelsCacheRef.current.set(item.name, models); setModelsList(models); } catch (err) { // 检测是否是 API 不支持的错误 (404 或特定错误消息) const errorMessage = err instanceof Error ? err.message : ''; if ( errorMessage.includes('404') || errorMessage.includes('not found') || errorMessage.includes('Not Found') ) { setModelsError('unsupported'); } else { showNotification(`${t('notification.load_failed')}: ${errorMessage}`, 'error'); } } finally { setModelsLoading(false); } }; // 检查模型是否被 OAuth 排除 const isModelExcluded = (modelId: string, providerType: string): boolean => { const providerKey = normalizeProviderKey(providerType); const excludedModels = excluded[providerKey] || excluded[providerType] || []; return excludedModels.some((pattern) => { if (pattern.includes('*')) { // 支持通配符匹配 const regex = new RegExp('^' + pattern.replace(/\*/g, '.*') + '$', 'i'); return regex.test(modelId); } return pattern.toLowerCase() === modelId.toLowerCase(); }); }; // 获取类型标签显示文本 const getTypeLabel = (type: string): string => { const key = `auth_files.filter_${type}`; const translated = t(key); if (translated !== key) return translated; if (type.toLowerCase() === 'iflow') return 'iFlow'; return type.charAt(0).toUpperCase() + type.slice(1); }; // 获取类型颜色 const getTypeColor = (type: string): ThemeColors => { const set = TYPE_COLORS[type] || TYPE_COLORS.unknown; return resolvedTheme === 'dark' && set.dark ? set.dark : set.light; }; // OAuth 排除相关方法 const openExcludedModal = (provider?: string) => { const normalizedProvider = normalizeProviderKey(provider || ''); const fallbackProvider = normalizedProvider || (filter !== 'all' ? normalizeProviderKey(String(filter)) : ''); const lookupKey = fallbackProvider ? excludedProviderLookup.get(fallbackProvider) : undefined; const models = lookupKey ? excluded[lookupKey] : []; setExcludedForm({ provider: lookupKey || fallbackProvider, modelsText: Array.isArray(models) ? models.join('\n') : '', }); setExcludedModalOpen(true); }; const saveExcludedModels = async () => { const provider = normalizeProviderKey(excludedForm.provider); if (!provider) { showNotification(t('oauth_excluded.provider_required'), 'error'); return; } const models = excludedForm.modelsText .split(/[\n,]+/) .map((item) => item.trim()) .filter(Boolean); setSavingExcluded(true); try { if (models.length) { await authFilesApi.saveOauthExcludedModels(provider, models); } else { await authFilesApi.deleteOauthExcludedEntry(provider); } await loadExcluded(); showNotification(t('oauth_excluded.save_success'), 'success'); setExcludedModalOpen(false); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('oauth_excluded.save_failed')}: ${errorMessage}`, 'error'); } finally { setSavingExcluded(false); } }; const deleteExcluded = async (provider: string) => { const providerLabel = provider.trim() || provider; showConfirmation({ title: t('oauth_excluded.delete_title', { defaultValue: 'Delete Exclusion' }), message: t('oauth_excluded.delete_confirm', { provider: providerLabel }), variant: 'danger', confirmText: t('common.confirm'), onConfirm: async () => { const providerKey = normalizeProviderKey(provider); if (!providerKey) { showNotification(t('oauth_excluded.provider_required'), 'error'); return; } try { await authFilesApi.deleteOauthExcludedEntry(providerKey); await loadExcluded(); showNotification(t('oauth_excluded.delete_success'), 'success'); } catch (err: unknown) { try { const current = await authFilesApi.getOauthExcludedModels(); const next: Record = {}; Object.entries(current).forEach(([key, models]) => { if (normalizeProviderKey(key) === providerKey) return; next[key] = models; }); await authFilesApi.replaceOauthExcludedModels(next); await loadExcluded(); showNotification(t('oauth_excluded.delete_success'), 'success'); } catch (fallbackErr: unknown) { const errorMessage = fallbackErr instanceof Error ? fallbackErr.message : err instanceof Error ? err.message : ''; showNotification(`${t('oauth_excluded.delete_failed')}: ${errorMessage}`, 'error'); } } }, }); }; // OAuth 模型映射相关方法 const normalizeMappingEntries = ( entries?: OAuthModelMappingEntry[] ): OAuthModelMappingFormEntry[] => { if (!Array.isArray(entries) || entries.length === 0) { return [buildEmptyMappingEntry()]; } return entries.map((entry) => ({ id: generateId(), name: entry.name ?? '', alias: entry.alias ?? '', fork: Boolean(entry.fork), })); }; const openMappingsModal = (provider?: string) => { const normalizedProvider = (provider || '').trim(); const fallbackProvider = normalizedProvider || (filter !== 'all' ? String(filter) : ''); const lookupKey = fallbackProvider ? mappingProviderLookup.get(fallbackProvider.toLowerCase()) : undefined; const mappings = lookupKey ? modelMappings[lookupKey] : []; const providerValue = lookupKey || fallbackProvider; const normalizedProviderKey = normalizeProviderKey(providerValue); const defaultModelsFileName = files .filter((file) => { const isRuntimeOnly = isRuntimeOnlyAuthFile(file); const isAistudio = (file.type || '').toLowerCase() === 'aistudio'; const canShowModels = !isRuntimeOnly || isAistudio; if (!canShowModels) return false; if (!normalizedProviderKey) return false; const typeKey = normalizeProviderKey(String(file.type || '')); const providerKey = normalizeProviderKey(String(file.provider || '')); return typeKey === normalizedProviderKey || providerKey === normalizedProviderKey; }) .map((file) => file.name) .sort((a, b) => a.localeCompare(b))[0]; setMappingForm({ provider: providerValue, mappings: normalizeMappingEntries(mappings), }); setMappingModelsFileName(defaultModelsFileName || ''); setMappingModelsList([]); setMappingModelsError(null); setMappingModalOpen(true); }; const updateMappingEntry = ( index: number, field: keyof OAuthModelMappingEntry, value: string | boolean ) => { setMappingForm((prev) => ({ ...prev, mappings: prev.mappings.map((entry, idx) => idx === index ? { ...entry, [field]: value } : entry ), })); }; const addMappingEntry = () => { setMappingForm((prev) => ({ ...prev, mappings: [...prev.mappings, buildEmptyMappingEntry()], })); }; const removeMappingEntry = (index: number) => { setMappingForm((prev) => { const next = prev.mappings.filter((_, idx) => idx !== index); return { ...prev, mappings: next.length ? next : [buildEmptyMappingEntry()], }; }); }; const saveModelMappings = async () => { const provider = mappingForm.provider.trim(); if (!provider) { showNotification(t('oauth_model_mappings.provider_required'), 'error'); return; } const seen = new Set(); const mappings = mappingForm.mappings .map((entry) => { const name = String(entry.name ?? '').trim(); const alias = String(entry.alias ?? '').trim(); if (!name || !alias) return null; const key = `${name.toLowerCase()}::${alias.toLowerCase()}::${entry.fork ? '1' : '0'}`; if (seen.has(key)) return null; seen.add(key); return entry.fork ? { name, alias, fork: true } : { name, alias }; }) .filter(Boolean) as OAuthModelMappingEntry[]; setSavingMappings(true); try { if (mappings.length) { await authFilesApi.saveOauthModelMappings(provider, mappings); } else { await authFilesApi.deleteOauthModelMappings(provider); } await loadModelMappings(); showNotification(t('oauth_model_mappings.save_success'), 'success'); setMappingModalOpen(false); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('oauth_model_mappings.save_failed')}: ${errorMessage}`, 'error'); } finally { setSavingMappings(false); } }; const deleteModelMappings = async (provider: string) => { showConfirmation({ title: t('oauth_model_mappings.delete_title', { defaultValue: 'Delete Mappings' }), message: t('oauth_model_mappings.delete_confirm', { provider }), variant: 'danger', confirmText: t('common.confirm'), onConfirm: async () => { try { await authFilesApi.deleteOauthModelMappings(provider); await loadModelMappings(); showNotification(t('oauth_model_mappings.delete_success'), 'success'); } catch (err: unknown) { const errorMessage = err instanceof Error ? err.message : ''; showNotification(`${t('oauth_model_mappings.delete_failed')}: ${errorMessage}`, 'error'); } }, }); }; // 渲染标签筛选器 const renderFilterTags = () => (
{existingTypes.map((type) => { const isActive = filter === type; const color = type === 'all' ? { bg: 'var(--bg-tertiary)', text: 'var(--text-primary)' } : getTypeColor(type); const activeTextColor = resolvedTheme === 'dark' ? '#111827' : '#fff'; return ( ); })}
); // 预计算所有认证文件的状态栏数据(避免每次渲染重复计算) const statusBarCache = useMemo(() => { const cache = new Map>(); files.forEach((file) => { const rawAuthIndex = file['auth_index'] ?? file.authIndex; const authIndexKey = normalizeAuthIndexValue(rawAuthIndex); if (authIndexKey) { // 过滤出属于该认证文件的 usage 明细 const filteredDetails = usageDetails.filter((detail) => { const detailAuthIndex = normalizeAuthIndexValue(detail.auth_index); return detailAuthIndex !== null && detailAuthIndex === authIndexKey; }); cache.set(authIndexKey, calculateStatusBarData(filteredDetails)); } }); return cache; }, [usageDetails, files]); // 渲染状态监测栏 const renderStatusBar = (item: AuthFileItem) => { // 认证文件使用 authIndex 来匹配 usage 数据 const rawAuthIndex = item['auth_index'] ?? item.authIndex; const authIndexKey = normalizeAuthIndexValue(rawAuthIndex); const statusData = (authIndexKey && statusBarCache.get(authIndexKey)) || calculateStatusBarData([]); const hasData = statusData.totalSuccess + statusData.totalFailure > 0; const rateClass = !hasData ? '' : statusData.successRate >= 90 ? styles.statusRateHigh : statusData.successRate >= 50 ? styles.statusRateMedium : styles.statusRateLow; return (
{statusData.blocks.map((state, idx) => { const blockClass = state === 'success' ? styles.statusBlockSuccess : state === 'failure' ? styles.statusBlockFailure : state === 'mixed' ? styles.statusBlockMixed : styles.statusBlockIdle; return
; })}
{hasData ? `${statusData.successRate.toFixed(1)}%` : '--'}
); }; // 渲染单个认证文件卡片 const renderFileCard = (item: AuthFileItem) => { const fileStats = resolveAuthFileStats(item, keyStats); const isRuntimeOnly = isRuntimeOnlyAuthFile(item); const isAistudio = (item.type || '').toLowerCase() === 'aistudio'; const showModelsButton = !isRuntimeOnly || isAistudio; const typeColor = getTypeColor(item.type || 'unknown'); return (
{getTypeLabel(item.type || 'unknown')} {item.name}
{t('auth_files.file_size')}: {item.size ? formatFileSize(item.size) : '-'} {t('auth_files.file_modified')}: {formatModified(item)}
{t('stats.success')}: {fileStats.success} {t('stats.failure')}: {fileStats.failure}
{/* 状态监测栏 */} {renderStatusBar(item)}
{showModelsButton && ( )} {!isRuntimeOnly && ( <> )} {!isRuntimeOnly && (
void handleStatusToggle(item, value)} />
)} {isRuntimeOnly && (
{t('auth_files.type_virtual') || '虚拟认证文件'}
)}
); }; const titleNode = (
{t('auth_files.title_section')} {files.length > 0 && {files.length}}
); return (

{t('auth_files.title')}

{t('auth_files.description')}

} > {error &&
{error}
} {/* 筛选区域 */}
{renderFilterTags()}
{ setSearch(e.target.value); setPage(1); }} placeholder={t('auth_files.search_placeholder')} />
commitPageSizeInput(e.currentTarget.value)} onKeyDown={(e) => { if (e.key === 'Enter') { e.currentTarget.blur(); } }} />
{/* 卡片网格 */} {loading ? (
{t('common.loading')}
) : pageItems.length === 0 ? ( ) : (
{pageItems.map(renderFileCard)}
)} {/* 分页 */} {!loading && filtered.length > pageSize && (
{t('auth_files.pagination_info', { current: currentPage, total: totalPages, count: filtered.length, })}
)} {/* OAuth 排除列表卡片 */} openExcludedModal()} disabled={disableControls || excludedError === 'unsupported'} > {t('oauth_excluded.add')} } > {excludedError === 'unsupported' ? ( ) : Object.keys(excluded).length === 0 ? ( ) : (
{Object.entries(excluded).map(([provider, models]) => (
{provider}
{models?.length ? t('oauth_excluded.model_count', { count: models.length }) : t('oauth_excluded.no_models')}
))}
)}
{/* OAuth 模型映射卡片 */} openMappingsModal()} disabled={disableControls || modelMappingsError === 'unsupported'} > {t('oauth_model_mappings.add')} } > {modelMappingsError === 'unsupported' ? ( ) : Object.keys(modelMappings).length === 0 ? ( ) : (
{Object.entries(modelMappings).map(([provider, mappings]) => (
{provider}
{mappings?.length ? t('oauth_model_mappings.model_count', { count: mappings.length }) : t('oauth_model_mappings.no_models')}
))}
)}
{/* 详情弹窗 */} setDetailModalOpen(false)} title={selectedFile?.name || t('auth_files.title_section')} footer={ <> } > {selectedFile && (
{JSON.stringify(selectedFile, null, 2)}
)}
{/* 模型列表弹窗 */} setModelsModalOpen(false)} title={ t('auth_files.models_title', { defaultValue: '支持的模型' }) + ` - ${modelsFileName}` } footer={ } > {modelsLoading ? (
{t('auth_files.models_loading', { defaultValue: '正在加载模型列表...' })}
) : modelsError === 'unsupported' ? ( ) : modelsList.length === 0 ? ( ) : (
{modelsList.map((model) => { const isExcluded = isModelExcluded(model.id, modelsFileType); return (
{ navigator.clipboard.writeText(model.id); showNotification( t('notification.link_copied', { defaultValue: '已复制到剪贴板' }), 'success' ); }} title={ isExcluded ? t('auth_files.models_excluded_hint', { defaultValue: '此模型已被 OAuth 排除', }) : t('common.copy', { defaultValue: '点击复制' }) } > {model.id} {model.display_name && model.display_name !== model.id && ( {model.display_name} )} {model.type && {model.type}} {isExcluded && ( {t('auth_files.models_excluded_badge', { defaultValue: '已排除' })} )}
); })}
)}
{/* prefix/proxy_url 编辑弹窗 */} setPrefixProxyEditor(null)} closeDisabled={prefixProxyEditor?.saving === true} width={720} title={ prefixProxyEditor?.fileName ? `${t('auth_files.prefix_proxy_button')} - ${prefixProxyEditor.fileName}` : t('auth_files.prefix_proxy_button') } footer={ <> } > {prefixProxyEditor && (
{prefixProxyEditor.loading ? (
{t('auth_files.prefix_proxy_loading')}
) : ( <> {prefixProxyEditor.error && (
{prefixProxyEditor.error}
)}