PHP实现远程图片下载并保存至本地的两种方法

/**
 * 功能：下载远程图片并保存到本地
 * @param string $url 远程图片地址
 * @param string $save_dir 保存目录
 * @param string $filename 保存文件名（留空则自动生成）
 * @param int $type 0=禁用cURL(使用readfile), 1=启用cURL
 * @return array 包含文件名、路径和状态码
 */
function downloadRemoteImage($url, $save_dir = '', $filename = '', $type = 1) {
    // 1. 参数校验
    if (trim($url) === '') {
        return ['file_name' => '', 'save_path' => '', 'error' => 'URL为空'];
    }
    
    // 设置默认保存目录
    if (trim($save_dir) === '') {
        $save_dir = './';
    }
    
    // 2. 确定文件名
    if (trim($filename) === '') {
        $ext = strrchr($url, '.');
        // 简单的格式过滤，建议根据Content-Type进一步校验
        if ($ext !== '.gif' && $ext !== '.jpg' && $ext !== '.jpeg' && $ext !== '.png') {
            return ['file_name' => '', 'save_path' => '', 'error' => '不支持的图片格式'];
        }
        $filename = time() . $ext;
    }
    
    // 3. 目录处理与安全创建
    if (substr($save_dir, -1) !== '/') {
        $save_dir .= '/';
    }
    
    if (!is_dir($save_dir)) {
        // 递归创建目录
        if (!mkdir($save_dir, 0755, true)) {
            return ['file_name' => '', 'save_path' => '', 'error' => '目录创建失败'];
        }
    }
    
    // 4. 获取文件内容
    $img = '';
    if ($type) {
        // 使用cURL
        $ch = curl_init();
        curl_setopt($ch, CURLOPT_URL, $url);
        curl_setopt($ch, CURLOPT_RETURNTRANSFER, 1);
        curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 5);
        curl_setopt($ch, CURLOPT_FOLLOWLOCATION, 1); // 允许重定向
        $img = curl_exec($ch);
        $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
        curl_close($ch);
        
        if ($httpCode != 200 || empty($img)) {
            return ['file_name' => '', 'save_path' => '', 'error' => '请求失败或返回数据为空'];
        }
    } else {
        // 备选方案：ob + readfile
        ob_start();
        readfile($url);
        $img = ob_get_contents();
        ob_end_clean();
    }
    
    // 5. 写入本地文件
    $fp = fopen($save_dir . $filename, 'wb'); // 使用wb模式确保二进制安全
    fwrite($fp, $img);
    fclose($fp);
    
    unset($img, $url);
    
    return [
        'file_name' => $filename,
        'save_path' => $save_dir . $filename,
        'error' => 0
    ];
}

function grabImageWithOB($url, $filename = "") {
    if ($url == "") return false;
    
    // 提取后缀并校验
    if ($filename == "") {
        $ext = strrchr($url, ".");
        if ($ext !== ".gif" && $ext !== ".jpg" && $ext !== ".jpeg" && $ext !== ".png") {
            return false;
        }
        $filename = date("dMYHis") . $ext;
    }
    
    // 启动输出控制
    ob_start();
    // 读取并输出
    readfile($url);
    // 获取缓存内容
    $img = ob_get_contents();
    // 清理并关闭
    ob_end_clean();
    
    if (!$img) return false;
    
    // 写入文件
    $fp = fopen($filename, "w");
    fwrite($fp, $img);
    fclose($fp);
    
    return $filename;
}

curl_setopt($ch, CURLOPT_USERAGENT, 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36');