package cn.iocoder.yudao.module.qcreport.service.aiimport.llm;

import cn.iocoder.yudao.module.qcreport.service.aiimport.document.QcReportDocumentExtract;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;

import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.List;

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

/**
 * 调用计划。
 * <p>
 * 这是「一个文件该发几次模型调用」这个控制流的唯一落点，所以它被抽成纯函数、可以在这里被完整跑一遍，
 * 而不必在单测里 mock 掉真正的外部依赖。
 * <p>
 * 另有一条接口细节必须钉死：交给 {@code AiChatApi.chatWithImage} 的必须是<b>裸 base64</b>。
 * 那边拿到参数直接 {@code Base64.getDecoder().decode(...)}，带 {@code data:image/png;base64,}
 * 前缀就会抛 {@code IllegalArgumentException}——这个失败在联调时才暴露，且报错信息不会提到前缀。
 */
class QcReportLlmCallPlannerTest {

    @Test
    @DisplayName("有文本层的文件整份拼一次调用，不逐页")
    void textChannelMakesSingleCall() {
        QcReportDocumentExtract extract = QcReportDocumentExtract.ofText(
                "第一页内容\n--- 第 2 页 ---\n第二页内容", 2, "pdfbox-text", List.of());

        List<QcReportLlmCall> calls = QcReportLlmCallPlanner.plan(extract, "报告.pdf");

        assertEquals(1, calls.size(), "文本通道逐页发调用既费额度又会让页间上下文断掉");
        assertEquals(QcReportLlmCall.Kind.TEXT, calls.get(0).kind());
        assertEquals("报告.pdf", calls.get(0).sourceLabel());
        assertTrue(calls.get(0).text().contains("第二页内容"), "整份正文必须一次带上");
    }

    @Test
    @DisplayName("扫描件逐页一次调用，标签带页码以便定位是哪一页出的问题")
    void imageChannelMakesOneCallPerPage() {
        QcReportDocumentExtract extract = QcReportDocumentExtract.ofImages(List.of(
                image("第 1 页"),
                image("第 2 页"),
                image("第 3 页")), 3, "pdfbox-render", List.of());

        List<QcReportLlmCall> calls = QcReportLlmCallPlanner.plan(extract, "扫描件.pdf");

        assertEquals(3, calls.size(), "多模态接口一次只收一张图，页数就是调用数");
        for (QcReportLlmCall call : calls) {
            assertEquals(QcReportLlmCall.Kind.IMAGE, call.kind());
            assertTrue(call.sourceLabel().startsWith("扫描件.pdf 第"), "标签必须能定位到具体页");
            assertEquals("image/png", call.mimeType());
            assertTrue(call.text().isEmpty(), "图片调用不该再带正文");
        }
        assertEquals(List.of("扫描件.pdf 第 1 页", "扫描件.pdf 第 2 页", "扫描件.pdf 第 3 页"),
                calls.stream().map(QcReportLlmCall::sourceLabel).toList());
    }

    @Test
    @DisplayName("图片是裸 base64：带 data: 前缀会让 AiChatApiImpl 的 Base64 解码直接抛异常")
    void imageBase64IsRaw() {
        byte[] png = "假装这是 PNG 字节".getBytes(StandardCharsets.UTF_8);
        QcReportDocumentExtract extract = QcReportDocumentExtract.ofImages(
                List.of(new QcReportDocumentExtract.ImagePart(png, "image/png", "第 1 页")),
                1, "raw-image", List.of());

        String base64 = QcReportLlmCallPlanner.plan(extract, "a.png").get(0).imageBase64();

        assertEquals(Base64.getEncoder().encodeToString(png), base64,
                "必须是裸 base64：接口那边不做剥前缀处理就直接解码，多一个字符都会炸");
        assertArrayEquals(png, Base64.getDecoder().decode(base64), "解码回来必须还是原字节");
        assertThrows(IllegalArgumentException.class,
                () -> Base64.getDecoder().decode("data:image/png;base64," + base64),
                "带前缀会解码失败——这条断言把那个坑钉在单测里，免得联调时才发现");
    }

    private static QcReportDocumentExtract.ImagePart image(String label) {
        return new QcReportDocumentExtract.ImagePart(label.getBytes(StandardCharsets.UTF_8), "image/png", label);
    }

}