package com.ruoyi.web.controller.tool;

import cn.hutool.core.util.StrUtil;
import com.ruoyi.common.utils.file.FileUploadUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;

@Slf4j
public class FaceBase64ToUrlUtil {
    /**
     * 处理访客人像（Base64转URL）
     * 使用若依的 FileUploadUtils，返回完整URL
     *
     * @param faceImageData Base64图片数据
     * @param idCard 身份证号（用于生成文件名）
     * @return 图片完整URL
     */
    public static String processFaceImage(String faceImageData, String idCard) {
        if (!StringUtils.hasText(faceImageData)) {
            return null;
        }

        try {
            // 判断是否为Base64格式
            if (isBase64Image(faceImageData)) {
                // 1. Base64转MultipartFile
                MultipartFile file = base64ToMultipartFile(faceImageData, "face_" + idCard);
                if (file == null) {
                    log.error("Base64转文件失败");
                    return null;
                }

                // 2. 使用若依的FileUploadUtils上传，返回完整URL
                String imageUrl = FileUploadUtils.upload(file);
                log.info("访客人像上传成功，URL: {}", imageUrl);
                return imageUrl;
            } else {
                // 如果已经是URL，直接返回
                log.info("访客人像已是URL格式，直接使用: {}", faceImageData);
                return faceImageData;
            }
        } catch (Exception e) {
            log.error("处理访客人像失败，身份证号: {}", idCard, e);
            return null;
        }
    }

    /**
     * 判断是否为Base64图片格式
     */
    private static boolean isBase64Image(String data) {
        if (!StringUtils.hasText(data)) {
            return false;
        }
        return data.contains("data:image/") ||
                data.contains("base64") ||
                (data.length() > 100 && !data.startsWith("http"));
    }

    /**
     * Base64转MultipartFile
     */
    private static MultipartFile base64ToMultipartFile(String base64Data, String fileName) {
        try {
            // 去除Base64前缀
            String[] base64Array = base64Data.split(",");
            String base64Str = base64Array.length > 1 ? base64Array[1] : base64Array[0];

            // 获取图片格式
            String imageType = getImageType(base64Data);
            if (StrUtil.isBlank(imageType)) {
                imageType = "jpg";
            }

            // 解码Base64
            byte[] imageBytes = cn.hutool.core.codec.Base64.decode(base64Str);

            // 创建临时文件
            String fullFileName = fileName + "." + imageType;
            File tempFile = File.createTempFile("upload_", "_" + fullFileName);
            tempFile.deleteOnExit();

            // 写入数据
            try (FileOutputStream fos = new FileOutputStream(tempFile)) {
                fos.write(imageBytes);
                fos.flush();
            }

            // 转换为MultipartFile
            return new CustomMultipartFile(tempFile, fullFileName);

        } catch (Exception e) {
            log.error("Base64转MultipartFile失败", e);
            return null;
        }
    }

    /**
     * 获取图片格式
     */
    private static String getImageType(String base64Data) {
        if (StrUtil.isBlank(base64Data)) {
            return "jpg";
        }

        if (base64Data.contains("data:image/")) {
            String prefix = base64Data.substring(0, base64Data.indexOf(","));
            if (prefix.contains("png")) {
                return "png";
            } else if (prefix.contains("jpg") || prefix.contains("jpeg")) {
                return "jpg";
            } else if (prefix.contains("gif")) {
                return "gif";
            } else if (prefix.contains("bmp")) {
                return "bmp";
            } else if (prefix.contains("webp")) {
                return "webp";
            }
        }

        return "jpg";
    }

    /**
     * 自定义MultipartFile实现
     */
    public static class CustomMultipartFile implements MultipartFile {
        private final File file;
        private final String fileName;

        public CustomMultipartFile(File file, String fileName) {
            this.file = file;
            this.fileName = fileName;
        }

        @Override
        public String getName() {
            return fileName;
        }

        @Override
        public String getOriginalFilename() {
            return fileName;
        }

        @Override
        public String getContentType() {
            String extension = getFileExtension(fileName);
            switch (extension) {
                case "png":
                    return "image/png";
                case "jpg":
                case "jpeg":
                    return "image/jpeg";
                case "gif":
                    return "image/gif";
                case "bmp":
                    return "image/bmp";
                case "webp":
                    return "image/webp";
                default:
                    return "image/jpeg";
            }
        }

        @Override
        public boolean isEmpty() {
            return file == null || file.length() == 0;
        }

        @Override
        public long getSize() {
            return file.length();
        }

        @Override
        public byte[] getBytes() throws IOException {
            byte[] bytes = new byte[(int) file.length()];
            try (FileInputStream fis = new FileInputStream(file)) {
                fis.read(bytes);
            }
            return bytes;
        }

        @Override
        public InputStream getInputStream() throws IOException {
            return new FileInputStream(file);
        }

        @Override
        public void transferTo(File dest) throws IOException, IllegalStateException {
            try (FileInputStream fis = new FileInputStream(file);
                 FileOutputStream fos = new FileOutputStream(dest)) {
                byte[] buffer = new byte[1024];
                int length;
                while ((length = fis.read(buffer)) > 0) {
                    fos.write(buffer, 0, length);
                }
                fos.flush();
            }
        }

        private String getFileExtension(String fileName) {
            if (fileName == null || fileName.isEmpty()) {
                return "";
            }
            int lastDotIndex = fileName.lastIndexOf(".");
            if (lastDotIndex > 0 && lastDotIndex < fileName.length() - 1) {
                return fileName.substring(lastDotIndex + 1).toLowerCase();
            }
            return "";
        }
    }
}