Merge pull request #48 from JacobLinCool/feat-cn-graphql-noj-go

feat: make endpoint of leatcode-cn configurable

Author: JacobLinCool

src/_tests/leetcode-cn.test.ts

@@ -67,5 +67,24 @@ describe("LeetCode", () => {
             });
             expect(data.user.username).toBe("LeetCode");
         });
+
+        it("should be able to use graphql noj-go", async () => {
+            const { data } = await lc.graphql(
+                {
+                    operationName: "data",
+                    variables: { username: "LeetCode" },
+                    query: `
+                query data($username: String!, $year: Int) {
+                    calendar: userCalendar(userSlug: $username, year: $year) {
+                      streak
+                      totalActiveDays
+                      submissionCalendar
+                    }
+                  }`,
+                },
+                "/graphql/noj-go/",
+            );
+            expect(typeof data.calendar.streak).toBe("number");
+        });
     });
 });

src/leetcode-cn.ts

@@ -93,15 +93,19 @@ export class LeetCodeCN extends EventEmitter {
     /**
      * Use GraphQL to query LeetCodeCN API.
      * @param query
+     * @param endpoint Maybe you want to use `/graphql/noj-go/` instead of `/graphql/`.
      * @returns
      */
-    public async graphql(query: LeetCodeGraphQLQuery): Promise<LeetCodeGraphQLResponse> {
+    public async graphql(
+        query: LeetCodeGraphQLQuery,
+        endpoint = "/graphql/",
+    ): Promise<LeetCodeGraphQLResponse> {
         await this.initialized;
 
         try {
             await this.limiter.lock();
             const BASE = BASE_URL_CN;
-            const res = await fetch(`${BASE}/graphql`, {
+            const res = await fetch(`${BASE}${endpoint}`, {
                 method: "POST",
                 headers: {
                     "content-type": "application/json",

src/types.ts

@@ -44,7 +44,7 @@ export interface ICredential {
 // LeetCode GraphQL
 export interface LeetCodeGraphQLQuery {
     operationName: string;
-    variables: { [key: string]: unknown };
+    variables?: { [key: string]: unknown };
     query: string;
 }