Update list command to fit recent leetcode change

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

Author: skygragon

lib/commands/list.js

@@ -79,11 +79,13 @@ cmd.handler = function(argv) {
       stat[problem.state] = (stat[problem.state] || 0) + 1;
       if (problem.locked) ++stat.locked;
 
-      console.log(sprintf('%s %s [%3d] %-60s %-6s (%s)',
+      console.log(sprintf('%s %s [%3d] %-60s %-6s (%.2f %%)',
             h.prettyState(problem.state),
             (problem.locked ? '🔒' : ' '),
-            problem.id, problem.name,
-            problem.level, problem.percent));
+            problem.id,
+            problem.name,
+            problem.level,
+            problem.percent));
     });
 
     if (argv.stat) {

lib/config.js

@@ -7,7 +7,8 @@ var defaultConfig = {
   // 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/',
+  PROBLEMS_URL: 'https://leetcode.com/api/problems/algorithms/',
+  PROBLEM_URL:  'https://leetcode.com/problems/$id',
   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/',

lib/helper.js

@@ -16,6 +16,15 @@ h.prettyYesNo = function(x) {
   return x ? '✔' : '✘';
 };
 
+h.levelToName = function(level) {
+  switch (level) {
+    case 1:  return 'Easy';
+    case 2:  return 'Medium';
+    case 3:  return 'Hard';
+    default: return ' ';
+  }
+};
+
 h.statusToName = function(sc) {
   switch (sc) {
     case 10: return 'Accepted';

lib/leetcode_client.js

@@ -1,4 +1,3 @@
-var _ = require('underscore');
 var cheerio = require('cheerio');
 var request = require('request');
 
@@ -24,26 +23,18 @@ leetcodeClient.getProblems = function(cb) {
     if (e) return cb(e);
     if (resp.statusCode !== 200) return cb('HTTP failed:' + resp.statusCode);
 
-    var $ = cheerio.load(body);
-    var problems = $('#problemList tbody tr').map(function() {
-      var tds = $(this).children();
-      var problem = {
-        state:   $(tds[0]).children('span').attr('class'),
-        id:      $(tds[1]).text(),
-        name:    $(tds[2]).children('a').text(),
-        link:    $(tds[2]).children('a').attr('href'),
-        locked:  ($(tds[2]).children('i[class="fa fa-lock"]').length === 1),
-        percent: $(tds[3]).text(),
-        level:   $(tds[6]).text()
+    var problems = JSON.parse(body).stat_status_pairs.map(function(p) {
+      return {
+        state:   p.status || 'None',
+        id:      p.stat.question_id,
+        name:    p.stat.question__title,
+        key:     p.stat.question__title_slug,
+        link:    config.PROBLEM_URL.replace('$id', p.stat.question__title_slug),
+        locked:  p.paid_only,
+        percent: p.stat.total_acs * 100 / p.stat.total_submitted,
+        level:   h.levelToName(p.difficulty.level)
       };
-
-      // fixup problem attributes
-      problem.id = parseInt(problem.id, 10);
-      problem.key = _.last(_.compact(problem.link.split('/')));
-      problem.link = config.BASE_URL + problem.link;
-
-      return problem;
-    }).get();
+    });
 
     return cb(null, problems);
   });