refs #10: delay a while if run code too soon

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

Author: skygragon

lib/leetcode_client.js

@@ -248,6 +248,7 @@ function runCode(opts, problem, cb) {
   opts.headers.Referer = problem.link;
   opts.headers['X-Requested-With'] = 'XMLHttpRequest';
   opts.json = true;
+  opts._delay = opts._delay || 1; // in seconds
 
   opts.body = opts.body || {};
   _.extendOwn(opts.body, {
@@ -261,6 +262,21 @@ function runCode(opts, problem, cb) {
     e = checkError(e, resp, 200);
     if (e) return cb(e);
 
+    if (body.error) {
+      if (body.error.indexOf('run code too soon') < 0)
+        return cb(body.error);
+
+      // hit 'run code too soon' error, have to wait a bit
+      console.log('\nWARN:', body.error);
+
+      // linear wait
+      ++opts._delay;
+      console.log('Will retry after %d seconds...', opts._delay);
+
+      var reRun = _.partial(runCode, opts, problem, cb);
+      return setTimeout(reRun, opts._delay * 1000);
+    }
+
     opts.json = false;
     opts.body = null;