Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.util;// package com.spice.profitmandi.print; // <- uncomment and adjust if you want this in a package

import com.itextpdf.text.Document;

import com.itextpdf.text.DocumentException;

import com.itextpdf.text.Paragraph;

import com.itextpdf.text.pdf.PdfWriter;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.model.CustomCustomer;

import com.spice.profitmandi.common.model.CustomOrderItem;

import com.spice.profitmandi.common.model.InvoicePdfModel;

import javax.print.*;

import java.io.ByteArrayOutputStream;

import java.nio.charset.Charset;

import java.util.ArrayList;

import java.util.List;

/**

 * Utility to format and print an InvoicePdfModel to a thermal printer width (58mm/80mm).

 * Compatible with Java 8.

 */

public class InvoiceFormatter {

    public static final int WIDTH_58MM = 32; // typical ~32 chars/line

    public static final int WIDTH_80MM = 48; // typical ~48 chars/line

    /**

     * Build printable lines for a thermal printer.

     */

    public static void getInvoice(InvoicePdfModel invoice, ByteArrayOutputStream baos, int lineWidth) throws ProfitMandiBusinessException {

        List<String> lines = new ArrayList<>();

        // ===== HEADER =====

        addIfNotEmpty(lines, center(safe(invoice.getTitle()), lineWidth));

        if (invoice.isCancelled()) {

            lines.add(center("*** INVOICE CANCELLED ***", lineWidth));

        }

        String docLabel = invoice.getTitle() != null && invoice.getTitle().toLowerCase().contains("challan") ? "DC No" : "Invoice No";

        addIfNotEmpty(lines, center(docLabel + ": " + safe(invoice.getInvoiceNumber()), lineWidth));

        addIfNotEmpty(lines, center("Date: " + safe(invoice.getInvoiceDate()), lineWidth));

        lines.add(repeat('-', lineWidth));

        // ===== CUSTOMER =====

        CustomCustomer c = invoice.getCustomer();

        if (c != null) {

            String fullName = (safe(c.getFirstName()) + " " + safe(c.getLastName())).trim();

            addIfNotEmpty(lines, "Customer: " + fullName);

            addIfNotEmpty(lines, "Mobile  : " + safe(c.getMobileNumber()));

            addIfNotEmpty(lines, "GSTIN   : " + safe(c.getGstNumber()));

            if (c.getAddress() != null) {

                List<String> addrLines = wrapWords("Address: " + c.getAddress().getAddressString(), lineWidth);

                lines.addAll(addrLines);

            }

            if (safe(invoice.getCustomerAddressStateCode()).length() > 0) {

                lines.add("State   : " + invoice.getCustomerAddressStateCode());

            }

            lines.add(repeat('-', lineWidth));

        }

        // ===== ITEMS HEADER =====

        // We'll allocate fixed widths for columns: Qty(4) Rate(9) Amount(10) -> trailer length ~ 4 + 1 + 9 + 1 + 10 = 25

        // Thus, description width = lineWidth - 25

        final String hdrTrailer = rightPad("Qty", 4) + " " + rightPad("Rate", 9) + " " + rightPad("Amount", 10);

        int descWidthForHeader = Math.max(0, lineWidth - hdrTrailer.length());

        lines.add(rightPad("Item", descWidthForHeader) + hdrTrailer);

        lines.add(repeat('-', lineWidth));

        // ===== ITEMS =====

        if (invoice.getOrderItems() != null) {

            boolean marginSeparatorAdded = false;

            for (CustomOrderItem item : invoice.getOrderItems()) {

                if (item != null && item.isMarginScheme() && !marginSeparatorAdded) {

                    lines.add(repeat('-', lineWidth));

                    lines.add(center("Margin Scheme - Rule 32(5)", lineWidth));

                    lines.add(repeat('-', lineWidth));

                    marginSeparatorAdded = true;

                }

                lines.addAll(formatItem(item, lineWidth));

            }

        }

        lines.add(repeat('-', lineWidth));

        // ===== TOTALS =====

        lines.add(right("TOTAL: " + formatAmount(invoice.getTotalAmount()), lineWidth));

        // T&Cs

        if (invoice.getTncs() != null && !invoice.getTncs().isEmpty()) {

            lines.add(repeat('-', lineWidth));

            lines.add("Terms & Conditions:");

            for (String t : invoice.getTncs()) {

                lines.addAll(wrapWords(safe(t), lineWidth));

            }

        }

        // ===== MARGIN SCHEME DECLARATION =====

        if (invoice.isHasMarginSchemeItems() && invoice.getMarginSchemeDeclarations() != null

                && !invoice.getMarginSchemeDeclarations().isEmpty()) {

            lines.add(repeat('-', lineWidth));

            lines.add("Margin Scheme Declaration:");

            for (String decl : invoice.getMarginSchemeDeclarations()) {

                lines.addAll(wrapWords("- " + safe(decl), lineWidth));

            }

        }

        lines.add(center("Thank you! Visit again", lineWidth));

        // Writer writes into the provided baos

        Document document = new Document();

        // Create PdfWriter that writes into baos

        try {

            PdfWriter.getInstance(document, baos);

        } catch (DocumentException e) {

            throw new ProfitMandiBusinessException("Could not Create instance", "Could not create instance", "Could not create instance");

        }

        document.open();

        // Add each string as a new line

        for (String line : lines) {

            try {

                document.add(new Paragraph(line));

            } catch (DocumentException e) {

                throw new RuntimeException(e);

            }

        }

        document.close(); // Very important: flushes to baos

    }

    public static void saveAsPdf(String filePath, List<String> lines) throws Exception {

        /*PdfWriter writer = new PdfWriter(filePath);

        // Create PDF document

        PdfDocument pdf = new PdfDocument(writer);

        // Add a document layout wrapper

        Document document = new Document(pdf);

        // Add each line as a paragraph

        for (String line : lines) {

            document.add(new Paragraph(line));

        }

        document.close();*/

    }

    /**

     * Format one item with word-wrapped description and aligned Qty/Rate/Amount.

     */

    public static List<String> formatItem(CustomOrderItem item, int lineWidth) {

        List<String> out = new ArrayList<String>();

        if (item == null) return out;

        // trailer columns: Qty(4), Rate(9 - incl. 2 decimals), Amount(10)

        String trailer = String.format("%4d %9.2f %10.2f",

                item.getQuantity(),

                item.getRate(),

                item.getNetAmount());

        int descWidth = Math.max(0, lineWidth - trailer.length());

        List<String> wrappedDesc = wrapWords(safe(item.getDescription()), descWidth);

        if (wrappedDesc.isEmpty()) wrappedDesc.add("");

        // First line: part of desc + trailer

        out.add(rightPad(wrappedDesc.get(0), descWidth) + trailer);

        // Remaining description lines

        for (int i = 1; i < wrappedDesc.size(); i++) {

            out.add(wrappedDesc.get(i));

        }

        // Optional meta lines under the item

        if (safe(item.getHsnCode()).length() > 0) {

            out.add("   HSN: " + item.getHsnCode());

        }

        float gstRate = safeFloat(item.getIgstRate()) + safeFloat(item.getCgstRate()) + safeFloat(item.getSgstRate());

        if (gstRate > 0.0f) {

            out.add(String.format("   GST: %.2f%% (IGST %.2f, CGST %.2f, SGST %.2f)",

                    gstRate, safeFloat(item.getIgstRate()), safeFloat(item.getCgstRate()), safeFloat(item.getSgstRate())));

        }

        if (safeFloat(item.getDiscount()) > 0.0f) {

            out.add("   Disc: " + formatAmount(item.getDiscount()));

        }

        if (item.isMarginScheme()) {

            out.add("   Purchase Price: " + formatAmount(item.getPurchasePrice()));

            out.add("   Selling Price : " + formatAmount(item.getSellingPrice()));

            out.add("   Margin (Taxable): " + formatAmount(item.getMargin()));

        }

        return out;

    }

// ===== Printing =====

    /**

     * Send lines to the DEFAULT system printer. Works if your thermal printer has a Windows/Linux driver.

     * If you need ESC/POS raw, see sendRawWithCut.

     */

    public static void sendToDefaultPrinter(List<String> lines) throws PrintException {

        PrintService svc = PrintServiceLookup.lookupDefaultPrintService();

        if (svc == null) throw new PrintException("No default print service found");

        DocPrintJob job = svc.createPrintJob();

        StringBuilder sb = new StringBuilder();

        for (String line : lines) {

            sb.append(line == null ? "" : line).append('\n');

        }

        byte[] data = sb.toString().getBytes(Charset.forName("UTF-8"));

        Doc doc = new SimpleDoc(data, DocFlavor.BYTE_ARRAY.AUTOSENSE, null);

        job.print(doc, null);

    }

    /**

     * Attempt to send raw text + ESC/POS full cut to the default printer.

     * This only works if the printer driver accepts raw bytes. If you see junk, switch to a RAW queue or an ESC/POS lib.

     */

    public static void sendRawWithCut(List<String> lines) throws PrintException {

        PrintService svc = PrintServiceLookup.lookupDefaultPrintService();

        if (svc == null) throw new PrintException("No default print service found");

        DocPrintJob job = svc.createPrintJob();

        // Build payload: text + some feeds + ESC/POS cut

        byte[] newline = new byte[]{'\n'};

        byte[] feed = new byte[]{0x1B, 'd', 0x05}; // ESC d 5  -> feed 5 lines

        byte[] cut = new byte[]{0x1D, 0x56, 0x42, 0x00}; // GS V B 0 -> full cut

        // join

        List<byte[]> chunks = new ArrayList<byte[]>();

        StringBuilder sb = new StringBuilder();

        for (String line : lines) {

            sb.append(line == null ? "" : line).append('\n');

        }

        chunks.add(sb.toString().getBytes(Charset.forName("UTF-8")));

        chunks.add(feed);

        chunks.add(cut);

        int total = 0;

        for (byte[] ch : chunks) total += ch.length;

        byte[] payload = new byte[total];

        int pos = 0;

        for (byte[] ch : chunks) {

            System.arraycopy(ch, 0, payload, pos, ch.length);

            pos += ch.length;

        }

        Doc doc = new SimpleDoc(payload, DocFlavor.BYTE_ARRAY.AUTOSENSE, null);

        job.print(doc, null);

    }

// ===== Helpers =====

    private static void addIfNotEmpty(List<String> list, String s) {

        if (s == null) return;

        if (s.trim().length() == 0) return;

        list.add(s);

    }

    private static String safe(String s) {

        return s == null ? "" : s;

    }

    private static float safeFloat(Float f) {

        return f == null ? 0.0f : f.floatValue();

    }

    private static String formatAmount(float v) {

        return String.format("%.2f", v);

    }

    private static String rightPad(String s, int width) {

        if (s == null) s = "";

        if (s.length() >= width) return s.substring(0, width);

        StringBuilder sb = new StringBuilder(s);

        while (sb.length() < width) sb.append(' ');

        return sb.toString();

    }

    private static String leftPad(String s, int width) {

        if (s == null) s = "";

        if (s.length() >= width) return s.substring(s.length() - width);

        StringBuilder sb = new StringBuilder();

        while (sb.length() + s.length() < width) sb.append(' ');

        sb.append(s);

        return sb.toString();

    }

    private static String center(String text, int width) {

        if (text == null) return "";

        if (text.length() >= width) return text;

        int pad = (width - text.length()) / 2;

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < pad; i++) sb.append(' ');

        sb.append(text);

        return sb.toString();

    }

    private static String right(String text, int width) {

        if (text == null) return "";

        if (text.length() >= width) return text;

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < width - text.length(); i++) sb.append(' ');

        sb.append(text);

        return sb.toString();

    }

    private static String repeat(char ch, int count) {

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < count; i++) sb.append(ch);

        return sb.toString();

    }

    private static String repeat(char ch, int count, boolean withNewline) {

        String s = repeat(ch, count);

        return withNewline ? s + "\n" : s;

    }

    private static String repeat(char ch1, char ch2, int count) {

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < count; i++) {

            sb.append(ch1).append(ch2);

        }

        return sb.toString();

    }

    private static String repeat(String s, int count) {

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < count; i++) sb.append(s);

        return sb.toString();

    }

    private static String repeat(char ch, int count, String trailer) {

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < count; i++) sb.append(ch);

        sb.append(trailer);

        return sb.toString();

    }

    private static String repeat(char ch, int count, String trailer, boolean newline) {

        String s = repeat(ch, count, trailer);

        return newline ? s + "\n" : s;

    }

    /**

     * Word wrap that prefers breaking at spaces; falls back to hard breaks for very long tokens.

     */

    public static List<String> wrapWords(String text, int width) {

        List<String> out = new ArrayList<String>();

        if (text == null) return out;

        text = text.trim();

        if (width <= 0) {

            out.add(text);

            return out;

        }

        String[] words = text.split("\\s+");

        StringBuilder line = new StringBuilder();

        int i;

        for (i = 0; i < words.length; i++) {

            String w = words[i];

            if (w.length() > width) {

                // Flush current line

                if (line.length() > 0) {

                    out.add(line.toString());

                    line.setLength(0);

                }

                // Hard wrap the long word

                int idx = 0;

                while (idx < w.length()) {

                    int end = Math.min(idx + width, w.length());

                    out.add(w.substring(idx, end));

                    idx = end;

                }

                continue;

            }

            if (line.length() == 0) {

                line.append(w);

            } else if (line.length() + 1 + w.length() <= width) {

                line.append(' ').append(w);

            } else {

                out.add(line.toString());

                line.setLength(0);

                line.append(w);

            }

        }

        if (line.length() > 0) out.add(line.toString());

        return out;

    }

}