主要改动: - 后端:重构 Gemini 配额服务,支持多层级配额策略(GCP Standard/Free, Google One, AI Studio, Code Assist) - 后端:优化 OAuth 服务,增强 tier_id 识别和存储逻辑 - 后端:改进用量统计服务,支持不同平台的配额查询 - 后端:优化限流服务,增加临时解除调度状态管理 - 前端:统一四种授权方式的用量显示格式和徽标样式 - 前端:增强账户配额信息展示,支持多种配额类型 - 前端:改进创建和重新授权模态框的用户体验 - 国际化:完善中英文配额相关文案 - 移除 CHANGELOG.md 文件 测试:所有单元测试通过
199 lines
7.3 KiB
Vue
199 lines
7.3 KiB
Vue
<template>
|
||
<div v-if="shouldShowQuota">
|
||
<!-- First line: Platform + Tier Badge -->
|
||
<div class="mb-1 flex items-center gap-1">
|
||
<span :class="['badge text-xs px-2 py-0.5 rounded font-medium', tierBadgeClass]">
|
||
{{ tierLabel }}
|
||
</span>
|
||
</div>
|
||
|
||
<!-- Usage status: unlimited flow or rate limit -->
|
||
<div class="text-xs text-gray-400 dark:text-gray-500">
|
||
<span v-if="!isRateLimited">
|
||
{{ t('admin.accounts.gemini.rateLimit.unlimited') }}
|
||
</span>
|
||
<span
|
||
v-else
|
||
:class="[
|
||
'font-medium',
|
||
isUrgent
|
||
? 'text-red-600 dark:text-red-400 animate-pulse'
|
||
: 'text-amber-600 dark:text-amber-400'
|
||
]"
|
||
>
|
||
{{ t('admin.accounts.gemini.rateLimit.limited', { time: resetCountdown }) }}
|
||
</span>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
|
||
<script setup lang="ts">
|
||
import { computed, ref, watch, onUnmounted } from 'vue'
|
||
import { useI18n } from 'vue-i18n'
|
||
import type { Account, GeminiCredentials } from '@/types'
|
||
|
||
const props = defineProps<{
|
||
account: Account
|
||
}>()
|
||
|
||
const { t } = useI18n()
|
||
|
||
const now = ref(new Date())
|
||
let timer: ReturnType<typeof setInterval> | null = null
|
||
|
||
// 是否为 Code Assist OAuth
|
||
// 判断逻辑与后端保持一致:project_id 存在即为 Code Assist
|
||
const isCodeAssist = computed(() => {
|
||
const creds = props.account.credentials as GeminiCredentials | undefined
|
||
// 显式为 code_assist,或 legacy 情况(oauth_type 为空但 project_id 存在)
|
||
return creds?.oauth_type === 'code_assist' || (!creds?.oauth_type && !!creds?.project_id)
|
||
})
|
||
|
||
// 是否为 Google One OAuth
|
||
const isGoogleOne = computed(() => {
|
||
const creds = props.account.credentials as GeminiCredentials | undefined
|
||
return creds?.oauth_type === 'google_one'
|
||
})
|
||
|
||
// 是否应该显示配额信息
|
||
const shouldShowQuota = computed(() => {
|
||
return props.account.platform === 'gemini'
|
||
})
|
||
|
||
// Tier 标签文本
|
||
const tierLabel = computed(() => {
|
||
const creds = props.account.credentials as GeminiCredentials | undefined
|
||
|
||
if (isCodeAssist.value) {
|
||
const tier = (creds?.tier_id || '').toString().trim().toLowerCase()
|
||
if (tier === 'gcp_enterprise') return 'GCP Enterprise'
|
||
if (tier === 'gcp_standard') return 'GCP Standard'
|
||
// Backward compatibility
|
||
const upper = (creds?.tier_id || '').toString().trim().toUpperCase()
|
||
if (upper.includes('ULTRA') || upper.includes('ENTERPRISE')) return 'GCP Enterprise'
|
||
if (upper) return `GCP ${upper}`
|
||
return 'GCP'
|
||
}
|
||
|
||
if (isGoogleOne.value) {
|
||
const tier = (creds?.tier_id || '').toString().trim().toLowerCase()
|
||
if (tier === 'google_ai_ultra') return 'Google AI Ultra'
|
||
if (tier === 'google_ai_pro') return 'Google AI Pro'
|
||
if (tier === 'google_one_free') return 'Google One Free'
|
||
// Backward compatibility
|
||
const upper = (creds?.tier_id || '').toString().trim().toUpperCase()
|
||
if (upper === 'AI_PREMIUM') return 'Google AI Pro'
|
||
if (upper === 'GOOGLE_ONE_UNLIMITED') return 'Google AI Ultra'
|
||
if (upper) return `Google One ${upper}`
|
||
return 'Google One'
|
||
}
|
||
|
||
// API Key: 显示 AI Studio
|
||
const tier = (creds?.tier_id || '').toString().trim().toLowerCase()
|
||
if (tier === 'aistudio_paid') return 'AI Studio Pay-as-you-go'
|
||
if (tier === 'aistudio_free') return 'AI Studio Free Tier'
|
||
return 'AI Studio'
|
||
})
|
||
|
||
// Tier Badge 样式(统一样式)
|
||
const tierBadgeClass = computed(() => {
|
||
const creds = props.account.credentials as GeminiCredentials | undefined
|
||
|
||
if (isCodeAssist.value) {
|
||
const tier = (creds?.tier_id || '').toString().trim().toLowerCase()
|
||
if (tier === 'gcp_enterprise') return 'bg-purple-100 text-purple-600 dark:bg-purple-900/40 dark:text-purple-300'
|
||
if (tier === 'gcp_standard') return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
|
||
// Backward compatibility
|
||
const upper = (creds?.tier_id || '').toString().trim().toUpperCase()
|
||
if (upper.includes('ULTRA') || upper.includes('ENTERPRISE')) return 'bg-purple-100 text-purple-600 dark:bg-purple-900/40 dark:text-purple-300'
|
||
return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
|
||
}
|
||
|
||
if (isGoogleOne.value) {
|
||
const tier = (creds?.tier_id || '').toString().trim().toLowerCase()
|
||
if (tier === 'google_ai_ultra') return 'bg-purple-100 text-purple-600 dark:bg-purple-900/40 dark:text-purple-300'
|
||
if (tier === 'google_ai_pro') return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
|
||
if (tier === 'google_one_free') return 'bg-gray-100 text-gray-600 dark:bg-gray-700 dark:text-gray-300'
|
||
// Backward compatibility
|
||
const upper = (creds?.tier_id || '').toString().trim().toUpperCase()
|
||
if (upper === 'GOOGLE_ONE_UNLIMITED') return 'bg-purple-100 text-purple-600 dark:bg-purple-900/40 dark:text-purple-300'
|
||
if (upper === 'AI_PREMIUM') return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
|
||
return 'bg-gray-100 text-gray-600 dark:bg-gray-700 dark:text-gray-300'
|
||
}
|
||
|
||
// AI Studio 默认样式:蓝色
|
||
const tier = (creds?.tier_id || '').toString().trim().toLowerCase()
|
||
if (tier === 'aistudio_paid') return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
|
||
if (tier === 'aistudio_free') return 'bg-gray-100 text-gray-600 dark:bg-gray-700 dark:text-gray-300'
|
||
return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
|
||
})
|
||
|
||
// 是否限流
|
||
const isRateLimited = computed(() => {
|
||
if (!props.account.rate_limit_reset_at) return false
|
||
const resetTime = Date.parse(props.account.rate_limit_reset_at)
|
||
// 防护:如果日期解析失败(NaN),则认为未限流
|
||
if (Number.isNaN(resetTime)) return false
|
||
return resetTime > now.value.getTime()
|
||
})
|
||
|
||
// 倒计时文本
|
||
const resetCountdown = computed(() => {
|
||
if (!props.account.rate_limit_reset_at) return ''
|
||
const resetTime = Date.parse(props.account.rate_limit_reset_at)
|
||
// 防护:如果日期解析失败,显示 "-"
|
||
if (Number.isNaN(resetTime)) return '-'
|
||
|
||
const diffMs = resetTime - now.value.getTime()
|
||
if (diffMs <= 0) return t('admin.accounts.gemini.rateLimit.now')
|
||
|
||
const diffSeconds = Math.floor(diffMs / 1000)
|
||
const diffMinutes = Math.floor(diffSeconds / 60)
|
||
const diffHours = Math.floor(diffMinutes / 60)
|
||
|
||
if (diffMinutes < 1) return `${diffSeconds}s`
|
||
if (diffHours < 1) {
|
||
const secs = diffSeconds % 60
|
||
return `${diffMinutes}m ${secs}s`
|
||
}
|
||
const mins = diffMinutes % 60
|
||
return `${diffHours}h ${mins}m`
|
||
})
|
||
|
||
// 是否紧急(< 1分钟)
|
||
const isUrgent = computed(() => {
|
||
if (!props.account.rate_limit_reset_at) return false
|
||
const resetTime = Date.parse(props.account.rate_limit_reset_at)
|
||
// 防护:如果日期解析失败,返回 false
|
||
if (Number.isNaN(resetTime)) return false
|
||
|
||
const diffMs = resetTime - now.value.getTime()
|
||
return diffMs > 0 && diffMs < 60000
|
||
})
|
||
|
||
// 监听限流状态,动态启动/停止定时器
|
||
watch(
|
||
() => isRateLimited.value,
|
||
(limited) => {
|
||
if (limited && !timer) {
|
||
// 进入限流状态,启动定时器
|
||
timer = setInterval(() => {
|
||
now.value = new Date()
|
||
}, 1000)
|
||
} else if (!limited && timer) {
|
||
// 解除限流,停止定时器
|
||
clearInterval(timer)
|
||
timer = null
|
||
}
|
||
},
|
||
{ immediate: true } // 立即执行,确保挂载时已限流的情况也能启动定时器
|
||
)
|
||
|
||
onUnmounted(() => {
|
||
if (timer !== null) {
|
||
clearInterval(timer)
|
||
timer = null
|
||
}
|
||
})
|
||
</script>
|