Do our best to guess the language in git init

Author: gkoberger

api.js

@@ -87,10 +87,12 @@ exports.api = function(args, opts) {
 };
 
 exports.load = function(action) {
-  //try {
-    return require(path.join(__dirname, 'lib', `${action}.js`));
-  //} catch(e) {
-    //console.log('Action not found');
-  //}
+  var file = path.join(__dirname, 'lib', `${action}.js`);
+  if(utils.fileExists(file)) {
+    return require(file);
+  }
+
+  console.log('Action not found.'.red);
+  console.log('Type ' + 'api help'.yellow + ' to see all commands');
 };
 

lib/init.js

@@ -74,7 +74,7 @@ exports.run = function(config, info) {
       choices: types,
       validate: function (value) {
         var pass = /.(json|yaml|yml)$/.test(value);
-        var doesntExist = !fileExists(value);
+        var doesntExist = !utils.fileExists(value);
 
         if (pass && doesntExist) {
           return true;
@@ -87,37 +87,7 @@ exports.run = function(config, info) {
           return 'This file already exists';
         }
       }
-    },
-    {
-      type: 'list',
-      name: 'language',
-      message: 'API primary language',
-      choices: [
-        'JavaScript',
-        'Ruby',
-        'Python',
-        'CoffeeScript',
-        'PHP',
-        'Java',
-        'Go',
-        'Other',
-      ],
-      validate: function (value) {
-        var pass = /.(json|yaml|yml)$/.test(value);
-        var doesntExist = !fileExists(value);
-
-        if (pass && doesntExist) {
-          return true;
-        }
-
-        if(!pass) {
-          return 'Your file must end with .json or .yaml';
-        }
-        if(!doesntExist) {
-          return 'This file already exists';
-        }
-      }
-    },
+    }
   ];
 
   inquirer.prompt(questions).then(function (answers) {
@@ -164,7 +134,7 @@ exports.run = function(config, info) {
     console.log("following syntax in a comment above the code:");
     console.log("");
 
-    console.log(utils.swaggerInlineExample(answers.language));
+    console.log(utils.swaggerInlineExample(utils.guessLanguage()));
 
     console.log('');
     console.log('For more information on this syntax, see https://github.com/readmeio/swagger-inline');
@@ -187,7 +157,7 @@ function writeFile(output, swagger) {
 
 function getDefaultSwagger() {
   var i = 0;
-  while(file = fileExists(_file(i))) {
+  while(file = utils.fileExists(_file(i))) {
     i++;
   }
   return _file(i);
@@ -197,10 +167,3 @@ function getDefaultSwagger() {
   }
 }
 
-function fileExists(file) {
-  try {
-    return fs.statSync(file).isFile();
-  } catch (err) {
-    return false;
-  }
-}

package.json

@@ -23,6 +23,7 @@
     "base-store": "^0.4.4",
     "colors": "^1.1.2",
     "git-utils": "^4.1.2",
+    "glob": "^7.1.0",
     "inquirer": "^1.2.1",
     "json2yaml": "^1.1.0",
     "jsonfile": "^2.3.1",

utils.js

@@ -1,6 +1,7 @@
 var fs = require('fs');
 var os = require('os');
 var path = require('path');
+var glob = require("glob")
 
 var _ = require('lodash');
 var git = require('git-utils');
@@ -85,6 +86,48 @@ exports.addId = function(file, id) {
   return true;
 };
 
+exports.fileExists = function(file) {
+  try {
+    return fs.statSync(file).isFile();
+  } catch (err) {
+    return false;
+  }
+};
+
+exports.guessLanguage = function(cb) {
+  // Really simple way at guessing the language.
+  // If we're wrong, it's not a big deal... and
+  // way better than asking them what language
+  // they're writing (since the UI was confusing).
+
+  var language = 'js';
+  var languages = {
+    rb: 0,
+    coffee: 0,
+    py: 0,
+    js: 0,
+    java: 0,
+    php: 0,
+    go: 0
+  };
+
+  var files = glob.sync("*");
+  _.each(files, function(f) {
+    var ext = f.split('.').slice(-1)[0];
+    if(typeof languages[ext] !== 'undefined') {
+      languages[ext]++;
+    }
+  });
+
+  _.each(languages, function(i, l) {
+    if(i > languages[language]) {
+      language = l;
+    }
+  });
+
+  return language;
+};
+
 exports.swaggerInlineExample = function(_lang) {
   var prefix = '    ';
 
@@ -99,12 +142,12 @@ exports.swaggerInlineExample = function(_lang) {
   ];
 
   var languages = {
-    'javascript': ['/*', ' * ', '*/', 'route.get("/pet/:petId", pet.show);'],
+    'js': ['/*', ' * ', '*/', 'route.get("/pet/:petId", pet.show);'],
     'java': ['/*', ' * ', '*/', 'public String getPet(id) {'],
     'php': ['/*', ' * ', '*/', 'function showPet($id) {'],
-    'coffeescript': ['###', '', '###', "route.get '/pet/:petId', pet.show"],
-    'ruby': ['=begin', '', '=end', "get '/pet/:petId' do"],
-    'python': ['"""', '', '"""', "def getPet(id):"],
+    'coffee': ['###', '', '###', "route.get '/pet/:petId', pet.show"],
+    'rb': ['=begin', '', '=end', "get '/pet/:petId' do"],
+    'py': ['"""', '', '"""', "def getPet(id):"],
     'go': ['/*', ' * ', '*/', 'func getPet(id) {'],
   };