package com.rongyichuang.judge;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.rongyichuang.common.entity.Media;
import com.rongyichuang.judge.entity.Judge;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.web.client.TestRestTemplate;
import org.springframework.boot.test.web.server.LocalServerPort;
import org.springframework.core.io.ClassPathResource;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.util.StreamUtils;

import javax.sql.DataSource;
import java.nio.charset.StandardCharsets;
import java.util.Collections;
import java.util.Map;

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

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class JudgeCreationE2ETest {

    @LocalServerPort
    private int port;

    @Autowired
    private TestRestTemplate restTemplate;

    @Autowired
    private DataSource dataSource;

    private JdbcTemplate jdbcTemplate;

    @BeforeEach
    void setUp() {
        jdbcTemplate = new JdbcTemplate(dataSource);
        // Clean up database before each test
        jdbcTemplate.execute("DELETE FROM t_judge_tag");
        jdbcTemplate.execute("DELETE FROM t_judge");
        jdbcTemplate.execute("DELETE FROM t_media");
    }

    @Test
    void testCreateJudgeEndToEnd() throws Exception {
        // 1. Save Media Info (Simulating image upload)
        String mediaQuery = readGraphQLQuery("graphql/test-save-media.graphql");
        String mediaFileName = "logo.png";
        String mediaFilePath = "test/logo.png";
        String formattedMediaQuery = String.format(mediaQuery, mediaFileName, mediaFilePath, 12345L, "image/png", "cos", "test-bucket", "ap-guangzhou");

        Map<String, String> mediaRequest = Collections.singletonMap("query", formattedMediaQuery);
        String mediaResponse = postGraphQL(mediaRequest);

        ObjectMapper objectMapper = new ObjectMapper();
        Map<String, Object> mediaResponseMap = objectMapper.readValue(mediaResponse, Map.class);
        String mediaId = ((Map<String, String>)((Map<String, Object>) mediaResponseMap.get("data")).get("saveMedia")).get("id");
        assertNotNull(mediaId, "Media ID should not be null");

        // 2. Verify Media in DB with JDBC
        Map<String, Object> mediaRecord = jdbcTemplate.queryForMap("SELECT * FROM t_media WHERE id = ?", mediaId);
        assertEquals(mediaFileName, mediaRecord.get("file_name"));
        assertEquals(mediaFilePath, mediaRecord.get("file_path"));

        // 3. Save Judge Info
        String judgeQuery = readGraphQLQuery("graphql/test-save-judge.graphql");
        String formattedJudgeQuery = String.format(judgeQuery, "测试评委", "13800138000", mediaId);
        Map<String, String> judgeRequest = Collections.singletonMap("query", formattedJudgeQuery);
        String judgeResponse = postGraphQL(judgeRequest);
        Map<String, Object> judgeResponseMap = objectMapper.readValue(judgeResponse, Map.class);
        String judgeId = ((Map<String, String>)((Map<String, Object>) judgeResponseMap.get("data")).get("saveJudge")).get("id");
        assertNotNull(judgeId, "Judge ID should not be null");

        // 4. Verify Judge in DB with JDBC
        Map<String, Object> judgeRecord = jdbcTemplate.queryForMap("SELECT * FROM t_judge WHERE id = ?", judgeId);
        assertEquals("测试评委", judgeRecord.get("name"));
        assertEquals("13800138000", judgeRecord.get("phone"));
        assertEquals(Long.valueOf(mediaId), judgeRecord.get("avatar_media_id"));

        System.out.println("--- E2E Test Passed ---");
        System.out.println("Judge created with ID: " + judgeId);
        System.out.println("Media created with ID: " + mediaId);
        System.out.println("--- Verification with JDBC successful ---");
    }

    private String postGraphQL(Map<String, String> requestBody) {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(org.springframework.http.MediaType.APPLICATION_JSON);
        HttpEntity<Map<String, String>> entity = new HttpEntity<>(requestBody, headers);
        String url = "http://localhost:" + port + "/graphql";
        org.springframework.http.ResponseEntity<String> response = restTemplate.postForEntity(url, entity, String.class);

        if (response.getStatusCode() != org.springframework.http.HttpStatus.OK || !response.hasBody()) {
            throw new RuntimeException("GraphQL request failed with status: " + response.getStatusCode() + " and body: " + response.getBody());
        }

        return response.getBody();
    }

    private String readGraphQLQuery(String path) throws Exception {
        ClassPathResource resource = new ClassPathResource(path);
        return StreamUtils.copyToString(resource.getInputStream(), StandardCharsets.UTF_8);
    }
}