package cn.iocoder.yudao.module.system.service.storage;

import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.system.dal.dataobject.storage.SystemAgentImportSessionDO;
import cn.iocoder.yudao.module.system.dal.dataobject.storage.SystemStorageAttachmentDO;
import cn.iocoder.yudao.module.system.dal.dataobject.storage.SystemStorageBlobDO;
import cn.iocoder.yudao.module.system.dal.mysql.storage.SystemAgentImportSessionMapper;
import cn.iocoder.yudao.module.system.dal.mysql.storage.SystemStorageAttachmentMapper;
import cn.iocoder.yudao.module.system.framework.agent.AgentBridgeException;
import cn.iocoder.yudao.module.system.framework.storage.config.StorageProperties;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.junit.jupiter.api.io.TempDir;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockedStatic;
import org.mockito.junit.jupiter.MockitoExtension;

import java.nio.file.Files;
import java.nio.file.Path;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
class SystemAgentImportBridgeServiceTest {

    @Mock private SystemAgentImportSessionMapper sessionMapper;
    @Mock private SystemStorageAttachmentMapper attachmentMapper;
    @Mock private SystemStorageBlobService blobService;
    @Mock private SystemStorageAttachmentService attachmentService;
    @Mock private StorageProperties storageProperties;
    @InjectMocks private SystemAgentImportBridgeService service;

    @Test
    void foreignSessionCannotReadAttachment() {
        SystemStorageAttachmentDO attachment = new SystemStorageAttachmentDO();
        attachment.setRecordType("agent_import_session");
        attachment.setRecordId(21L);
        attachment.setStorageBlobId(90L);
        when(attachmentMapper.selectById(5L)).thenReturn(attachment);
        SystemAgentImportSessionDO session = new SystemAgentImportSessionDO();
        session.setOwnerUserId(8L);
        session.setStatus("active");
        when(sessionMapper.selectById(21L)).thenReturn(session);
        try (MockedStatic<SecurityFrameworkUtils> auth = mockStatic(SecurityFrameworkUtils.class)) {
            auth.when(SecurityFrameworkUtils::getLoginUserId).thenReturn(7L);
            assertThrows(AgentBridgeException.class, () -> service.requireOwnedAttachmentRef(5L));
        }
        verifyNoInteractions(blobService);
    }

    @Test
    void ownSessionCannotBindForeignUpload() {
        SystemAgentImportSessionDO session = new SystemAgentImportSessionDO();
        session.setOwnerUserId(7L);
        session.setStatus("active");
        when(sessionMapper.selectById(21L)).thenReturn(session);
        SystemStorageBlobDO blob = new SystemStorageBlobDO();
        blob.setCreator("8");
        blob.setByteSize(100L);
        when(blobService.getStorageBlob(90L)).thenReturn(blob);
        try (MockedStatic<SecurityFrameworkUtils> auth = mockStatic(SecurityFrameworkUtils.class)) {
            auth.when(SecurityFrameworkUtils::getLoginUserId).thenReturn(7L);
            assertThrows(AgentBridgeException.class, () -> service.bindAttachment(21L, 90L));
        }
        verifyNoInteractions(attachmentService);
    }

    @Test
    void ownerReadsBoundAttachment(@TempDir Path tempDir) throws Exception {
        Path blobDir = Files.createDirectories(tempDir.resolve("2026/1022"));
        byte[] content = "code,name\nA,物料A\n".getBytes(java.nio.charset.StandardCharsets.UTF_8);
        Files.write(blobDir.resolve("uuid-materials.csv"), content);
        when(storageProperties.getPath()).thenReturn(tempDir.toString());

        SystemStorageAttachmentDO attachment = new SystemStorageAttachmentDO();
        attachment.setRecordType("agent_import_session");
        attachment.setRecordId(21L);
        attachment.setStorageBlobId(90L);
        when(attachmentMapper.selectById(5L)).thenReturn(attachment);
        SystemAgentImportSessionDO session = new SystemAgentImportSessionDO();
        session.setOwnerUserId(7L);
        session.setStatus("active");
        when(sessionMapper.selectById(21L)).thenReturn(session);
        SystemStorageBlobDO blob = new SystemStorageBlobDO();
        blob.setCreator("7");
        blob.setByteSize((long) content.length);
        blob.setOriginalFilename("materials.csv");
        blob.setPath("2026/1022");
        blob.setUidFilename("uuid-materials.csv");
        when(blobService.getStorageBlob(90L)).thenReturn(blob);
        try (MockedStatic<SecurityFrameworkUtils> auth = mockStatic(SecurityFrameworkUtils.class)) {
            auth.when(SecurityFrameworkUtils::getLoginUserId).thenReturn(7L);
            SystemAgentAttachmentRef ref = service.requireOwnedAttachmentRef(5L);
            assertEquals("materials.csv", ref.originalFilename());
            assertEquals(21L, ref.sessionId());
            assertEquals(64, service.sha256OfAttachment(5L).length());
        }
    }
}