Change to new problem list API

Author: aaaahern

lib/commands/show.js

@@ -27,7 +27,7 @@ var cmd = {
 };
 
 cmd.handler = function(argv) {
-  core.getProblem(argv.keyword, function(e, problem) {
+  core.getProblem(argv.keyword.toString(), function(e, problem) {
     if (e) return console.log('ERROR:', e);
 
     var msg = '';

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/',
   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/leetcode_client.js

@@ -24,31 +24,36 @@ 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 problemsListJson = JSON.parse(body);
+    var problems = [];
+    problemsListJson.stat_status_pairs.forEach(function(problemData) {
       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()
+        state:   problemData['status'],
+        id:      problemData['stat']['question_id'].toString(),
+        name:    problemData['stat']['question__title'],
+        key:     problemData['stat']['question__title_slug'],
+        link:    config.PROBLEM_URL + problemData['stat']['question__title_slug'],
+        locked:  problemData['paid_only'],
+        level:   parseDifficulty(problemData['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();
+      var percent = parseInt(problemData['stat']['total_acs'], 10) / parseInt(problemData['stat']['total_submitted'], 10) * 100;
+      problem.percent = percent.toFixed(1).toString() + '%';
+      problems.push(problem);
+    });
 
     return cb(null, problems);
   });
 };
 
+function parseDifficulty(level) {
+  switch (level) {
+    case 1 : return 'Easy';
+    case 2 : return 'Medium';
+    case 3 : return 'Hard';
+    default : return 'Unknown Difficulty';
+  }
+}
+
 // hacking ;P
 var aceCtrl = {
   init: function() {