style(gen:endpoint):

Author: Awk34

src/generators/endpoint/index.js

@@ -5,7 +5,6 @@ import {Base} from 'yeoman-generator';
 import {genNamedBase} from '../generator-base';
 
 export class Generator extends Base {
-
   constructor(...args) {
     super(...args);
 
@@ -33,69 +32,68 @@ export class Generator extends Base {
   }
 
   prompting() {
-    var promptCb = (props) => {
+    let promptCb = props => {
       if(props.route.charAt(0) !== '/') {
-        props.route = '/' + props.route;
+        props.route = `/${props.route}`;
       }
 
       this.route = props.route;
 
-      if (props.models) {
+      if(props.models) {
         delete this.filters.mongoose;
         delete this.filters.mongooseModels;
         delete this.filters.sequelize;
         delete this.filters.sequelizeModels;
 
         this.filters[props.models] = true;
-        this.filters[props.models + 'Models'] = true;
+        this.filters[`${props.models}Models`] = true;
       }
     };
 
-    if (this.options.route) {
-      if (this.filters.mongoose && this.filters.sequelize) {
-        if (this.options.models) {
-          return promptCb(this.options);
-        }
+    if(this.options.route) {
+      if(this.filters.mongoose && this.filters.sequelize && this.options.models) {
+        return promptCb(this.options);
       } else {
-        if (this.filters.mongooseModels) { this.options.models = 'mongoose'; }
-        else if (this.filters.sequelizeModels) { this.options.models = 'sequelize'; }
-        else { delete this.options.models; }
+        if(this.filters.mongooseModels) this.options.models = 'mongoose';
+        else if(this.filters.sequelizeModels) this.options.models = 'sequelize';
+        else delete this.options.models;
         return promptCb(this.options);
       }
     }
 
     var name = this.name;
 
     var base = this.config.get('routesBase') || '/api/';
-    if(base.charAt(base.length-1) !== '/') {
-      base = base + '/';
+    if(base.charAt(base.length - 1) !== '/') {
+      base = `${base}/`;
     }
 
     // pluralization defaults to true for backwards compat
-    if (this.config.get('pluralizeRoutes') !== false) {
-      name = name + 's';
+    if(this.config.get('pluralizeRoutes') !== false) {
+      name = `${name}s`;
     }
 
     var prompts = [{
       name: 'route',
       message: 'What will the url of your endpoint be?',
-      default: base + name
+      default: `${base}${name}`
     }, {
       type: 'list',
       name: 'models',
       message: 'What would you like to use for the endpoint\'s models?',
-      choices: [ 'Mongoose', 'Sequelize' ],
+      choices: ['Mongoose', 'Sequelize'],
       default: this.filters.sequelizeModels ? 1 : 0,
-      filter: (val) => val.toLowerCase(),
+      filter: val => val.toLowerCase(),
       when: () => this.filters.mongoose && this.filters.sequelize
     }];
 
     return this.prompt(prompts).then(promptCb);
   }
 
   configuring() {
-    this.routeDest = path.join(this.options.endpointDirectory ||
-      this.config.get('endpointDirectory') || 'server/api/', this.name);
+    this.routeDest = path.join(this.options.endpointDirectory
+      || this.config.get('endpointDirectory')
+      || 'server/api/', this.name);
   }
 
   writing() {
@@ -111,34 +109,34 @@ export class Generator extends Base {
         file: routesFile,
         needle: this.config.get('routesNeedle'),
         splicable: [
-          "app.use(\'" + this.route +"\', require(\'" + reqPath + "\'));"
+          `app.use('${this.route}', require('${reqPath}'));`
         ]
       };
       this.rewriteFile(routeConfig);
     }
 
-    if (this.filters.socketio && this.config.get('insertSockets')) {
+    if(this.filters.socketio && this.config.get('insertSockets')) {
       var socketsFile = this.config.get('registerSocketsFile');
       var reqPath = this.relativeRequire(this.routeDest + '/' + this.basename +
         '.socket', socketsFile);
       var socketConfig = {
         file: socketsFile,
         needle: this.config.get('socketsNeedle'),
         splicable: [
-          "require(\'" + reqPath + "\').register(socket);"
+          `require('${reqPath}').register(socket);`
         ]
       };
       this.rewriteFile(socketConfig);
     }
 
-    if (this.filters.sequelize && this.config.get('insertModels')) {
+    if(this.filters.sequelize && this.config.get('insertModels')) {
       var modelsFile = this.config.get('registerModelsFile');
-      var reqPath = this.relativeRequire(this.routeDest + '/' + this.basename + '.model', modelsFile);
+      var reqPath = this.relativeRequire(`${this.routeDest}/${this.basename}.model`, modelsFile);
       var modelConfig = {
         file: modelsFile,
         needle: this.config.get('modelsNeedle'),
         splicable: [
-          "db." + this.classedName + " = db.sequelize.import(\'" + reqPath +"\');"
+          `db.${this.classedName} = db.sequelize.import('${reqPath}');`
         ]
       };
       this.rewriteFile(modelConfig);