package com.doumee.core.utils;


import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;
import org.apache.velocity.runtime.RuntimeConstants;
import org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.*;
import java.util.Date;
import java.util.*;

// 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
public class CodeGenerator {
public static  String packName ="com.doumee";
    public static void main(String[] args) {
        // 配置数据库连接
        String url = "jdbc:mysql://192.168.0.211:3306/dmmall_full";
        String username = "root";
        String password = "Doumee@168";
        String database = "dmmall_full";
        String tableName = "car_driver";
        // 获取表信息并生成代码
        Map<String, Object> tableInfo = getTableInfo(url, username, password, database, tableName);
        generateCode(tableInfo);
    }

    public static Map<String, Object> getTableInfo(String url, String username, String password, String database, String tableName) {
        Map<String, Object> map = new HashMap<>();
        try (Connection conn = DriverManager.getConnection(url, username, password)) {
            // 查询表注释
            String tableQuery = "SELECT table_comment FROM information_schema.tables WHERE table_schema = ? AND table_name = ?";
            try (PreparedStatement ps = conn.prepareStatement(tableQuery)) {
                ps.setString(1, database);
                ps.setString(2, tableName);
                ResultSet rs = ps.executeQuery();
                if (rs.next()) {
                    map.put("tableComment", rs.getString("table_comment"));
                }
            }
            // 查询字段信息 auto_increment
            String columnQuery = "SELECT extra,column_name, column_type, column_comment FROM information_schema.columns WHERE table_schema = ? AND table_name = ? order by ordinal_position";
            List<Map<String, String>> columns = new ArrayList<>();
            try (PreparedStatement ps = conn.prepareStatement(columnQuery)) {
                ps.setString(1, database);
                ps.setString(2, tableName);
                ResultSet rs = ps.executeQuery();
                int index = 1;
                while (rs.next()) {
                    String javaName = toCamelCase(rs.getString("column_name"));
                    String getJavaName = javaName.substring(0,1).toUpperCase()+javaName.substring(1);
                    Map<String, String> columnInfo = new HashMap<>();
                    if(rs.getString("extra") !=null
                            && rs.getString("extra").contains("auto_increment")){
                        //如果是自增长
                        columnInfo.put("auto","1");  // 转换为 Java 变量名
                    }else{
                        columnInfo.put("auto","0");  // 转换为 Java 变量名
                    }
                    columnInfo.put("index",index++ +"");
                    columnInfo.put("columnName", rs.getString("column_name"));
                    columnInfo.put("javaName",javaName);  // 转换为 Java 变量名
                    columnInfo.put("getJavaName",getJavaName);  // 转换为 Java 变量名
                    columnInfo.put("javaType", sqlTypeToJavaType(rs.getString("column_type")));  // 转换为 Java 类型
                    columnInfo.put("comment", rs.getString("column_comment"));
                    columns.add(columnInfo);
                }
            }
            map.put("columns", columns);
            map.put("tableName", tableName);
            map.put("entityName", toCamelCase(tableName));
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return map;
    }

    private static String sqlTypeToJavaType(String sqlType) {
        sqlType = sqlType.toLowerCase();
        if (sqlType.contains("int")) {
            return "Integer";
        } else if (sqlType.contains("bigint")) {
            return "Long";
        } else if (sqlType.contains("char") || sqlType.contains("text") || sqlType.contains("varchar")) {
            return "String";
        } else if (sqlType.contains("datetime") || sqlType.contains("timestamp")) {
            return "Date";
        } else if (sqlType.contains("date")) {
            return "Date";
        } else if (sqlType.contains("decimal") || sqlType.contains("double")) {
            return "BigDecimal";
        } else {
            return "String";
        }
    }

    private static String toCamelCase(String name) {
        StringBuilder result = new StringBuilder();
        String[] parts = name.split("_");
        for (int i = 0; i < parts.length; i++) {
            if (i == 0) {
                result.append(parts[i].toLowerCase());
            } else {
                result.append(Character.toUpperCase(parts[i].charAt(0))).append(parts[i].substring(1).toLowerCase());
            }
        }
        return result.toString();
    }
    private static void generateCode(Map<String, Object> tableInfo) {
        // 读取表信息
        String entityName = (String) tableInfo.get("entityName");
        String tableName = (String) tableInfo.get("tableName");
        String tableComment = (String) tableInfo.get("tableComment");
        List<Map<String, String>> columns = (List<Map<String, String>>) tableInfo.get("columns");

        String templateDir = "D:\\code\\idea\\doumee_code\\server\\src\\main\\resources\\templates"; // 模板文件的路径
        Properties p = new Properties();
        // 指定模板的加载位置
        p.setProperty(VelocityEngine.FILE_RESOURCE_LOADER_PATH, templateDir);
        // 指定输入编码
        p.setProperty(VelocityEngine.INPUT_ENCODING, "UTF-8");

        p.setProperty(RuntimeConstants.RESOURCE_LOADER, "file");
        p.setProperty("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.FileResourceLoader");
        p.setProperty("velocimacro.library", "macro_library.vm"); // 设置宏库文件名
        p.setProperty("velocimacro.library.autoreload", "true"); // 允许自动重新加载宏库
        p.setProperty("velocimacro.permissions.allow.inline.to.replace.global", "true"); // 允许内联宏替换全局宏
        p.setProperty("velocimacro.library", "/templates/macro_library.vm"); // 设置宏库文件名
        p.setProperty("velocimacro.library.autoreload", "true"); // 允许自动重新加载宏库

        // 设置 Velocity 配置
        VelocityEngine velocityEngine = new VelocityEngine();
        velocityEngine.setProperty(RuntimeConstants.RESOURCE_LOADER, "classpath");
        velocityEngine.setProperty("classpath.resource.loader.class", ClasspathResourceLoader.class.getName());
        velocityEngine.init();

        // 配置模板目录


        // 设置输出目录
        String outputDir = "D:\\code\\idea\\doumee_code\\server\\src\\main\\java\\com\\testcom"; // 输出的代码文件夹路径

        String baseName = entityName.substring(0, 1).toUpperCase() + entityName.substring(1);
        // 创建 Velocity 上下文
        Map<String,String> pack = new HashMap<>();
        pack.put("Base",packName);
        pack.put("Entity",packName+".dao.businees.model");
        pack.put("Controller",packName+".api.businees");
        pack.put("Mapper",packName+".dao.businees");
        pack.put("Service",packName+".service.businees");
        pack.put("ServiceImpl",packName+".service.businees.impl");
        VelocityContext context = new VelocityContext();
        context.put("package",pack); // 设置包路径
        context.put("entityName", baseName);
        context.put("entityNameLower", entityName);
        context.put("entityNameLowerFull", entityName.toLowerCase());
        context.put("tableName", tableName);
        context.put("tableComment", tableComment);
        context.put("columns", columns);
        context.put("nowDate", DateUtil.getPlusTime2(new Date()));

        // 生成 Entity 文件
        generateFile(velocityEngine, context,  "entity.vm", baseName, outputDir + "/dao/businees/model/" , ".java");

        // 生成 Mapper 文件
        generateFile(velocityEngine, context,  "mapper.vm",  baseName, outputDir + "/dao/businees/" , "Mapper.java");

        // 生成 Service 接口文件
        generateFile(velocityEngine, context,  "service.vm",  baseName, outputDir + "/service/businees/" , "Service.java");

        // 生成 ServiceImpl 文件
        generateFile(velocityEngine, context,  "serviceImpl.vm",  baseName, outputDir + "/service/businees/impl/" , "ServiceImpl.java");

        // 生成 Controller 文件
        generateFile(velocityEngine, context,  "controller.vm",  baseName, outputDir + "/api/businees/", "Controller.java");
    }

    private static void generateFile(VelocityEngine velocityEngine, VelocityContext context,  String templateName, String baseName,String outputFilePath,String endName)  {
        // 获取模板
        File file = new File(outputFilePath);
        if(!file.isDirectory()){
            file.mkdirs();
        }
        System.out.println(file.isFile());
        Template template = velocityEngine.getTemplate(  "templates/"+templateName);

        // 创建文件输出流
        try{
            File f = new File(outputFilePath+baseName+endName);
            if(f.isFile()){
                f.delete();
            }
            f.createNewFile();
            // 将上下文数据渲染到模板中
            FileWriter writer = new FileWriter(outputFilePath+baseName+endName);
            template.merge(context, writer);
            writer.flush();
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}