package com.ruoyi.logistics.util;

import cn.hutool.core.img.ImgUtil;
import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

/**
 * 二维码工具类
 * 
 * @author RuiJing
 * @date 2026-02-11
 */
public class QRCodeUtils {

    /**
     * 生成二维码图片并返回字节数组
     * 
     * @param content 二维码内容
     * @param width 二维码宽度
     * @param height 二维码高度
     * @return 二维码图片字节数组
     */
    public static byte[] generateQRCodeBytes(String content, int width, int height) {
        // 设置二维码配置
        QrConfig config = new QrConfig();
        config.setWidth(width);
        config.setHeight(height);
        // 设置容错级别
        config.setErrorCorrection(ErrorCorrectionLevel.M);
        // 设置边距
        config.setMargin(1);
        
        // 生成二维码图片
        BufferedImage image = QrCodeUtil.generate(content, config);
        
        // 将BufferedImage转换为字节数组
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        try {
            ImgUtil.write(image, "png", outputStream);
            return outputStream.toByteArray();
        } catch (Exception e) {
            throw new RuntimeException("生成二维码失败", e);
        } finally {
            try {
                outputStream.close();
            } catch (IOException e) {
                // 忽略关闭流的异常
            }
        }
    }

    /**
     * 生成二维码图片并返回Base64编码字符串
     * 
     * @param content 二维码内容
     * @param width 二维码宽度
     * @param height 二维码高度
     * @return Base64编码的二维码图片字符串
     */
    public static String generateQRCodeBase64(String content, int width, int height) {
        byte[] qrCodeBytes = generateQRCodeBytes(content, width, height);
        return java.util.Base64.getEncoder().encodeToString(qrCodeBytes);
    }
}