package jnpf.dmsMapper;

import com.fasterxml.jackson.databind.ObjectMapper;
import jnpf.dmsDatabase.DmsUuidTypeHandler;
import jnpf.dmsEntity.permission.entity.DmsFileVersionEntity;
import jnpf.dmsService.DmsVersionPropertiesService;
import org.apache.ibatis.builder.xml.XMLMapperBuilder;
import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
import org.apache.ibatis.mapping.Environment;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;

import java.io.StringReader;
import java.nio.charset.StandardCharsets;
import java.sql.SQLException;
import java.time.LocalDate;
import java.util.UUID;

import static org.junit.jupiter.api.Assertions.*;

/** Uses the target schema's real constraints/guard on a connection-local table; always rolls back. */
@EnabledIfEnvironmentVariable(named = "DMS_PROPERTIES_TEST_URL", matches = ".+")
class DmsVersionPropertiesPostgresTest {
    @Test void insertsSnapshotAtomicallyWithoutRelaxingExistingVersionGuards() throws Exception {
        var dataSource = new UnpooledDataSource("org.postgresql.Driver", System.getenv("DMS_PROPERTIES_TEST_URL"),
                System.getenv("DMS_PROPERTIES_TEST_USER"), System.getenv("DMS_PROPERTIES_TEST_PASSWORD"));
        var config = new Configuration(new Environment("properties-test", new JdbcTransactionFactory(), dataSource));
        config.getTypeHandlerRegistry().register(DmsUuidTypeHandler.class);
        config.setDefaultStatementTimeout(15);
        String path = "mapper/DmsVersionPropertiesMapper.xml";
        try (var input = getClass().getResourceAsStream("/" + path)) {
            assertNotNull(input);
            String xml = new String(input.readAllBytes(), StandardCharsets.UTF_8)
                    .replace("public.dms_file_versions", "pg_temp.dms_version_properties_test");
            new XMLMapperBuilder(new StringReader(xml), config, path, config.getSqlFragments()).parse();
        }
        try (var session = new SqlSessionFactoryBuilder().build(config).openSession(false)) {
            try {
                var connection = session.getConnection();
                try (var statement = connection.createStatement()) {
                    statement.execute("CREATE TEMP TABLE dms_version_properties_test "
                            + "(LIKE public.dms_file_versions INCLUDING DEFAULTS INCLUDING CONSTRAINTS INCLUDING INDEXES) ON COMMIT DROP");
                    statement.execute("CREATE TRIGGER metadata_guard BEFORE UPDATE ON pg_temp.dms_version_properties_test "
                            + "FOR EACH ROW EXECUTE FUNCTION public.dms_guard_file_version_metadata_completion()");
                }
                var mapper = session.getMapper(DmsVersionPropertiesMapper.class);
                var service = new DmsVersionPropertiesService(mapper);
                var json = new ObjectMapper();
                var properties = json.readTree("{\"prepared_by_name\":\"管理员\",\"reviewers\":[{\"user_id\":\"859053987238827589\",\"name\":\"\"}]}");
                var row = version("018f0000-0000-7000-8000-000000000701", 1);
                service.insertVersion(row, properties);
                assertEquals(properties, mapper.find("properties-test", row.getId()).getExtendedProperties());
                service.verifyRetry("properties-test", row.getId(), properties);
                assertNull(mapper.find("another-tenant", row.getId()));

                var legacy = version("018f0000-0000-7000-8000-000000000702", 2);
                service.insertVersion(legacy, null);
                assertNull(mapper.find("properties-test", legacy.getId()).getExtendedProperties());
                try (var statement = connection.createStatement();
                     var result = statement.executeQuery("SELECT count(*) FROM pg_temp.dms_version_properties_test WHERE created_at IS NOT NULL")) {
                    assertTrue(result.next());
                    assertEquals(2, result.getInt(1));
                }
                // Both the previously failing snapshot-only update and identity mutation stay forbidden.
                for (String assignment : new String[]{"extended_properties = '{}'::jsonb", "file_name = 'changed.docx'"}) {
                    var savepoint = connection.setSavepoint();
                    try (var statement = connection.createStatement()) {
                        SQLException error = assertThrows(SQLException.class, () -> statement.executeUpdate(
                                "UPDATE pg_temp.dms_version_properties_test SET " + assignment + " WHERE version_no = 1"));
                        assertEquals("55000", error.getSQLState());
                    } finally {
                        connection.rollback(savepoint);
                    }
                }
                session.clearCache();
                assertEquals(properties, mapper.find("properties-test", row.getId()).getExtendedProperties());
            } finally {
                session.rollback(true);
            }
        }
    }

    private DmsFileVersionEntity version(String id, long number) {
        var row = new DmsFileVersionEntity();
        row.setId(UUID.fromString(id));
        row.setTenantId("properties-test");
        row.setDocumentId(UUID.fromString("018f0000-0000-7000-8000-000000000700"));
        row.setVersionNo(number);
        row.setVersionLabel(number == 1 ? "D(00)" : "D(01)");
        row.setFileName("test.docx");
        row.setSizeBytes(10L);
        row.setChecksumSha256("a".repeat(64));
        row.setPreparedBy("test-user");
        row.setPreparedOn(LocalDate.of(2026, 10, 9));
        row.setShifouShoukong("yes");
        row.setCreatedBy("test-user");
        return row;
    }
}