1. 程式人生 > >php採集遠端圖片儲存到本地

php採集遠端圖片儲存到本地

/**
 * 採集遠端圖片
 * @param string $url 遠端檔案地址
 * @param string $filename 儲存後的檔名(為空時則為隨機生成的檔名,否則為原檔名)
 * @param array $fileType 允許的檔案型別
 * @param string $dirName 檔案儲存的路徑
 * @param int $type 遠端獲取檔案的方式
 * @return json 返回檔名、檔案的儲存路徑
 */
function download_image($url, $fileName = '', $dirName, $fileType = array('jpg', 'gif', 'png'), $type = 1)
{
    if ($url == '')
    {
        return false;
    }

    // 獲取檔案原檔名
    $defaultFileName = basename($url);

    // 獲取檔案型別
    $suffix = substr(strrchr($url, '.'), 1);
    if (!in_array($suffix, $fileType))
    {
        return false;
    }

    // 設定儲存後的檔名
    $fileName = $fileName == '' ? time() . rand(0, 9) . '.' . $suffix : $defaultFileName;

    // 獲取遠端檔案資源
    if ($type)
    {
        $ch = curl_init();
        $timeout = 30;
        curl_setopt($ch, CURLOPT_URL, $url);
        curl_setopt($ch, CURLOPT_RETURNTRANSFER, 1);
        curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, $timeout);
        $file = curl_exec($ch);
        curl_close($ch);
    }
    else
    {
        ob_start();
        readfile($url);
        $file = ob_get_contents();
        ob_end_clean();
    }

    // 設定檔案儲存路徑
    $dirName = $dirName;
    if (!file_exists($dirName))
    {
        mkdir($dirName, 0777, true);
    }

    // 儲存檔案
    $res = fopen($dirName . '/' . $fileName, 'a');
    fwrite($res, $file);
    fclose($res);

    return array(
        'fileName' => $fileName,
        'saveDir' => $dirName
    );
}