feat: 支持配置微信用户已通关关卡

app/api/wx-users/level-progress/route.ts

@@ -1,13 +1,12 @@
 import { NextRequest, NextResponse } from 'next/server'
-import { prisma } from '@/lib/prisma'
 import { auth } from '@/lib/auth'
+import { prisma } from '@/lib/prisma'
+import { v4 as uuidv4 } from 'uuid'
 
 export const dynamic = 'force-dynamic'
 
-// DELETE /api/wx-users/level-progress - Batch delete level progress
-export async function DELETE(
-  request: NextRequest,
-) {
+// PUT /api/wx-users/level-progress - Replace a user's completed levels
+export async function PUT(request: NextRequest) {
   try {
     const session = await auth.api.getSession({
       headers: request.headers,
@@ -17,28 +16,101 @@ export async function DELETE(
       return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
     }
 
-    const { ids } = await request.json()
+    const body = await request.json()
+    const { userId, levelIds } = body
 
-    if (!Array.isArray(ids) || ids.length === 0 || !ids.every((id) => typeof id === 'string')) {
+    if (typeof userId !== 'string' || !userId) {
+      return NextResponse.json({ error: 'userId is required' }, { status: 400 })
+    }
+
+    if (!Array.isArray(levelIds) || !levelIds.every((id) => typeof id === 'string')) {
       return NextResponse.json(
-        { error: 'ids must be a non-empty array of strings' },
+        { error: 'levelIds must be an array of strings' },
         { status: 400 }
       )
     }
 
-    const result = await prisma.wxUserLevelProgress.deleteMany({
-      where: {
-        id: {
-          in: ids,
-        },
-      },
+    const uniqueLevelIds = Array.from(new Set(levelIds))
+
+    const [user, levels] = await Promise.all([
+      prisma.wxUser.findUnique({
+        where: { id: userId },
+        select: { id: true },
+      }),
+      prisma.level.findMany({
+        where: { id: { in: uniqueLevelIds } },
+        select: { id: true },
+      }),
+    ])
+
+    if (!user) {
+      return NextResponse.json({ error: 'User not found' }, { status: 404 })
+    }
+
+    if (levels.length !== uniqueLevelIds.length) {
+      return NextResponse.json(
+        { error: 'Some levels do not exist' },
+        { status: 400 }
+      )
+    }
+
+    await prisma.$transaction(async (tx) => {
+      await tx.wxUserLevelProgress.deleteMany({
+        where: { userId },
+      })
+
+      if (uniqueLevelIds.length > 0) {
+        await tx.wxUserLevelProgress.createMany({
+          data: uniqueLevelIds.map((levelId) => ({
+            id: uuidv4(),
+            userId,
+            levelId,
+            timeSpent: 0,
+          })),
+        })
+      }
     })
 
-    return NextResponse.json({ deleted: result.count })
+    return NextResponse.json({
+      success: true,
+      completedLevelCount: uniqueLevelIds.length,
+    })
   } catch (error) {
-    console.error('Error deleting level progress:', error)
+    console.error('Error updating level progress:', error)
     return NextResponse.json(
-      { error: 'Failed to delete level progress' },
+      { error: 'Failed to update level progress' },
+      { status: 500 }
+    )
+  }
+}
+
+// DELETE /api/wx-users/level-progress - Clear all completed levels for a user
+export async function DELETE(request: NextRequest) {
+  try {
+    const session = await auth.api.getSession({
+      headers: request.headers,
+    })
+
+    if (!session) {
+      return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
+    }
+
+    const body = await request.json()
+    const { userId } = body
+
+    if (typeof userId !== 'string' || !userId) {
+      return NextResponse.json({ error: 'userId is required' }, { status: 400 })
+    }
+
+    const result = await prisma.wxUserLevelProgress.deleteMany({
+      where: { userId },
+    })
+
+    return NextResponse.json({ success: true, deleted: result.count })
+  } catch (error) {
+    console.error('Error clearing level progress:', error)
+    return NextResponse.json(
+      { error: 'Failed to clear level progress' },
       { status: 500 }
     )
   }