feat: align with upstream 1.5.9.9

2026-09-12 18:21:11 +08:00
commit 34f0679117
805 changed files with 376275 additions and 0 deletions
@@ -0,0 +1,282 @@
+import { createError, defineEventHandler } from 'h3'
+import { db } from '~/drizzle/db'
+import {
+  apiKeyPermissions,
+  apiKeys,
+  apiLogs,
+  collaborationLogs,
+  emailTemplates,
+  notificationSettings,
+  notifications,
+  playTimes,
+  requestTimes,
+  schedules,
+  semesters,
+  songBlacklists,
+  songCollaborators,
+  songReplayRequests,
+  songs,
+  systemSettings,
+  userIdentities,
+  users,
+  userStatusLogs,
+  votes
+} from '~/drizzle/schema'
+import { ne, sql } from 'drizzle-orm'
+
+function getFirstRow<T>(result: any): T | undefined {
+  return result?.rows?.[0] ?? result?.[0]
+}
+
+function quoteIdentifierPart(value: string) {
+  const normalized =
+    value.startsWith('"') && value.endsWith('"') ? value.slice(1, -1).replace(/""/g, '"') : value
+
+  return `"${normalized.replace(/"/g, '""')}"`
+}
+
+function toQualifiedIdentifier(value: string) {
+  return sql.raw(value.split('.').map(quoteIdentifierPart).join('.'))
+}
+
+function toPgRelationName(value: string) {
+  return value.split('.').map(quoteIdentifierPart).join('.')
+}
+
+function toRegclass(value: string) {
+  return sql.raw(`'${value.replace(/'/g, "''")}'::regclass`)
+}
+
+// 重置所有表的自增序列
+async function resetAutoIncrementSequences() {
+  const tables = [
+    { name: 'users', dbName: 'User' },
+    { name: 'songs', dbName: 'Song' },
+    { name: 'votes', dbName: 'Vote' },
+    { name: 'schedules', dbName: 'Schedule' },
+    { name: 'notifications', dbName: 'Notification' },
+    { name: 'notificationSettings', dbName: 'NotificationSettings' },
+    { name: 'playTimes', dbName: 'PlayTime' },
+    { name: 'semesters', dbName: 'Semester' },
+    { name: 'systemSettings', dbName: 'SystemSettings' },
+    { name: 'songBlacklists', dbName: 'SongBlacklist' },
+    { name: 'userStatusLogs', dbName: 'user_status_logs' },
+    { name: 'userIdentities', dbName: 'UserIdentity' },
+    { name: 'requestTimes', dbName: 'RequestTime' },
+    { name: 'songReplayRequests', dbName: 'song_replay_requests' },
+    { name: 'emailTemplates', dbName: 'EmailTemplate' }
+  ]
+
+  const results = []
+
+  for (const table of tables) {
+    try {
+      // 获取表的最大ID
+      const maxIdResult = await db.execute(
+        sql`SELECT MAX(id) as max_id FROM ${toQualifiedIdentifier(table.dbName)}`
+      )
+      const maxId = Number(
+        getFirstRow<{ max_id: number | string | null }>(maxIdResult)?.max_id || 0
+      )
+
+      // 获取序列名称
+      const sequenceNameResult = await db.execute(
+        sql`SELECT pg_get_serial_sequence(${toPgRelationName(table.dbName)}, 'id') as sequence_name`
+      )
+      const sequenceName = getFirstRow<{ sequence_name: string | null }>(
+        sequenceNameResult
+      )?.sequence_name
+
+      if (sequenceName) {
+        if (maxId === 0) {
+          await db.execute(
+            sql`ALTER SEQUENCE ${toQualifiedIdentifier(sequenceName)} RESTART WITH 1`
+          )
+        } else {
+          const newSequenceValue = maxId
+          await db.execute(sql`SELECT setval(${toRegclass(sequenceName)}, ${newSequenceValue})`)
+        }
+        results.push({ table: table.name, success: true })
+      } else {
+        results.push({ table: table.name, success: false, message: 'Sequence not found' })
+      }
+    } catch (error) {
+      console.warn(`重置 ${table.name} 表序列失败: ${error.message}`)
+      results.push({ table: table.name, success: false, error: error.message })
+    }
+  }
+  return results
+}
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    const user = event.context.user
+    if (!user || !['ADMIN', 'SUPER_ADMIN'].includes(user.role)) {
+      throw createError({
+        statusCode: 403,
+        message: '权限不足'
+      })
+    }
+
+    const resetResults = {
+      success: true,
+      message: '数据库重置完成',
+      details: {
+        tablesCleared: 0,
+        recordsDeleted: 0,
+        sequencesReset: 0,
+        errors: [],
+        preservedData: []
+      }
+    }
+
+    try {
+      // 使用事务确保数据一致性
+      await db.transaction(async (tx) => {
+        // 按照外键依赖顺序删除数据
+
+        // 1. API 日志与权限 (依赖于 API 密钥)
+        const deletedApiLogs = await tx.delete(apiLogs).returning({ id: apiLogs.id })
+        resetResults.details.recordsDeleted += deletedApiLogs.length
+        resetResults.details.tablesCleared++
+
+        const deletedApiKeyPermissions = await tx
+          .delete(apiKeyPermissions)
+          .returning({ id: apiKeyPermissions.id })
+        resetResults.details.recordsDeleted += deletedApiKeyPermissions.length
+        resetResults.details.tablesCleared++
+
+        // 2. API 密钥 (依赖于用户)
+        const deletedApiKeys = await tx.delete(apiKeys).returning({ id: apiKeys.id })
+        resetResults.details.recordsDeleted += deletedApiKeys.length
+        resetResults.details.tablesCleared++
+
+        // 3. 通知与设置 (依赖于用户、歌曲)
+        const deletedNotifications = await tx
+          .delete(notifications)
+          .returning({ id: notifications.id })
+        resetResults.details.recordsDeleted += deletedNotifications.length
+        resetResults.details.tablesCleared++
+
+        const deletedNotificationSettings = await tx
+          .delete(notificationSettings)
+          .returning({ id: notificationSettings.id })
+        resetResults.details.recordsDeleted += deletedNotificationSettings.length
+        resetResults.details.tablesCleared++
+
+        // 4. 协作日志 (依赖于歌曲协作者)
+        const deletedCollaborationLogs = await tx
+          .delete(collaborationLogs)
+          .returning({ id: collaborationLogs.id })
+        resetResults.details.recordsDeleted += deletedCollaborationLogs.length
+        resetResults.details.tablesCleared++
+
+        // 5. 歌曲协作者 (依赖于歌曲、用户)
+        const deletedSongCollaborators = await tx
+          .delete(songCollaborators)
+          .returning({ id: songCollaborators.id })
+        resetResults.details.recordsDeleted += deletedSongCollaborators.length
+        resetResults.details.tablesCleared++
+
+        // 6. 歌曲重播申请 (依赖于歌曲、用户)
+        const deletedReplayRequests = await tx
+          .delete(songReplayRequests)
+          .returning({ id: songReplayRequests.id })
+        resetResults.details.recordsDeleted += deletedReplayRequests.length
+        resetResults.details.tablesCleared++
+
+        // 7. 排期 (依赖于歌曲、播出时段)
+        const deletedSchedules = await tx.delete(schedules).returning({ id: schedules.id })
+        resetResults.details.recordsDeleted += deletedSchedules.length
+        resetResults.details.tablesCleared++
+
+        // 8. 投票 (依赖于歌曲、用户)
+        const deletedVotes = await tx.delete(votes).returning({ id: votes.id })
+        resetResults.details.recordsDeleted += deletedVotes.length
+        resetResults.details.tablesCleared++
+
+        // 9. 歌曲 (依赖于用户、播出时段)
+        const deletedSongs = await tx.delete(songs).returning({ id: songs.id })
+        resetResults.details.recordsDeleted += deletedSongs.length
+        resetResults.details.tablesCleared++
+
+        // 10. 黑名单 (依赖于用户 [创建者])
+        const deletedBlacklists = await tx
+          .delete(songBlacklists)
+          .returning({ id: songBlacklists.id })
+        resetResults.details.recordsDeleted += deletedBlacklists.length
+        resetResults.details.tablesCleared++
+
+        // 11. 用户状态日志 (依赖于用户)
+        const deletedStatusLogs = await tx
+          .delete(userStatusLogs)
+          .returning({ id: userStatusLogs.id })
+        resetResults.details.recordsDeleted += deletedStatusLogs.length
+        resetResults.details.tablesCleared++
+
+        // 12. 第三方身份关联 (依赖于用户)
+        const deletedUserIdentities = await tx
+          .delete(userIdentities)
+          .returning({ id: userIdentities.id })
+        resetResults.details.recordsDeleted += deletedUserIdentities.length
+        resetResults.details.tablesCleared++
+
+        // 13. 邮件模板 (依赖于用户 [更新者])
+        const deletedEmailTemplates = await tx
+          .delete(emailTemplates)
+          .returning({ id: emailTemplates.id })
+        resetResults.details.recordsDeleted += deletedEmailTemplates.length
+        resetResults.details.tablesCleared++
+
+        // 14. 用户 (除自己外)
+        const deletedUsers = await tx
+          .delete(users)
+          .where(ne(users.id, user.id))
+          .returning({ id: users.id })
+        resetResults.details.recordsDeleted += deletedUsers.length
+        resetResults.details.tablesCleared++
+        resetResults.details.preservedData.push(`保留当前管理员账户: ${user.name}`)
+
+        // 15. 播出时段
+        const deletedPlayTimes = await tx.delete(playTimes).returning({ id: playTimes.id })
+        resetResults.details.recordsDeleted += deletedPlayTimes.length
+        resetResults.details.tablesCleared++
+
+        // 16. 学期
+        const deletedSemesters = await tx.delete(semesters).returning({ id: semesters.id })
+        resetResults.details.recordsDeleted += deletedSemesters.length
+        resetResults.details.tablesCleared++
+
+        // 17. 请求时段
+        const deletedRequestTimes = await tx.delete(requestTimes).returning({ id: requestTimes.id })
+        resetResults.details.recordsDeleted += deletedRequestTimes.length
+        resetResults.details.tablesCleared++
+
+        // 18. 系统设置
+        const deletedSystemSettings = await tx
+          .delete(systemSettings)
+          .returning({ id: systemSettings.id })
+        resetResults.details.recordsDeleted += deletedSystemSettings.length
+        resetResults.details.tablesCleared++
+      })
+
+      // 重置自增序列
+      const sequenceResults = await resetAutoIncrementSequences()
+      resetResults.details.sequencesReset = sequenceResults.filter((r) => r.success).length
+    } catch (error) {
+      console.error('重置数据库失败:', error)
+      throw createError({
+        statusCode: 500,
+        message: `重置数据库失败：${error.message}`
+      })
+    }
+
+    return resetResults
+  } catch (error) {
+    throw createError({
+      statusCode: error.statusCode || 500,
+      message: error.message || '服务器内部错误'
+    })
+  }
+})