初次提交

2026-09-06 08:19:02 +08:00
commit 4ce4649aaf
1061 changed files with 337258 additions and 0 deletions
@@ -0,0 +1,201 @@
+import { db } from '~/drizzle/db'
+import { sql } from 'drizzle-orm'
+
+// 修复单个表的序列
+async function fixTableSequence(table: string, dbTableName: string) {
+  try {
+    // 获取表的最大ID
+    const maxIdResult = await db.execute(sql.raw(`SELECT MAX(id) as max_id FROM "${dbTableName}"`))
+    const maxId = Number((maxIdResult as any)[0]?.max_id || 0)
+
+    if (maxId === 0) {
+      return {
+        success: true,
+        table: table,
+        message: `表 ${table} 为空，无需修复序列`,
+        data: {
+          table: table,
+          maxId: 0,
+          skipped: true
+        }
+      }
+    }
+
+    // 获取序列名称
+    const sequenceNameResult = await db.execute(
+      sql.raw(`SELECT pg_get_serial_sequence('"${dbTableName}"', 'id') as sequence_name`)
+    )
+    const sequenceName = (sequenceNameResult as any)[0]?.sequence_name
+
+    if (!sequenceName) {
+      return {
+        success: false,
+        table: table,
+        error: `无法找到表 ${table}.id 的序列`
+      }
+    }
+
+    // 获取当前序列值
+    const currentSeqResult = await db.execute(sql.raw(`SELECT last_value FROM ${sequenceName}`))
+    const currentSeqValue = Number((currentSeqResult as any)[0]?.last_value || 0)
+
+    // 重置序列值到最大ID + 1
+    const newSequenceValue = maxId + 1
+
+    // 检查序列是否已经正确
+    if (currentSeqValue >= newSequenceValue) {
+      return {
+        success: true,
+        table: table,
+        message: `表 ${table} 序列值已正确，无需修复`,
+        data: {
+          table: table,
+          maxId: maxId,
+          currentSequenceValue: currentSeqValue,
+          alreadyCorrect: true
+        }
+      }
+    }
+
+    await db.execute(sql.raw(`SELECT setval('${sequenceName}', ${newSequenceValue})`))
+
+    return {
+      success: true,
+      table: table,
+      message: `表 ${table} 的序列已成功重置 (${currentSeqValue} → ${newSequenceValue})`,
+      data: {
+        table: table,
+        maxId: maxId,
+        oldSequenceValue: currentSeqValue,
+        newSequenceValue: newSequenceValue,
+        sequenceName: sequenceName
+      }
+    }
+  } catch (error) {
+    return {
+      success: false,
+      table: table,
+      error: error instanceof Error ? error.message : '未知错误'
+    }
+  }
+}
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    const user = event.context.user
+    if (!user || !['ADMIN', 'SUPER_ADMIN'].includes(user.role)) {
+      throw createError({
+        statusCode: 403,
+        statusMessage: '需要管理员权限'
+      })
+    }
+
+    const body = await readBody(event)
+    const { table } = body
+
+    // 支持的表列表和表名映射
+    const supportedTables = [
+      'Song',
+      'User',
+      'UserIdentity',
+      'UserStatusLog',
+      'Vote',
+      'Schedule',
+      'Notification',
+      'NotificationSettings',
+      'PlayTime',
+      'Semester',
+      'SystemSettings',
+      'SongBlacklist',
+      'SongReplayRequest',
+      'RequestTime',
+      'EmailTemplate'
+    ]
+
+    // Prisma模型名到数据库表名的映射
+    const tableNameMap: Record<string, string> = {
+      Song: 'Song',
+      User: 'User',
+      UserIdentity: 'UserIdentity',
+      UserStatusLog: 'user_status_logs',
+      Vote: 'Vote',
+      Schedule: 'Schedule',
+      Notification: 'Notification',
+      NotificationSettings: 'NotificationSettings',
+      PlayTime: 'PlayTime',
+      Semester: 'Semester',
+      SystemSettings: 'SystemSettings',
+      SongBlacklist: 'SongBlacklist',
+      SongReplayRequest: 'song_replay_requests',
+      RequestTime: 'RequestTime',
+      EmailTemplate: 'EmailTemplate'
+    }
+
+    // 如果没有指定表名或指定为'all'，则修复所有表
+    if (!table || table === 'all') {
+      const results = []
+      let hasError = false
+      let fixedCount = 0
+      let skippedCount = 0
+      let errorCount = 0
+
+      for (const tableName of supportedTables) {
+        try {
+          const result = await fixTableSequence(tableName, tableNameMap[tableName])
+          results.push(result)
+
+          if (result.success) {
+            if (result.data?.skipped || result.data?.alreadyCorrect) {
+              skippedCount++
+            } else {
+              fixedCount++
+            }
+          } else {
+            hasError = true
+            errorCount++
+          }
+        } catch (error) {
+          const errorResult = {
+            success: false,
+            table: tableName,
+            error: error instanceof Error ? error.message : '未知错误'
+          }
+          results.push(errorResult)
+          hasError = true
+          errorCount++
+        }
+      }
+
+      return {
+        success: !hasError,
+        message: hasError ? `序列修复完成，但有 ${errorCount} 个表失败` : `所有表序列修复完成`,
+        summary: {
+          total: supportedTables.length,
+          fixed: fixedCount,
+          skipped: skippedCount,
+          errors: errorCount
+        },
+        results: results
+      }
+    }
+
+    if (!supportedTables.includes(table)) {
+      return {
+        success: false,
+        error: `不支持的表名。支持的表: ${supportedTables.join(', ')}, 或使用 'all' 修复所有表`
+      }
+    }
+
+    const dbTableName = tableNameMap[table]
+
+    // 使用辅助函数修复单个表
+    return await fixTableSequence(table, dbTableName)
+  } catch (error) {
+    console.error('Fix sequence error:', error)
+    return {
+      success: false,
+      error: error instanceof Error ? error.message : '未知错误'
+    }
+  }
+})