package cn.iocoder.yudao.module.mes.framework.qrcode;

import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import cn.iocoder.yudao.framework.common.util.http.HttpUtils;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;

import javax.imageio.ImageIO;
import java.awt.Color;
import java.awt.Font;
import java.awt.FontMetrics;
import java.awt.Graphics2D;
import java.awt.GraphicsEnvironment;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.List;
import java.util.Set;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * 二维码标签图片工具类
 * <p>
 * 生成二维码在上、码值在下的紧凑标签。二维码内容仍使用调用方传入的溯源链接。
 */
@Slf4j
public final class QrCodeImageUtils {

    private QrCodeImageUtils() {
    }

    /** 批量导出二维码数量上限（防止一次性生成过多导致内存/响应过大） */
    public static final int MAX_QR_EXPORT = 2000;

    /** 二维码边长 */
    private static final int QR_SIZE = 260;
    /** 图片四周留白 */
    private static final int PADDING = 16;
    /** 二维码与文本间距 */
    private static final int GAP = 4;
    /** 字体大小 */
    private static final int FONT_SIZE = 24;
    /** 行高 */
    private static final int LINE_HEIGHT = 32;

    /**
     * 中文字体候选（按优先级）。
     * <p>
     * 只允许收录可渲染中文的字体：纯拉丁字体（如 Arial）一旦入选，中文会全部渲染成方框，
     * 且导出过程不报错，属于最隐蔽的一类事故。
     */
    private static final String[] FONT_CANDIDATES = {
            "Noto Sans CJK SC", "Source Han Sans SC", "WenQuanYi Zen Hei", "PingFang SC",
            "Hiragino Sans GB", "Microsoft YaHei UI", "Microsoft YaHei", "SimSun"
    };

    /** 中文字形探针字符（用于探测字体是否具备中文字形） */
    private static final char CJK_PROBE = '袋';

    /** 解析出的中文字体族名；为 null 表示当前环境没有任何可渲染中文的字体 */
    private static final String FONT_NAME = resolveFontName();

    /**
     * 生成二维码标签图片
     *
     * @param code        二维码内容（溯源链接或纯码值）
     * @param displayCode 二维码下方显示的原始码值
     * @return 合成图片
     */
    public static BufferedImage createQrLabel(String code, String displayCode) {
        QrConfig config = new QrConfig();
        config.setErrorCorrection(ErrorCorrectionLevel.H);
        config.setMargin(4); // 保留二维码静区，保证打印后的识读率
        config.setWidth(QR_SIZE);
        config.setHeight(QR_SIZE);
        BufferedImage qrImage = QrCodeUtil.generate(code, config);

        int width = PADDING * 2 + QR_SIZE;
        int height = PADDING * 2 + QR_SIZE + GAP + LINE_HEIGHT;
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = image.createGraphics();
        try {
            g.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
            g.setColor(Color.WHITE);
            g.fillRect(0, 0, width, height);
            g.drawImage(qrImage, PADDING, PADDING, null);
            String text = StrUtil.nullToEmpty(displayCode);
            int fontSize = FONT_SIZE;
            g.setFont(new Font(Font.SANS_SERIF, Font.BOLD, fontSize));
            while (g.getFontMetrics().stringWidth(text) > QR_SIZE && fontSize > 1) {
                g.setFont(new Font(Font.SANS_SERIF, Font.BOLD, --fontSize));
            }
            FontMetrics metrics = g.getFontMetrics();
            int textX = (width - metrics.stringWidth(text)) / 2;
            int textY = PADDING + QR_SIZE + GAP
                    + (LINE_HEIGHT - metrics.getHeight()) / 2 + metrics.getAscent();
            g.setColor(Color.BLACK);
            g.drawString(text, textX, textY);
        } finally {
            g.dispose();
        }
        return image;
    }

    /**
     * 将多张二维码标签图片写入 ZIP 响应（流式，逐张生成，内存只占一张）
     *
     * @param response 响应
     * @param zipName  ZIP 文件名（不含后缀）
     * @param labels   标签列表
     */
    public static void writeZip(HttpServletResponse response, String zipName, List<QrLabelInfo> labels) throws IOException {
        response.setContentType("application/zip");
        response.setHeader("Content-Disposition",
                "attachment;filename=" + HttpUtils.encodeUtf8(zipName + ".zip"));
        ZipOutputStream zip = new ZipOutputStream(response.getOutputStream());
        try {
            for (QrLabelInfo label : labels) {
                BufferedImage image = createQrLabel(label.getCode(), label.getFileName());
                zip.putNextEntry(new ZipEntry(label.getFileName() + ".png"));
                ImageIO.write(image, "png", zip);
                zip.closeEntry();
            }
            zip.finish();
            zip.flush();
        } finally {
            // 不 close ZipOutputStream，避免关闭底层 ServletOutputStream（与 FileTypeUtils 同理）
        }
    }

    // ==================== 私有方法 ====================

    /**
     * 解析当前环境可用的中文字体族名。
     * <p>
     * 不能按字体族名做精确匹配：同一字体在不同系统上的族名可能不同（例如 Windows 上实际注册为
     * {@code Microsoft YaHei UI} 而非 {@code Microsoft YaHei}），精确匹配落空后会一路退到纯拉丁字体，
     * 中文即渲染为方框。因此这里统一用 {@link Font#canDisplay(char)} 探测中文字形。
     *
     * @return 可渲染中文的字体族名；无可用字体时返回 {@code null}
     */
    private static String resolveFontName() {
        Set<String> families = Set.of(GraphicsEnvironment.getLocalGraphicsEnvironment().getAvailableFontFamilyNames());
        for (String candidate : FONT_CANDIDATES) {
            if (families.contains(candidate) && canDisplayCjk(candidate)) {
                return candidate;
            }
        }
        // 逻辑字体由 JDK 负责字形回退，多数环境可渲染中文
        if (canDisplayCjk(Font.SANS_SERIF)) {
            return Font.SANS_SERIF;
        }
        return null;
    }

    /**
     * 探测指定字体族能否渲染中文
     */
    private static boolean canDisplayCjk(String family) {
        return new Font(family, Font.PLAIN, FONT_SIZE).canDisplay(CJK_PROBE);
    }

    /**
     * 校验当前环境存在可渲染中文的字体，否则抛出异常。
     * <p>
     * 供启动期调用：宁可在启动时失败，也不能把中文渲染成方框的标签交付印刷。
     */
    public static void assertCjkFontAvailable() {
        if (FONT_NAME == null) {
            throw new IllegalStateException("当前环境没有任何可渲染中文的字体，二维码标签的中文将显示为方框，"
                    + "拒绝生成标签。请在服务器安装中文字体（如 fonts-noto-cjk）后重启。");
        }
    }

}