Implement 'submit'

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

Author: skygragon

lib/commands/show.js

@@ -40,7 +40,7 @@ cmd.handler = function(argv) {
       if (!template)
         return console.log('Failed to generate source file: unknown language', argv.lang);
 
-      var f = problem.key + h.fileExt(argv.lang);
+      var f = problem.key + h.langToExt(argv.lang);
       fs.writeFileSync(f, template.defaultCode);
       msg = sprintf('(File: %s)', f);
     }

lib/commands/submit.js

@@ -0,0 +1,52 @@
+var  util = require('util');
+
+var core = require('../core'),
+    h = require('../helper');
+
+var cmd = {
+  command: 'submit <filename>',
+  desc: 'Submit solution to leetcode.',
+  builder: {
+    filename: {
+      describe: 'Problem source file name.'
+    }
+  }
+};
+
+cmd.handler = function(argv) {
+  var keyword = h.getFilename(argv.filename);
+  core.getProblem(keyword, function(e, problem){
+    if (e) return console.log('ERROR:', e);
+
+    problem.file = argv.filename;
+
+    core.submitProblem(problem, function(e, results){
+      if (e) return console.log('ERROR:', e);
+
+      var result = results[0];
+      var ok = (result.total_correct == result.total_testcases) &&
+        (result.status_code == 10);
+      var run_ok = result.run_success;
+
+      if (!ok && run_ok) {
+        console.log(util.format('input:\n%s',   result['input']));
+      }
+
+      console.log(util.format('    %s %s',
+            h.prettyYesNo(ok),
+            h.statusToName(result.status_code)));
+      console.log(util.format('    %s %d/%d cases passed (%s)',
+            h.prettyYesNo(ok),
+            result.total_correct,
+            result.total_testcases,
+            result.status_runtime));
+
+      if (!ok && run_ok) {
+        console.log(util.format('    ✘ output:   %s', result['code_output']));
+        console.log(util.format('    ✔ expected: %s', result['expected_output']));
+      }
+    });
+  });
+}
+
+module.exports = cmd;

lib/config.js

@@ -1,9 +1,12 @@
 module.exports = {
+  // usually you don't wanna change those
   BASE_URL:     'https://leetcode.com',
   LOGIN_URL:    'https://leetcode.com/accounts/login/',
   PROBLEMS_URL: 'https://leetcode.com/problems/',
   TEST_URL:     'https://leetcode.com/problems/$key/interpret_solution/',
+  SUBMIT_URL:   'https://leetcode.com/problems/$key/submit/',
   VERIFY_URL:   'https://leetcode.com/submissions/detail/$id/check/',
 
+  // but you will want change these
   LANG:         'cpp' // others: c,csharp,golang,java,javascript,python,ruby,swift
 };

lib/core.js

@@ -39,6 +39,10 @@ core.testProblem = function(problem, cb) {
   client.testProblem(problem, cb);
 };
 
+core.submitProblem = function(problem, cb) {
+  client.submitProblem(problem, cb);
+};
+
 core.login = function(user, cb) {
   client.login(user, function(e, user){
     if (e) return cb(e);

lib/helper.js

@@ -11,7 +11,26 @@ h.prettyState = function(state) {
   };
 };
 
-h.fileExt = function(lang) {
+h.prettyYesNo = function(x) {
+  return x ? '✔' : '✘';
+};
+
+h.statusToName = function(sc) {
+  switch(sc) {
+    case 10: return 'Accepted';              break;
+    case 11: return 'Wrong Answer';          break;
+    case 12: return 'Memory Limit Exceeded'; break;
+    case 13: return 'Output Limit Exceeded'; break;
+    case 14: return 'Time Limit Exceeded';   break;
+    case 15: return 'Runtime Error';         break;
+    case 16: return 'Internal Error';        break;
+    case 20: return 'Compile Error';         break;
+    case 21: return 'Unknown Error';         break;
+    default: return 'Unknown';               break;
+  }
+};
+
+h.langToExt = function(lang) {
   switch(lang) {
     case 'c':          return '.c';     break;
     case 'cpp':        return '.cpp';   break;
@@ -26,7 +45,7 @@ h.fileExt = function(lang) {
   }
 };
 
-h.fileLang = function(fullpath) {
+h.extToLang = function(fullpath) {
   var ext = path.extname(fullpath);
   switch(ext) {
     case '.c':     return 'c';          break;

lib/leetcode_client.js

@@ -11,7 +11,7 @@ function makeOpts(url) {
   if (core.isLogin()) {
     var user = core.getUser();
     opts.headers['Cookie'] = 'PHPSESSID='+user.session_id+';csrftoken='+user.session_csrf+';';
-    //opts.headers['X-CSRFToken'] = user.session_csrf;
+    opts.headers['X-CSRFToken'] = user.session_csrf;
   }
   return opts;
 }
@@ -132,7 +132,7 @@ leetcode_client.testProblem = function(problem, cb) {
   opts.json = true;
   opts.body = {
     data_input:  problem.testcase,
-    lang:        h.fileLang(problem.file),
+    lang:        h.extToLang(problem.file),
     question_id: parseInt(problem.id),
     test_mode:   false,
     typed_code:  h.fileData(problem.file)
@@ -153,4 +153,31 @@ leetcode_client.testProblem = function(problem, cb) {
   });
 };
 
+leetcode_client.submitProblem = function(problem, cb) {
+  var opts = makeOpts();
+  opts.url = config.SUBMIT_URL.replace('$key', problem.key);
+  opts.headers['Origin'] = config.BASE_URL;
+  opts.headers['Referer'] = problem.link;
+  opts.headers['X-Requested-With'] = 'XMLHttpRequest';
+  opts.json = true;
+  opts.body = {
+    judge_type:  'large',
+    lang:        h.extToLang(problem.file),
+    question_id: parseInt(problem.id),
+    test_mode:   false,
+    typed_code:  h.fileData(problem.file)
+  };
+
+  request.post(opts, function(e, resp, body){
+    if (e) return cb(e);
+    if (resp.statusCode != 200) return cb('HTTP failed:' + resp.statusCode);
+
+    opts.json = false;
+    opts.body = null;
+
+    var jobs = [ { name: 'Your', id: body.submission_id } ];
+    verify_test(opts, jobs, [], cb);
+  });
+};
+
 module.exports = leetcode_client;