Implement 'show -g -l'

* generate source file with template

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

Author: skygragon

lib/commands/show.js

@@ -1,27 +1,55 @@
-var sprintf = require('sprintf-js').sprintf;
+var sprintf = require('sprintf-js').sprintf,
+    _ = require('underscore'),
+    fs = require('fs');
 
-var core = require('../core');
+var core = require('../core'),
+    config = require('../config'),
+    h = require('../helper');
 
 var cmd = {
-  command: 'show <keyword>',
+  command: 'show <keyword> [--gen|-g] [--lang|-l]',
   desc: 'Show problem details.',
   builder: {
     keyword: {
       describe: 'Problem keyword, e.g. name, index, or URI path.'
+    },
+    gen: {
+      alias: 'g',
+      type: 'boolean',
+      describe: 'Generate template source file.'
+    },
+    lang: {
+      alias: 'l',
+      type: 'string',
+      default: config.LANG,
+      describe: 'Language to use, used with -g.'
     }
   }
 };
 
 cmd.handler = function(argv) {
+
   core.getProblem(argv.keyword, function(e, problem){
     if (e) return console.log('ERROR:', e);
 
-    console.log(sprintf('[%d] %s\n', problem.id, problem.name));
+    var msg = '';
+    if (argv.gen) {
+      var template = _.find(problem.templates, function(x){
+        return x.value == argv.lang;
+      });
+      if (!template)
+        return console.log('Failed to generate source file: unknown language', argv.lang);
+
+      var f = problem.key + h.fileExt(argv.lang);
+      fs.writeFileSync(f, template.defaultCode);
+      msg = sprintf('(File: %s)', f);
+    }
+
+    console.log(sprintf('[%d] %s\t%s\n', problem.id, problem.name, msg));
     console.log(sprintf('%s\n', problem.link));
     console.log(sprintf('* %s (%s)', problem.level, problem.percent));
     console.log(sprintf('* Total Accepted: %d', problem.total_ac));
-    console.log(sprintf('* Total Submissions: %d', problem.total_submit));
-    console.log();
+    console.log(sprintf('* Total Submissions: %d\n', problem.total_submit));
     console.log(problem.desc);
   });
 }

lib/config.js

@@ -1,5 +1,6 @@
 module.exports = {
   BASE_URL:     'https://leetcode.com',
   LOGIN_URL:    'https://leetcode.com/accounts/login/',
-  PROBLEMS_URL: 'https://leetcode.com/problems/'
+  PROBLEMS_URL: 'https://leetcode.com/problems/',
+  LANG:         'cpp' // others: c,csharp,golang,java,javascript,python,ruby,swift
 };

lib/helper.js

@@ -6,7 +6,22 @@ h.prettyState = function(state) {
     case 'notac': return '✘'; break;
     default:      return ' '; break;
   };
-}
+};
+
+h.fileExt = function(lang) {
+  switch(lang) {
+    case 'c':          return '.c';     break;
+    case 'cpp':        return '.cpp';   break;
+    case 'csharp':     return '.cs';    break;
+    case 'golang':     return '.go';    break;
+    case 'java':       return '.java';  break;
+    case 'javascript': return '.js';    break;
+    case 'python':     return '.py';    break;
+    case 'ruby':       return '.rb';    break;
+    case 'swift':      return '.swift'; break;
+    default:           return '.raw';   break;
+  }
+};
 
 h.getSetCookieValue = function(resp, key) {
   var cookies = resp.headers['set-cookie'];
@@ -18,6 +33,6 @@ h.getSetCookieValue = function(resp, key) {
       if (kv[0] == key) return kv[1];
     }
   }
-}
+};
 
 module.exports = h;

lib/leetcode_client.js

@@ -45,6 +45,9 @@ leetcode_client.getProblems = function(cb) {
   });
 };
 
+// hacking ;P
+var aceCtrl = { init: function(){ return Array.prototype.slice.call(arguments); } };
+
 leetcode_client.getProblem = function(problem, cb) {
   request(problem.link, function(e, resp, body){
     if (e) return cb(e);
@@ -53,9 +56,14 @@ leetcode_client.getProblem = function(problem, cb) {
     var $ = cheerio.load(body);
     var info = $('div[class="question-info text-info"] ul li strong');
 
-    problem.desc = $('meta[property="og:description"]').attr('content');
     problem.total_ac = $(info[0]).text();
     problem.total_submit = $(info[1]).text();
+    problem.desc = $('meta[property="og:description"]').attr('content');
+
+    var raw = $('div[ng-controller="AceCtrl as aceCtrl"]').attr('ng-init');
+    raw = raw.replace(/\n/g,''); // FIXME: might break test cases!
+    var args = eval(raw);
+    problem.templates = args[0];
 
     return cb(null, problem);
   });