package com.ruoyi.web.controller.tool;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.binary.Base64;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

@Slf4j
@Component
public class ImageUtil {

    /**
     * 将网络图片URL转为Base64编码
     *
     * @param imageUrl 网络图片地址
     * @return Base64编码的图片字符串
     */
    public static String imageUrlToBase64(String imageUrl) {
        if (StringUtils.isEmpty(imageUrl)) {
            return null;
        }

        InputStream inputStream = null;
        ByteArrayOutputStream outputStream = null;

        try {
            // 1. 创建URL连接
            URL url = new URL(imageUrl);
            HttpURLConnection connection = (HttpURLConnection) url.openConnection();
            connection.setRequestMethod("GET");
            connection.setConnectTimeout(5000);
            connection.setReadTimeout(10000);
            connection.setDoInput(true);

            // 2. 获取图片数据
            inputStream = connection.getInputStream();
            outputStream = new ByteArrayOutputStream();

            byte[] buffer = new byte[8192];
            int len;
            while ((len = inputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, len);
            }

            // 3. 转换为Base64
            byte[] imageBytes = outputStream.toByteArray();
            String base64 = Base64.encodeBase64String(imageBytes);

            // 4. 根据图片类型添加前缀
            String contentType = connection.getContentType();
            if (contentType != null && contentType.startsWith("image/")) {
                return "data:" + contentType + ";base64," + base64;
            }

            return "data:image/jpeg;base64," + base64;

        } catch (Exception e) {
            log.error("图片转Base64失败，URL：{}", imageUrl, e);
            return null;
        } finally {
            try {
                if (inputStream != null) {
                    inputStream.close();
                }
                if (outputStream != null) {
                    outputStream.close();
                }
            } catch (Exception e) {
                log.error("关闭流失败", e);
            }
        }
    }
}