PHP图像压缩后下载文件类型不支持问题的解决方案

<?php

/**
 * 压缩图像并提供下载
 *
 * @param string $source_url 源图像文件路径
 * @param string $dest 目标图像保存路径
 * @param int $quality 图像质量 (JPEG: 0-100, PNG: 0-9)
 * @param string $type 目标输出类型 ('jpg' 或 'png')
 * @throws \Exception 如果无法打开文件
 */
function compress_image($source_url, $dest, $quality, $type) {
    // 1. 获取图像信息并创建图像资源
    $info = getimagesize($source_url);
    if ($info === false) {
        throw new \Exception('无法获取图像信息或文件不是有效图像: ' . $source_url);
    }

    $image = null;
    switch ($info['mime']) {
        case 'image/jpeg':
            $image = imagecreatefromjpeg($source_url);
            break;
        case 'image/gif':
            $image = imagecreatefromgif($source_url);
            break;
        case 'image/png':
            $image = imagecreatefrompng($source_url);
            break;
        default:
            throw new \Exception('不支持的图像MIME类型: ' . $info['mime']);
    }

    if ($image === false) {
        throw new \Exception('无法从源文件创建图像资源: ' . $source_url);
    }

    // 2. 将压缩后的图像保存到指定路径
    if ($type == "jpg") {
        imagejpeg($image, $dest, $quality);
    } else {
        // PNG质量范围为0-9，通常将100分制转换为9分制
        $png_quality = round($quality / 10); // 原始代码是 $quality/10，这里更精确地四舍五入
        imagepng($image, $dest, $png_quality);
    }

    // 释放内存
    imagedestroy($image);

    // 3. 设置HTTP头并流式传输文件内容供下载
    if (file_exists($dest) && ($fh = fopen($dest, 'r')) !== false) {
        // 确保在输出任何内容之前设置所有头
        header("Content-Type: application/force-download"); // 通用下载MIME类型
        // 也可以根据实际文件类型设置：
        // header("Content-Type: " . mime_content_type($dest)); 
        header("Content-Disposition: attachment; filename=\"" . basename($dest) . "\";");
        header("Content-Length: " . filesize($dest)); // 告知浏览器文件大小

        // 清除任何可能存在的输出缓冲区，避免额外内容干扰文件下载
        if (ob_get_level()) {
            ob_end_clean();
        }

        // 将文件内容直接输出到浏览器
        fpassthru($fh); // 推荐使用 fpassthru，它更高效
        fclose($fh);
    } else {
        throw new \Exception('无法打开或找到已保存的文件: ' . $dest);
    }
}

// 示例调用（在您的上传处理逻辑中）
if (isset($_REQUEST['submit_form'])) {
    foreach ($_FILES['image_file']['tmp_name'] as $key => $value) {
        $file_name = $_FILES['image_file']['name'][$key];
        $temp_name = $_FILES['image_file']['tmp_name'][$key];

        // 假设用户选择了质量和类型，这里使用默认值
        $quality = isset($_REQUEST['quality']) ? (int)$_REQUEST['quality'] : 80;
        $output_type = isset($_REQUEST['type']) ? ($_REQUEST['type'] == 1 ? "png" : "jpg") : "jpg";

        // 确保上传目录存在且可写
        $upload_dir = './upload/';
        if (!is_dir($upload_dir)) {
            mkdir($upload_dir, 0777, true);
        }

        $dest_path = $upload_dir . basename($file_name); // 使用 basename 避免路径注入

        try {
            compress_image($temp_name, $dest_path, $quality, $output_type);
            // 如果需要下载多个文件，这里可能需要调整逻辑，例如将文件打包成ZIP
            // 或者每次只处理一个文件并立即下载。
            // 对于多个文件，通常是先全部处理保存，然后提供一个ZIP下载链接。
        } catch (\Exception $e) {
            error_log("图像处理或下载失败: " . $e->getMessage());
            // 根据需要向用户显示错误信息
            echo "处理文件 " . htmlspecialchars($file_name) . " 时发生错误: " . htmlspecialchars($e->getMessage());
        }
    }
    exit; // 确保在文件下载后终止脚本执行
}

?>