fix: leetcode-cn can not use GitHub log in bug (#46)

leetcode-tools · Jun 27, 2020 · 1c45e40 · 1c45e40
1 parent 404420e
commit 1c45e40
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/plugins/leetcode.js b/lib/plugins/leetcode.js
@@ -540,10 +540,16 @@ plugin.login = function(user, cb) {
 };
 
 function parseCookie(cookie, body, cb) {
+  const isCN = config.app === 'leetcode.cn';
   const SessionPattern = /LEETCODE_SESSION=(.+?)(;|$)/;
-  const csrfPattern = /csrftoken=(.+?)(;|$)/;
+  let csrfPattern;
+  if (isCN) {
+    csrfPattern = /name="csrfmiddlewaretoken" value="(.*?)"/;
+  } else {
+    csrfPattern = /csrftoken=(.+?)(;|$)/;
+  }
+  const reCsrfResult = csrfPattern.exec(isCN? body: cookie);
   const reSessionResult = SessionPattern.exec(cookie);
-  const reCsrfResult = csrfPattern.exec(cookie);
   if (reSessionResult === null || reCsrfResult === null) {
     return cb('invalid cookie?');
   }