package cn.iocoder.byzs.module.infra.framework.file.core.utils;

import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.byzs.framework.common.util.http.HttpUtils;
import jakarta.servlet.http.HttpServletResponse;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.tika.Tika;
import org.apache.tika.mime.MimeTypeException;
import org.apache.tika.mime.MimeTypes;

import java.io.IOException;

/**
 * 文件类型 Utils
 *
 * @author 博雅智算源码
 */
@Slf4j
public class FileTypeUtils {

    private static final Tika TIKA = new Tika();

    /**
     * 获得文件的 mineType，对于 doc，jar 等文件会有误差
     *
     * @param data 文件内容
     * @return mineType 无法识别时会返回“application/octet-stream”
     */
    @SneakyThrows
    public static String getMineType(byte[] data) {
        return TIKA.detect(data);
    }

    /**
     * 已知文件名，获取文件类型，在某些情况下比通过字节数组准确，例如使用 jar 文件时，通过名字更为准确
     *
     * @param name 文件名
     * @return mineType 无法识别时会返回“application/octet-stream”
     */
    public static String getMineType(String name) {
        return TIKA.detect(name);
    }

    /**
     * 在拥有文件和数据的情况下，最好使用此方法，最为准确
     *
     * @param data 文件内容
     * @param name 文件名
     * @return mineType 无法识别时会返回“application/octet-stream”
     */
    public static String getMineType(byte[] data, String name) {
        return TIKA.detect(data, name);
    }

    /**
     * 根据 mineType 获得文件后缀
     *
     * 注意：如果获取不到，或者发生异常，都返回 null
     *
     * @param mineType 类型
     * @return 后缀，例如说 .pdf
     */
    public static String getExtension(String mineType) {
        try {
            return MimeTypes.getDefaultMimeTypes().forName(mineType).getExtension();
        } catch (MimeTypeException e) {
            log.warn("[getExtension][获取文件后缀({}) 失败]", mineType, e);
            return null;
        }
    }

    /**
     * 返回附件（优化版：支持缓存和视频分片）
     *
     * @param response 响应
     * @param filename 文件名
     * @param content  内容（完整或分片）
     */
    public static void writeAttachment(HttpServletResponse response, String filename, byte[] content) throws IOException {
        // 1. 基础头设置
        String contentType = getMineType(content, filename);

        // 2. 设置响应头（根据文件类型）
//        if (filename.endsWith(".m3u8")) {
//            contentType = "application/x-mpegURL";
//        } else if (filename.endsWith(".ts")) {
//            contentType = "video/MP2T";
//        }
        response.setContentType(contentType);
        response.setContentLength(content.length);

        // 2. 缓存设置（核心：添加缓存头）
        // 视频缓存30天，其他文件缓存7天（可按需调整）
        long maxAge = StrUtil.containsIgnoreCase(contentType, "video") ? 30 * 24 * 60 * 60 : 7 * 24 * 60 * 60;
        response.setHeader("Cache-Control", "public, max-age=" + maxAge); // 公共缓存，有效期30天
        response.setHeader("Pragma", "public"); // 兼容老浏览器

        // 3. 文件名处理（视频用 inline 播放，非视频用 attachment 下载）
        String disposition = StrUtil.containsIgnoreCase(contentType, "video")
                ? "inline;filename=" + HttpUtils.encodeUtf8(filename) // 视频：在线播放
                : "attachment;filename=" + HttpUtils.encodeUtf8(filename); // 其他：下载
        response.setHeader("Content-Disposition", disposition);

        // 4. 视频特殊处理（修正原逻辑错误）
        if (StrUtil.containsIgnoreCase(contentType, "video")) {
            response.setHeader("Accept-Ranges", "bytes"); // 支持分片
            // 原代码的 Content-Range 格式错误，已在 getFileContent 中正确设置，此处无需重复
        }

        // 5. 输出内容
        IoUtil.write(response.getOutputStream(), false, content);
    }

}
