Support outdir in 'submission'

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

Author: skygragon

lib/commands/submission.js

@@ -10,17 +10,27 @@ var queue = require('../queue');
 
 var cmd = {
   command: 'submission [keyword]',
-  desc:    'Retrieve earlier submission by name or index',
+  desc:    'retrieve earlier submission by name or index',
   builder: {
     all: {
       alias:    'a',
       type:     'boolean',
       describe: 'Retrieve for all problems'
+    },
+    outdir: {
+      alias:    'o',
+      type:     'string',
+      describe: 'Where to save the submissions',
+      default:  '.'
     }
   }
 };
 
 function getSubmissionDone(e, msg, problem, cb) {
+  // NOTE: msg color means different purpose:
+  // - red: error
+  // - yellow: accepted, new download
+  // - white: existed already, no download
   console.log(sprintf('[%3d] %-60s %s',
         problem.id,
         problem.name,
@@ -29,7 +39,7 @@ function getSubmissionDone(e, msg, problem, cb) {
   if (cb) cb(e);
 }
 
-function getSubmission(problem, cb) {
+function getSubmission(argv, problem, cb) {
   var done = _.partial(getSubmissionDone, _, _, problem, cb);
 
   core.getSubmissions(problem, function(e, submissions) {
@@ -45,7 +55,8 @@ function getSubmission(problem, cb) {
     // if no accepted, use the latest non-accepted one
     submission = submission || submissions[0];
 
-    var filename = sprintf('%s.%s.%s%s',
+    var filename = sprintf('%s/%s.%s.%s%s',
+        argv.outdir,
         problem.key,
         submission.id,
         problem.state,
@@ -66,6 +77,8 @@ function getSubmission(problem, cb) {
 }
 
 cmd.handler = function(argv) {
+  var doTask = _.partial(getSubmission, argv, _, _);
+
   if (argv.all) {
     core.getProblems(function(e, problems) {
       if (e) return console.log('ERROR:', e);
@@ -74,7 +87,7 @@ cmd.handler = function(argv) {
         return q.state === 'ac';
       });
 
-      queue.run(problems, getSubmission);
+      queue.run(problems, doTask);
     });
     return;
   }
@@ -85,7 +98,7 @@ cmd.handler = function(argv) {
   core.getProblem(argv.keyword, function(e, problem) {
     if (e) return console.log('ERROR:', e);
 
-    queue.run([problem], getSubmission);
+    queue.run([problem], doTask);
   });
 };