Use level weight in stat -c.

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

Author: skygragon

lib/commands/stat.js

@@ -125,7 +125,7 @@ function showGraph(problems) {
       ' ', ICONS.none + '  Remaining');
 }
 
-function showCal() {
+function showCal(problems) {
   const MONTHS = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];
   const WEEKDAYS = ['Sun', 'Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat'];
   const ICONS = [
@@ -142,15 +142,22 @@ function showCal() {
 
   const now = moment();
 
+  const SCORES = {easy: 1, medium: 2, hard: 5};
+  function toScore(sum, id) {
+    const problem = problems.find(x => x.fid === id);
+    if (problem) sum += (SCORES[problem.level.toLowerCase()] || 1);
+    return sum;
+  }
+
   // load historical stats
   const graph = [];
   const stats = require('../cache').get(h.KEYS.stat) || {};
   for (let k of _.keys(stats)) {
-    const v = (stats[k]['ac.set'] || []).length;
-    if (v === 0) continue;
+    const score = (stats[k]['ac.set'] || []).reduce(toScore, 0);
+    if (score === 0) continue;
 
     const d = moment(k, 'YYYY-MM-DD');
-    graph[now.diff(d, 'days')] = v;
+    graph[now.diff(d, 'days')] = score;
   }
 
   // print header
@@ -208,7 +215,7 @@ cmd.handler = function(argv) {
 
     log.info();
     if (argv.graph)    showGraph(problems);
-    else if (argv.cal) showCal();
+    else if (argv.cal) showCal(problems);
     else               showProgress(problems);
     log.info();
   });