From ba16523c0376aef8ecea1f1f0f81a13603b97b9c Mon Sep 17 00:00:00 2001
From: cfngc4594 <cfngc4594@gmail.com>
Date: Fri, 7 Mar 2025 16:34:51 +0800
Subject: [PATCH] feat(problemset): add difficulty color mapping based on
 problem difficulty

---
 src/app/(app)/problemset/page.tsx | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/app/(app)/problemset/page.tsx b/src/app/(app)/problemset/page.tsx
index 0317998..93d8de9 100644
--- a/src/app/(app)/problemset/page.tsx
+++ b/src/app/(app)/problemset/page.tsx
@@ -8,6 +8,20 @@ import {
   TableRow,
 } from "@/components/ui/table";
 import prisma from "@/lib/prisma";
+import { Difficulty } from "@prisma/client";
+
+const getDifficultyColor = (difficulty: Difficulty) => {
+  switch (difficulty) {
+    case "EASY":
+      return "text-green-500";
+    case "MEDIUM":
+      return "text-yellow-500";
+    case "HARD":
+      return "text-red-500";
+    default:
+      return "text-gray-500";
+  }
+};
 
 export default async function ProblemsetPage() {
   const problems = await prisma.problem.findMany({
@@ -46,7 +60,9 @@ export default async function ProblemsetPage() {
                 {problem.title}
               </Link>
             </TableCell>
-            <TableCell className="py-2.5">
+            <TableCell
+              className={`py-2.5 ${getDifficultyColor(problem.difficulty)}`}
+            >
               {problem.difficulty}
             </TableCell>
           </TableRow>