sudoku leader board fixed

server/routes/sudoku.ts

@@ -51,34 +51,71 @@ router.post('/record', auth, async (req: Request, res: Response) => {
 router.get('/leaderboard/:difficulty', async (req: Request, res: Response) => {
   try {
     const { difficulty } = req.params;
-    const { page = '1', limit = '10' } = req.query;
+    const { page = '1', limit = '10', mode = 'all' } = req.query;
 
     const validDifficulties: SudokuDifficulty[] = ['easy', 'medium', 'hard'];
+    const validModes = ['all', 'standard', 'irregular'];
+
     if (!validDifficulties.includes(difficulty as SudokuDifficulty)) {
       return res.status(400).json({ error: '无效的难度级别' });
     }
 
+    if (!validModes.includes(mode as string)) {
+      return res.status(400).json({ error: '无效的游戏模式' });
+    }
+
     const pageNum = parseInt(page as string, 10);
     const pageSize = parseInt(limit as string, 10);
     const skipCount = (pageNum - 1) * pageSize;
+    const gameMode = mode as 'standard' | 'irregular' | 'all';
 
     const records = await SudokuRecord.getLeaderboard(
       difficulty as SudokuDifficulty,
+      gameMode,
       skipCount,
       pageSize
     );
 
-    const totalUsers = await SudokuRecord.distinct('userId', {
+    const totalMatch: any = {
       difficulty,
       won: true
-    });
+    };
+
+    if (gameMode === 'standard') {
+      totalMatch.$or = [
+        { gameMode: 'standard' },
+        { gameMode: { $exists: false } }
+      ];
+    } else if (gameMode === 'irregular') {
+      totalMatch.gameMode = 'irregular';
+    }
+
+    const totalCountResult = await SudokuRecord.aggregate([
+      {
+        $match: totalMatch
+      },
+      {
+        $group: {
+          _id: {
+            userId: '$userId',
+            gameMode: { $ifNull: ['$gameMode', 'standard'] }
+          }
+        }
+      },
+      {
+        $count: 'count'
+      }
+    ]);
+
+    const total = totalCountResult[0]?.count || 0;
 
     res.json({
       records,
-      total: totalUsers.length,
+      total,
       currentPage: pageNum,
-      totalPages: Math.ceil(totalUsers.length / pageSize),
-      difficulty
+      totalPages: Math.ceil(total / pageSize),
+      difficulty,
+      mode: gameMode
     });
   } catch (error) {
     console.error('获取数独排行榜失败:', error);