feat: align with upstream 1.5.9.9

server/utils/submissionLimit.ts

@@ -0,0 +1,110 @@
+import { db, songs } from '~/drizzle/db'
+import { and, count, eq, gte, isNull, lte } from 'drizzle-orm'
+import {
+  getBeijingEndOfDay,
+  getBeijingEndOfWeek,
+  getBeijingEndOfMonth,
+  getBeijingStartOfDay,
+  getBeijingStartOfWeek,
+  getBeijingStartOfMonth
+} from '~/utils/timeUtils'
+
+export type LimitType = 'daily' | 'weekly' | 'monthly'
+
+type SubmissionCountOptions = {
+  excludeCardCodeRequests?: boolean
+}
+
+type CardCodeLimitSettings = {
+  enableSubmissionLimit?: boolean | null
+  enableCardCodeRequests?: boolean | null
+  requireCardCodeForRequests?: boolean | null
+  enableCardCodeLimitBypass?: boolean | null
+}
+
+export function isCardCodeLimitBypassActive(
+  settings: CardCodeLimitSettings | null | undefined
+): boolean {
+  return !!(
+    settings?.enableSubmissionLimit &&
+    settings.enableCardCodeLimitBypass &&
+    settings.enableCardCodeRequests === true
+  )
+}
+
+/**
+ * 获取指定限额类型的时间窗口（北京时间）
+ * @param type 限额类型
+ * @returns { start: Date, end: Date }
+ */
+export function getTimeRange(type: LimitType) {
+  switch (type) {
+    case 'weekly':
+      return {
+        start: getBeijingStartOfWeek(),
+        end: getBeijingEndOfWeek()
+      }
+    case 'monthly':
+      return {
+        start: getBeijingStartOfMonth(),
+        end: getBeijingEndOfMonth()
+      }
+    case 'daily':
+    default:
+      // 'daily' 是默认情况，为保证健壮性处理任何意外值。
+      return {
+        start: getBeijingStartOfDay(),
+        end: getBeijingEndOfDay()
+      }
+  }
+}
+
+/**
+ * 获取用户在指定周期内的投稿数量
+ * @param dbOrTx 数据库实例或事务
+ * @param userId 用户ID
+ * @param type 限额类型
+ * @param options 投稿统计选项
+ * @returns 投稿数量
+ */
+export async function getSubmissionCount(
+  dbOrTx: typeof db,
+  userId: number,
+  type: LimitType,
+  options: SubmissionCountOptions = {}
+): Promise<number> {
+  const { start, end } = getTimeRange(type)
+
+  const [result] = await dbOrTx
+    .select({ count: count() })
+    .from(songs)
+    .where(
+      and(
+        eq(songs.requesterId, userId),
+        gte(songs.createdAt, start),
+        lte(songs.createdAt, end),
+        options.excludeCardCodeRequests ? isNull(songs.cardCodeId) : undefined
+      )
+    )
+
+  return result?.count || 0
+}
+
+/**
+ * 检查用户是否达到投稿限额
+ * @param dbOrTx 数据库实例或事务
+ * @param userId 用户ID
+ * @param type 限额类型
+ * @param limit 限额数量
+ * @param options 投稿统计选项
+ * @returns boolean true 表示已达到限额（不允许继续投稿），false 表示未达到
+ */
+export async function isLimitReached(
+  dbOrTx: typeof db,
+  userId: number,
+  type: LimitType,
+  limit: number,
+  options: SubmissionCountOptions = {}
+): Promise<boolean> {
+  return (await getSubmissionCount(dbOrTx, userId, type, options)) >= limit
+}