package jnpf.fileStorage.dms;

import jnpf.base.ActionResult;
import jnpf.dmsEntity.download.model.DmsFileDownloadTicket;
import jnpf.dmsEntity.preview.model.DmsFilePreviewTicket;
import jnpf.dmsEntity.preview.model.DmsWatermarkSnapshot;
import jnpf.exception.DataException;
import jnpf.onlyoffice.PageWatermarkApi;
import jnpf.onlyoffice.PageWatermarkRequest;
import org.junit.Test;
import java.io.ByteArrayOutputStream;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.atomic.AtomicReference;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
import static org.mockito.ArgumentMatchers.*;

public class DmsFileWatermarkServiceTest {
    @Test
    public void previewPrintAndDownloadDelegateCleanSourceAndKeepAllPagesWatermarksIndependent() throws Exception {
        for (String scope : List.of(DmsFilePreviewTicket.SCOPE, DmsFilePreviewTicket.PRINT_SCOPE, DmsFileDownloadTicket.SCOPE)) {
            DmsDocxWatermarkService docx = mock(DmsDocxWatermarkService.class);
            PageWatermarkApi renderer = mock(PageWatermarkApi.class);
            AtomicReference<DmsFilePreviewTicket> delegated = new AtomicReference<>();
            DmsFileWatermarkService service = new DmsFileWatermarkService(docx, renderer, ticket -> {
                delegated.set(ticket);
                return "server-only-source-ticket";
            });
            byte[] source = {1, 2, 3};
            DmsWatermarkSnapshot custom = new DmsWatermarkSnapshot();
            custom.setContent("REC-001");
            custom.setRenderPages(List.of(1, 3, 5));
            DmsWatermarkSnapshot everyPage = new DmsWatermarkSnapshot();
            everyPage.setContent("Original watermark");
            List<PageWatermarkRequest.Overlay> overlays = List.of(new PageWatermarkRequest.Overlay(List.of(1, 3, 5), List.of()));
            when(docx.pageOverlays(source, List.of(custom))).thenReturn(overlays);
            when(renderer.render(any())).thenAnswer(invocation -> {
                PageWatermarkRequest request = invocation.getArgument(0);
                assertEquals(overlays, request.getOverlays());
                assertEquals("server-only-source-ticket", request.getSourceTicket());
                ActionResult<byte[]> result = new ActionResult<>();
                result.setCode(200);
                result.setData(drawings(request.getMarkerPrefix()));
                return result;
            });
            when(docx.apply(any(byte[].class), eq(List.of(everyPage)))).thenReturn(new byte[]{4, 5, 6});
            DmsFilePreviewTicket ticket = new DmsFilePreviewTicket();
            ticket.setScope(scope);
            ticket.setFileVersionId(UUID.randomUUID());
            ticket.setTenantId("tenant-1");
            ticket.setUserId("user-1");
            ticket.setFileName("original.docx");
            ticket.setObjectName("original.docx");
            ticket.setWatermarks(List.of(custom, everyPage));
            byte[] result;
            if (DmsFileDownloadTicket.SCOPE.equals(scope)) {
                DmsFileDownloadTicket download = new DmsFileDownloadTicket();
                org.springframework.beans.BeanUtils.copyProperties(ticket, download);
                result = service.apply(source, download);
            } else result = service.apply(source, ticket);
            assertArrayEquals(new byte[]{4, 5, 6}, result);
            assertEquals(DmsFilePreviewTicket.SCOPE, delegated.get().getScope());
            assertEquals(ticket.getFileVersionId(), delegated.get().getFileVersionId());
            assertEquals("tenant-1", delegated.get().getTenantId());
            assertEquals("original.docx", delegated.get().getObjectName());
            assertTrue(delegated.get().getWatermarks().isEmpty());
            assertTrue(delegated.get().getExpiresAt() > System.currentTimeMillis());
            assertEquals(List.of(custom, everyPage), ticket.getWatermarks());
            assertEquals(scope, ticket.getScope());
            verify(docx).apply(any(byte[].class), eq(List.of(everyPage)));
        }
    }

    @Test
    public void aFailedPageJobNeverFallsBackToEveryPageOrUnwatermarkedSource() throws Exception {
        DmsDocxWatermarkService docx = mock(DmsDocxWatermarkService.class);
        PageWatermarkApi renderer = mock(PageWatermarkApi.class);
        DmsFileWatermarkService service = new DmsFileWatermarkService(docx, renderer, ignored -> "source-ticket");
        DmsWatermarkSnapshot mark = new DmsWatermarkSnapshot();
        mark.setRenderPages(List.of(2));
        DmsFilePreviewTicket ticket = new DmsFilePreviewTicket();
        ticket.setFileName("original.docx");
        ticket.setWatermarks(List.of(mark));
        when(docx.pageOverlays(any(), any())).thenReturn(List.of(new PageWatermarkRequest.Overlay(List.of(2), List.of())));
        when(renderer.render(any())).thenThrow(new DataException("水印页码 2 超过文档总页数 1"));
        assertThrows(DataException.class, () -> service.apply(new byte[]{1}, ticket));
        verify(docx, never()).apply(any(), any());
    }

    @Test
    public void previewAndDownloadImagesUseRasterWatermarkWithoutOnlyOffice() {
        DmsDocxWatermarkService docx = mock(DmsDocxWatermarkService.class);
        DmsImageWatermarkService image = mock(DmsImageWatermarkService.class);
        PageWatermarkApi renderer = mock(PageWatermarkApi.class);
        DmsFileWatermarkService service = new DmsFileWatermarkService(docx, image, renderer,
                ignored -> "unused-ticket");
        byte[] source = {1, 2, 3};
        byte[] watermarked = {4, 5, 6};
        DmsWatermarkSnapshot mark = new DmsWatermarkSnapshot();
        List<DmsWatermarkSnapshot> marks = List.of(mark);
        when(image.apply(source, "photo.PNG", null, marks)).thenReturn(watermarked);

        DmsFilePreviewTicket preview = new DmsFilePreviewTicket();
        preview.setFileName("photo.PNG");
        preview.setWatermarks(marks);
        DmsFileDownloadTicket download = new DmsFileDownloadTicket();
        download.setFileName("photo.PNG");
        download.setWatermarks(marks);

        assertArrayEquals(watermarked, service.apply(source, preview));
        assertArrayEquals(watermarked, service.apply(source, download));
        verify(image, times(2)).apply(source, "photo.PNG", null, marks);
        verifyNoInteractions(docx, renderer);
    }

    @Test
    public void previewPrintAndDownloadPdfUsePdfWatermark() {
        DmsDocxWatermarkService docx = mock(DmsDocxWatermarkService.class);
        DmsPdfWatermarkService pdf = mock(DmsPdfWatermarkService.class);
        PageWatermarkApi renderer = mock(PageWatermarkApi.class);
        DmsFileWatermarkService service = new DmsFileWatermarkService(docx, null, pdf, renderer,
                ignored -> "unused-ticket");
        byte[] source = {1, 2, 3};
        byte[] watermarked = {4, 5, 6};
        DmsWatermarkSnapshot mark = new DmsWatermarkSnapshot();
        List<DmsWatermarkSnapshot> marks = List.of(mark);
        when(pdf.apply(source, marks)).thenReturn(watermarked);

        for (String scope : List.of(DmsFilePreviewTicket.SCOPE, DmsFilePreviewTicket.PRINT_SCOPE)) {
            DmsFilePreviewTicket ticket = new DmsFilePreviewTicket();
            ticket.setScope(scope);
            ticket.setFileName("report.pdf");
            ticket.setWatermarks(marks);
            assertArrayEquals(watermarked, service.apply(source, ticket));
        }
        DmsFileDownloadTicket download = new DmsFileDownloadTicket();
        download.setFileName("report.pdf");
        download.setWatermarks(marks);
        assertArrayEquals(watermarked, service.apply(source, download));
        verify(pdf, times(3)).apply(source, marks);
        verifyNoInteractions(docx, renderer);
    }

    private byte[] drawings(String prefix) throws Exception {
        StringBuilder xml = new StringBuilder("<w:document xmlns:w=\"http://schemas.openxmlformats.org/wordprocessingml/2006/main\" "
                + "xmlns:wp=\"http://schemas.openxmlformats.org/drawingml/2006/wordprocessingDrawing\">");
        for (int page : List.of(1, 3, 5)) {
            xml.append("<wp:anchor layoutInCell=\"1\" simplePos=\"0\"><wp:simplePos x=\"0\" y=\"0\"/>")
                    .append("<wp:positionH relativeFrom=\"page\"><wp:posOffset>0</wp:posOffset></wp:positionH>")
                    .append("<wp:positionV relativeFrom=\"page\"><wp:posOffset>0</wp:posOffset></wp:positionV>")
                    .append("<wp:docPr name=\"").append(prefix).append("_0_").append(page)
                    .append("\"/></wp:anchor>");
        }
        xml.append("</w:document>");
        ByteArrayOutputStream bytes = new ByteArrayOutputStream();
        try (ZipOutputStream zip = new ZipOutputStream(bytes)) {
            zip.putNextEntry(new ZipEntry("word/document.xml"));
            zip.write(xml.toString().getBytes(StandardCharsets.UTF_8));
            zip.closeEntry();
        }
        return bytes.toByteArray();
    }
}