Download non-ac submission if necessary

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

Author: skygragon

lib/commands/submission.js

@@ -29,8 +29,9 @@ var cmd = {
 function getSubmissionDone(e, msg, problem, cb) {
   // NOTE: msg color means different purpose:
   // - red: error
-  // - yellow: accepted, new download
-  // - white: existed already, no download
+  // - green: accepted, fresh download
+  // - yellow: not ac-ed, fresh download
+  // - white: existed already, skip download
   console.log(sprintf('[%3d] %-60s %s',
         problem.id,
         problem.name,
@@ -71,7 +72,11 @@ function getSubmission(argv, problem, cb) {
       if (e) return done(e);
 
       fs.writeFileSync(filename, submission.code);
-      done(null, chalk.yellow.underline(filename));
+
+      if (submission.state === 'Accepted')
+        done(null, chalk.green.underline(filename));
+      else
+        done(null, chalk.yellow.underline(filename));
     });
   });
 }
@@ -84,7 +89,7 @@ cmd.handler = function(argv) {
       if (e) return console.log('ERROR:', e);
 
       problems = problems.filter(function(q) {
-        return q.state === 'ac';
+        return q.state === 'ac' || q.state === 'notac';
       });
 
       queue.run(problems, doTask);