package jnpf.bizcommon.onlyoffice.service.impl;

import com.google.zxing.BinaryBitmap;
import com.google.zxing.MultiFormatReader;
import com.google.zxing.client.j2se.BufferedImageLuminanceSource;
import com.google.zxing.common.HybridBinarizer;
import jnpf.bizcommon.onlyoffice.entity.OnlyOfficeProperties;
import jnpf.dmsEntity.preview.model.DmsFilePreviewTicket;
import jnpf.dmsEntity.preview.model.DmsWatermarkPositionSnapshot;
import jnpf.dmsEntity.preview.model.DmsWatermarkSnapshot;
import jnpf.exception.DataException;
import org.apache.pdfbox.Loader;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.common.PDRectangle;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.junit.Test;

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.math.BigDecimal;
import java.util.List;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

public class DmsOnlyOfficePrintPdfComposerTest {
    static {
        System.setProperty("java.awt.headless", "true");
    }

    @Test
    public void extractsOnlySelectedPagesAndStampsEachCopyBarcode() throws Exception {
        DmsOnlyOfficePrintPdfComposer composer = new DmsOnlyOfficePrintPdfComposer(new OnlyOfficeProperties());

        byte[] result = composer.compose(basePdf(3), List.of(
                ticket("1,3", "COPY-1"), ticket("1,3", "COPY-2")));

        try (PDDocument pdf = Loader.loadPDF(result)) {
            assertEquals(4, pdf.getNumberOfPages());
            PDFRenderer renderer = new PDFRenderer(pdf);
            assertEquals("COPY-1", decode(renderer, 0));
            assertEquals("COPY-1", decode(renderer, 1));
            assertEquals("COPY-2", decode(renderer, 2));
            assertEquals("COPY-2", decode(renderer, 3));
        }
    }

    @Test
    public void placesPrintBarcodeAtLandscapeCoordinatesOnRotatedPages() throws Exception {
        DmsOnlyOfficePrintPdfComposer composer = new DmsOnlyOfficePrintPdfComposer(new OnlyOfficeProperties());
        DmsFilePreviewTicket ticket = ticket("1", "ROTATED");
        DmsWatermarkPositionSnapshot position = ticket.getWatermarks().get(0).getPositions().get(0);
        position.setLandscapeXMm(new BigDecimal("180"));
        position.setLandscapeYMm(new BigDecimal("40"));
        DmsWatermarkPositionSnapshot a3 = new DmsWatermarkPositionSnapshot();
        a3.setPaperType("A3");
        a3.setPaperWidthMm(new BigDecimal("297"));
        a3.setPaperHeightMm(new BigDecimal("420"));
        a3.setPortraitXMm(new BigDecimal("10"));
        a3.setPortraitYMm(new BigDecimal("20"));
        a3.setLandscapeXMm(new BigDecimal("330"));
        a3.setLandscapeYMm(new BigDecimal("60"));
        ticket.getWatermarks().get(0).setPositions(List.of(position, a3));

        for (int rotation : List.of(90, 180, 270)) {
            boolean isA3 = rotation == 270;
            byte[] source;
            try (PDDocument document = new PDDocument();
                 ByteArrayOutputStream output = new ByteArrayOutputStream()) {
                PDPage page = new PDPage(isA3 ? PDRectangle.A3 : PDRectangle.A4);
                page.setRotation(rotation);
                document.addPage(page);
                document.save(output);
                source = output.toByteArray();
            }
            try (PDDocument stamped = Loader.loadPDF(composer.composePreview(source, ticket))) {
                PDFRenderer renderer = new PDFRenderer(stamped);
                assertEquals("ROTATED", decode(renderer, 0));
                BufferedImage image = renderer.renderImageWithDPI(0, 72);
                int visible = 0;
                int xMm = rotation == 180 ? 65 : isA3 ? 330 : 180;
                int yMm = rotation == 180 ? 100 : isA3 ? 60 : 40;
                for (int y = (int) (yMm * 72D / 25.4D); y < (int) ((yMm + 80) * 72D / 25.4D); y++) {
                    for (int x = (int) (xMm * 72D / 25.4D); x < (int) ((xMm + 80) * 72D / 25.4D); x++) {
                        if ((image.getRGB(x, y) & 0x00ffffff) < 0x00808080) visible++;
                    }
                }
                assertTrue("Expected barcode at landscape coordinates", visible > 0);
            }
        }
    }

    @Test
    public void filtersPagesBeyondRenderedDocumentForPrintAndPreview() throws Exception {
        DmsOnlyOfficePrintPdfComposer composer = new DmsOnlyOfficePrintPdfComposer(new OnlyOfficeProperties());

        byte[] printed = composer.compose(basePdf(3), List.of(ticket("1-5", "PRINT")));
        byte[] previewed = composer.composePreview(basePdf(3), ticket("2,4", "PREVIEW"));

        try (PDDocument pdf = Loader.loadPDF(printed)) {
            assertEquals(3, pdf.getNumberOfPages());
        }
        try (PDDocument pdf = Loader.loadPDF(previewed)) {
            assertEquals(1, pdf.getNumberOfPages());
        }
    }

    @Test(expected = DataException.class)
    public void rejectsPrintWhenAllRequestedPagesAreBeyondRenderedDocument() throws Exception {
        DmsOnlyOfficePrintPdfComposer composer = new DmsOnlyOfficePrintPdfComposer(new OnlyOfficeProperties());

        composer.compose(basePdf(3), List.of(ticket("4-5", "PRINT")));
    }

    @Test(expected = DataException.class)
    public void rejectsBatchOverConfiguredPageLimit() throws Exception {
        OnlyOfficeProperties properties = new OnlyOfficeProperties();
        properties.getDmsPrint().setMaxBatchPages(3);
        DmsOnlyOfficePrintPdfComposer composer = new DmsOnlyOfficePrintPdfComposer(properties);

        composer.compose(basePdf(3), List.of(ticket("1,3", "COPY-1"), ticket("1,3", "COPY-2")));
    }

    @Test(expected = DataException.class)
    public void appliesBatchByteLimitToSingleCopy() throws Exception {
        OnlyOfficeProperties properties = new OnlyOfficeProperties();
        properties.getDmsPrint().setMaxBatchPdfBytes(1);
        DmsOnlyOfficePrintPdfComposer composer = new DmsOnlyOfficePrintPdfComposer(properties);

        composer.compose(basePdf(1), List.of(ticket("1", "COPY-1")));
    }

    private String decode(PDFRenderer renderer, int pageIndex) throws Exception {
        BinaryBitmap bitmap = new BinaryBitmap(new HybridBinarizer(
                new BufferedImageLuminanceSource(renderer.renderImageWithDPI(pageIndex, 144))));
        return new MultiFormatReader().decode(bitmap).getText();
    }

    private byte[] basePdf(int pages) throws Exception {
        try (PDDocument document = new PDDocument();
             ByteArrayOutputStream output = new ByteArrayOutputStream()) {
            for (int page = 0; page < pages; page++) document.addPage(new PDPage(PDRectangle.A4));
            document.save(output);
            return output.toByteArray();
        }
    }

    private DmsFilePreviewTicket ticket(String pages, String content) {
        DmsFilePreviewTicket ticket = new DmsFilePreviewTicket();
        ticket.setPrintPages(pages);
        ticket.setWatermarks(List.of(watermark(content)));
        return ticket;
    }

    private DmsWatermarkSnapshot watermark(String content) {
        DmsWatermarkPositionSnapshot position = new DmsWatermarkPositionSnapshot();
        position.setPaperWidthMm(new BigDecimal("210"));
        position.setPaperHeightMm(new BigDecimal("297"));
        position.setPortraitXMm(new BigDecimal("65"));
        position.setPortraitYMm(new BigDecimal("100"));

        DmsWatermarkSnapshot watermark = new DmsWatermarkSnapshot();
        watermark.setCustomBarcode(true);
        watermark.setBarcodeFormat("QR_CODE");
        watermark.setContent(content);
        watermark.setShowBarcodeText(false);
        watermark.setMarkWidthMm(new BigDecimal("80"));
        watermark.setMarkHeightMm(new BigDecimal("80"));
        watermark.setOpacity(100);
        watermark.setPositions(List.of(position));
        return watermark;
    }
}