package jnpf.dmsService.search;

import jnpf.dmsEntity.search.model.DmsSearchIndexSource;
import jnpf.dmsMapper.DmsSearchIndexSourceMapper;
import jnpf.dmsService.search.extract.DmsTextExtractionResult;
import jnpf.dmsService.search.extract.DmsTextExtractionService;
import jnpf.dmsService.search.file.DmsSearchFileSourceService;
import jnpf.dmsService.search.file.DmsTemporarySearchFile;
import jnpf.dmsService.search.meili.DmsMeilisearchClient;
import jnpf.dmsService.search.meili.DmsSearchIndexDocument;
import jnpf.dmsService.search.ocr.DmsOcrAdapterRegistry;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;

import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Optional;
import java.util.UUID;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

class DmsSearchIndexProcessorTest {

    @Test
    void extractsUploadedFileAndWritesMeilisearchDocument() throws Exception {
        DmsSearchIndexSourceMapper sourceMapper = mock(DmsSearchIndexSourceMapper.class);
        DmsSearchFileSourceService fileSource = mock(DmsSearchFileSourceService.class);
        DmsTextExtractionService extraction = mock(DmsTextExtractionService.class);
        DmsOcrAdapterRegistry ocr = mock(DmsOcrAdapterRegistry.class);
        DmsSearchIndexWriter meilisearch = mock(DmsSearchIndexWriter.class);
        DmsSearchIndexProcessor processor = new DmsSearchIndexProcessor(sourceMapper, fileSource,
                extraction, ocr, meilisearch);
        DmsSearchIndexTask task = task();
        DmsSearchIndexSource source = source(task);
        Path temporary = Files.createTempFile("dms-search-test-", ".docx");

        when(sourceMapper.selectSource(task.getTenantId(), task.getDocumentId(),
                task.getFileVersionId())).thenReturn(source);
        when(fileSource.fetch(task)).thenReturn(new DmsTemporarySearchFile(
                temporary, "quality.docx", "application/docx"));
        when(extraction.extract(temporary, "quality.docx"))
                .thenReturn(new DmsTextExtractionResult("检验操作规程", false));
        when(ocr.configuredAdapter()).thenReturn(Optional.empty());

        processor.process(task);

        ArgumentCaptor<DmsSearchIndexDocument> captor =
                ArgumentCaptor.forClass(DmsSearchIndexDocument.class);
        verify(meilisearch).publish(captor.capture());
        assertEquals("检验操作规程", captor.getValue().getContent());
        assertEquals(task.getFileVersionId().toString(), captor.getValue().getId());
    }

    private DmsSearchIndexTask task() {
        DmsSearchIndexTask task = new DmsSearchIndexTask();
        task.setTenantId("tenant-1");
        task.setDocumentId(UUID.randomUUID());
        task.setFileVersionId(UUID.randomUUID());
        task.setChecksumSha256("a".repeat(64));
        return task;
    }

    private DmsSearchIndexSource source(DmsSearchIndexTask task) {
        DmsSearchIndexSource source = new DmsSearchIndexSource();
        source.setTenantId(task.getTenantId());
        source.setDocumentId(task.getDocumentId());
        source.setFileVersionId(task.getFileVersionId());
        source.setDocumentName("检验操作规程");
        source.setFileName("quality.docx");
        return source;
    }
}