Update stat.ac when submit succeed.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

lib/commands/stat.js

@@ -43,7 +43,8 @@ var cmd = {
       .example(chalk.yellow('leetcode stat'), 'Show progress status')
       .example(chalk.yellow('leetcode stat --no-lock'), 'Show progress status without locked questions')
       .example(chalk.yellow('leetcode stat -t algorithms'), 'Show progress status of algorithms questions')
-      .example(chalk.yellow('leetcode stat -g'), 'Show detailed status in graph');
+      .example(chalk.yellow('leetcode stat -g'), 'Show detailed status in graph')
+      .example(chalk.yellow('leetcode stat -c'), 'Show accepted status in calendar');
   }
 };
 

lib/commands/submit.js

@@ -44,7 +44,6 @@ function printLine() {
 
 cmd.handler = function(argv) {
   session.argv = argv;
-
   if (!fs.existsSync(argv.filename))
     return log.error('File ' + argv.filename + ' not exist!');
 
@@ -66,8 +65,8 @@ cmd.handler = function(argv) {
       printLine(result, '%d/%d cases passed (%s)',
           result.passed, result.total, result.runtime);
 
-      // show beat ratio
       if (result.ok) {
+        session.updateStat('ac', 1);
         core.getSubmission({id: result.id}, function(e, submission) {
           if (e || !submission || !submission.distributionChart)
             return log.warn('Failed to get submission beat ratio.');

lib/session.js

@@ -1,3 +1,4 @@
+var moment = require('moment');
 var _ = require('underscore');
 
 var cache = require('./cache');
@@ -32,4 +33,14 @@ session.isLogin = function() {
   return this.getUser() !== null;
 };
 
+session.updateStat = function(k, v) {
+  // TODO: use other storage if too many stat data
+  var today = moment().format('YYYY-MM-DD');
+  var stats = cache.get(h.KEYS.stat) || {};
+  var stat = stats[today] = stats[today] || {};
+  stat[k] = stat[k] || 0;
+  stat[k] += v;
+  cache.set(h.KEYS.stat, stats);
+};
+
 module.exports = session;