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

我们在通过 wp 发布文章时经常会复制别处的文章，文章里通常也带有图片，那么如何把图片粘贴过来的时候自动本地化上传到媒体库呢？

因为最近在给一个客户开发一个自动本地化+自动水印+自动 FTP 上传到另一台服务器的功能，所以在这里给大家说明一下。

/**
* 钩子函数：将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);