Implement 'test -i -t'

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

Author: skygragon

lib/commands/test.js

@@ -0,0 +1,63 @@
+var  util = require('util'),
+     _ = require('underscore');
+
+var core = require('../core'),
+    h = require('../helper');
+
+var cmd = {
+  command: 'test <filename> [--testcase|-t] [-i]',
+  desc: 'Run test case to leetcode.',
+  builder: {
+    filename: {
+      describe: 'Problem source file name.'
+    },
+    testcase: {
+      alias: 't',
+      type: 'string',
+      describe: 'Provide test case.'
+    },
+    i: {
+      type: 'boolean',
+      describe: 'Provide test case interactively.'
+    }
+  }
+};
+
+function prettyLine(actual, expected, key) {
+  var sym = '✔';
+  if (!actual.run_success) { sym = '✘'; }
+  else if (expected && key != 'status_runtime' && !_.isEqual(actual[key], expected[key])) { sym = '✘'; }
+  return util.format('    %s %s: %s', sym, key.split('_').pop(), actual[key]);
+}
+
+cmd.handler = function(argv) {
+  var testcase = argv.testcase;
+  if (argv.i) {
+    testcase = h.readStdin();
+  }
+
+  var keyword = h.getFilename(argv.filename);
+  core.getProblem(keyword, function(e, problem){
+    if (e) return console.log('ERROR:', e);
+
+    problem.file = argv.filename;
+    problem.testcase = testcase.replace('\\n', '\n');
+
+    console.log('\nInput data:');
+    console.log(problem.testcase);
+
+    core.testProblem(problem, function(e, results){
+      if (e) return console.log('ERROR:', e);
+
+      for (var i=0; i< results.length; ++i) {
+        console.log();
+        console.log(results[i].name);
+        console.log(prettyLine(results[i], results[i+1], 'status_runtime'));
+        console.log(prettyLine(results[i], results[i+1], 'code_answer'));
+        console.log(prettyLine(results[i], results[i+1], 'code_output'));
+      }
+    });
+  });
+}
+
+module.exports = cmd;

lib/config.js

@@ -2,5 +2,8 @@ module.exports = {
   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/',
+  VERIFY_URL:   'https://leetcode.com/submissions/detail/$id/check/',
+
   LANG:         'cpp' // others: c,csharp,golang,java,javascript,python,ruby,swift
 };

lib/core.js

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

lib/helper.js

@@ -1,3 +1,6 @@
+var fs = require('fs'),
+    path = require('path');
+
 var h = {};
 
 h.prettyState = function(state) {
@@ -23,6 +26,35 @@ h.fileExt = function(lang) {
   }
 };
 
+h.fileLang = function(fullpath) {
+  var ext = path.extname(fullpath);
+  switch(ext) {
+    case '.c':     return 'c';          break;
+    case '.cpp':   return 'cpp';        break;
+    case '.cs':    return 'csharp';     break;
+    case '.go':    return 'golang';     break;
+    case '.java':  return 'java';       break;
+    case '.js':    return 'javascript'; break;
+    case '.py':    return 'python';     break;
+    case '.rb':    return 'ruby';       break;
+    case '.swift': return 'swift';      break;
+    default:       return 'unknown';    break;
+  }
+};
+
+h.fileData = function(fullpath) {
+  return fs.readFileSync(fullpath).toString();
+};
+
+h.getFilename = function(fullpath) {
+  return path.basename(fullpath, path.extname(fullpath));
+};
+
+h.readStdin = function() {
+  // FIXME: not work for win32
+  return fs.readFileSync('/dev/stdin').toString();
+};
+
 h.getSetCookieValue = function(resp, key) {
   var cookies = resp.headers['set-cookie'];
   if (!cookies) return null;

lib/leetcode_client.js

@@ -11,6 +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;
   }
   return opts;
 }
@@ -102,4 +103,54 @@ leetcode_client.login = function(user, cb) {
   });
 };
 
+function verify_test(opts, jobs, results, cb) {
+  if (jobs.length == 0)
+    return cb(null, results);
+
+  opts.url = config.VERIFY_URL.replace('$id', jobs[0].id);
+  request.get(opts, function(e, resp, body){
+    if (e) return cb(e);
+    if (resp.statusCode != 200) return cb('HTTP failed:' + resp.statusCode);
+
+    var result = JSON.parse(body);
+    if (result.state == 'SUCCESS') {
+      result.name = jobs[0].name;
+      results.push(result);
+      jobs.shift();
+    }
+
+    setImmediate(verify_test, opts, jobs, results, cb);
+  });
+}
+
+leetcode_client.testProblem = function(problem, cb) {
+  var opts = makeOpts();
+  opts.url = config.TEST_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 = {
+    data_input:  problem.testcase,
+    lang:        h.fileLang(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.interpret_id },
+      { name: 'Expected', id: body.interpret_expected_id }
+    ];
+    verify_test(opts, jobs, [], cb);
+  });
+};
+
 module.exports = leetcode_client;