import PDFDocument from "pdfkit";
import type { Response } from "express";
import { PdfHeader } from "./components/PdfHeader.js";

const COLORS = {
    primary: "#6F1D1B",
    accent: "#A4243B",
    text: "#212529",
    muted: "#6C757D",
    border: "#DEE2E6",
    rowAlt: "#FFF5F5",
    white: "#FFFFFF",
};

const PAGE_MARGIN = 40;

function fmt(value: number | string | undefined | null): string {
    const n = Number(value ?? 0);
    return isNaN(n) ? "-" : n.toLocaleString("en-IN", { minimumFractionDigits: 2, maximumFractionDigits: 2 });
}

function fmtQty(value: number | string | undefined | null): string {
    const n = Number(value ?? 0);
    return isNaN(n) ? "-" : n.toLocaleString("en-IN", { minimumFractionDigits: 3, maximumFractionDigits: 3 });
}

export class SupplierReturnsPdf {
    static async generate(
        returns: any[],
        res: Response,
        filters?: {
            search?: string;
            startDate?: string;
            endDate?: string;
        }
    ) {
        const doc = new PDFDocument({
            size: "A4",
            margin: PAGE_MARGIN,
            bufferPages: true,
        });

        res.setHeader("Content-Type", "application/pdf");
        res.setHeader(
            "Content-Disposition",
            `attachment; filename="supplier-returns-${Date.now()}.pdf"`
        );
        res.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
        doc.pipe(res);

        const pageWidth = doc.page.width - PAGE_MARGIN * 2;

        await PdfHeader.draw(doc, pageWidth, PAGE_MARGIN, "SUPPLIER RETURNS REPORT");

        this.drawFilterLine(doc, pageWidth, filters);
        this.drawTable(doc, returns, pageWidth);
        this.drawTotals(doc, returns, pageWidth);
        this.drawFooter(doc, pageWidth);

        doc.end();
    }

    private static drawFilterLine(
        doc: PDFKit.PDFDocument,
        pageWidth: number,
        filters?: { search?: string; startDate?: string; endDate?: string }
    ) {
        const parts: string[] = [];
        if (filters?.startDate)
            parts.push(`From: ${new Date(filters.startDate).toLocaleDateString("en-GB").replace(/\//g, "-")}`);
        if (filters?.endDate)
            parts.push(`To: ${new Date(filters.endDate).toLocaleDateString("en-GB").replace(/\//g, "-")}`);
        if (filters?.search) parts.push(`Search: "${filters.search}"`);

        if (parts.length) {
            doc
                .font("Helvetica")
                .fontSize(8.5)
                .fillColor(COLORS.muted)
                .text(parts.join("   |   "), PAGE_MARGIN, doc.y, {
                    width: pageWidth,
                    align: "right",
                });
            doc.y += 4;
        }

        doc.y += 10;
    }

    private static drawTable(
        doc: PDFKit.PDFDocument,
        returns: any[],
        pageWidth: number
    ) {
        const pw = pageWidth;
        const cols = {
            no:         { x: PAGE_MARGIN + pw * 0.000, width: pw * 0.045 },
            date:       { x: PAGE_MARGIN + pw * 0.045, width: pw * 0.115 },
            supplier:   { x: PAGE_MARGIN + pw * 0.160, width: pw * 0.210 },
            vegetable:  { x: PAGE_MARGIN + pw * 0.370, width: pw * 0.155 },
            lotNo:      { x: PAGE_MARGIN + pw * 0.525, width: pw * 0.105 },
            qty:        { x: PAGE_MARGIN + pw * 0.630, width: pw * 0.100 },
            bags:       { x: PAGE_MARGIN + pw * 0.730, width: pw * 0.080 },
            amount:     { x: PAGE_MARGIN + pw * 0.810, width: pw * 0.190 },
        };

        const rowHeight = 26;
        const headerH = 26;

        const drawTableHeader = () => {
            const y = doc.y;
            doc
                .font("Helvetica-Bold")
                .fontSize(8)
                .fillColor(COLORS.primary)
                .text("#",          cols.no.x,        y + 6, { width: cols.no.width,        align: "center" })
                .text("DATE",       cols.date.x,      y + 6, { width: cols.date.width })
                .text("SUPPLIER",   cols.supplier.x,  y + 6, { width: cols.supplier.width })
                .text("VEGETABLE",  cols.vegetable.x, y + 6, { width: cols.vegetable.width })
                .text("LOT NO",     cols.lotNo.x,     y + 6, { width: cols.lotNo.width })
                .text("QTY (kg)",   cols.qty.x,       y + 6, { width: cols.qty.width,       align: "right" })
                .text("BAGS",       cols.bags.x,      y + 6, { width: cols.bags.width,      align: "right" })
                .text("AMOUNT",     cols.amount.x,    y + 6, { width: cols.amount.width - 4, align: "right" });

            doc.y = y + headerH;

            doc
                .moveTo(PAGE_MARGIN, doc.y)
                .lineTo(PAGE_MARGIN + pw, doc.y)
                .strokeColor(COLORS.primary)
                .lineWidth(1)
                .stroke();

            doc.y += 4;
        };

        const ensureSpace = (needed: number) => {
            const bottomLimit = doc.page.height - PAGE_MARGIN - 60;
            if (doc.y + needed > bottomLimit) {
                doc.addPage();
                doc.y = PAGE_MARGIN;
                drawTableHeader();
            }
        };

        drawTableHeader();

        returns.forEach((r, i) => {
            ensureSpace(rowHeight);
            const y = doc.y;

            if (i % 2 === 1) {
                doc.rect(PAGE_MARGIN, y, pw, rowHeight).fill(COLORS.rowAlt);
            }

            const dateStr = r.returnDate
                ? new Date(r.returnDate).toLocaleDateString("en-GB").replace(/\//g, "-")
                : "-";
            const supplierName = r.purchase?.supplier?.name ?? "-";
            const vegetableName = r.purchase?.vegetable?.name ?? "-";
            const lotNo = r.purchase?.lotNo ?? "-";

            doc
                .font("Helvetica-Bold")
                .fontSize(8)
                .fillColor(COLORS.accent)
                .text(`${i + 1}`, cols.no.x, y + 6, { width: cols.no.width, align: "center" });

            doc.font("Helvetica").fontSize(8.5).fillColor(COLORS.text);
            doc.text(dateStr,       cols.date.x,      y + 6, { width: cols.date.width - 4,      lineBreak: false });
            doc.text(supplierName,  cols.supplier.x,  y + 6, { width: cols.supplier.width - 4,  ellipsis: true, lineBreak: false });
            doc.text(vegetableName, cols.vegetable.x, y + 6, { width: cols.vegetable.width - 4, ellipsis: true, lineBreak: false });
            doc.text(lotNo,         cols.lotNo.x,     y + 6, { width: cols.lotNo.width - 4,     ellipsis: true, lineBreak: false });

            doc.font("Helvetica").fontSize(8.5).fillColor(COLORS.text);
            doc.text(fmtQty(r.quantity), cols.qty.x,  y + 6, { width: cols.qty.width - 4,  align: "right", lineBreak: false });
            doc.text(r.bags ? fmtQty(r.bags) : "-", cols.bags.x, y + 6, { width: cols.bags.width - 4, align: "right", lineBreak: false });

            doc
                .font("Helvetica-Bold")
                .fontSize(8.5)
                .fillColor(COLORS.primary)
                .text(`₹ ${fmt(r.returnAmount)}`, cols.amount.x, y + 6, { width: cols.amount.width - 6, align: "right", lineBreak: false });

            doc.y = y + rowHeight;
        });

        doc
            .moveTo(PAGE_MARGIN, doc.y)
            .lineTo(PAGE_MARGIN + pw, doc.y)
            .strokeColor(COLORS.border)
            .lineWidth(0.5)
            .stroke();

        doc.y += 14;
    }

    private static drawTotals(
        doc: PDFKit.PDFDocument,
        returns: any[],
        pageWidth: number
    ) {
        const totalQty = returns.reduce((s, r) => s + Number(r.quantity ?? 0), 0);
        const totalBags = returns.reduce((s, r) => s + Number(r.bags ?? 0), 0);
        const totalAmount = returns.reduce((s, r) => s + Number(r.returnAmount ?? 0), 0);

        const boxWidth = pageWidth * 0.50;
        const labelX = PAGE_MARGIN + pageWidth - boxWidth;
        const y = doc.y;

        doc
            .moveTo(labelX, y)
            .lineTo(PAGE_MARGIN + pageWidth, y)
            .strokeColor(COLORS.primary)
            .lineWidth(0.75)
            .stroke();

        doc.y = y + 6;

        doc.font("Helvetica-Bold").fontSize(10).fillColor(COLORS.primary);

        doc.text("TOTAL QTY (kg):", labelX, doc.y, { width: boxWidth * 0.62 });
        doc.text(fmtQty(totalQty), labelX, doc.y - 13, { width: boxWidth, align: "right" });
        doc.y += 4;

        doc.text("TOTAL BAGS:", labelX, doc.y, { width: boxWidth * 0.62 });
        doc.text(fmtQty(totalBags), labelX, doc.y - 13, { width: boxWidth, align: "right" });
        doc.y += 4;

        doc.fontSize(11);
        doc.text("TOTAL RETURN AMOUNT:", labelX, doc.y, { width: boxWidth * 0.62 });
        doc.text(`₹ ${fmt(totalAmount)}`, labelX, doc.y - 13.5, { width: boxWidth, align: "right" });
        doc.y += 18;
    }

    private static drawFooter(doc: PDFKit.PDFDocument, pageWidth: number) {
        const y = doc.page.height - PAGE_MARGIN - 20;
        doc
            .font("Helvetica-Oblique")
            .fontSize(8.5)
            .fillColor(COLORS.muted)
            .text(
                "This is a computer-generated supplier returns report.",
                PAGE_MARGIN,
                y,
                { width: pageWidth, align: "center" }
            );
    }
}