Update local cache after submit problem.

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

Author: skygragon

lib/commands/submit.js

@@ -48,6 +48,9 @@ cmd.handler = function(argv) {
         console.log(util.format(
               '    ✔ expected: %s', result.expected_output));
       }
+
+      // update this problem status in local cache
+      core.updateProblem(problem, {state: (ok ? 'ac' : 'notac')});
     });
   });
 };

lib/core.js

@@ -45,6 +45,19 @@ core.submitProblem = function(problem, cb) {
   client.submitProblem(problem, cb);
 };
 
+core.updateProblem = function(problem, kv) {
+  var problems = cache.get('all');
+  if (!problems) return false;
+
+  var old_problem = _.find(problems, function(x) {
+    return x.id === problem.id;
+  });
+  if (!old_problem) return false;
+
+  _.extend(old_problem, kv);
+  return cache.set('all', problems);
+};
+
 core.login = function(user, cb) {
   client.login(user, function(e, user) {
     if (e) return cb(e);