Merge pull request #11 from wong2/star

Add command to star a problem

Author: skygragon

lib/commands/star.js

@@ -0,0 +1,20 @@
+var log = require('loglevel');
+var core = require('../core');
+var h = require('../helper');
+
+var cmd = {
+  command: 'star <keyword>',
+  desc:    'star problem by name or index'
+};
+
+cmd.handler = function(argv) {
+  core.getProblem(argv.keyword, function(e, problem) {
+    if (e) return log.fail(e);
+    core.starProblem(problem, function(e) {
+      if (e) return log.fail(e);
+      log.info(h.prettyText(' Starred', true));
+    });
+  });
+};
+
+module.exports = cmd;

lib/config.js

@@ -13,6 +13,7 @@ var DEFAULT_CONFIG = {
   URL_SUBMISSIONS: 'https://leetcode.com/problems/$key/submissions/',
   URL_SUBMISSION:  'https://leetcode.com/submissions/detail/$id/',
   URL_VERIFY:      'https://leetcode.com/submissions/detail/$id/check/',
+  URL_STAR:        'https://leetcode.com/problems/favor/',
 
   // but you will want change these
   LANG:        'cpp', // avail: [c,cpp,csharp,golang,java,javascript,python,ruby,swift]

lib/core.js

@@ -71,6 +71,10 @@ core.updateProblem = function(problem, kv) {
   return cache.set('all', problems);
 };
 
+core.starProblem = function(problem, cb) {
+  client.starProblem(problem, cb);
+};
+
 core.login = function(user, cb) {
   var self = this;
   client.login(user, function(e, user) {

lib/leetcode_client.js

@@ -315,4 +315,19 @@ leetcodeClient.submitProblem = function(problem, cb) {
   });
 };
 
+leetcodeClient.starProblem = function(problem, cb) {
+  var opts = makeOpts(config.URL_STAR);
+  opts.method = 'POST';
+  opts.headers.Origin = config.URL_BASE;
+  opts.headers.Referer = problem.link;
+  opts.headers['X-Requested-With'] = 'XMLHttpRequest';
+  opts.json = true;
+  opts.body = {'qid': problem.id};
+
+  requestWithReLogin(opts, function(e, resp, body) {
+    e = checkError(e, resp, 200);
+    cb(e);
+  });
+};
+
 module.exports = leetcodeClient;

test/test_core.js

@@ -180,6 +180,17 @@ describe('core', function() {
       });
     });
 
+    it('should starProblem ok', function(done) {
+      client.starProblem = function(problem, cb) {
+        return cb(null);
+      };
+
+      core.starProblem(PROBLEMS[0], function(e) {
+        assert.equal(e, null);
+        done();
+      });
+    });
+
     // dummy test
     it('should testProblem ok', function(done) {
       client.testProblem = function(problem, cb) {

test/test_leetcode_client.js

@@ -253,6 +253,28 @@ describe('leetcode_client', function() {
     });
   }); // #submitProblem
 
+  describe('#starProblem', function() {
+    it('should ok', function(done) {
+      var problem = {
+        id:     389,
+        name:   'Find the Difference',
+        key:    'find-the-difference',
+        link:   'https://leetcode.com/problems/find-the-difference',
+        locked: false,
+        file:   '/dev/null'
+      };
+
+      nock('https://leetcode.com')
+        .post('/problems/favor/')
+        .reply(200, '{"is_favor": true}');
+
+      client.starProblem(problem, function(e) {
+        assert.equal(e, null);
+        done();
+      });
+    });
+  }); // #starProblem
+
   describe('#getSubmissions', function() {
     it('should ok', function(done) {
       var problem = {