package cn.iocoder.yudao.module.mdm.service.agent;

import cn.iocoder.yudao.module.mdm.controller.admin.agent.vo.MdmAgentImportPreflightReqVO;
import cn.iocoder.yudao.module.mdm.controller.admin.agent.vo.MdmAgentImportPreflightRespVO;
import cn.iocoder.yudao.module.mdm.controller.admin.item.vo.MdmItemImportExcelVO;
import cn.iocoder.yudao.module.mdm.controller.admin.item.vo.MdmItemSaveReqVO;
import cn.iocoder.yudao.module.mdm.dal.dataobject.item.MdmItemDO;
import cn.iocoder.yudao.module.mdm.service.item.MdmItemImportContext;
import cn.iocoder.yudao.module.mdm.service.item.MdmItemRowResolution;
import cn.iocoder.yudao.module.mdm.service.item.MdmItemService;
import cn.iocoder.yudao.module.system.framework.agent.AgentPrincipal;
import cn.iocoder.yudao.module.system.service.storage.SystemAgentAttachmentRef;
import cn.iocoder.yudao.module.system.service.storage.SystemAgentImportBridgeService;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.mockito.junit.jupiter.MockitoSettings;
import org.mockito.quality.Strictness;

import java.nio.file.Path;
import java.util.List;
import java.util.Map;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
@MockitoSettings(strictness = Strictness.LENIENT)
class MdmAgentImportPreflightServiceTest {

    private static final String SHA = "a".repeat(64);

    @Mock private SystemAgentImportBridgeService bridgeService;
    @Mock private MdmItemService itemService;
    @Mock private MdmAgentImportPolicy policy;
    @InjectMocks private MdmAgentImportPreflightService service;

    private final AgentPrincipal principal = new AgentPrincipal(7L, 0L, "live");

    @Test
    void classifiesCreateSkipAndBlockedRows() {
        stubAttachmentAndPolicy();
        MdmItemDO existing = new MdmItemDO();
        existing.setId(99L);
        when(itemService.getItemListByCode("A")).thenReturn(List.of());
        when(itemService.getItemListByCode("B")).thenReturn(List.of(existing));

        MdmAgentImportPreflightRespVO result = service.preflight(
                request("A", "B", "C"), principal);

        assertEquals(3, result.getSummary().getTotal());
        assertEquals(1, result.getSummary().getCreate());
        assertEquals(1, result.getSummary().getSkip());
        assertEquals(1, result.getSummary().getBlocked());
        assertEquals("CREATE", result.getRows().get(0).getAction());
        assertEquals("SKIP", result.getRows().get(1).getAction());
        assertEquals(99L, result.getRows().get(1).getTargetItemId());
        assertFalse(result.getRows().get(1).getWarnings().isEmpty());
        assertEquals("BLOCKED", result.getRows().get(2).getAction());
        assertEquals("CATEGORY_NOT_FOUND", result.getRows().get(2).getBlockers().get(0).getCode());
        assertFalse(result.getCanCommit());
    }

    @Test
    void duplicateCodeInFileIsBlocked() {
        stubAttachmentAndPolicy();
        when(itemService.getItemListByCode(anyString())).thenReturn(List.of());

        MdmAgentImportPreflightRespVO result = service.preflight(
                request("A", "A"), principal);

        assertEquals(2, result.getSummary().getBlocked());
        assertEquals("DUPLICATE_CODE_IN_FILE", result.getRows().get(1).getBlockers().get(0).getCode());
    }

    @Test
    void hashIsStableAcrossIdenticalRuns() {
        stubAttachmentAndPolicy();
        when(itemService.getItemListByCode("A")).thenReturn(List.of());

        MdmAgentImportPreflightRespVO first = service.preflight(request("A"), principal);
        MdmAgentImportPreflightRespVO second = service.preflight(request("A"), principal);

        assertTrue(first.getCanCommit());
        assertEquals(first.getPreflightHash(), second.getPreflightHash());
    }

    private void stubAttachmentAndPolicy() {
        when(bridgeService.requireOwnedAttachmentRef(5L))
                .thenReturn(new SystemAgentAttachmentRef(5L, 21L, "materials.csv", 100L, Path.of("x")));
        when(bridgeService.sha256OfAttachment(5L)).thenReturn(SHA);
        when(policy.getMaxRows()).thenReturn(5000);
        when(policy.isAllowMesSync()).thenReturn(false);
        MdmItemImportContext context = new MdmItemImportContext(
                Map.of("RAW", 11L), Map.of("RAW", "原料"),
                Map.of("PCS", 22L), Map.of("PCS", "个"));
        when(itemService.buildImportContext(any())).thenReturn(context);
        when(itemService.resolveImportRow(any(MdmItemImportExcelVO.class), any()))
                .thenAnswer(invocation -> MdmItemRowResolution.success(new MdmItemSaveReqVO()));
    }

    private static MdmAgentImportPreflightReqVO request(String... codes) {
        MdmAgentImportPreflightReqVO reqVO = new MdmAgentImportPreflightReqVO();
        reqVO.setAttachmentId(5L);
        reqVO.setSessionId(21L);
        reqVO.setSourceSha256(SHA);
        reqVO.setProposalRevision("rev-1");
        reqVO.setRows(java.util.stream.IntStream.range(0, codes.length).mapToObj(index -> {
            String code = codes[index];
            MdmAgentImportPreflightReqVO.Row row = new MdmAgentImportPreflightReqVO.Row();
            row.setCandidateId("c-" + index);
            row.setRowNo(index + 1);
            row.setCode(code);
            row.setName("物料" + code);
            row.setCategoryCode("C".equals(code) ? "NOPE" : "RAW");
            row.setUnitMeasureCode("PCS");
            return row;
        }).toList());
        return reqVO;
    }
}