feat: add getTitleSlugQuestionNumberMapping method (#15)

Author: codewithsathya

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@leetnotion/leetcode-api",
-	"version": "1.8.3",
+	"version": "1.8.4",
 	"description": "Get user profiles, submissions, and problems on LeetCode.",
 	"type": "module",
 	"types": "lib/index.d.ts",

src/_tests/leetcode-advanced.test.ts

@@ -73,17 +73,22 @@ describe('LeetCode Advanced', { timeout: 60_000 * 60 }, () => {
 			expect(problems.length).toBeGreaterThan(3000);
 		});
 
-		// it('should be able to get detailed problems', async () => {
-		// 	const problems = await lc.detailedProblems({
-		// 		category: '',
-		// 		offset: 0,
-		// 		limit: 10,
-		// 	});
-		// 	expect(problems.length).equals(10);
-		// 	expect(problems[0].questionFrontendId).toBeTruthy();
-		// 	expect(problems[0].title).toBeTruthy();
-		// 	expect(problems[0].difficulty).toBeTruthy();
-		// });
+		it('should be able to get title slug question number mapping', async () => {
+			const mapping = await lc.getTitleSlugQuestionNumberMapping();
+			expect(Object.keys(mapping).length).toBeGreaterThan(3000);
+		});
+
+		it('should be able to get detailed problems', async () => {
+			const problems = await lc.detailedProblems({
+				category: '',
+				offset: 0,
+				limit: 10,
+			});
+			expect(problems.length).equals(10);
+			expect(problems[0].questionFrontendId).toBeTruthy();
+			expect(problems[0].title).toBeTruthy();
+			expect(problems[0].difficulty).toBeTruthy();
+		});
 	});
 
 	describe('Authenticated', () => {

src/graphql/title-slug-question-number-mapping.graphql

@@ -0,0 +1,19 @@
+query problemsetQuestionList(
+	$categorySlug: String
+	$limit: Int
+	$skip: Int
+	$filters: QuestionListFilterInput
+) {
+	problemsetQuestionList: questionList(
+		categorySlug: $categorySlug
+		limit: $limit
+		skip: $skip
+		filters: $filters
+	) {
+		total: totalNum
+		questions: data {
+			questionFrontendId
+			titleSlug
+		}
+	}
+}

src/leetcode-advanced.ts

@@ -9,6 +9,7 @@ import IS_EASTER_EGG_COLLECTED from './graphql/is-easter-egg-collected.graphql?r
 import MINIMAL_COMPANY_TAGS from './graphql/minimal-company-tags.graphql?raw';
 import NO_OF_QUESTIONS from './graphql/no-of-problems.graphql?raw';
 import QUESTION_FRONTEND_IDS from './graphql/question-frontend-ids.graphql?raw';
+import TITLE_SLUG_QUESTION_NUMBER_MAPPING_QUERY from './graphql/title-slug-question-number-mapping.graphql?raw';
 import TOPIC_TAGS from './graphql/topic-tags.graphql?raw';
 import { LeetCode } from './leetcode';
 import {
@@ -365,6 +366,29 @@ export class LeetCodeAdvanced extends LeetCode {
 		return problems;
 	}
 
+	/**
+	 * Get title slug question number mapping for all leetcode questions
+	 * @returns Mapping
+	 */
+	public async getTitleSlugQuestionNumberMapping(): Promise<Record<string, string>> {
+		await this.initialized;
+		const { data } = await this.graphql({
+			query: TITLE_SLUG_QUESTION_NUMBER_MAPPING_QUERY,
+			variables: {
+				categorySlug: '',
+				filters: {},
+				skip: 0,
+				limit: 100000,
+			},
+		});
+		const problems = data.problemsetQuestionList.questions as LeetcodeProblem[];
+		const mapping: Record<string, string> = {};
+		problems.forEach((problem) => {
+			mapping[problem.titleSlug] = problem.questionFrontendId;
+		});
+		return mapping;
+	}
+
 	private convertRecentSubmissionToSubmissionType(recentSubmission: RecentSubmission): Submission {
 		return {
 			id: parseInt(recentSubmission.id, 10),