C++题目生成添加头文件以及main函数

Author: luchenqun

.gitignore

@@ -39,3 +39,4 @@ jspm_packages
 tmp/
 *.swp
 .DS_Store
+package-lock.json

lib/commands/show.js

@@ -83,11 +83,11 @@ function genFileName(problem, opts) {
 
   // try to use a new filename to avoid overwrite by mistake
   let i = 0;
-  let name;
-  do {
-    name = path.join(opts.outdir, params.join('.').replace(/\.+/g, '.'));
-    params[2] = i++;
-  } while (fs.existsSync(name));
+  let name = path.join(opts.outdir, params.join('.').replace(/\.+/g, '.'));
+  // do {
+  //   name = path.join(opts.outdir, params.join('.').replace(/\.+/g, '.'));
+  //   params[2] = i++;
+  // } while (fs.existsSync(name));
   return name;
 }
 
@@ -105,11 +105,35 @@ function showProblem(problem, argv) {
   let code;
   const needcode = argv.gen || argv.codeonly;
   if (needcode) {
-    const template = problem.templates.find(x => x.value === argv.lang);
+    var template = problem.templates.find(x => x.value === argv.lang);
     if (!template) {
       log.fail('Not supported language "' + argv.lang + '"');
       log.warn('Supported languages: ' + langlist);
       return;
+    } else {
+      if(argv.lang === 'cpp'){
+        var include = `
+#include <iostream>
+#include <string>
+#include <vector>
+using namespace std;
+
+`;
+
+        template.defaultCode = include + template.defaultCode;
+
+        var main = `
+
+int main()
+{
+    Solution *s = new Solution();
+
+    delete s;
+    return 0;
+}
+        `
+        template.defaultCode += main;
+      }
     }
 
     const opts = {
@@ -124,6 +148,8 @@ function showProblem(problem, argv) {
   if (argv.gen) {
     filename = genFileName(problem, argv);
     h.mkdir(argv.outdir);
+    code = code.replace(/ \* \r\n/g, '');
+    log.debug('code ', filename, JSON.stringify(code));
     fs.writeFileSync(filename, code);
 
     if (argv.editor !== undefined) {

lib/core.js

@@ -125,11 +125,12 @@ core.exportProblem = function(problem, opts) {
   const tplfile = path.join(h.getCodeDir('templates'), opts.tpl + '.tpl');
   let output = _.template(h.getFileData(tplfile))(input);
 
-  if (h.isWindows()) {
-    output = output.replace(/\n/g, '\r\n');
-  } else {
-    output = output.replace(/\r\n/g, '\n');
-  }
+  // if (h.isWindows()) {
+  //   output = output.replace(/\n/g, '\r\n');
+  // } else {
+  //   output = output.replace(/\r\n/g, '\n');
+  // }
+  
   return output;
 };