expose more code to plugins.

* leetcode.cn/lintcode could reuse this.

Signed-off-by: Eric Wang <skygragon@gmail.com>

Author: skygragon

lib/plugins/leetcode.js

@@ -20,24 +20,24 @@ const plugin = new Plugin(10, 'leetcode', '',
 var spin;
 
 // update options with user credentials
-function signOpts(opts, user) {
+plugin.signOpts = function(opts, user) {
   opts.headers.Cookie = 'LEETCODE_SESSION=' + user.sessionId +
                         ';csrftoken=' + user.sessionCSRF + ';';
   opts.headers['X-CSRFToken'] = user.sessionCSRF;
   opts.headers['X-Requested-With'] = 'XMLHttpRequest';
-}
+};
 
-function makeOpts(url) {
+plugin.makeOpts = function(url) {
   const opts = {};
   opts.url = url;
   opts.headers = {};
 
   if (session.isLogin())
-    signOpts(opts, session.getUser());
+    plugin.signOpts(opts, session.getUser());
   return opts;
-}
+};
 
-function checkError(e, resp, expectedStatus) {
+plugin.checkError = function(e, resp, expectedStatus) {
   if (!e && resp && resp.statusCode !== expectedStatus) {
     const code = resp.statusCode;
     log.debug('http error: ' + code);
@@ -49,7 +49,7 @@ function checkError(e, resp, expectedStatus) {
     }
   }
   return e;
-}
+};
 
 plugin.init = function() {
   config.app = 'leetcode';
@@ -80,11 +80,11 @@ plugin.getProblems = function(cb) {
 
 plugin.getCategoryProblems = function(category, cb) {
   log.debug('running leetcode.getCategoryProblems: ' + category);
-  const opts = makeOpts(config.sys.urls.problems.replace('$category', category));
+  const opts = plugin.makeOpts(config.sys.urls.problems.replace('$category', category));
 
   spin.text = 'Downloading category ' + category;
   request(opts, function(e, resp, body) {
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     const json = JSON.parse(body);
@@ -123,7 +123,7 @@ plugin.getProblem = function(problem, cb) {
   const user = session.getUser();
   if (problem.locked && !user.paid) return cb('failed to load locked problem!');
 
-  const opts = makeOpts(config.sys.urls.graphql);
+  const opts = plugin.makeOpts(config.sys.urls.graphql);
   opts.headers.Origin = config.sys.urls.base;
   opts.headers.Referer = problem.link;
 
@@ -149,7 +149,7 @@ plugin.getProblem = function(problem, cb) {
   const spin = h.spin('Downloading ' + problem.slug);
   request.post(opts, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     const q = body.data.question;
@@ -191,7 +191,7 @@ function runCode(opts, problem, cb) {
   const spin = h.spin('Sending code to judge');
   request(opts, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     if (body.error) {
@@ -224,7 +224,7 @@ function verifyResult(task, queue, cb) {
   const spin = h.spin('Waiting for judge result');
   request(opts, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     let result = JSON.parse(body);
@@ -273,7 +273,7 @@ function formatResult(result) {
 
 plugin.testProblem = function(problem, cb) {
   log.debug('running leetcode.testProblem');
-  const opts = makeOpts(config.sys.urls.test.replace('$slug', problem.slug));
+  const opts = plugin.makeOpts(config.sys.urls.test.replace('$slug', problem.slug));
   opts.body = {data_input: problem.testcase};
 
   runCode(opts, problem, function(e, task) {
@@ -292,7 +292,7 @@ plugin.testProblem = function(problem, cb) {
 
 plugin.submitProblem = function(problem, cb) {
   log.debug('running leetcode.submitProblem');
-  const opts = makeOpts(config.sys.urls.submit.replace('$slug', problem.slug));
+  const opts = plugin.makeOpts(config.sys.urls.submit.replace('$slug', problem.slug));
   opts.body = {judge_type: 'large'};
 
   runCode(opts, problem, function(e, task) {
@@ -308,11 +308,11 @@ plugin.submitProblem = function(problem, cb) {
 
 plugin.getSubmissions = function(problem, cb) {
   log.debug('running leetcode.getSubmissions');
-  const opts = makeOpts(config.sys.urls.submissions.replace('$slug', problem.slug));
+  const opts = plugin.makeOpts(config.sys.urls.submissions.replace('$slug', problem.slug));
   opts.headers.Referer = config.sys.urls.problem.replace('$slug', problem.slug);
 
   request(opts, function(e, resp, body) {
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     // FIXME: this only return the 1st 20 submissions, we should get next if necessary.
@@ -326,10 +326,10 @@ plugin.getSubmissions = function(problem, cb) {
 
 plugin.getSubmission = function(submission, cb) {
   log.debug('running leetcode.getSubmission');
-  const opts = makeOpts(config.sys.urls.submission.replace('$id', submission.id));
+  const opts = plugin.makeOpts(config.sys.urls.submission.replace('$id', submission.id));
 
   request(opts, function(e, resp, body) {
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     let re = body.match(/submissionCode:\s('[^']*')/);
@@ -343,7 +343,7 @@ plugin.getSubmission = function(submission, cb) {
 
 plugin.starProblem = function(problem, starred, cb) {
   log.debug('running leetcode.starProblem');
-  const opts = makeOpts();
+  const opts = plugin.makeOpts();
   opts.headers.Origin = config.sys.urls.base;
   opts.headers.Referer = problem.link;
 
@@ -364,7 +364,7 @@ plugin.starProblem = function(problem, starred, cb) {
   }
 
   request(opts, function(e, resp, body) {
-    e = checkError(e, resp, 204);
+    e = plugin.checkError(e, resp, 204);
     if (e) return cb(e);
 
     cb(null, starred);
@@ -373,12 +373,12 @@ plugin.starProblem = function(problem, starred, cb) {
 
 plugin.getFavorites = function(cb) {
   log.debug('running leetcode.getFavorites');
-  const opts = makeOpts(config.sys.urls.favorites);
+  const opts = plugin.makeOpts(config.sys.urls.favorites);
 
   const spin = h.spin('Retrieving user favorites');
   request(opts, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     const favorites = JSON.parse(body);
@@ -388,7 +388,7 @@ plugin.getFavorites = function(cb) {
 
 plugin.getUserInfo = function(cb) {
   log.debug('running leetcode.getUserInfo');
-  const opts = makeOpts(config.sys.urls.graphql);
+  const opts = plugin.makeOpts(config.sys.urls.graphql);
   opts.headers.Origin = config.sys.urls.base;
   opts.headers.Referer = config.sys.urls.base;
   opts.json = true;
@@ -407,7 +407,7 @@ plugin.getUserInfo = function(cb) {
   const spin = h.spin('Retrieving user profile');
   request.post(opts, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     const user = body.data.user;
@@ -416,15 +416,15 @@ plugin.getUserInfo = function(cb) {
 };
 
 function runSession(method, data, cb) {
-  const opts = makeOpts(config.sys.urls.session);
+  const opts = plugin.makeOpts(config.sys.urls.session);
   opts.json = true;
   opts.method = method;
   opts.body = data;
 
   const spin = h.spin('Waiting session result');
   request(opts, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e && e.statusCode === 302) e = session.errors.EXPIRED;
 
     return e ? cb(e) : cb(null, body.sessions);
@@ -459,7 +459,7 @@ plugin.signin = function(user, cb) {
   const spin = h.spin('Signing in leetcode.com');
   request(config.sys.urls.login, function(e, resp, body) {
     spin.stop();
-    e = checkError(e, resp, 200);
+    e = plugin.checkError(e, resp, 200);
     if (e) return cb(e);
 
     user.loginCSRF = h.getSetCookieValue(resp, 'csrftoken');