WordPress 自动将文章里复制粘贴的外链图片本地化到媒体库

/**

* 钩子函数：将post_content中本站服务器域名外的img上传至服务器并替换url

*

* @param Int $post_id

* @param Object $post

*

*/

function ecp_save_post($post_id, $post) {

// wordpress 全局变量 wpdb类

global $wpdb;

// 只有在点击发布/更新时才执行以下动作

if($post->post_status == 'publish') {

// 匹配<img>、src，存入$matches数组,

$p = '/<img.*[\s]src=[\"|\'](.*)[\"|\'].*>/iU';

$num = preg_match_all($p, $post->post_content, $matches);

if ($num) {

// 本地上传路径信息(数组)，用来构造url

$wp_upload_dir = wp_upload_dir();

// 脚本执行不限制时间

set_time_limit(0);

// 构造curl，配置参数

$ch = curl_init();

curl_setopt($ch, CURLOPT_HEADER, false);

curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);

curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);

// 抓取时如果发生301，302跳转，则进行跳转抓取

curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);

// 最多跳转20次

curl_setopt($ch, CURLOPT_MAXREDIRS,20);

// 发起连接前最长等待时间

curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 30);

$ecp_options = $_SERVER['HTTP_HOST'];

foreach ($matches[1] as $src) {

if (isset($src) && strpos($src, $ecp_options) === false) {

// 如果图片域名不是本站域名

// 检查src中的url有无扩展名，没有则重新给定文件名

// 注意：如果url中有扩展名但格式为webp，那么返回的file_info数组为 ['ext' =>'','type' =>'']

$file_info = wp_check_filetype(basename($src), null);

if ($file_info['ext'] == false) {

// 无扩展名和webp格式的图片会被作为无扩展名文件处理

date_default_timezone_set('PRC');

$file_name = date('YmdHis-').dechex(mt_rand(100000, 999999)).'.tmp';

} else {

// 有扩展名的图片重新给定文件名防止与本地文件名冲突

$file_name = dechex(mt_rand(100000, 999999)) . '-' . basename($src);

}

// 抓取图片, 将图片写入本地文件

curl_setopt($ch, CURLOPT_URL, $src);

$file_path = $wp_upload_dir['path'] . '/' . $file_name;

$img = fopen($file_path, 'wb');

// curl写入$img

curl_setopt($ch, CURLOPT_FILE, $img);

$img_data = curl_exec($ch);

fclose($img);

if (file_exists($file_path) && filesize($file_path) > 0) {

// 将扩展名为tmp和webp的图片转换为jpeg文件并重命名

$t = curl_getinfo($ch, CURLINFO_CONTENT_TYPE);

$arr = explode('/', $t);

// 对url地址中没有扩展名或扩展名为webp的图片进行处理

if (pathinfo($file_path, PATHINFO_EXTENSION) == 'tmp') {

$file_path = ecp_handle_ext($file_path, $arr[1], $wp_upload_dir['path'], $file_name, 'tmp');

} elseif (pathinfo($file_path, PATHINFO_EXTENSION) == 'webp') {

$file_path = ecp_handle_ext($file_path, $arr[1], $wp_upload_dir['path'], $file_name, 'webp');

}

// 替换文章内容中的src

$post->post_content = str_replace($src, $wp_upload_dir['url'] . '/' . basename($file_path), $post->post_content);

// 构造附件post参数并插入媒体库(作为一个post插入到数据库)

$attachment = ecp_get_attachment_post(basename($file_path), $wp_upload_dir['url'] . '/' . basename($file_path));

// 生成并更新图片的metadata信息

$attach_id = wp_insert_attachment($attachment, ltrim($wp_upload_dir['subdir'] . '/' . basename($file_path), '/'), 0);

$attach_data = wp_generate_attachment_metadata($attach_id, $file_path);

// 直接调用wordpress函数，将metadata信息写入数据库

$ss = wp_update_attachment_metadata($attach_id, $attach_data);

}

}

}

curl_close($ch);

// 更新posts数据表的post_content字段

$wpdb->update( $wpdb->posts, array('post_content' => $post->post_content), array('ID' => $post->ID));

}

}

}

/**

* 处理没有扩展名的图片:转换格式或更改扩展名

* @param string $file 图片本地绝对路径

* @param string $type 图片mimetype

* @param string $file_dir 图片在本地的文件夹

* @param string $file_name 图片名称

* @param string $ext 图片扩展名

* @return string 处理后的本地图片绝对路径

*/

function ecp_handle_ext($file, $type, $file_dir, $file_name, $ext) {

switch ($ext) {

case 'tmp':

if (rename($file, str_replace('tmp', $type, $file))) {

if ('webp' == $type) {

// 将webp格式的图片转换为jpeg格式

return ecp_image_convert('webp', 'jpeg', $file_dir . '/' . str_replace('tmp', $type, $file_name));

}

return $file_dir . '/' . str_replace('tmp', $type, $file_name);

}

case 'webp':

if ('webp' == $type) {

// 将webp格式的图片转换为jpeg格式

return ecp_image_convert('webp', 'jpeg', $file);

} else {

if (rename($file, str_replace('webp', $type, $file))) {

return $file_dir . '/' . str_replace('webp', $type, $file_name);

}

}

default:

return $file;

}

}

/**

* 图片格式转换，暂只能从webp转换为jpeg

*

* @param string $from

* @param string $to

* @param string $image 图片本地绝对路径

* @return string 转换后的图片绝对路径

*/

function ecp_image_convert($from='webp', $to='jpeg', $image) {

// 加载 WebP 文件

$im = imagecreatefromwebp($image);

// 以 100% 的质量转换成 jpeg 格式并将原webp格式文件删除

if (imagejpeg($im, str_replace('webp', 'jpeg', $image), 100)) {

try {

unlink($image);

} catch (Exception $e) {

$error_msg = sprintf('Error removing local file %s: %s', $image,

$e->getMessage());

error_log($error_msg);

}

}

imagedestroy($im);

return str_replace('webp', 'jpeg', $image);

}

/**

* 构造图片post参数

*

* @param string $filename

* @param string $url

* @return array 图片post参数数组

*/

function ecp_get_attachment_post($filename, $url) {

$file_info = wp_check_filetype($filename, null);

return array(

'guid' => $url,

'post_type' => 'attachement',

'post_mime_type' => $file_info['type'],

'post_title' => preg_replace('/\.[^.]+$/', '', $filename),

'post_content' => '',

'post_status' => 'inherit'

);

}

// 钩子, 发布/草稿/预览时触发

add_action('save_post', 'ecp_save_post', 120, 2);