Merge pull request #24 from SPerekrestova/chore/cleanup

Chore/cleanup

Author: SPerekrestova

package-lock.json

@@ -69,7 +69,6 @@
         "leetcode-query": "^2.0.0",
         "minimist": "^1.2.8",
         "pino": "^9.6.0",
-        "ts-node": "^10.9.2",
         "zod": "^3.24.2"
     },
     "devDependencies": {
@@ -84,7 +83,6 @@
         "globals": "^16.0.0",
         "husky": "^9.1.7",
         "lint-staged": "^15.5.1",
-        "msw": "^2.12.7",
         "pino-pretty": "^13.0.0",
         "prettier": "^3.5.3",
         "prettier-plugin-organize-imports": "^4.1.0",

package.json

@@ -15,12 +15,27 @@ const LANGUAGE_MAP: Record<string, string> = {
     java: "java",
     python: "python3",
     python3: "python3",
+    c: "c",
     cpp: "cpp",
     "c++": "cpp",
+    csharp: "csharp",
+    "c#": "csharp",
     javascript: "javascript",
     js: "javascript",
     typescript: "typescript",
-    ts: "typescript"
+    ts: "typescript",
+    php: "php",
+    swift: "swift",
+    kotlin: "kotlin",
+    dart: "dart",
+    golang: "golang",
+    go: "golang",
+    ruby: "ruby",
+    scala: "scala",
+    rust: "rust",
+    racket: "racket",
+    erlang: "erlang",
+    elixir: "elixir"
 };
 
 /**
@@ -79,25 +94,13 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
         return { submissions };
     }
 
-    /**
-     * 获取用户最近的提交记录
-     * @param username
-     * @param limit
-     * @returns
-     */
     async fetchUserRecentSubmissions(
         username: string,
         limit?: number
     ): Promise<any> {
         return await this.leetCodeApi.recent_submissions(username, limit);
     }
 
-    /**
-     * 获取用户最近 AC 的提交记录
-     * @param username
-     * @param limit
-     * @returns
-     */
     async fetchUserRecentACSubmissions(
         username: string,
         limit?: number
@@ -149,11 +152,17 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
         attended: boolean = true
     ): Promise<any> {
         const contestInfo = await this.leetCodeApi.user_contest_info(username);
-        if (contestInfo.userContestRankingHistory && attended) {
-            contestInfo.userContestRankingHistory =
-                contestInfo.userContestRankingHistory.filter((contest: any) => {
-                    return contest && contest.attended;
-                });
+        if (contestInfo.userContestRankingHistory) {
+            if (attended) {
+                contestInfo.userContestRankingHistory =
+                    contestInfo.userContestRankingHistory.filter(
+                        (contest: any) => {
+                            return contest && contest.attended;
+                        }
+                    );
+            }
+        } else {
+            contestInfo.userContestRankingHistory = [];
         }
         return contestInfo;
     }
@@ -175,10 +184,7 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
         const filteredTopicTags =
             problem.topicTags?.map((tag: any) => tag.slug) || [];
 
-        const filteredCodeSnippets =
-            problem.codeSnippets?.filter((snippet: any) =>
-                ["cpp", "python3", "java"].includes(snippet.langSlug)
-            ) || [];
+        const filteredCodeSnippets = problem.codeSnippets || [];
 
         let parsedSimilarQuestions: any[] = [];
         if (problem.similarQuestions) {
@@ -435,7 +441,7 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
                     return {
                         accepted: false,
                         statusMessage: "Error",
-                        errorMessage: `Unexpected submission state: \${result.state}`
+                        errorMessage: `Unexpected submission state: ${result.state}`
                     };
                 }
 
@@ -448,6 +454,10 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
                             accepted: true,
                             runtime: result.runtime,
                             memory: result.memory,
+                            runtimePercentile: result.runtime_percentile,
+                            memoryPercentile: result.memory_percentile,
+                            totalCorrect: result.total_correct,
+                            totalTestcases: result.total_testcases,
                             statusMessage: "Accepted"
                         };
                     } else {
@@ -456,16 +466,27 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
                         if (result.input) {
                             failedTestCase = `Input: ${result.input}`;
                             if (result.expected_answer && result.code_answer) {
-                                failedTestCase += `\\nExpected: ${JSON.stringify(result.expected_answer)}`;
-                                failedTestCase += `\\nGot: ${JSON.stringify(result.code_answer)}`;
+                                failedTestCase += `\nExpected: ${JSON.stringify(result.expected_answer)}`;
+                                failedTestCase += `\nGot: ${JSON.stringify(result.code_answer)}`;
                             }
                         }
 
+                        // Use the most specific error message available
+                        const errorMessage =
+                            result.full_compile_error ||
+                            result.compile_error ||
+                            result.full_runtime_error ||
+                            result.runtime_error ||
+                            result.std_output ||
+                            undefined;
+
                         return {
                             accepted: false,
                             statusMessage: result.status_msg,
                             failedTestCase,
-                            errorMessage: result.std_output
+                            errorMessage,
+                            totalCorrect: result.total_correct,
+                            totalTestcases: result.total_testcases
                         };
                     }
                 }
@@ -535,7 +556,7 @@ export class LeetCodeGlobalService implements LeetcodeServiceInterface {
         const question = response.data.data?.question;
         if (!question) {
             throw new Error(
-                `Problem slug "\${problemSlug}" not found or invalid.`
+                `Problem slug "${problemSlug}" not found or invalid.`
             );
         }
         return question.questionId;

src/leetcode/leetcode-global-service.ts

@@ -80,7 +80,7 @@ Offer browser-specific guidance:
 - "LeetCode doesn't provide an official authentication API for third-party tools like this MCP server. Cookie extraction is the standard approach used by LeetCode CLI tools and integrations. Your credentials are stored securely and locally on your machine."
 
 **If user is concerned about security:**
-- "Your credentials are encrypted and stored locally in ~/.leetcode-mcp/credentials.json. They're never sent anywhere except directly to LeetCode's API to make requests on your behalf. This is the same method used by official LeetCode CLI tools."
+- "Your credentials are stored locally in ~/.leetcode-mcp/credentials.json with restricted file permissions (owner-only access). They're never sent anywhere except directly to LeetCode's API to make requests on your behalf. This is the same method used by official LeetCode CLI tools."
 
 **If user has trouble copying the entire value:**
 - "Make sure you're clicking on the value field (not the name), and that you can see the full text is selected before copying. The value is typically quite long (50+ characters)."
@@ -163,7 +163,7 @@ Once you're logged in, I'll walk you through getting two cookie values we need f
 - Credentials typically expire after 7-14 days
 - LeetCode allows only one active session per account at a time
 - The MCP server validates credentials by making a test API call to LeetCode's GraphQL endpoint
-- Credentials are stored encrypted in ~/.leetcode-mcp/credentials.json`;
+- Credentials are stored locally in ~/.leetcode-mcp/credentials.json with restricted file permissions`;
 
                 return {
                     messages: [

src/mcp/prompts/auth-prompts.ts

@@ -194,12 +194,17 @@ Keep the user engaged and learning at each step. Adjust the level of guidance ba
             "c#": ".cs",
             csharp: ".cs",
             go: ".go",
+            golang: ".go",
             rust: ".rs",
             ruby: ".rb",
             swift: ".swift",
             kotlin: ".kt",
             scala: ".scala",
-            php: ".php"
+            php: ".php",
+            dart: ".dart",
+            erlang: ".erl",
+            elixir: ".ex",
+            racket: ".rkt"
         };
 
         return extensionMap[language.toLowerCase()] || ".txt";

src/mcp/prompts/learning-prompts.ts

@@ -18,18 +18,35 @@ export class ProblemToolRegistry extends ToolRegistry {
                     "Retrieves today's LeetCode Daily Challenge problem with complete details, including problem description, constraints, and examples. After fetching, invoke the leetcode_learning_mode and leetcode_problem_workflow prompts before helping the user work on it."
             },
             async () => {
-                const data = await this.leetcodeService.fetchDailyChallenge();
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: JSON.stringify({
-                                date: new Date().toISOString().split("T")[0],
-                                problem: data
-                            })
-                        }
-                    ]
-                };
+                try {
+                    const data =
+                        await this.leetcodeService.fetchDailyChallenge();
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify({
+                                    date:
+                                        data?.date ??
+                                        new Date().toISOString().split("T")[0],
+                                    problem: data
+                                })
+                            }
+                        ]
+                    };
+                } catch (error: any) {
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify({
+                                    error: "Failed to fetch daily challenge",
+                                    message: error.message
+                                })
+                            }
+                        ]
+                    };
+                }
             }
         );
 
@@ -49,21 +66,35 @@ export class ProblemToolRegistry extends ToolRegistry {
                 }
             },
             async ({ titleSlug }) => {
-                const data =
-                    await this.leetcodeService.fetchProblemSimplified(
-                        titleSlug
-                    );
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: JSON.stringify({
-                                titleSlug,
-                                problem: data
-                            })
-                        }
-                    ]
-                };
+                try {
+                    const data =
+                        await this.leetcodeService.fetchProblemSimplified(
+                            titleSlug
+                        );
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify({
+                                    titleSlug,
+                                    problem: data
+                                })
+                            }
+                        ]
+                    };
+                } catch (error: any) {
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify({
+                                    error: "Failed to fetch problem details",
+                                    message: error.message
+                                })
+                            }
+                        ]
+                    };
+                }
             }
         );
 
@@ -120,26 +151,44 @@ export class ProblemToolRegistry extends ToolRegistry {
                 offset,
                 searchKeywords
             }) => {
-                const data = await this.leetcodeService.searchProblems(
-                    category,
-                    tags,
-                    difficulty,
-                    limit,
-                    offset,
-                    searchKeywords
-                );
-                return {
-                    content: [
-                        {
-                            type: "text",
-                            text: JSON.stringify({
-                                filters: { tags, difficulty, searchKeywords },
-                                pagination: { limit, offset },
-                                problems: data
-                            })
-                        }
-                    ]
-                };
+                try {
+                    const data = await this.leetcodeService.searchProblems(
+                        category,
+                        tags,
+                        difficulty,
+                        limit,
+                        offset,
+                        searchKeywords
+                    );
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify({
+                                    filters: {
+                                        tags,
+                                        difficulty,
+                                        searchKeywords
+                                    },
+                                    pagination: { limit, offset },
+                                    problems: data
+                                })
+                            }
+                        ]
+                    };
+                } catch (error: any) {
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify({
+                                    error: "Failed to search problems",
+                                    message: error.message
+                                })
+                            }
+                        ]
+                    };
+                }
             }
         );
     }