Add 'submission' command.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

lib/commands/submission.js

@@ -0,0 +1,50 @@
+var _ = require('underscore');
+var chalk = require('chalk');
+var fs = require('fs');
+
+var sprintf = require('sprintf-js').sprintf;
+
+var core = require('../core');
+var h = require('../helper');
+
+var cmd = {
+  command: 'submission <keyword>',
+  desc:    'Retrieve earlier submission by name or index',
+  builder: {
+    // TODO: retrieve all?? That would be very time costing...
+  }
+};
+
+cmd.handler = function(argv) {
+  core.getProblem(argv.keyword, function(e, problem) {
+    if (e) return console.log('ERROR:', e);
+
+    core.getSubmissions(problem, function(e, submissions) {
+      if (e) return console.log('ERROR:', e);
+
+      console.log('Total: %s submissions', chalk.yellow(submissions.length));
+
+      // 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';
+      });
+
+      if (!submission) {
+        console.log('No Accepted found?');
+        return;
+      }
+
+      core.getSubmission(submission, function(e, submission) {
+        if (e) return console.log('ERROR:', e);
+
+        var f = problem.key + '.' + submission.id + h.langToExt(submission.lang);
+        fs.writeFileSync(f, submission.code);
+
+        console.log(sprintf('Saved: %s', chalk.yellow.underline(f)));
+      });
+    });
+  });
+};
+
+module.exports = cmd;

lib/commands/version.js

@@ -36,7 +36,7 @@ cmd.handler = function(argv) {
   var sprintf = require('sprintf-js').sprintf;
   console.log('\n[Configuration]');
   Object.getOwnPropertyNames(config).sort().forEach(function(k) {
-    console.log(sprintf('%-13s %s', k + ':', config[k]));
+    console.log(sprintf('%-16s %s', k + ':', config[k]));
   });
 };
 

lib/config.js

@@ -4,13 +4,15 @@ var h = require('./helper');
 
 var DEFAULT_CONFIG = {
   // usually you don't wanna change those
-  URL_BASE:     'https://leetcode.com',
-  URL_LOGIN:    'https://leetcode.com/accounts/login/',
-  URL_PROBLEMS: 'https://leetcode.com/api/problems/algorithms/',
-  URL_PROBLEM:  'https://leetcode.com/problems/$id',
-  URL_TEST:     'https://leetcode.com/problems/$key/interpret_solution/',
-  URL_SUBMIT:   'https://leetcode.com/problems/$key/submit/',
-  URL_VERIFY:   'https://leetcode.com/submissions/detail/$id/check/',
+  URL_BASE:        'https://leetcode.com',
+  URL_LOGIN:       'https://leetcode.com/accounts/login/',
+  URL_PROBLEMS:    'https://leetcode.com/api/problems/algorithms/',
+  URL_PROBLEM:     'https://leetcode.com/problems/$id',
+  URL_TEST:        'https://leetcode.com/problems/$key/interpret_solution/',
+  URL_SUBMIT:      'https://leetcode.com/problems/$key/submit/',
+  URL_SUBMISSIONS: 'https://leetcode.com/problems/$key/submissions/',
+  URL_SUBMISSION:  'https://leetcode.com/submissions/detail/$id/',
+  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]

lib/core.js

@@ -42,6 +42,14 @@ core.getProblem = function(keyword, cb) {
   });
 };
 
+core.getSubmissions = function(problem, cb) {
+  client.getSubmissions(problem, cb);
+};
+
+core.getSubmission = function(submission, cb) {
+  client.getSubmission(submission, cb);
+};
+
 core.testProblem = function(problem, cb) {
   client.testProblem(problem, cb);
 };

lib/leetcode_client.js

@@ -1,3 +1,4 @@
+var _ = require('underscore');
 var cheerio = require('cheerio');
 var request = require('request');
 
@@ -137,6 +138,51 @@ leetcodeClient.getProblem = function(problem, cb) {
   });
 };
 
+leetcodeClient.getSubmissions = function(problem, cb) {
+  var opts = makeOpts();
+  opts.url = config.URL_SUBMISSIONS.replace('$key', problem.key);
+
+  request(opts, function(e, resp, body) {
+    e = checkError(e, resp, 200);
+    // FIXME: if session expired, this will still return 200
+    if (e) return cb(e);
+
+    var $ = cheerio.load(body);
+    var submissions = $('table[id="result-testcases"] tbody tr')
+      .map(function() {
+        var tds = $(this).children();
+        var submission = {
+          path:    $(tds[2]).children('a').attr('href'),
+          state:   $(tds[2]).children('a').children('strong').text(),
+          runtime: $(tds[3]).text().trim(),
+          lang:    $(tds[4]).text()
+        };
+
+        submission.id = _.last(_.compact(submission.path.split('/')));
+
+        return submission;
+      }).get();
+
+    return cb(null, submissions);
+  });
+};
+
+leetcodeClient.getSubmission = function(submission, cb) {
+  var opts = makeOpts();
+  opts.url = config.URL_SUBMISSION.replace('$id', submission.id);
+
+  request(opts, function(e, resp, body) {
+    e = checkError(e, resp, 200);
+    if (e) return cb(e);
+
+    var re = body.match(/submissionCode:\s('[^']*')/);
+    if (re) {
+      submission.code = eval(re[1]);
+    }
+    return cb(null, submission);
+  });
+};
+
 leetcodeClient.login = function(user, cb) {
   request(config.URL_LOGIN, function(e, resp, body) {
     e = checkError(e, resp, 200);