refs #18: auto retry when session expired in submissions.

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

Author: skygragon

lib/commands/submission.js

@@ -43,7 +43,7 @@ function onTaskDone(e, msg, problem, cb) {
   log.info(sprintf('[%3d] %-60s %s',
         problem.id,
         problem.name,
-        (e ? chalk.red('ERROR: ' + e) : msg)
+        (e ? chalk.red('ERROR: ' + (e.msg || e)) : msg)
         ));
   if (cb) cb(e);
 }

lib/commands/user.js

@@ -44,7 +44,7 @@ cmd.handler = function(argv) {
     });
   } else if (argv.logout) {
     // logout
-    user = core.logout(null);
+    user = core.logout(true);
     if (user)
       log.info('Successfully logout as', chalk.yellow(user.name));
     else

lib/core.js

@@ -146,7 +146,7 @@ core.exportProblem = function(problem, f, codeOnly) {
 };
 
 core.login = function(user, cb) {
-  this.logout();
+  this.logout(false);
   client.login(user, function(e, user) {
     if (e) return cb(e);
 
@@ -166,13 +166,11 @@ core.login = function(user, cb) {
   });
 };
 
-core.logout = function(user) {
-  user = this.getUser();
-  if (user) {
-    // NOTE: need invalidate any user related cache
-    cache.del('.user');
-    cache.del('all');
-  }
+core.logout = function(purge) {
+  var user = this.getUser();
+  if (purge) cache.del('.user');
+  // NOTE: need invalidate any user related cache
+  cache.del('all');
   return user;
 };
 

lib/leetcode_client.js

@@ -150,7 +150,6 @@ leetcodeClient.getProblem = function(user, problem, cb) {
   var opts = makeOpts();
   opts.url = problem.link;
   requestWithReLogin(opts, function(e, resp, body) {
-    // FIXME: if session expired, this will still return 200
     if (e) return cb(e);
 
     var $ = cheerio.load(body);
@@ -185,7 +184,6 @@ leetcodeClient.getSubmissions = function(problem, cb) {
   opts.headers.Referer = config.URL_PROBLEM.replace('$id', problem.key);
 
   requestWithReLogin(opts, function(e, resp, body) {
-    // FIXME: if session expired, this will still return 200
     if (e) return cb(e);
 
     // FIXME: this only return the 1st 20 submissions, we should get next if necessary.