118 lines
4.2 KiB
TypeScript
118 lines
4.2 KiB
TypeScript
import { db } from '~/drizzle/db'
|
|
import { cardCodes } from '~/drizzle/schema'
|
|
import { inArray } from 'drizzle-orm'
|
|
|
|
export default defineEventHandler(async (event) => {
|
|
const user = event.context.user
|
|
if (!user) {
|
|
throw createError({ statusCode: 401, message: '未授权访问' })
|
|
}
|
|
if (!['SONG_ADMIN', 'ADMIN', 'SUPER_ADMIN'].includes(user.role)) {
|
|
throw createError({ statusCode: 403, message: '权限不足' })
|
|
}
|
|
|
|
const body = await readBody(event) ?? {}
|
|
|
|
const rawCodes = Array.isArray(body.codes) ? body.codes : body.codes ? [body.codes] : []
|
|
const codes = rawCodes.map((c: string) => String(c).trim().toUpperCase()).filter(Boolean)
|
|
|
|
const MAX_BATCH_COUNT = 10000
|
|
const parsedCount = Number(body.count)
|
|
const requestedBatchCount = Number.isInteger(parsedCount) ? parsedCount : 0
|
|
if (requestedBatchCount > MAX_BATCH_COUNT) {
|
|
throw createError({
|
|
statusCode: 400,
|
|
message: `单次最多生成 ${MAX_BATCH_COUNT} 个点歌券`
|
|
})
|
|
}
|
|
const batchCount = requestedBatchCount
|
|
const prefix = typeof body.prefix === 'string' ? body.prefix.trim().toUpperCase() : ''
|
|
const parsedLength = Number(body.length)
|
|
const length = Number.isInteger(parsedLength) ? parsedLength : 12
|
|
if (length > 64) {
|
|
throw createError({ statusCode: 400, message: '点歌券随机长度不能超过 64 位' })
|
|
}
|
|
const charsetInput = typeof body.charset === 'string' && body.charset.trim() ? body.charset.trim().toUpperCase() : 'ABCDEFGHJKLMNPQRSTUVWXYZ23456789'
|
|
const charset = [...new Set(charsetInput.split(''))].join('')
|
|
if (!charset) {
|
|
throw createError({ statusCode: 400, message: '字符集不能为空' })
|
|
}
|
|
|
|
const manualCodeSet = new Set(codes)
|
|
const generatedCodeSet = new Set<string>()
|
|
if (batchCount > 0) {
|
|
const actualLength = Math.max(4, length)
|
|
const maxPossibleCodes = charset.length ** actualLength
|
|
if (maxPossibleCodes < batchCount) {
|
|
throw createError({
|
|
statusCode: 400,
|
|
message: '可用字符集过小或长度不足，无法生成足够数量的唯一点歌券，请增大字符集或长度'
|
|
})
|
|
}
|
|
|
|
const makeRandom = () => {
|
|
let code = prefix
|
|
for (let i = 0; i < actualLength; i++) {
|
|
code += charset.charAt(Math.floor(Math.random() * charset.length))
|
|
}
|
|
return code
|
|
}
|
|
|
|
let attempts = 0
|
|
const maxAttempts = batchCount * 100
|
|
while (generatedCodeSet.size < batchCount && attempts < maxAttempts) {
|
|
attempts++
|
|
const next = makeRandom()
|
|
if (!generatedCodeSet.has(next) && !manualCodeSet.has(next)) {
|
|
generatedCodeSet.add(next)
|
|
}
|
|
}
|
|
if (generatedCodeSet.size < batchCount) {
|
|
throw createError({
|
|
statusCode: 400,
|
|
message: '可用字符集过小或长度不足，无法生成足够数量的唯一点歌券，请增大字符集或长度'
|
|
})
|
|
}
|
|
}
|
|
|
|
const finalCodes = [...codes, ...generatedCodeSet]
|
|
if (!finalCodes.length) {
|
|
throw createError({ statusCode: 400, message: '请提供要创建的点歌券或生成数量' })
|
|
}
|
|
if (finalCodes.length > MAX_BATCH_COUNT) {
|
|
throw createError({
|
|
statusCode: 400,
|
|
message: `单次最多创建 ${MAX_BATCH_COUNT} 个点歌券`
|
|
})
|
|
}
|
|
|
|
try {
|
|
const uniqueCodes = [...new Set(finalCodes)]
|
|
const existingRows = await db
|
|
.select({ code: cardCodes.code })
|
|
.from(cardCodes)
|
|
.where(inArray(cardCodes.code, uniqueCodes))
|
|
const existingCodes = new Set(existingRows.map((row) => row.code))
|
|
const insertCodes = uniqueCodes.filter((code) => !existingCodes.has(code))
|
|
|
|
if (!insertCodes.length) {
|
|
throw createError({ statusCode: 400, message: '这些点歌券已经存在，无需重复创建' })
|
|
}
|
|
|
|
const inserts = insertCodes.map((c: string) => ({
|
|
code: c,
|
|
status: 'AVAILABLE',
|
|
note: typeof body.note === 'string' ? body.note.trim() || null : null
|
|
}))
|
|
const res = await db.insert(cardCodes).values(inserts).returning()
|
|
return {
|
|
success: true,
|
|
data: res,
|
|
skipped: uniqueCodes.length - insertCodes.length
|
|
}
|
|
} catch (err: any) {
|
|
console.error('创建点歌券失败', err)
|
|
throw createError({ statusCode: err.statusCode || 500, message: err.message || '创建点歌券失败' })
|
|
}
|
|
})
|