Merge pull request #50 from fanzhenya/master

Add beat ratio display after AC

Author: skygragon

lib/commands/submit.js

@@ -54,6 +54,28 @@ cmd.handler = function(argv) {
           result.status_runtime);
       log.info(INDENT + h.prettyText(line, ok));
 
+      // show beat ratio
+      if (ok) {
+        core.getSubmission({id: result.id}, function(e, submission) {
+          if (submission.distributionChart) {
+            var beatRatio = 0.0;
+            var chart = submission.distributionChart.distribution;
+            var lang = submission.distributionChart.lang;
+            for (var i in chart) {
+              if (parseFloat(chart[i][0]) <= parseFloat(result.display_runtime)) {
+                continue;
+              } else {
+                beatRatio += parseFloat(chart[i][1]);
+              }
+            }
+            line = util.format(' Your runtime beats %d %% of %s submissions',
+                beatRatio.toFixed(2),
+                lang);
+            log.info(INDENT + h.prettyText(line, ok));
+          }
+        });
+      }
+
       // show testcase
       var testcase = result.input || result.last_testcase;
       if (!ok && testcase) {

lib/leetcode_client.js

@@ -209,6 +209,10 @@ leetcodeClient.getSubmission = function(submission, cb) {
     if (re) {
       submission.code = eval(re[1]);
     }
+    re = body.match(/distribution_formatted:\s('[^']*')/);
+    if (re) {
+      submission.distributionChart = JSON.parse(eval(re[1]));
+    }
     return cb(null, submission);
   });
 };
@@ -273,6 +277,7 @@ function verifyResult(opts, jobs, results, cb) {
     var result = JSON.parse(body);
     if (result.state === 'SUCCESS') {
       result.name = jobs[0].name;
+      result.id = jobs[0].id;
       results.push(result);
       jobs.shift();
     }

test/test_leetcode_client.js

@@ -280,8 +280,8 @@ describe('leetcode_client', function() {
         assert.equal(e, null);
         assert.deepEqual(results,
           [
-            {name: 'Your', state: 'SUCCESS'},
-            {name: 'Expected', state: 'SUCCESS'}
+            {id: 'id2', name: 'Your', state: 'SUCCESS'},
+            {id: 'id1', name: 'Expected', state: 'SUCCESS'}
           ]);
         done();
       });
@@ -311,7 +311,7 @@ describe('leetcode_client', function() {
 
       client.submitProblem(PROBLEM, function(e, results) {
         assert.equal(e, null);
-        assert.deepEqual(results, [{name: 'Your', state: 'SUCCESS'}]);
+        assert.deepEqual(results, [{id: 'id1', name: 'Your', state: 'SUCCESS'}]);
         done();
       });
     });
@@ -335,7 +335,7 @@ describe('leetcode_client', function() {
 
       client.submitProblem(PROBLEM, function(e, results) {
         assert.equal(e, null);
-        assert.deepEqual(results, [{name: 'Your', state: 'SUCCESS'}]);
+        assert.deepEqual(results, [{id: 'id1', name: 'Your', state: 'SUCCESS'}]);
         done();
       });
     });