doggybee · andersonjoseph · Sep 2, 2025
diff --git a/src/graphql/queries.ts b/src/graphql/queries.ts
@@ -220,4 +220,57 @@ export const userContestRankingQuery = `
       }
     }
   }
-`;
+`;
+
+export const randomProblemQuery = `
+query randomQuestion($categorySlug: String, $filters: QuestionListFilterInput) {
+  randomQuestion(categorySlug: $categorySlug, filters: $filters) {
+      questionId
+      questionFrontendId
+      boundTopicId
+      title
+      titleSlug
+      content
+      translatedTitle
+      difficulty
+      likes
+      dislikes
+      isLiked
+      similarQuestions
+      exampleTestcases
+      contributors {
+        username
+        profileUrl
+        avatarUrl
+      }
+      topicTags {
+        name
+        slug
+      }
+      companyTagStats
+      codeSnippets {
+        lang
+        langSlug
+        code
+      }
+      stats
+      hints
+      solution {
+        id
+        canSeeDetail
+        paidOnly
+        hasVideoSolution
+      }
+      status
+      sampleTestCase
+      metaData
+      judgerAvailable
+      judgeType
+      mysqlSchemas
+      enableRunCode
+      enableTestMode
+      enableDebugger
+      envInfo
+  }
+}
+`;
diff --git a/src/services/leetcode-service.ts b/src/services/leetcode-service.ts
@@ -6,7 +6,8 @@ import {
   problemDetailsQuery,
   searchProblemsQuery,
   contestDetailsQuery,
-  userContestRankingQuery
+  userContestRankingQuery,
+  randomProblemQuery
 } from "../graphql/queries.js";
 
 export class LeetCodeService {
@@ -80,6 +81,19 @@ export class LeetCodeService {
     return this.executeQuery(problemDetailsQuery, { titleSlug });
   }
 
+  /**
+   * Fetch a random problem based on the given criteria
+   */
+  async fetchRandomProblem(tags?: string, difficulty?: string) {
+    return this.executeQuery(randomProblemQuery, {
+      categorySlug: "",
+      filters: {
+	tags: tags ? tags.split("+") : [],
+	difficulty: difficulty || null,
+      }
+    });
+  }
+
   /**
    * Search for problems matching criteria
    */
@@ -108,4 +122,4 @@ export class LeetCodeService {
   async fetchUserContestRanking(username: string) {
     return this.executeQuery(userContestRankingQuery, { username });
   }
-}
+}
diff --git a/src/tools/problem-tools.ts b/src/tools/problem-tools.ts
@@ -78,4 +78,30 @@ export function registerProblemTools(server: McpServer, leetcodeService: LeetCod
       }
     }
   );
-}
+
+  // Random Problem
+  server.tool(
+    "random-problem",
+    {
+      tags: z.string().optional().describe("Tags to filter by, separated by '+' (e.g., 'array+dynamic-programming')"),
+      difficulty: z.enum(["EASY", "MEDIUM", "HARD"]).optional().describe("Difficulty level")
+    },
+    async ({tags, difficulty}) => {
+      try {
+        const data = await leetcodeService.fetchRandomProblem(tags, difficulty);
+        return {
+          content: [{ 
+            type: "text", 
+            text: JSON.stringify(data, null, 2)
+          }]
+        };
+      } catch (error: unknown) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        return {
+          content: [{ type: "text", text: `Error: ${errorMessage}` }],
+          isError: true
+        };
+      }
+    }
+  );
+}