fix: no csrf token in set-cookie

JacobLinCool · JacobLinCool · commit bb47140ace98 · 2024-03-11T01:56:22.000+08:00
diff --git a/.changeset/metal-shirts-cry.md b/.changeset/metal-shirts-cry.md
@@ -0,0 +1,5 @@
+---
+"leetcode-query": patch
+---
+
+Works with no cookie presented in the response
diff --git a/src/credential-cn.ts b/src/credential-cn.ts
@@ -16,7 +16,10 @@ async function get_csrf() {
             query: "query nojGlobalData {\n  siteRegion\n  chinaHost\n  websocketUrl\n}\n",
         }),
     });
-    const cookies_raw = res.headers.get("set-cookie") as string;
+    const cookies_raw = res.headers.get("set-cookie");
+    if (!cookies_raw) {
+        return undefined;
+    }
 
     const csrf_token = parse_cookie(cookies_raw).csrftoken;
     return csrf_token;
diff --git a/src/credential.ts b/src/credential.ts
@@ -4,11 +4,14 @@ import type { ICredential } from "./types";
 import { parse_cookie } from "./utils";
 
 async function get_csrf() {
-    const cookies_raw = (await fetch(BASE_URL, {
+    const cookies_raw = await fetch(BASE_URL, {
         headers: {
             "user-agent": USER_AGENT,
         },
-    }).then((res) => res.headers.get("set-cookie"))) as string;
+    }).then((res) => res.headers.get("set-cookie"));
+    if (!cookies_raw) {
+        return undefined;
+    }
 
     const csrf_token = parse_cookie(cookies_raw).csrftoken;
     return csrf_token;
diff --git a/src/leetcode-cn.ts b/src/leetcode-cn.ts
@@ -127,7 +127,7 @@ export class LeetCodeCN extends EventEmitter {
             this.emit("receive-graphql", res.clone());
 
             if (res.headers.has("set-cookie")) {
-                const cookies = parse_cookie(res.headers.get("set-cookie") as string);
+                const cookies = parse_cookie(res.headers.get("set-cookie") || "");
 
                 if (cookies["csrftoken"]) {
                     this.credential.csrf = cookies["csrftoken"];
diff --git a/src/leetcode.ts b/src/leetcode.ts
@@ -372,7 +372,7 @@ export class LeetCode extends EventEmitter {
             this.emit("receive-graphql", res.clone());
 
             if (res.headers.has("set-cookie")) {
-                const cookies = parse_cookie(res.headers.get("set-cookie") as string);
+                const cookies = parse_cookie(res.headers.get("set-cookie") || "");
 
                 if (cookies["csrftoken"]) {
                     this.credential.csrf = cookies["csrftoken"];

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +---
 +"leetcode-query": patch
 +---
++
 +Works with no cookie presented in the response