package cn.iocoder.yudao.framework.common.util.io;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileTypeUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.file.FileNameUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.digest.DigestUtil;
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.common.exception.enums.GlobalErrorCodeConstants;
import lombok.SneakyThrows;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLEncoder;
import java.util.Date;

/**
 * 文件工具类
 *
 * @author 芋道源码
 */
public class FileUtils {

    /**
     * 创建临时文件
     * 该文件会在 JVM 退出时，进行删除
     *
     * @param data 文件内容
     * @return 文件
     */
    @SneakyThrows
    public static File createTempFile(String data) {
        File file = createTempFile();
        // 写入内容
        FileUtil.writeUtf8String(data, file);
        return file;
    }

    /**
     * 创建临时文件
     * 该文件会在 JVM 退出时，进行删除
     *
     * @param data 文件内容
     * @return 文件
     */
    @SneakyThrows
    public static File createTempFile(byte[] data) {
        File file = createTempFile();
        // 写入内容
        FileUtil.writeBytes(data, file);
        return file;
    }

    /**
     * 创建临时文件，无内容
     * 该文件会在 JVM 退出时，进行删除
     *
     * @return 文件
     */
    @SneakyThrows
    public static File createTempFile() {
        // 创建文件，通过 UUID 保证唯一
        File file = File.createTempFile(IdUtil.simpleUUID(), null);
        // 标记 JVM 退出时，自动删除
        file.deleteOnExit();
        return file;
    }

    /**
     * 生成文件路径
     *
     * @param content      文件内容
     * @param originalName 原始文件名
     * @return path，唯一不可重复
     */
    public static String generatePath(byte[] content, String originalName) {
        String sha256Hex = DigestUtil.sha256Hex(content);
        // 情况一：如果存在 name，则优先使用 name 的后缀
        if (StrUtil.isNotBlank(originalName)) {
            String extName = FileNameUtil.extName(originalName);
            //return StrUtil.isBlank(extName) ? sha256Hex : sha256Hex + "." + extName;
            return StrUtil.isBlank(extName) ? sha256Hex : originalName;
        }
        // 情况二：基于 content 计算
        return sha256Hex + '.' + FileTypeUtil.getType(new ByteArrayInputStream(content));
    }

    public static long getFileSizeMB(String fileUrl){
        try {
            URL url = new URL(fileUrl);
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setRequestMethod("HEAD");
            conn.setRequestProperty("Accept-Encoding", "identity"); // 禁用压缩

            // 尝试解析Content-Length头
            String contentLength = conn.getHeaderField("Content-Length");
            if (contentLength != null) {
                return Long.parseLong(contentLength);
            }

            // 处理分块传输或未知大小
            if ("chunked".equalsIgnoreCase(conn.getHeaderField("Transfer-Encoding"))) {
                System.out.println("分块传输编码，无法直接获取大小");
                return -1;
            }

            conn.disconnect();
            return -1; // 其他未知情况
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public static String getFileNameByUrlParse(String urlPath)  {
        // 1. 解析 URL 路径，获取完整路径部分（path）
        URL url = null;
        try {
            url = new URL(urlPath);
        } catch (MalformedURLException e) {
            throw new ServiceException(new ErrorCode(100,"文件名称解析错误"));
        }
        String fullPath = url.getPath(); // 示例：/admin-api/infra/file/29/get/wechat_2025-08-19_221137_550.png

        // 2. 找到最后一个 "/" 的位置，截取其后的字符串（即文件名）
        int lastSlashIndex = fullPath.lastIndexOf("/");
        if (lastSlashIndex == -1) {
            // 若路径中无 "/"，则整个路径视为文件名（极端场景，如：https://example.com/filename.png）
            return fullPath;
        }
        return fullPath.substring(lastSlashIndex + 1);
    }

    public static String encodePathSegment(String pathSegment) {
        try {
            // 使用UTF-8编码，符合现代Web标准
            String encoded = URLEncoder.encode(pathSegment, "UTF-8");
            // 替换空格编码：URLEncoder会将空格转为+，而URL路径中推荐使用%20
            return encoded.replace("+", "%20");
        } catch (UnsupportedEncodingException e) {
            // UTF-8是Java标准编码，不会抛出此异常
            throw new RuntimeException("URL编码失败", e);
        }
    }
}
