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,71 @@
+import { eq } from 'drizzle-orm'
+import { scheduleSongPool, songs, users } from '~/drizzle/schema'
+import { getServerDate } from '~~/server/utils/serverTime'
+
+/**
+ * 恢复单条备选池记录（含 songId + addedBy 外键映射校验）
+ * 返回 void；歌曲不存在时记录警告并跳过
+ */
+export async function restoreScheduleSongPoolRecord(
+  tx: any,
+  record: any,
+  songIdMapping: Map<number, number>,
+  userIdMapping: Map<number, number>,
+  stats: { warnings: string[]; created?: number },
+  onInserted?: () => void
+): Promise<void> {
+  let validPoolSongId = record.songId
+  const mappedPoolSongId = songIdMapping.get(record.songId)
+  if (mappedPoolSongId) {
+    validPoolSongId = mappedPoolSongId
+  } else {
+    const songExists = await tx
+      .select()
+      .from(songs)
+      .where(eq(songs.id, record.songId))
+      .limit(1)
+    if (songExists.length === 0) {
+      console.warn(`备选池记录的歌曲ID ${record.songId} 不存在，跳过此记录`)
+      stats.warnings.push(`备选池歌曲ID ${record.songId} 不存在`)
+      return
+    }
+    // mapping 未命中但歌曲存在时，使用数据库查到的 ID（merge 模式下可能已被重新分配）
+    validPoolSongId = songExists[0].id
+  }
+
+  let validPoolAddedBy = record.addedBy || null
+  if (validPoolAddedBy) {
+    const mappedAddedBy = userIdMapping.get(validPoolAddedBy)
+    if (mappedAddedBy) {
+      validPoolAddedBy = mappedAddedBy
+    } else {
+      const userExists = await tx
+        .select()
+        .from(users)
+        .where(eq(users.id, validPoolAddedBy))
+        .limit(1)
+      if (userExists.length === 0) {
+        validPoolAddedBy = null
+      }
+    }
+  }
+
+  let createdAt = getServerDate()
+  if (record.createdAt) {
+    const d = new Date(record.createdAt)
+    if (!Number.isNaN(d.getTime())) {
+      createdAt = d
+    } else {
+      console.warn(`备选池记录 createdAt "${record.createdAt}" 非法，使用当前时间`)
+    }
+  }
+
+  const poolData = {
+    songId: validPoolSongId,
+    addedBy: validPoolAddedBy,
+    createdAt
+  }
+
+  const inserted = await tx.insert(scheduleSongPool).values(poolData).onConflictDoNothing({ target: [scheduleSongPool.songId] }).returning()
+  if (inserted.length > 0) onInserted?.()
+}