import cn.com.digitalhainan.devtools.mbg.factory.IDatabases;
import cn.com.digitalhainan.devtools.mbg.factory.MysqlDatabases;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

@SpringBootTest
class MbgIntegrationTest {

    @Autowired
    private MysqlDatabases mysqlDatabases;

    @Test
    void contextLoads() {
        assert mysqlDatabases != null;
    }
}

package cn.com.digitalhainan.devtools.mbg.common;

import lombok.Builder;
import lombok.Data;
import lombok.experimental.Accessors;
import java.io.Serializable;
import java.util.List;

@Data
@Accessors(chain = true)
public class MbgInfo implements Serializable {
    /** 表前缀（生成类名时去除） */
    private String tablePrefix;
    /** 基础包名（如 "cn.com.digitalhainan.project"） */
    private String basePackage;
    /** 模块名（如 "system"，用于子包路径和 mapper xml 目录） */
    private String model;
    /** 指定生成的表名列表（null 则生成全部表） */
    private List<String> tables;
    /** 作者名（写入生成文件注释） */
    private String author;
}

package cn.com.digitalhainan.devtools.mbg.common;

import lombok.Builder;
import lombok.Data;
import lombok.experimental.Accessors;
import java.io.Serializable;

@Data
@Accessors(chain = true)
public class DatabasesInfo implements Serializable {
    /** 数据库 JDBC 地址 */
    private String url;
    /** 数据库用户名 */
    private String userName;
    /** 数据库密码（AES 加密） */
    private String password;
    /** 数据库类型（目前仅支持 "mysql"） */
    private String dbType;
}

package cn.com.digitalhainan.devtools.mbg.factory;

import cn.com.digitalhainan.devtools.mbg.common.DatabasesInfo;
import cn.com.digitalhainan.devtools.mbg.common.MbgInfo;

public interface IDatabases {
    /** 生成代码并返回 ZIP 字节数组 */
    byte[] generatorCode(MbgInfo gmbVO, DatabasesInfo databasesDO);

    /** 生成代码并写入指定文件路径 */
    void generatorCode(MbgInfo gmbVO, DatabasesInfo databasesDO, String file);

    /** 使用主数据源生成代码并写入文件（需 DynamicRoutingDataSource） */
    void generatorCode(MbgInfo gmbVO, String file);
}

import cn.com.digitalhainan.devtools.mbg.factory.MysqlDatabases;
import cn.com.digitalhainan.devtools.mbg.common.MbgInfo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Arrays;

@Service
public class CodeGeneratorService {

    @Autowired
    private MysqlDatabases mysqlDatabases;

    public void generateCode() {
        MbgInfo mbgInfo = new MbgInfo()
                .setTablePrefix("t_")                          // 表前缀，生成类名时去除
                .setBasePackage("cn.com.digitalhainan.myapp")  // 基础包名
                .setModel("system")                            // 模块名
                .setTables(Arrays.asList("t_sys_user", "t_sys_role")) // 指定表
                .setAuthor("dev-team");

        // 写入文件
        mysqlDatabases.generatorCode(mbgInfo, "D:/output/code.zip");
    }
}

import cn.com.digitalhainan.devtools.mbg.factory.MysqlDatabases;
import cn.com.digitalhainan.devtools.mbg.common.MbgInfo;
import cn.com.digitalhainan.devtools.mbg.common.DatabasesInfo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Arrays;

@Service
public class ExternalCodeGeneratorService {

    @Autowired
    private MysqlDatabases mysqlDatabases;

    public byte[] generateFromExternalDB() {
        // 1. 构造生成参数
        MbgInfo mbgInfo = new MbgInfo()
                .setTablePrefix("biz_")
                .setBasePackage("cn.com.digitalhainan.newmodule")
                .setModel("order")
                .setTables(Arrays.asList("biz_order", "biz_order_item"))
                .setAuthor("code-gen");

        // 2. 构造数据库连接信息
        DatabasesInfo dbInfo = new DatabasesInfo()
                .setUrl("jdbc:mysql://192.168.1.100:3306/target_db?useSSL=false&characterEncoding=utf8")
                .setUserName("root")
                .setPassword("encrypted_password")
                .setDbType("mysql");

        // 3. 生成代码（返回 ZIP 字节数组）
        return mysqlDatabases.generatorCode(mbgInfo, dbInfo);
    }
}

import cn.com.digitalhainan.devtools.mbg.factory.DatabasesFactory;
import cn.com.digitalhainan.devtools.mbg.factory.IDatabases;
import cn.com.digitalhainan.devtools.mbg.common.MbgInfo;
import cn.com.digitalhainan.devtools.mbg.common.DatabasesInfo;

// 工厂方式获取（通过 MbgContext 兼容 Spring/非 Spring 环境）
IDatabases databases = DatabasesFactory.getDatabases("mysql");

MbgInfo mbgInfo = new MbgInfo()
        .setTablePrefix("")
        .setBasePackage("cn.com.digitalhainan.demo")
        .setModel("demo")
        .setAuthor("dev");

DatabasesInfo dbInfo = new DatabasesInfo()
        .setUrl("jdbc:mysql://localhost:3306/demo_db")
        .setUserName("root")
        .setPassword("password");

byte[] zipBytes = databases.generatorCode(mbgInfo, dbInfo);

createTime, version, updateTime, creator, updator, disable

import cn.com.digitalhainan.devtools.mbg.constants.DbTypeEnum;

// 当前仅支持 mysql，扩展示例：
// postgresql("org.postgresql.Driver")

import cn.com.digitalhainan.devtools.mbg.factory.AbstractDatabases;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

public class PostgresqlDatabases extends AbstractDatabases {

    @Override
    public ResultSet getTableResultSet(Statement statement) throws SQLException {
        // 实现 PostgreSQL 查询表信息的 SQL
        return statement.executeQuery(
            "SELECT table_name as \"tableName\", obj_description(relfilenode, 'pg_class') as \"tableComment\" " +
            "FROM pg_class c JOIN pg_namespace n ON n.oid = c.relnamespace " +
            "WHERE c.relkind = 'r' AND n.nspname = 'public'"
        );
    }

    @Override
    public ResultSet getColumnResultSet(Statement statement) throws SQLException {
        // 实现 PostgreSQL 查询列信息的 SQL
        return statement.executeQuery(
            "SELECT column_name, data_type, column_default, is_nullable " +
            "FROM information_schema.columns WHERE table_schema = 'public'"
        );
    }

    @Override
    protected String transformColumnAttrType(String columnType) {
        // 实现 PostgreSQL 类型到 Java 类型的映射
        return columnType;
    }

    @Override
    protected String driverClassName() {
        return "org.postgresql.Driver";
    }
}

import cn.com.digitalhainan.devtools.mbg.factory.DatabasesFactory;
import cn.com.digitalhainan.devtools.mbg.factory.IDatabases;

// dataMap.put("postgresql", PostgresqlDatabases.class);