Add 'submission -a'

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

Author: skygragon

lib/commands/submission.js

@@ -6,44 +6,78 @@ var sprintf = require('sprintf-js').sprintf;
 
 var core = require('../core');
 var h = require('../helper');
+var queue = require('../queue');
 
 var cmd = {
-  command: 'submission <keyword>',
+  command: 'submission [keyword]',
   desc:    'Retrieve earlier submission by name or index',
   builder: {
-    // TODO: retrieve all?? That would be very time costing...
+    all: {
+      alias:    'a',
+      type:     'boolean',
+      describe: 'Retrieve for all problems'
+    }
   }
 };
 
-cmd.handler = function(argv) {
-  core.getProblem(argv.keyword, function(e, problem) {
-    if (e) return console.log('ERROR:', e);
+function getSubmissionDone(e, msg, problem, cb) {
+  console.log(sprintf('[%3d] %-60s %s',
+        problem.id,
+        problem.name,
+        (e ? chalk.red('ERROR: ' + e) : chalk.yellow.underline(msg))
+        ));
+  if (cb) cb(e);
+}
 
-    core.getSubmissions(problem, function(e, submissions) {
-      if (e) return console.log('ERROR:', e);
+function getSubmission(problem, cb) {
+  var done = _.partial(getSubmissionDone, _, _, problem, cb);
 
-      console.log('Total: %s submissions', chalk.yellow(submissions.length));
+  core.getSubmissions(problem, function(e, submissions) {
+    if (e) return done(e);
+    if (submissions.length === 0) return done('no submissions?');
 
-      // Find the latest accepted one
-      var submission = _.find(submissions, function(x) {
-        // TODO: select by lang? or always select the latest one?
-        return x.state === 'Accepted';
-      });
+    // find the latest accepted one
+    var submission = _.find(submissions, function(x) {
+      // TODO: select by lang?
+      return x.state === 'Accepted';
+    });
 
-      if (!submission) {
-        console.log('No Accepted found?');
-        return;
-      }
+    // if no accepted, use the latest non-accepted one
+    submission = submission || submissions[0];
 
-      core.getSubmission(submission, function(e, submission) {
-        if (e) return console.log('ERROR:', e);
+    core.getSubmission(submission, function(e, submission) {
+      if (e) return done(e);
 
-        var f = problem.key + '.' + submission.id + h.langToExt(submission.lang);
-        fs.writeFileSync(f, submission.code);
+      var f = sprintf('%s.%s.%s%s', problem.key, submission.id,
+                      problem.state, h.langToExt(submission.lang));
+      fs.writeFileSync(f, submission.code);
 
-        console.log(sprintf('Saved: %s', chalk.yellow.underline(f)));
+      done(null, f);
+    });
+  });
+}
+
+cmd.handler = function(argv) {
+  if (argv.all) {
+    core.getProblems(function(e, problems) {
+      if (e) return console.log('ERROR:', e);
+
+      problems = problems.filter(function(q) {
+        return q.state === 'ac';
       });
+
+      queue.run(problems, getSubmission);
     });
+    return;
+  }
+
+  if (!argv.keyword)
+    return console.log('ERROR: missing keyword?');
+
+  core.getProblem(argv.keyword, function(e, problem) {
+    if (e) return console.log('ERROR:', e);
+
+    queue.run([problem], getSubmission);
   });
 };
 

lib/config.js

@@ -15,9 +15,10 @@ var DEFAULT_CONFIG = {
   URL_VERIFY:      'https://leetcode.com/submissions/detail/$id/check/',
 
   // but you will want change these
-  LANG:       'cpp', // avail: [c,cpp,csharp,golang,java,javascript,python,ruby,swift]
-  USE_COLOR:  true,
-  AUTO_LOGIN: false
+  LANG:        'cpp', // avail: [c,cpp,csharp,golang,java,javascript,python,ruby,swift]
+  USE_COLOR:   true,
+  AUTO_LOGIN:  false,
+  MAX_WORKERS: 10
 };
 
 function Config() {}

lib/leetcode_client.js

@@ -158,7 +158,8 @@ leetcodeClient.getSubmissions = function(problem, cb) {
           lang:    $(tds[4]).text()
         };
 
-        submission.id = _.last(_.compact(submission.path.split('/')));
+        if (submission.path)
+          submission.id = _.last(_.compact(submission.path.split('/')));
 
         return submission;
       }).get();

lib/queue.js

@@ -0,0 +1,33 @@
+var config = require('./config');
+
+var queue = {};
+
+function startWorker(ctx) {
+  // no more tasks, quit now
+  if (ctx.tasks.length === 0) {
+    if (--ctx.workers === 0 && ctx.cb)
+      ctx.cb();
+    return;
+  }
+
+  var task = ctx.tasks.shift();
+  ctx.doTask(task, function(e) {
+    // TODO: could retry failed task here.
+    startWorker(ctx);
+  });
+}
+
+queue.run = function(tasks, doTask, cb) {
+  var ctx = {
+    tasks:   tasks,
+    doTask:  doTask,
+    cb:      cb,
+    workers: config.MAX_WORKERS
+  };
+
+  for (var i = 0; i < ctx.workers; ++i) {
+    startWorker(ctx);
+  }
+};
+
+module.exports = queue;