support GitHub and LinkedIn login for leetcode-cn.com

Author: yihong0618

lib/plugins/leetcode.cn.js

@@ -17,20 +17,24 @@ var plugin = new Plugin(15, 'leetcode.cn', '2018.11.25',
 
 plugin.init = function() {
   config.app = 'leetcode.cn';
-  config.sys.urls.base            = 'https://leetcode-cn.com';
-  config.sys.urls.login           = 'https://leetcode-cn.com/accounts/login/';
-  config.sys.urls.problems        = 'https://leetcode-cn.com/api/problems/$category/';
-  config.sys.urls.problem         = 'https://leetcode-cn.com/problems/$slug/description/';
-  config.sys.urls.graphql         = 'https://leetcode-cn.com/graphql';
-  config.sys.urls.problem_detail  = 'https://leetcode-cn.com/graphql';
-  config.sys.urls.test            = 'https://leetcode-cn.com/problems/$slug/interpret_solution/';
-  config.sys.urls.session         = 'https://leetcode-cn.com/session/';
-  config.sys.urls.submit          = 'https://leetcode-cn.com/problems/$slug/submit/';
-  config.sys.urls.submissions     = 'https://leetcode-cn.com/api/submissions/$slug';
-  config.sys.urls.submission      = 'https://leetcode-cn.com/submissions/detail/$id/';
-  config.sys.urls.verify          = 'https://leetcode-cn.com/submissions/detail/$id/check/';
-  config.sys.urls.favorites       = 'https://leetcode-cn.com/list/api/questions';
-  config.sys.urls.favorite_delete = 'https://leetcode-cn.com/list/api/questions/$hash/$id';
+  config.sys.urls.base              = 'https://leetcode-cn.com';
+  config.sys.urls.login             = 'https://leetcode-cn.com/accounts/login/';
+  config.sys.urls.problems          = 'https://leetcode-cn.com/api/problems/$category/';
+  config.sys.urls.problem           = 'https://leetcode-cn.com/problems/$slug/description/';
+  config.sys.urls.graphql           = 'https://leetcode-cn.com/graphql';
+  config.sys.urls.problem_detail    = 'https://leetcode-cn.com/graphql';
+  config.sys.urls.test              = 'https://leetcode-cn.com/problems/$slug/interpret_solution/';
+  config.sys.urls.session           = 'https://leetcode-cn.com/session/';
+  config.sys.urls.submit            = 'https://leetcode-cn.com/problems/$slug/submit/';
+  config.sys.urls.submissions       = 'https://leetcode-cn.com/api/submissions/$slug';
+  config.sys.urls.submission        = 'https://leetcode-cn.com/submissions/detail/$id/';
+  config.sys.urls.verify            = 'https://leetcode-cn.com/submissions/detail/$id/check/';
+  config.sys.urls.favorites         = 'https://leetcode-cn.com/list/api/questions';
+  config.sys.urls.favorite_delete   = 'https://leetcode-cn.com/list/api/questions/$hash/$id';
+  // third parties
+  config.sys.urls.github_login      = 'https://leetcode-cn.com/accounts/github/login/?next=%2F';
+  config.sys.urls.linkedin_login    = 'https://leetcode-cn.com/accounts/linkedin_oauth2/login/?next=%2F';
+  config.sys.urls.leetcode_redirect = 'https://leetcode-cn.com/';
 };
 
 // FIXME: refactor those

lib/plugins/leetcode.js

@@ -552,13 +552,31 @@ function parseCookie(cookie, cb) {
   };
 }
 
+// leetcode-cn.com Cookie is not the same as leetcode.com in third parties
+function parseCNCookie(cookie, body, cb) {
+  const SessionPattern = /LEETCODE_SESSION=(.+?)(;|$)/;
+  const csrfPattern = /name="csrfmiddlewaretoken" value="(.*?)"/;
+  const reSessionResult = SessionPattern.exec(cookie);
+  const reCsrfResult = csrfPattern.exec(body);
+  if (reSessionResult === null || reCsrfResult === null) {
+    return cb('invalid cookie?');
+  }
+  return {
+    sessionId:   reSessionResult[1],
+    sessionCSRF: reCsrfResult[1],
+  };
+}
+
 function requestLeetcodeAndSave(request, leetcodeUrl, user, cb, party) {
   request.get({url: leetcodeUrl}, function(e, resp, body) {
     const redirectUri = resp.request.uri.href;
     if (redirectUri !== config.sys.urls.leetcode_redirect) {
       return cb(`${party} login failed or ${party} did not connect to LeetCode`);
     }
-    const cookieData = parseCookie(resp.request.headers.cookie, cb);
+    let cookieData = {}
+    if (leetcodeUrl.includes('cn')) {
+      cookieData = parseCNCookie(resp.request.headers.cookie, body, cb);
+    } else cookieData = parseCookie(resp.request.headers.cookie, cb)
     user.sessionId = cookieData.sessionId;
     user.sessionCSRF = cookieData.sessionCSRF;
     session.saveUser(user);