Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.document;

import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.PageSize;
import com.itextpdf.text.pdf.PdfWriter;

import com.spice.profitmandi.common.document.section.FooterSection;
import com.spice.profitmandi.common.document.section.HeaderSection;
import com.spice.profitmandi.common.document.section.ItemsTableSection;
import com.spice.profitmandi.common.document.section.PartiesSection;
import com.spice.profitmandi.common.document.section.Section;
import com.spice.profitmandi.common.document.section.SummarySection;
import com.spice.profitmandi.common.document.support.CancelledStamp;
import com.spice.profitmandi.common.util.EWayBillPDF;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.io.ByteArrayOutputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * Renders a list of {@link DocumentData} into a single PDF by driving the canonical section
 * pipeline (V2's look) for each document. Replaces the duplicated bodies of the legacy
 * {@code generateInvoiceV2 / generateAndWrite / generateAndWriteDebitNote /
 * generateAndWriteCustomerCreditNotes} methods.
 */
public class DocumentRenderer {

    private static final Logger LOGGER = LogManager.getLogger(DocumentRenderer.class);

    // Stateless, reusable. Order defines the page layout.
    private static final List<Section> SECTIONS = Arrays.asList(
            new HeaderSection(),
            new PartiesSection(),
            new ItemsTableSection(),
            new SummarySection(),
            new FooterSection());

    public void render(List<DocumentData> docs, OutputStream out) {
        try {
            ByteArrayOutputStream pdfBuffer = new ByteArrayOutputStream();
            boolean cancelledPages = false;
            List<Integer> cancelledPageList = new ArrayList<>();

            Document document = new Document(PageSize.A4, 28, 28, 34, 34);
            PdfWriter pdfWriter = PdfWriter.getInstance(document, pdfBuffer);
            document.open();

            for (DocumentData data : docs) {
                if (data.isCancelled()) {
                    cancelledPageList.add(1);
                    cancelledPages = true;
                } else {
                    cancelledPageList.add(0);
                }

                PdfContext ctx = new PdfContext(document, data);
                for (Section section : SECTIONS) {
                    section.render(ctx);
                }

                document.newPage();

                if (data.eWayBill() != null) {
                    EWayBillPDF.generateDocument(document, pdfWriter, data.eWayBill());
                }
            }

            document.close();
            if (cancelledPages) {
                CancelledStamp.apply(pdfBuffer, cancelledPageList);
            }
            pdfBuffer.writeTo(out);
        } catch (DocumentException e) {
            LOGGER.error("Unable to write data to pdf file : ", e);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}