135 lines
5.1 KiB
TypeScript
135 lines
5.1 KiB
TypeScript
import { db } from '~/drizzle/db'
|
|
import { requestTimes } from '~/drizzle/schema'
|
|
import { and, eq, gt, lte } from 'drizzle-orm'
|
|
import { getBeijingTimeISOString } from '~/utils/timeUtils'
|
|
import { getSubmissionCount, isCardCodeLimitBypassActive } from '~~/server/utils/submissionLimit'
|
|
import { getSystemSettingsCached } from '~~/server/utils/system-settings-helper'
|
|
import { createApiError } from '~~/server/utils/apiError'
|
|
import { SERVER_ERROR_CODES } from '~~/server/config/constants'
|
|
|
|
export default defineEventHandler(async (event) => {
|
|
// 检查用户认证
|
|
const user = event.context.user
|
|
|
|
if (!user) {
|
|
throw createApiError(401, SERVER_ERROR_CODES.SONG_LOGIN_REQUIRED_VIEW_STATUS, '需要登录才能查看投稿状态')
|
|
}
|
|
|
|
try {
|
|
// 获取系统设置
|
|
const systemSettingsData = await getSystemSettingsCached()
|
|
|
|
// 超级管理员、管理员、歌曲管理员不受投稿限制
|
|
const isAdmin = ['SUPER_ADMIN', 'ADMIN', 'SONG_ADMIN'].includes(user.role)
|
|
|
|
// 基础返回结构
|
|
const status: any = {
|
|
limitEnabled: systemSettingsData?.enableSubmissionLimit || false,
|
|
dailyLimit: null,
|
|
weeklyLimit: null,
|
|
monthlyLimit: null,
|
|
dailyUsed: 0,
|
|
weeklyUsed: 0,
|
|
monthlyUsed: 0,
|
|
dailyRemaining: null,
|
|
weeklyRemaining: null,
|
|
monthlyRemaining: null,
|
|
submissionClosed: !!(systemSettingsData?.forceBlockAllRequests && !isAdmin),
|
|
timeLimitationEnabled: systemSettingsData?.enableRequestTimeLimitation || false,
|
|
currentTimePeriod: null
|
|
}
|
|
|
|
// 检查投稿时段限制
|
|
if (status.timeLimitationEnabled) {
|
|
const currentTime = getBeijingTimeISOString()
|
|
|
|
const activeTimePeriod = await db
|
|
.select()
|
|
.from(requestTimes)
|
|
.where(
|
|
and(
|
|
lte(requestTimes.startTime, currentTime),
|
|
gt(requestTimes.endTime, currentTime),
|
|
eq(requestTimes.enabled, true)
|
|
)
|
|
)
|
|
.limit(1)
|
|
|
|
if (activeTimePeriod.length > 0) {
|
|
status.currentTimePeriod = activeTimePeriod[0]
|
|
} else if (!isAdmin) {
|
|
// 如果没有活跃时段且不是管理员，则视为投稿已关闭
|
|
status.submissionClosed = true
|
|
}
|
|
}
|
|
|
|
if (!status.limitEnabled) {
|
|
return status
|
|
}
|
|
|
|
// 确定生效的限额类型（三选一逻辑）
|
|
const dailyLimit = systemSettingsData.dailySubmissionLimit
|
|
const weeklyLimit = systemSettingsData.weeklySubmissionLimit
|
|
const monthlyLimit = systemSettingsData.monthlySubmissionLimit
|
|
|
|
let effectiveLimit: number | null = null
|
|
let limitType: 'daily' | 'weekly' | 'monthly' | null = null
|
|
|
|
if (dailyLimit !== null && dailyLimit !== undefined) {
|
|
effectiveLimit = dailyLimit
|
|
limitType = 'daily'
|
|
} else if (weeklyLimit !== null && weeklyLimit !== undefined) {
|
|
effectiveLimit = weeklyLimit
|
|
limitType = 'weekly'
|
|
} else if (monthlyLimit !== null && monthlyLimit !== undefined) {
|
|
effectiveLimit = monthlyLimit
|
|
limitType = 'monthly'
|
|
}
|
|
|
|
// 检查是否设置了限额为0（关闭投稿）
|
|
if (effectiveLimit === 0) {
|
|
status.dailyLimit = limitType === 'daily' ? effectiveLimit : null
|
|
status.weeklyLimit = limitType === 'weekly' ? effectiveLimit : null
|
|
status.monthlyLimit = limitType === 'monthly' ? effectiveLimit : null
|
|
status.dailyRemaining = 0
|
|
status.weeklyRemaining = 0
|
|
status.monthlyRemaining = 0
|
|
status.submissionClosed = !isAdmin // 管理员不受投稿关闭限制
|
|
return status
|
|
}
|
|
|
|
// 只计算生效的限额类型的使用量
|
|
let dailyUsed = 0
|
|
let weeklyUsed = 0
|
|
let monthlyUsed = 0
|
|
|
|
const excludeCardCodeRequests = isCardCodeLimitBypassActive(systemSettingsData)
|
|
|
|
if (limitType === 'daily' && effectiveLimit && effectiveLimit > 0) {
|
|
dailyUsed = await getSubmissionCount(db, user.id, 'daily', { excludeCardCodeRequests })
|
|
} else if (limitType === 'weekly' && effectiveLimit && effectiveLimit > 0) {
|
|
weeklyUsed = await getSubmissionCount(db, user.id, 'weekly', { excludeCardCodeRequests })
|
|
} else if (limitType === 'monthly' && effectiveLimit && effectiveLimit > 0) {
|
|
monthlyUsed = await getSubmissionCount(db, user.id, 'monthly', { excludeCardCodeRequests })
|
|
}
|
|
|
|
status.dailyLimit = limitType === 'daily' ? effectiveLimit : null
|
|
status.weeklyLimit = limitType === 'weekly' ? effectiveLimit : null
|
|
status.monthlyLimit = limitType === 'monthly' ? effectiveLimit : null
|
|
status.dailyUsed = dailyUsed
|
|
status.weeklyUsed = weeklyUsed
|
|
status.monthlyUsed = monthlyUsed
|
|
status.dailyRemaining =
|
|
limitType === 'daily' && effectiveLimit ? Math.max(0, effectiveLimit - dailyUsed) : null
|
|
status.weeklyRemaining =
|
|
limitType === 'weekly' && effectiveLimit ? Math.max(0, effectiveLimit - weeklyUsed) : null
|
|
status.monthlyRemaining =
|
|
limitType === 'monthly' && effectiveLimit ? Math.max(0, effectiveLimit - monthlyUsed) : null
|
|
|
|
return status
|
|
} catch (error) {
|
|
console.error('获取投稿状态失败:', error)
|
|
throw createApiError(500, 'SONG_FETCH_STATUS_FAILED', '获取投稿状态失败')
|
|
}
|
|
})
|