Handle locked problems correctly.

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

Author: skygragon

lib/commands/list.js

@@ -54,20 +54,23 @@ cmd.handler = function(argv) {
       });
     }
 
-    var stat = {};
+    var stat = {locked: 0};
     problems.forEach(function(problem) {
       stat[problem.level] = (stat[problem.level] || 0) + 1;
       stat[problem.state] = (stat[problem.state] || 0) + 1;
+      if (problem.locked) ++stat.locked;
 
-      console.log(sprintf('%s [%3d] %-60s %-6s (%s)',
-            h.prettyState(problem.state), problem.id,
-            problem.name, problem.level, problem.percent));
+      console.log(sprintf('%s %s [%3d] %-60s %-6s (%s)',
+            h.prettyState(problem.state),
+            (problem.locked ? '🔒' : ' '),
+            problem.id, problem.name,
+            problem.level, problem.percent));
     });
 
     if (argv.stat) {
       console.log();
-      console.log(sprintf('  All:  %-9d Listed: %-9d',
-            all, problems.length));
+      console.log(sprintf('  All:  %-9d Listed: %-9d Lock: %-9d',
+            all, problems.length, stat.locked));
       console.log(sprintf('  AC:   %-9d Not-AC: %-9d New:  %-9d',
             (stat.ac || 0), (stat.notac || 0), (stat.None || 0)));
       console.log(sprintf('  Easy: %-9d Medium: %-9d Hard: %-9d',

lib/core.js

@@ -31,6 +31,8 @@ core.getProblem = function(keyword, cb) {
     if (cached) return cb(null, cached);
 
     client.getProblem(problem, function(e, problem) {
+      if (e) return cb(e);
+
       cache.set(problem.key, problem);
       return cb(null, problem);
     });

lib/leetcode_client.js

@@ -32,6 +32,7 @@ leetcodeClient.getProblems = function(cb) {
         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()
       };
@@ -68,6 +69,10 @@ leetcodeClient.getProblem = function(problem, cb) {
     problem.desc = $('meta[property="og:description"]').attr('content');
 
     var raw = $('div[ng-controller="AceCtrl as aceCtrl"]').attr('ng-init');
+    if (!raw)
+      return cb('failed to load' + (problem.locked ? ' locked ' : ' ') +
+                'problem!');
+
     raw = raw.replace(/\n/g, ''); // FIXME: might break test cases!
     var args = eval(raw);
     problem.templates = args[0];