mp-java/src/test/java/com/gxwebsoft/generator/CmsGenerator.java

package com.gxwebsoft.generator;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.gxwebsoft.generator.engine.BeetlTemplateEnginePlus;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.io.*;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.regex.Pattern;
import java.util.regex.Matcher;

/**
 * CMS模块-代码生成工具
 *
 * @author WebSoft
 * @since 2021-09-05 00:31:14
 */
public class CmsGenerator {
  // 输出位置
  private static final String OUTPUT_LOCATION = System.getProperty("user.dir");
  //private static final String OUTPUT_LOCATION = "D:/codegen";  // 不想生成到项目中可以写磁盘路径
  // 输出目录
  private static final String OUTPUT_DIR = "/src/main/java";
  // Vue文件输出位置
  private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/site";
  // UniApp文件输出目录
  private static final String OUTPUT_LOCATION_UNIAPP = "/Users/gxwebsoft/VUE/template-10550";
  // Vue文件输出目录
  private static final String OUTPUT_DIR_VUE = "/src";
  // 作者名称
  private static final String AUTHOR = "科技小王子";
  // 是否在xml中添加二级缓存配置
  private static final boolean ENABLE_CACHE = false;
  // 数据库连接配置
  private static final String DB_URL = "jdbc:mysql://47.119.165.234:3308/modules?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8";
  private static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver";
  private static final String DB_USERNAME = "modules";
  private static final String DB_PASSWORD = "8YdLnk7KsPAyDXGA";
  // 包名
  private static final String PACKAGE_NAME = "com.gxwebsoft";
  // 模块名
  private static final String MODULE_NAME = "cms";
  // 需要生成的表
  private static final String[] TABLE_NAMES = new String[]{
//            "cms_article",
//            "cms_article_category",
//            "cms_article_like",
//            "cms_article_comment",
//              "cms_article_content",
//            "cms_docs",
//                  "cms_docs_book",
//                  "cms_docs_content",
//            "cms_ad",
//            "cms_ad_record",
//            "cms_navigation",
//            "cms_design",
//            "cms_design_record",
//            "cms_website",
//            "cms_website_field",
//              "cms_form",
//              "cms_form_record",
//              "cms_domain",
//              "cms_mp_menu"
//    "cms_mp_pages",
//    "cms_mp"
//    "cms_mp_field"
//      "cms_mp_ad"
//    "cms_components"
//              "cms_mp_official_menu",
//    "cms_order",
//      "cms_model"
//    "cms_lang",
//    "cms_lang_log",
//    "cms_website_setting",
//    "cms_statistics"

  };
  // 需要去除的表前缀
  private static final String[] TABLE_PREFIX = new String[]{
    "tb_"
  };
  // 不需要作为查询参数的字段
  private static final String[] PARAM_EXCLUDE_FIELDS = new String[]{
    "tenant_id",
    "create_time",
    "update_time"
  };
  // 查询参数使用String的类型
  private static final String[] PARAM_TO_STRING_TYPE = new String[]{
    "Date",
    "LocalDate",
    "LocalTime",
    "LocalDateTime"
  };
  // 查询参数使用EQ的类型
  private static final String[] PARAM_EQ_TYPE = new String[]{
    "Integer",
    "Boolean",
    "BigDecimal"
  };
  // 是否添加权限注解
  private static final boolean AUTH_ANNOTATION = true;
  // 是否添加日志注解
  private static final boolean LOG_ANNOTATION = true;
  // controller的mapping前缀
  private static final String CONTROLLER_MAPPING_PREFIX = "/api";
  // 模板所在位置
  private static final String TEMPLATES_DIR = "/src/test/java/com/gxwebsoft/generator/templates";

  public static void main(String[] args) {
    // 代码生成器
    AutoGenerator mpg = new AutoGenerator();

    // 全局配置
    GlobalConfig gc = new GlobalConfig();
    gc.setOutputDir(OUTPUT_LOCATION + OUTPUT_DIR);
    gc.setAuthor(AUTHOR);
    gc.setOpen(false);
    gc.setFileOverride(true);
    gc.setEnableCache(ENABLE_CACHE);
    gc.setSwagger2(true);
    gc.setIdType(IdType.AUTO);
    gc.setServiceName("%sService");
    mpg.setGlobalConfig(gc);

    // 数据源配置
    DataSourceConfig dsc = new DataSourceConfig();
    dsc.setUrl(DB_URL);
    // dsc.setSchemaName("public");
    dsc.setDriverName(DB_DRIVER);
    dsc.setUsername(DB_USERNAME);
    dsc.setPassword(DB_PASSWORD);
    mpg.setDataSource(dsc);

    // 包配置
    PackageConfig pc = new PackageConfig();
    pc.setModuleName(MODULE_NAME);
    pc.setParent(PACKAGE_NAME);
    mpg.setPackageInfo(pc);

    // 策略配置
    StrategyConfig strategy = new StrategyConfig();
    strategy.setNaming(NamingStrategy.underline_to_camel);
    strategy.setColumnNaming(NamingStrategy.underline_to_camel);
    strategy.setInclude(TABLE_NAMES);
    strategy.setTablePrefix(TABLE_PREFIX);
    strategy.setSuperControllerClass(PACKAGE_NAME + ".common.core.web.BaseController");
    strategy.setEntityLombokModel(true);
    strategy.setRestControllerStyle(true);
    strategy.setControllerMappingHyphenStyle(true);
    strategy.setLogicDeleteFieldName("deleted");
    mpg.setStrategy(strategy);

    // 模板配置
    TemplateConfig templateConfig = new TemplateConfig();
    templateConfig.setController(TEMPLATES_DIR + "/controller.java");
    templateConfig.setEntity(TEMPLATES_DIR + "/entity.java");
    templateConfig.setMapper(TEMPLATES_DIR + "/mapper.java");
    templateConfig.setXml(TEMPLATES_DIR + "/mapper.xml");
    templateConfig.setService(TEMPLATES_DIR + "/service.java");
    templateConfig.setServiceImpl(TEMPLATES_DIR + "/serviceImpl.java");
    mpg.setTemplate(templateConfig);
    mpg.setTemplateEngine(new BeetlTemplateEnginePlus());

    // 自定义模板配置
    InjectionConfig cfg = new InjectionConfig() {
      @Override
      public void initMap() {
        Map<String, Object> map = new HashMap<>();
        map.put("packageName", PACKAGE_NAME);
        map.put("paramExcludeFields", PARAM_EXCLUDE_FIELDS);
        map.put("paramToStringType", PARAM_TO_STRING_TYPE);
        map.put("paramEqType", PARAM_EQ_TYPE);
        map.put("authAnnotation", AUTH_ANNOTATION);
        map.put("logAnnotation", LOG_ANNOTATION);
        map.put("controllerMappingPrefix", CONTROLLER_MAPPING_PREFIX);
        this.setMap(map);
      }
    };
    String templatePath = TEMPLATES_DIR + "/param.java.btl";
    List<FileOutConfig> focList = new ArrayList<>();
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION + OUTPUT_DIR + "/"
          + PACKAGE_NAME.replace(".", "/")
          + "/" + pc.getModuleName() + "/param/"
          + tableInfo.getEntityName() + "Param" + StringPool.DOT_JAVA;
      }
    });
    /**
     * 以下是生成VUE项目代码
     * 生成文件的路径 /api/shop/goods/index.ts
     */
    templatePath = TEMPLATES_DIR + "/index.ts.btl";

    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE
          + "/api/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "index.ts";
      }
    });
    focList.add(new FileOutConfig() {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE
          + "/api/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "index.ts";
      }
    });
    // 生成TS文件 (/api/shop/goods/model/index.ts)
    templatePath = TEMPLATES_DIR + "/model.ts.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE
          + "/api/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/model/" + "index.ts";
      }
    });
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE
          + "/api/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/model/" + "index.ts";
      }
    });
    // 生成Vue文件(/views/shop/goods/index.vue)
    templatePath = TEMPLATES_DIR + "/index.vue.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE
          + "/views/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "index.vue";
      }
    });

    // 生成components文件(/views/shop/goods/components/edit.vue)
    templatePath = TEMPLATES_DIR + "/components.edit.vue.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE
          + "/views/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/components/" + tableInfo.getEntityPath() + "Edit.vue";
      }
    });

    // 生成components文件(/views/shop/goods/components/search.vue)
    templatePath = TEMPLATES_DIR + "/components.search.vue.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE
          + "/views/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/components/" + "search.vue";
      }
    });

    // ========== 移动端页面文件生成 ==========
    // 生成移动端列表页面配置文件 (/src/cms/article/index.config.ts)
    templatePath = TEMPLATES_DIR + "/index.config.ts.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE
          + "/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "index.config.ts";
      }
    });

    // 生成移动端列表页面组件文件 (/src/cms/article/index.tsx)
    templatePath = TEMPLATES_DIR + "/index.tsx.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE
          + "/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "index.tsx";
      }
    });

    // 生成移动端新增/编辑页面配置文件 (/src/cms/article/add.config.ts)
    templatePath = TEMPLATES_DIR + "/add.config.ts.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE
          + "/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "add.config.ts";
      }
    });

    // 生成移动端新增/编辑页面组件文件 (/src/cms/article/add.tsx)
    templatePath = TEMPLATES_DIR + "/add.tsx.btl";
    focList.add(new FileOutConfig(templatePath) {
      @Override
      public String outputFile(TableInfo tableInfo) {
        return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE
          + "/" + pc.getModuleName() + "/"
          + tableInfo.getEntityPath() + "/" + "add.tsx";
      }
    });

    cfg.setFileOutConfigList(focList);
    mpg.setCfg(cfg);

    mpg.execute();

    // 自动更新 app.config.ts
    updateAppConfig(TABLE_NAMES, MODULE_NAME);
  }

  /**
   * 自动更新 app.config.ts 文件，添加新生成的页面路径
   */
  private static void updateAppConfig(String[] tableNames, String moduleName) {
    String appConfigPath = OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE + "/app.config.ts";

    try {
      // 读取原文件内容
      String content = new String(Files.readAllBytes(Paths.get(appConfigPath)));

      // 为每个表生成页面路径
      StringBuilder newPages = new StringBuilder();
      for (String tableName : tableNames) {
        String entityPath = tableName.replaceAll("_", "");
        // 转换为驼峰命名
        String[] parts = tableName.split("_");
        StringBuilder camelCase = new StringBuilder(parts[0]);
        for (int i = 1; i < parts.length; i++) {
          camelCase.append(parts[i].substring(0, 1).toUpperCase()).append(parts[i].substring(1));
        }
        entityPath = camelCase.toString();

        newPages.append("      '").append(entityPath).append("/index',\n");
        newPages.append("      '").append(entityPath).append("/add',\n");
      }

      // 查找对应模块的子包配置
      String modulePattern = "\"root\":\\s*\"" + moduleName + "\",\\s*\"pages\":\\s*\\[([^\\]]*)]";
      Pattern pattern = Pattern.compile(modulePattern, Pattern.DOTALL);
      Matcher matcher = pattern.matcher(content);

      if (matcher.find()) {
        String existingPages = matcher.group(1);

        // 检查页面是否已存在，避免重复添加
        boolean needUpdate = false;
        String[] newPageArray = newPages.toString().split("\n");
        for (String newPage : newPageArray) {
          if (!newPage.trim().isEmpty() && !existingPages.contains(newPage.trim().replace("      ", "").replace(",", ""))) {
            needUpdate = true;
            break;
          }
        }

        if (needUpdate) {
          // 备份原文件
          String backupPath = appConfigPath + ".backup." + System.currentTimeMillis();
          Files.copy(Paths.get(appConfigPath), Paths.get(backupPath));
          System.out.println("已备份原文件到: " + backupPath);

          // 在现有页面列表末尾添加新页面
          String updatedPages = existingPages.trim();
          if (!updatedPages.endsWith(",")) {
            updatedPages += ",";
          }
          updatedPages += "\n" + newPages.toString().trim();

          // 替换内容
          String updatedContent = content.replace(matcher.group(1), updatedPages);

          // 写入更新后的内容
          Files.write(Paths.get(appConfigPath), updatedContent.getBytes());

          System.out.println("✅ 已自动更新 app.config.ts，添加了以下页面路径:");
          System.out.println(newPages.toString());
        } else {
          System.out.println("ℹ️ app.config.ts 中已包含所有页面路径，无需更新");
        }
      } else {
        System.out.println("⚠️ 未找到 " + moduleName + " 模块的子包配置，请手动添加页面路径");
      }

    } catch (Exception e) {
      System.err.println("❌ 更新 app.config.ts 失败: " + e.getMessage());
      e.printStackTrace();
    }
  }

}