retry.js

var _ = require('underscore');

var config = require('../config');
var log = require('../log');
var Plugin = require('../plugin');
var session = require('../session');

var plugin = new Plugin(30, 'retry', '',
    'Plugin to retry last failed request if AUTO_LOGIN is on.');

var count = {};

function canRetry(e, name) {
  return e && config.AUTO_LOGIN && (e === session.errors.EXPIRED) && count[name] < 1;
}

plugin.init = function() {
  var names = [
    'getProblems',
    'getProblem',
    'getSubmissions',
    'getSubmission',
    'getFavorites',
    'testProblem',
    'submitProblem',
    'starProblem'
  ];

  names.forEach(function(name) {
    count[name] = 0;
    plugin[name] = function() {
      var args = _.toArray(arguments);
      var cb = args.pop();

      var _cb = function() {
        var results = _.toArray(arguments);
        var e = results[0];
        if (!canRetry(e, name)) {
          count[name] = 0;
          return cb.apply(null, results);
        }

        ++count[name];
        plugin.relogin(function() {
          // for now we don't care result, just blindly retry
          plugin[name].apply(plugin, args.concat(cb));
        });
      };

      var next = plugin.next;
      next[name].apply(next, args.concat(_cb));
    };
  });
};

// leetcode.com is limiting one session alive in the same time,
// which means once you login on web, your cli session will get
// expired immediately. In that case we will try to re-login in
// the backend to give a seamless user experience.
plugin.relogin = function(cb) {
  log.debug('session expired, try to re-login...');

  var user = session.getUser();
  if (!user) {
    log.debug('relogin failed: no user found, please login again');
    return cb();
  }

  this.login(user, function(e) {
    if (e) {
      log.debug('login failed:' + e.msg);
    } else {
      log.debug('login successfully, cont\'d...');
    }
    return cb();
  });
};

module.exports = plugin;