package com.battery.business.runtime;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
import org.junit.jupiter.api.Test;
import org.springframework.core.io.ClassPathResource;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.init.ScriptUtils;
import com.battery.device.api.model.AdapterCapabilities;
import com.battery.device.api.registry.AdapterInstanceConfig;
import com.battery.device.api.registry.AdapterRegistryException;
import com.battery.device.api.registry.DeviceAdapterRegistry;

/**
 * 适配器实例按存储形态放宽与来源形态配置列的迁移契约：
 * 迁移把 {@code source_config} 加在能力列之后，运行库上该列确实存在且可空；
 * 类型白名单放宽到直连厂商库，未知类型仍然拒绝。
 */
class VendorSourceInstanceMigrationContractTest
{
    @Test
    void migrationAddsNullableSourceConfigColumnWithoutTouchingExistingRows() throws Exception
    {
        String path = "db/migration-final/V22__vendor_source_instance_config.sql";
        String migration = resource(path);
        assertTrue(migration.contains("ALTER TABLE t_device_adapter ADD COLUMN source_config JSON NULL"));
        assertTrue(migration.contains("AFTER capabilities"));
        // 迁移只加列：不改既有列、不写入任何数据，更不写入凭据。
        assertFalse(migration.matches("(?s).*\\b(INSERT|UPDATE|DELETE|DROP)\\b.*"));
        assertFalse(migration.toLowerCase().contains("password"));

        var source = new UnpooledDataSource("org.h2.Driver",
                "jdbc:h2:mem:" + UUID.randomUUID() + ";MODE=MySQL;DATABASE_TO_LOWER=TRUE", "sa", "");
        try (var connection = source.getConnection())
        {
            JdbcTemplate jdbc = new JdbcTemplate(source);
            jdbc.execute("create table t_device_adapter(adapter_id bigint auto_increment primary key,"
                    + "adapter_code varchar(64), capabilities json)");
            ScriptUtils.executeSqlScript(connection, new ClassPathResource(path));

            // 来源形态配置只承载非敏感项；连接口令由环境变量提供，不入库。
            jdbc.update("insert into t_device_adapter(adapter_code, capabilities, source_config) values(?,?,?)",
                    "neware76", "{\"control\":false}", "{\"database\":\"bts63\",\"user\":\"visitor\"}");
            assertEquals(1, jdbc.queryForObject(
                    "select count(*) from t_device_adapter where source_config is not null", Integer.class));
            // 没有来源形态配置的行（列可空）仍允许存在。
            jdbc.update("insert into t_device_adapter(adapter_code, capabilities) values(?,?)", "legacy-row",
                    "{\"control\":false}");
            assertEquals(1, jdbc.queryForObject(
                    "select count(*) from t_device_adapter where source_config is null", Integer.class));
        }
    }

    @Test
    void sourceConfigCarriesOnlyNonSensitiveOptionsAndUnknownTypesStayRejected()
    {
        AdapterInstanceConfig config = new AdapterInstanceConfig("neware76", "NEWARE_MYSQL", "26.1.2.3", 3306,
                60000, "Asia/Hong_Kong", true, AdapterCapabilities.observeOnly(),
                Map.of("database", "bts63", "user", "visitor", "passwordEnv", "BATTERY_VENDOR_DB_PASSWORD"));

        assertEquals("bts63", config.sourceOption("database"));
        assertTrue(config.capabilities().discovery());
        assertFalse(config.capabilities().control());

        try (DeviceAdapterRegistry registry = new DeviceAdapterRegistry(List.of()))
        {
            // 放宽不等于接受任何类型：未注册的存储形态仍然明确失败。
            assertThrows(AdapterRegistryException.class,
                    () -> registry.synchronize(new AdapterInstanceConfig("mystery", "SOME_OTHER_VENDOR",
                            "127.0.0.1", 3306, 60000, "Asia/Hong_Kong", true,
                            AdapterCapabilities.observeOnly(), Map.of())));
        }
    }

    private static String resource(String path)
    {
        try (InputStream stream = VendorSourceInstanceMigrationContractTest.class.getClassLoader()
                .getResourceAsStream(path))
        {
            if (stream == null) throw new IllegalStateException("缺少迁移文件：" + path);
            return new String(stream.readAllBytes(), StandardCharsets.UTF_8);
        }
        catch (java.io.IOException exception)
        {
            throw new IllegalStateException("无法读取迁移文件：" + path, exception);
        }
    }
}