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 ShopGenerator {
  // 输出位置
  private static final String OUTPUT_LOCATION = System.getProperty("user.dir");
  //private static final String OUTPUT_LOCATION = "D:/codegen";  // 不想生成到项目中可以写磁盘路径
  // JAVA输出目录
  private static final String OUTPUT_DIR = "/src/main/java";
  // Vue文件输出位置
  private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/mp-vue";
  // 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://8.134.169.209:13306/cms_demo?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 = "cms_demo";
  private static final String DB_PASSWORD = "EtzJFr4A3c4THZjY";
  // 包名
  private static final String PACKAGE_NAME = "com.gxwebsoft";
  // 模块名
  private static final String MODULE_NAME = "oa";
  // 需要生成的表
  private static final String[] TABLE_NAMES = new String[]{
//    "shop_spec",
//    "shop_spec_value",
//    "shop_goods",
//    "shop_category",
//    "shop_goods_spec",
//    "shop_goods_sku",
//    "shop_goods_category",
//      "shop_coupon",
//    "shop_goods_description",
//    "shop_goods_log",
//    "shop_goods_relation",
//    "shop_goods_comment",
//    "shop_goods_rule",
//    "shop_brand",
//    "shop_merchant",
//    "shop_merchant_type",
//    "shop_merchant_apply",
//    "shop_merchant_account",
//    "shop_chat_message",
//    "shop_chat_conversation",
//    "shop_user_collection",
//    "shop_goods_role_commission"
//    "shop_commission_role"
//    "shop_order",
//    "shop_order_info",
//    "shop_order_info_log",
//    "shop_order_goods",
//    "shop_wechat_deposit",
//    "shop_users",
//    "shop_user_address",
//    "shop_user_balance_log",
//      "shop_recharge_order",
    // 经销商表
//    "shop_dealer_apply",
//    "shop_dealer_capital",
//    "shop_dealer_order",
//    "shop_dealer_referee",
//    "shop_dealer_setting",
//    "shop_dealer_user",
//      "shop_user_referee",
//    "shop_dealer_withdraw",
//      "shop_user_coupon",
//    "shop_cart",
//    "shop_count",
//    "shop_order_delivery",
//    "shop_order_delivery_goods",
//    "shop_order_extract",
//    "shop_splash",
//    "shop_goods_income_config"
//      "shop_express",
//      "shop_express_template",
//      "shop_express_template_detail",
//      "shop_gift"
        "shop_article"
  };
  // 需要去除的表前缀
  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);
        // 添加项目类型标识，用于模板中的条件判断
        map.put("isUniApp", false); // Vue 项目
        map.put("isVueAdmin", true); // 后台管理项目
        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";
      }
    });
    // UniApp 使用专门的模板
    String uniappTemplatePath = TEMPLATES_DIR + "/index.ts.uniapp.btl";
    focList.add(new FileOutConfig(uniappTemplatePath) {
      @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";
      }
    });
    // UniApp 使用专门的 model 模板
    String uniappModelTemplatePath = TEMPLATES_DIR + "/model.ts.uniapp.btl";
    focList.add(new FileOutConfig(uniappModelTemplatePath) {
      @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/shop/goods/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/shop/goods/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/shop/goods/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/shop/goods/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();
    }
  }

}
