package com.doumee.core.utils;

import lombok.extern.slf4j.Slf4j;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * 图片水印工具类
 * 在图片左下角叠加时间 + 地址水印
 *
 * @author rk
 * @date 2026/04/21
 */
@Slf4j
public class ImageWatermarkUtil {

    /**
     * 为图片添加水印（时间 + 地址 + 经纬度）
     *
     * @param imageInput  原图输入流
     * @param timeText    时间文案（如 "2026-04-21 14:30:00"）
     * @param addressText 地址文案（如 "北京市海淀区xxx路"）
     * @param latitude    纬度
     * @param longitude   经度
     * @return 添加水印后的输入流，水印失败时返回原图
     */
    public static InputStream addWatermark(InputStream imageInput, String timeText, String addressText,
                                           Double latitude, Double longitude) {
        try {
            BufferedImage srcImage = ImageIO.read(imageInput);
            if (srcImage == null) {
                log.warn("水印: 无法读取图片，跳过水印处理");
                imageInput.reset();
                return imageInput;
            }

            int srcWidth = srcImage.getWidth();
            int srcHeight = srcImage.getHeight();

            // 字体大小按图片宽度自适应（约3%），最小16px
            int fontSize = Math.max(16, srcWidth * 3 / 100);
            int padding = fontSize;
            int lineHeight = fontSize + fontSize / 3;

            // 经纬度行
            String locationText = "";
            if (latitude != null && longitude != null) {
                locationText = String.format("经度:%.6f 纬度:%.6f", longitude, latitude);
            }
            int lineCount = 2 + (locationText.isEmpty() ? 0 : 1);

            // 计算水印区域背景高度
            int bgHeight = lineHeight * lineCount + padding;

            // 创建图片
            BufferedImage destImage = new BufferedImage(srcWidth, srcHeight, BufferedImage.TYPE_INT_RGB);
            Graphics2D g2d = destImage.createGraphics();

            // 绘制原图
            g2d.drawImage(srcImage, 0, 0, null);

            // 绘制半透明背景（左下角）
            g2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.5f));
            g2d.setColor(Color.BLACK);
            g2d.fillRect(0, srcHeight - bgHeight, srcWidth, bgHeight);

            // 绘制水印文字
            g2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.9f));
            g2d.setColor(Color.WHITE);
            Font font = new Font(Font.SANS_SERIF, Font.PLAIN, fontSize);
            g2d.setFont(font);

            // 开启抗锯齿
            g2d.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);

            int textX = padding;
            int textY = srcHeight - bgHeight + padding + fontSize;
            FontMetrics fm = g2d.getFontMetrics();
            int maxTextWidth = srcWidth - padding * 2;

            // 第一行：时间
            g2d.drawString(timeText, textX, textY);

            // 第二行：地址（地址过长时截断）
            String displayAddr = addressText;
            while (fm.stringWidth(displayAddr) > maxTextWidth && displayAddr.length() > 0) {
                displayAddr = displayAddr.substring(0, displayAddr.length() - 1);
            }
            g2d.drawString(displayAddr, textX, textY + lineHeight);

            // 第三行：经纬度
            if (!locationText.isEmpty()) {
                g2d.drawString(locationText, textX, textY + lineHeight * 2);
            }

            g2d.dispose();

            // 输出为字节流
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageIO.write(destImage, "jpg", baos);
            return new ByteArrayInputStream(baos.toByteArray());
        } catch (Exception e) {
            log.error("添加水印失败，返回原图", e);
            try {
                imageInput.reset();
            } catch (Exception ignored) {}
            return imageInput;
        }
    }
}