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', '获取投稿状态失败')
  }
})