Show reasonable AC count in stat -c.

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

Author: skygragon

lib/commands/stat.js

@@ -150,7 +150,7 @@ function showCal() {
   const graph = [];
   const stats = require('../cache').get(h.KEYS.stat) || {};
   for (let k of _.keys(stats)) {
-    const v = stats[k].ac || 0;
+    const v = (stats[k]['ac.set'] || []).length;
     if (v === 0) continue;
 
     const d = moment(k, 'YYYY-MM-DD');

lib/commands/submit.js

@@ -66,6 +66,7 @@ cmd.handler = function(argv) {
 
       if (result.ok) {
         session.updateStat('ac', 1);
+        session.updateStat('ac.set', problem.id);
         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

@@ -39,8 +39,15 @@ session.updateStat = function(k, v) {
   const today = moment().format('YYYY-MM-DD');
   const stats = cache.get(h.KEYS.stat) || {};
   const stat = stats[today] = stats[today] || {};
-  stat[k] = stat[k] || 0;
-  stat[k] += v;
+
+  if (k.endsWith('.set')) {
+    const s = new Set(stat[k] || []);
+    s.add(v);
+    stat[k] = Array.from(s);
+  } else {
+    stat[k] = (stat[k] || 0) + v;
+  }
+
   cache.set(h.KEYS.stat, stats);
 };
 

test/test_session.js

@@ -23,18 +23,18 @@ describe('session', function() {
     session.__set__('moment', moment);
   });
 
+  beforeEach(function() {
+    stats = null;
+  });
+
   describe('#updateStat', function() {
-    it('should ok', function() {
+    it('should update number ok', function() {
       now = '2017.12.13';
       session.updateStat('ac', 10);
-      assert.deepEqual(stats, {
-        '2017.12.13': {ac: 10}
-      });
+      assert.deepEqual(stats, {'2017.12.13': {ac: 10}});
 
       session.updateStat('ac', 20);
-      assert.deepEqual(stats, {
-        '2017.12.13': {ac: 30}
-      });
+      assert.deepEqual(stats, {'2017.12.13': {ac: 30}});
 
       now = '2017.12.14';
       session.updateStat('ac', 40);
@@ -43,5 +43,15 @@ describe('session', function() {
         '2017.12.14': {ac: 40}
       });
     });
+
+    it('should update set ok', function() {
+      now = '2017.12.13';
+      session.updateStat('ac.set', 101);
+      assert.deepEqual(stats, {'2017.12.13': {'ac.set': [101]}});
+      session.updateStat('ac.set', 100);
+      assert.deepEqual(stats, {'2017.12.13': {'ac.set': [101, 100]}});
+      session.updateStat('ac.set', 101);
+      assert.deepEqual(stats, {'2017.12.13': {'ac.set': [101, 100]}});
+    });
   });
 });