Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.util;

import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.pdf.PdfCopy;
import com.itextpdf.text.pdf.PdfReader;
import com.spice.profitmandi.common.document.DocumentData;
import com.spice.profitmandi.common.document.DocumentDataAdapter;
import com.spice.profitmandi.common.document.DocumentRenderer;
import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
import com.spice.profitmandi.common.model.CreditNotePdfModel;
import com.spice.profitmandi.common.model.DebitNotePdfModel;
import com.spice.profitmandi.common.model.InvoicePdfModel;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;

/**
 * Entry points for invoice/note PDF generation. All A4 document rendering is delegated to
 * {@link DocumentRenderer} (package {@code com.spice.profitmandi.common.document}); these methods
 * are thin adapters kept for their existing callers. Thermal (58/80mm) printing still goes through
 * {@link InvoiceFormatter}.
 */
public class PdfUtils {

    public static final String INVOICE_TITLE = "TAX INVOICE";
    public static final String DEBIT_NOTE_TITLE = "DEBIT NOTE";
    public static final String SECURITY_DEPOSIT = "SECURITY DEPOSIT RECEIPT";

    /** A4 → full document renderer; 58/80mm → thermal formatter. */
    public static void generateAndWrite(List<InvoicePdfModel> pdfModels, PrinterType printerType, ByteArrayOutputStream outputStream) throws ProfitMandiBusinessException {
        if (PrinterType.A4.equals(printerType)) {
            generateAndWrite(pdfModels, outputStream);
        } else {
            if (PrinterType.W80.equals(printerType)) {
                InvoiceFormatter.getInvoice(pdfModels.get(0), outputStream, InvoiceFormatter.WIDTH_80MM);
            } else if (PrinterType.W58.equals(printerType)) {
                InvoiceFormatter.getInvoice(pdfModels.get(0), outputStream, InvoiceFormatter.WIDTH_58MM);
            }
        }
    }

    public static void generateAndWrite(List<InvoicePdfModel> pdfModels, ByteArrayOutputStream outputStream) {
        List<DocumentData> docs = new ArrayList<>();
        for (InvoicePdfModel m : pdfModels) docs.add(DocumentDataAdapter.from(m));
        new DocumentRenderer().render(docs, outputStream);
    }

    public static void generateMarginSchemeInvoice(List<InvoicePdfModel> pdfModels, ByteArrayOutputStream outputStream) {
        generateInvoiceV2(pdfModels, outputStream);
    }

    public static void generateInvoiceV2(List<InvoicePdfModel> pdfModels, ByteArrayOutputStream outputStream) {
        List<DocumentData> docs = new ArrayList<>();
        for (InvoicePdfModel m : pdfModels) docs.add(DocumentDataAdapter.from(m));
        new DocumentRenderer().render(docs, outputStream);
    }

    public static void generateAndWriteDebitNote(List<DebitNotePdfModel> debitNotePdfModels, OutputStream outputStream) {
        List<DocumentData> docs = new ArrayList<>();
        for (DebitNotePdfModel m : debitNotePdfModels) docs.add(DocumentDataAdapter.from(m));
        new DocumentRenderer().render(docs, outputStream);
    }

    public static void generateAndWriteCustomerCreditNotes(List<CreditNotePdfModel> creditNotes, OutputStream outputStream) {
        List<DocumentData> docs = new ArrayList<>();
        for (CreditNotePdfModel m : creditNotes) docs.add(DocumentDataAdapter.from(m));
        new DocumentRenderer().render(docs, outputStream);
    }

    public static byte[] mergePdfFiles(List<byte[]> pdfFiles) throws IOException, DocumentException {
        ByteArrayOutputStream mergedOutputStream = new ByteArrayOutputStream();
        Document document = new Document();
        PdfCopy copy = new PdfCopy(document, mergedOutputStream);
        document.open();

        for (byte[] pdf : pdfFiles) {
            PdfReader reader = new PdfReader(new ByteArrayInputStream(pdf));
            int n = reader.getNumberOfPages();
            for (int i = 1; i <= n; i++) {
                copy.addPage(copy.getImportedPage(reader, i));
            }
            copy.freeReader(reader);
            reader.close();
        }

        document.close();
        return mergedOutputStream.toByteArray();
    }
}