代码生成增加模块属性

src/main/java/com/stylefeng/guns/core/template/config/ControllerConfig.java

@@ -11,15 +11,13 @@ import java.util.List;
  */
 public class ControllerConfig {
 
-    private String controllerPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\system\\controller\\{}Controller.java";
+    private ContextConfig contextConfig;
+
+    private String controllerPathTemplate;
     private String packageName;//包名称
     private List<String> imports;//所引入的包
 
-    public ControllerConfig(){
-        init();
-    }
-
-    private void init(){
+    public void init() {
         ArrayList<String> imports = new ArrayList<>();
         imports.add("com.stylefeng.guns.common.controller.BaseController");
         imports.add("org.springframework.stereotype.Controller");
@@ -28,7 +26,8 @@ public class ControllerConfig {
         imports.add("org.springframework.ui.Model");
         imports.add("org.springframework.web.bind.annotation.PathVariable");
         this.imports = imports;
-        this.packageName = "com.stylefeng.guns.modular.system.controller";
+        this.packageName = "com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".controller";
+        this.controllerPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\" + contextConfig.getModuleName() + "\\controller\\{}Controller.java";
     }
 
     public String getPackageName() {
@@ -54,4 +53,12 @@ public class ControllerConfig {
     public void setControllerPathTemplate(String controllerPathTemplate) {
         this.controllerPathTemplate = controllerPathTemplate;
     }
+
+    public ContextConfig getContextConfig() {
+        return contextConfig;
+    }
+
+    public void setContextConfig(ContextConfig contextConfig) {
+        this.contextConfig = contextConfig;
+    }
 }

src/main/java/com/stylefeng/guns/core/template/config/DaoConfig.java

@@ -8,10 +8,18 @@ package com.stylefeng.guns.core.template.config;
  */
 public class DaoConfig {
 
-    private String daoPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\system\\dao\\{}Dao.java";
-    private String xmlPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\system\\dao\\mapping\\{}Dao.xml";
+    private ContextConfig contextConfig;
 
-    private String packageName = "com.stylefeng.guns.modular.system.dao";
+    private String daoPathTemplate;
+    private String xmlPathTemplate;
+
+    private String packageName;
+
+    public void init() {
+        this.daoPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\" + contextConfig.getModuleName() + "\\dao\\{}Dao.java";
+        this.xmlPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\" + contextConfig.getModuleName() + "\\dao\\mapping\\{}Dao.xml";
+        this.packageName = "com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".dao";
+    }
 
     public String getPackageName() {
         return packageName;
@@ -36,4 +44,12 @@ public class DaoConfig {
     public void setXmlPathTemplate(String xmlPathTemplate) {
         this.xmlPathTemplate = xmlPathTemplate;
     }
+
+    public ContextConfig getContextConfig() {
+        return contextConfig;
+    }
+
+    public void setContextConfig(ContextConfig contextConfig) {
+        this.contextConfig = contextConfig;
+    }
 }

src/main/java/com/stylefeng/guns/core/template/config/PageConfig.java

@@ -8,11 +8,21 @@ package com.stylefeng.guns.core.template.config;
  */
 public class PageConfig {
 
-    private String pagePathTemplate = "\\src\\main\\webapp\\WEB-INF\\view\\system\\{}\\{}.html";
-    private String pageAddPathTemplate = "\\src\\main\\webapp\\WEB-INF\\view\\system\\{}\\{}_add.html";
-    private String pageEditPathTemplate = "\\src\\main\\webapp\\WEB-INF\\view\\system\\{}\\{}_edit.html";
-    private String pageJsPathTemplate = "\\src\\main\\webapp\\static\\modular\\system\\{}\\{}.js";
-    private String pageInfoJsPathTemplate = "\\src\\main\\webapp\\static\\modular\\system\\{}\\{}_info.js";
+    private ContextConfig contextConfig;
+
+    private String pagePathTemplate;
+    private String pageAddPathTemplate;
+    private String pageEditPathTemplate;
+    private String pageJsPathTemplate;
+    private String pageInfoJsPathTemplate;
+
+    public void init() {
+        pagePathTemplate = "\\src\\main\\webapp\\WEB-INF\\view\\" + contextConfig.getModuleName() + "\\{}\\{}.html";
+        pageAddPathTemplate = "\\src\\main\\webapp\\WEB-INF\\view\\" + contextConfig.getModuleName() + "\\{}\\{}_add.html";
+        pageEditPathTemplate = "\\src\\main\\webapp\\WEB-INF\\view\\" + contextConfig.getModuleName() + "\\{}\\{}_edit.html";
+        pageJsPathTemplate = "\\src\\main\\webapp\\static\\modular\\" + contextConfig.getModuleName() + "\\{}\\{}.js";
+        pageInfoJsPathTemplate = "\\src\\main\\webapp\\static\\modular\\" + contextConfig.getModuleName() + "\\{}\\{}_info.js";
+    }
 
     public String getPagePathTemplate() {
         return pagePathTemplate;
@@ -53,4 +63,12 @@ public class PageConfig {
     public void setPageInfoJsPathTemplate(String pageInfoJsPathTemplate) {
         this.pageInfoJsPathTemplate = pageInfoJsPathTemplate;
     }
+
+    public ContextConfig getContextConfig() {
+        return contextConfig;
+    }
+
+    public void setContextConfig(ContextConfig contextConfig) {
+        this.contextConfig = contextConfig;
+    }
 }

src/main/java/com/stylefeng/guns/core/template/config/ServiceConfig.java

@@ -13,23 +13,21 @@ public class ServiceConfig {
 
     private ContextConfig contextConfig;
 
-    private String servicePathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\system\\service\\I{}Service.java";
-    private String serviceImplPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\system\\service\\impl\\{}ServiceImpl.java";
+    private String servicePathTemplate;
+    private String serviceImplPathTemplate;
 
-    private String packageName = "com.stylefeng.guns.modular.system.service";
+    private String packageName;
 
     private List<String> serviceImplImports;
 
-    public ServiceConfig(ContextConfig contextConfig) {
-        this.contextConfig = contextConfig;
-        init();
-    }
-
-    private void init() {
+    public void init() {
         ArrayList<String> imports = new ArrayList<>();
         imports.add("org.springframework.stereotype.Service");
-        imports.add("com.stylefeng.guns.modular.system.service.I" + contextConfig.getBizEnBigName() + "Service");
+        imports.add("com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service.I" + contextConfig.getBizEnBigName() + "Service");
         this.serviceImplImports = imports;
+        this.servicePathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\" + contextConfig.getModuleName() + "\\service\\I{}Service.java";
+        this.serviceImplPathTemplate = "\\src\\main\\java\\com\\stylefeng\\guns\\modular\\" + contextConfig.getModuleName() + "\\service\\impl\\{}ServiceImpl.java";
+        this.packageName = "com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service";
     }
 
 

src/main/java/com/stylefeng/guns/core/template/engine/base/AbstractTemplateEngine.java

@@ -18,20 +18,32 @@ public class AbstractTemplateEngine {
 
     public void initConfig() {
         if (this.contextConfig == null) {
-            contextConfig = new ContextConfig();
+            this.contextConfig = new ContextConfig();
         }
         if (this.controllerConfig == null) {
-            controllerConfig = new ControllerConfig();
+            this.controllerConfig = new ControllerConfig();
         }
         if (this.pageConfig == null) {
-            pageConfig = new PageConfig();
+            this.pageConfig = new PageConfig();
         }
         if (this.daoConfig == null) {
-            daoConfig = new DaoConfig();
+            this.daoConfig = new DaoConfig();
         }
         if (this.serviceConfig == null) {
-            serviceConfig = new ServiceConfig(contextConfig);
+            this.serviceConfig = new ServiceConfig();
         }
+
+        this.controllerConfig.setContextConfig(this.contextConfig);
+        this.controllerConfig.init();
+
+        this.serviceConfig.setContextConfig(this.contextConfig);
+        this.serviceConfig.init();
+
+        this.daoConfig.setContextConfig(this.contextConfig);
+        this.daoConfig.init();
+
+        this.pageConfig.setContextConfig(this.contextConfig);
+        this.pageConfig.init();
     }
 
     public PageConfig getPageConfig() {

src/main/resources/gunsTemplate/Controller.java.btl

@@ -14,7 +14,7 @@ import ${import};
 @RequestMapping("/${context.bizEnName}")
 public class ${context.bizEnBigName}Controller extends BaseController {
 
-    private String PREFIX = "/system/${context.bizEnName}/";
+    private String PREFIX = "/${context.moduleName}/${context.bizEnName}/";
 
     /**
      * 跳转到${context.bizChName}首页

src/main/resources/gunsTemplate/page.html.btl

@@ -34,5 +34,5 @@
         </div>
     </div>
 </div>
-<script src="\${ctxPath}/static/modular/system/${context.bizEnName}/${context.bizEnName}.js"></script>
+<script src="\${ctxPath}/static/modular/${context.moduleName}/${context.bizEnName}/${context.bizEnName}.js"></script>
 @}

src/main/resources/gunsTemplate/page_add.html.btl

@@ -25,5 +25,5 @@
 
     </div>
 </div>
-<script src="\${ctxPath}/static/modular/system/${context.bizEnName}/${context.bizEnName}_info.js"></script>
+<script src="\${ctxPath}/static/modular/${context.moduleName}/${context.bizEnName}/${context.bizEnName}_info.js"></script>
 @}

src/main/resources/gunsTemplate/page_edit.html.btl

@@ -25,5 +25,5 @@
 
     </div>
 </div>
-<script src="\${ctxPath}/static/modular/system/${context.bizEnName}/${context.bizEnName}_info.js"></script>
+<script src="\${ctxPath}/static/modular/${context.moduleName}/${context.bizEnName}/${context.bizEnName}_info.js"></script>
 @}

src/test/java/com/stylefeng/guns/template/TemplateGenerator.java

@@ -18,9 +18,11 @@ public class TemplateGenerator {
         ContextConfig contextConfig = new ContextConfig();
         contextConfig.setBizChName("啊哈");
         contextConfig.setBizEnName("haha");
+        contextConfig.setModuleName("tk");
+        contextConfig.setProjectPath("D:\\tmp\\guns");
 
-        contextConfig.setAddPageSwitch(false);
-        contextConfig.setEditPageSwitch(false);
+        //contextConfig.setAddPageSwitch(false);
+        //contextConfig.setEditPageSwitch(false);
 
         GunsTemplateEngine gunsTemplateEngine = new SimpleTemplateEngine();
         gunsTemplateEngine.setContextConfig(contextConfig);