package com.ruoyi;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import org.springframework.boot.autoconfigure.SpringBootApplication;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * MyBatis-Plus 代码生成器
 * 功能：根据数据库表结构自动生成Entity、Mapper、Service、Controller等代码
 * 修复：解决Freemarker模板中generateService变量缺失的问题
 */
@SpringBootApplication
public class PlusCodeGenerator {

    // 数据库配置
    private static final String DB_URL = "jdbc:postgresql://localhost:5432/ruoyi-java";
    private static final String DB_USERNAME = "postgres";
    private static final String DB_PASSWORD = "root";

    // 项目基础配置
    private static final String BASE_PACKAGE = "com.ruoyi";
    private static final String MODULE_NAME = "basic"; // 模块名

    public static void main(String[] args) {
        String projectPath = System.getProperty("user.dir"); // 获取项目根路径
        String path = "ruoyi-system"; // 模块名称
        String table = "province,city,district"; // 表名，多个表逗号隔开

        // 代码输出路径配置
        String outputBasePath = Paths.get(projectPath, path, "src", "main", "java").toString();
        String outputResourcePath = Paths.get(projectPath, path, "src", "main", "resources", "mapper").toString();

        // 代码生成核心配置
        FastAutoGenerator.create(DB_URL, DB_USERNAME, DB_PASSWORD)
                .globalConfig(builder -> {
                    builder.author("chenhj") // 作者信息
                            .outputDir(outputBasePath) // 代码输出目录
                            .dateType(DateType.ONLY_DATE) // 日期类型
                            .commentDate("yyyy-MM-dd") // 注释日期格式
                            .disableOpenDir(); // 禁止自动打开输出目录
                })
                .packageConfig(builder -> {
                    builder.parent(BASE_PACKAGE) // 基础包名
                            .moduleName(MODULE_NAME) // 模块名（根据具体修改）
                            .entity("entity") // Entity包名
                            .mapper("mapper") // Mapper包名
                            .service("service") // Service包名
                            .serviceImpl("service.impl") // Service实现类包名
                            .controller("controller") // Controller包名
                            .pathInfo(Collections.singletonMap(
                                    OutputFile.xml,
                                    outputResourcePath // Mapper XML输出路径
                            ));
                })
                .strategyConfig(builder -> {
                    builder.addInclude(table) // 要生成的表名
                            .addTablePrefix("t_", "sys_") // 过滤表前缀

                            // 实体类配置
                            .entityBuilder()
                            .superClass("com.ruoyi.common.core.domain.BaseEntity") // 继承基类
                            .addSuperEntityColumns("create_by", "create_time", "update_by", "update_time") // 排除基类字段
                            .enableLombok() // 启用Lombok
                            .naming(NamingStrategy.underline_to_camel) // 表名转驼峰
                            .columnNaming(NamingStrategy.underline_to_camel) // 列名转驼峰
                            .enableFileOverride() // 允许覆盖文件
                            .logicDeleteColumnName("deleted") // 逻辑删除字段
                            .addTableFills(
                                    // 自动填充配置
                                    new Column("create_by", FieldFill.INSERT), // 创建人，插入时填充
                                    new Column("update_by", FieldFill.INSERT_UPDATE), // 更新人，插入和更新时填充
                                    new Column("create_time", FieldFill.INSERT), // 创建时间，插入时填充
                                    new Column("update_time", FieldFill.INSERT_UPDATE) // 更新时间，插入和更新时填充
                            )
                            .idType(IdType.AUTO) // 主键策略
                            .enableTableFieldAnnotation() // 启用字段注解

                            // 控制器配置
                            .controllerBuilder()
                            .enableFileOverride() // 允许覆盖
                            .enableRestStyle() // 生成RESTful风格控制器

                            // Service配置
                            .serviceBuilder()
                            .formatServiceFileName("%sService") // Service接口命名格式
                            .enableFileOverride() // 允许覆盖
                            .formatServiceImplFileName("%sServiceImpl") // Service实现类命名格式
                            .enableFileOverride() // 允许覆盖

                            // Mapper配置
                            .mapperBuilder()
                            .enableFileOverride() // 允许覆盖
                            .enableMapperAnnotation() // 启用@Mapper注解
                            .enableBaseResultMap() // 启用基础ResultMap
                            .enableBaseColumnList() // 启用基础ColumnList
                            .formatMapperFileName("%sMapper") // Mapper接口命名格式
                            .formatXmlFileName("%sMapper") // Mapper XML命名格式
                            .enableFileOverride(); // 允许覆盖
                })
                // 配置自定义模板
                .templateConfig(builder -> {
                    builder
                            .entity("/templates/entity.java") // 实体类模板
                            .xml("/templates/mapper.xml") // Mapper XML模板
                            .controller("/templates/controller.java") // 控制器模板
                            .serviceImpl("/templates/serviceImpl.java"); // Service实现类模板
                })
                // 注入模板变量
                .injectionConfig(builder -> {
                    Map<String, Object> customMap = new HashMap<>();
                    customMap.put("superEntityColumns", Arrays.asList(
                            "create_by", "create_time", "update_by", "update_time"
                    )); // 基类字段
                    customMap.put("idType", "AUTO"); // 主键类型
                    customMap.put("superEntityClass", "com.ruoyi.common.core.domain.BaseEntity"); // 基类全路径
                    customMap.put("author", "ruoyi"); // 作者信息
                    customMap.put("packageName", BASE_PACKAGE + "." + MODULE_NAME); // 包名
                    customMap.put("tableName", table); // 表名

                    // 新增：解决Freemarker模板中generateService变量缺失的问题
                    // 控制Service实现类是否实现Service接口
                    boolean generateService = true;
                    customMap.put("generateService", generateService);

                    builder.customMap(customMap);
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker模板引擎
                .execute(); // 执行代码生成

        // 后处理：修复实体类的完全限定名问题
        try {
            Path entityPath = Paths.get(outputBasePath, "com/ruoyi/basic/entity/Test.java");
            if (Files.exists(entityPath)) {
                String content = Files.readString(entityPath);
                content = content.replace(
                        "extends com.ruoyi.common.core.domain.BaseEntity",
                        "extends BaseEntity"
                ); // 简化基类引用
                Files.writeString(entityPath, content);
            }
        } catch (IOException e) {
            System.err.println("⚠️ 修复实体类失败: " + e.getMessage());
        }

        System.out.println("✅ 代码生成完成！文件已输出到：" + outputBasePath);
    }
}