Merge pull request #1 from woung717/fix-cloudflare-issue

fix: added some request headers and make it always.

Author: woung717

lib/plugins/leetcode.cn.js

@@ -39,20 +39,28 @@ plugin.init = function() {
 
 // FIXME: refactor those
 // update options with user credentials
-function signOpts(opts, user) {
+function setupHeaders(opts, user) {
   opts.headers.Cookie = 'LEETCODE_SESSION=' + user.sessionId +
                         ';csrftoken=' + user.sessionCSRF + ';';
   opts.headers['X-CSRFToken'] = user.sessionCSRF;
+  opts.headers['x-csrftoken'] = user.sessionCSRF;
   opts.headers['X-Requested-With'] = 'XMLHttpRequest';
+
+  opts.headers['User-Agent'] = 'Mozilla/5.0 (X11; Linux x86_64; rv:123.0) Gecko/20100101 Firefox/123.0';
+  opts.headers['Referer'] = 'https://leetcode.cn';
+  opts.headers['Origin'] = 'https://leetcode.cn/';
+  opts.headers['Host'] = 'leetcode.cn';
+  opts.headers['Content-Type'] = 'application/json';
+  opts.headers['Accept'] = 'application/json';
 }
 
 function makeOpts(url) {
   const opts = {};
   opts.url = url;
   opts.headers = {};
 
-  if (session.isLogin())
-    signOpts(opts, session.getUser());
+  setupHeaders(opts, session.getUser());
+  
   return opts;
 }
 

lib/plugins/leetcode.js

@@ -23,16 +23,24 @@ plugin.signOpts = function(opts, user) {
   opts.headers.Cookie = 'LEETCODE_SESSION=' + user.sessionId +
                         ';csrftoken=' + user.sessionCSRF + ';';
   opts.headers['X-CSRFToken'] = user.sessionCSRF;
+  opts.headers['x-csrftoken'] = user.sessionCSRF;
   opts.headers['X-Requested-With'] = 'XMLHttpRequest';
+
+  opts.headers['User-Agent'] = 'Mozilla/5.0 (X11; Linux x86_64; rv:123.0) Gecko/20100101 Firefox/123.0';
+  opts.headers['Referer'] = 'https://leetcode.com';
+  opts.headers['Origin'] = 'https://leetcode.com/';
+  opts.headers['Host'] = 'leetcode.com';
+  opts.headers['Content-Type'] = 'application/json';
+  opts.headers['Accept'] = 'application/json';
 };
 
 plugin.makeOpts = function(url) {
   const opts = {};
   opts.url = url;
   opts.headers = {};
 
-  if (session.isLogin())
-    plugin.signOpts(opts, session.getUser());
+  plugin.signOpts(opts, session.getUser());
+  
   return opts;
 };
 

lib/plugins/solution.discuss.js

@@ -16,6 +16,21 @@ var plugin = new Plugin(200, 'solution.discuss', '2019.02.03',
 var URL_DISCUSSES = 'https://leetcode.com/graphql';
 var URL_DISCUSS = 'https://leetcode.com/problems/$slug/discuss/$id';
 
+function setupHeaders(opts, user) {
+  opts.headers.Cookie = 'LEETCODE_SESSION=' + user.sessionId +
+                        ';csrftoken=' + user.sessionCSRF + ';';
+  opts.headers['X-CSRFToken'] = user.sessionCSRF;
+  opts.headers['x-csrftoken'] = user.sessionCSRF;
+  opts.headers['X-Requested-With'] = 'XMLHttpRequest';
+
+  opts.headers['User-Agent'] = 'Mozilla/5.0 (X11; Linux x86_64; rv:123.0) Gecko/20100101 Firefox/123.0';
+  opts.headers['Referer'] = 'https://leetcode.cn';
+  opts.headers['Origin'] = 'https://leetcode.cn/';
+  opts.headers['Host'] = 'leetcode.cn';
+  opts.headers['Content-Type'] = 'application/json';
+  opts.headers['Accept'] = 'application/json';
+}
+
 function getSolution(problem, lang, cb) {
   if (!problem) return cb();
 
@@ -60,6 +75,11 @@ function getSolution(problem, lang, cb) {
       })
     }
   };
+
+  opts.headers = {};
+  
+  setupHeaders(opts, session.getUser());
+
   request(opts, function(e, resp, body) {
     if (e) return cb(e);
     if (resp.statusCode !== 200)