feat(scaffold): integrate egg-tx (#4)

Author: LinYuanBaoBao

.gitignore

@@ -7,7 +7,7 @@ node_modules/
 run/
 logs/
 package-lock.json
-
+.idea
 *.lcov
 
 push.sh

generators/app/handler/package_json_th.js

@@ -35,6 +35,9 @@ class PackageJsonTemplateHandler extends AbstractTemplateHandler {
     if (this.props.dependencies.swagger) {
       this._extendDependencies(pkgJson, 'egg-swagger-doc', '^2.3.1');
     }
+    if (this.props.dependencies.tx) {
+      this._extendDependencies(pkgJson, 'egg-tx', '^1.0.2');
+    }
     switch (this.props.db) {
       case 'mongo': {
         this._extendDependencies(pkgJson, 'egg-mongoose', '^3.1.3');

generators/app/index.js

@@ -27,7 +27,8 @@ const obj = {
         } else {
           return '名称不能为空';
         }
-      } },
+      }
+    },
     option: { desc: '作者', type: String, default: '' }
   },
   router: {
@@ -126,7 +127,8 @@ module.exports = require('yo-power-generator').getGenerator(obj, {
     props.dependencies = {
       utils: true,
       eureka: true,
-      swagger: true
+      swagger: true,
+      tx: (props.db !== 'none')
     }
   }
 });

generators/app/templates/config/config.default.tmpl.js

@@ -75,7 +75,7 @@ module.exports = appInfo => {
     maxFileSize: 5 * 1024
   };
 
-  <%
+<%
   if (dependencies.eureka) {
     print(`
   config.eureka = {
@@ -131,6 +131,15 @@ module.exports = appInfo => {
     `);
   }
 
+  if (dependencies.tx) {
+    print(`
+  config.tx = {
+    reqAction: ['POST','PUT','DELETE'], 
+    dbType: '${db}',
+  };    
+    `);
+  }
+
   switch (db) {
     case 'mongo': {
       print(`
@@ -180,7 +189,7 @@ module.exports = appInfo => {
       break;
     }
   }
-  %>
+%>
 
   return {
     ...config,

generators/app/templates/config/plugin.tmpl.js

@@ -40,6 +40,14 @@ module.exports = {
   swaggerdoc: {
     enable: true,
     package: 'egg-swagger-doc',
+  },
+      `);
+    }
+    if(dependencies.tx){
+      print(`
+  tx: {
+    enable: true,
+    package: 'egg-tx',
   },
       `);
     }

generators/test/app/index.test.js

@@ -141,12 +141,15 @@ describe('package.json content', () => {
 
       it('should have dependencies', () => {
         assert(pkg.dependencies['egg-mongoose']);
+        assert(pkg.dependencies['egg-tx']);
       })
 
       it('should have config', () => {
         assert.fileContent([
           ['config/config.default.js', /config.mongoose.*=/],
-          ['config/plugin.js', /mongoose.*:/]
+          ['config/plugin.js', /mongoose.*:/],
+          ['config/config.default.js', /config.tx.*=\s.*\s*.*\s*dbType:\s'mongo'/],
+          ['config/plugin.js', /tx.*:/]
         ])
       })
     })
@@ -168,12 +171,15 @@ describe('package.json content', () => {
       it('should have dependencies', () => {
         assert(pkg.dependencies['egg-sequelize']);
         assert(pkg.dependencies['mysql2']);
+        assert(pkg.dependencies['egg-tx']);
       })
 
       it('should have config', () => {
         assert.fileContent([
           ['config/config.default.js', /config.sequelize.*=/],
-          ['config/plugin.js', /sequelize.*:/]
+          ['config/plugin.js', /sequelize.*:/],
+          ['config/config.default.js', /config.tx.*=\s.*\s*.*\s*dbType:\s'mysql'/],
+          ['config/plugin.js', /tx.*:/]
         ])
       })
     })
@@ -195,14 +201,17 @@ describe('package.json content', () => {
       it('should have not dependencies', () => {
         assert(!pkg.dependencies['egg-mongoose']);
         assert(!pkg.dependencies['egg-sequelize']);
+        assert(!pkg.dependencies['egg-tx']);
       })
 
       it('should have not config', () => {
         assert.noFileContent([
           ['config/config.default.js', /config.mongoose.*=/],
           ['config/plugin.js', /mongoose.*:/],
           ['config/config.default.js', /config.sequelize.*=/],
-          ['config/plugin.js', /sequelize.*:/]
+          ['config/plugin.js', /sequelize.*:/],
+          ['config/config.default.js', /config.tx.*=/],
+          ['config/plugin.js', /tx.*:/]
         ])
       })
     })