460 lines
10 KiB
PHP
Executable File
460 lines
10 KiB
PHP
Executable File
<?php
|
||
/**
|
||
* @author 友人a丶
|
||
* @date 2022/8/27
|
||
* 图片本地化封装类
|
||
*/
|
||
|
||
class Nicen_local {
|
||
|
||
|
||
private static $self;
|
||
private $is_sava_database; //本地化是否保存到数据库
|
||
private $root_path; //从文件系统根目录开始的绝对路径
|
||
private $site_path; //从站点根目录开始的绝对路径
|
||
private $is_add_domain; //是否需要添加域名
|
||
|
||
|
||
private function __construct() {
|
||
$this->is_sava_database = nicen_make_config( 'nicen_make_plugin_local' );
|
||
$this->site_path = nicen_make_config( 'nicen_make_plugin_path' ); //站点目录
|
||
$this->is_add_domain = nicen_make_config( 'nicen_make_plugin_add_domain' ); //是否需要域名
|
||
$this->root_path = nicen_local_image_root . $this->site_path; //站点目录
|
||
$this->add_thumd = nicen_make_config( 'nicen_make_plugin_add_thumb' );//生成缩略图
|
||
}
|
||
|
||
/**
|
||
* 本地化图片
|
||
*
|
||
* @param $flag false时return，true终止脚本直接输出
|
||
* */
|
||
function localImage( $url, $flag = true ) {
|
||
|
||
$upload_root = $this->root_path; //站点目录
|
||
$upload = $this->site_path; //主题路径
|
||
$log = Nicen_Log::getInstance(); //获取日志对象
|
||
|
||
/**
|
||
* 指定的图片保存目录是否存在
|
||
* 不存在则创建
|
||
* 创建不成功直接退出
|
||
* */
|
||
if ( ! file_exists( $upload_root ) ) {
|
||
if ( ! mkdir( $upload_root ) ) {
|
||
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => - 1,
|
||
'result' => $upload . '指定的图片保存目录创建失败！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => - 1,
|
||
'result' => $upload . '指定的图片保存目录创建失败！'
|
||
] );
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 判断目录是否可写
|
||
* */
|
||
if ( ! is_writable( $upload_root ) ) {
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => - 1,
|
||
'result' => $upload . '指定的图片保存目录不可写，替换失败！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => - 1,
|
||
'result' => $upload . '指定的图片保存目录不可写，替换失败！'
|
||
] );
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 判断是否传递图片
|
||
* */
|
||
if ( empty( $url ) ) {
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 0,
|
||
'result' => '图片链接为空！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 0,
|
||
'result' => '图片链接为空！'
|
||
] );
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 判断是否传递图片
|
||
* */
|
||
if ( strpos( $url, 'http' ) === false ) {
|
||
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 0,
|
||
'result' => '图片链接不规范！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 0,
|
||
'result' => '图片链接不规范！'
|
||
] );
|
||
}
|
||
}
|
||
|
||
|
||
$url = html_entity_decode( $url ); //反转义出真实链接
|
||
|
||
/**
|
||
* 判断是否处于白名单
|
||
* */
|
||
if ( $this->is_white( $url ) ) {
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 0,
|
||
'result' => "白名单链接，无需进行本地化"
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 0,
|
||
'result' => "白名单链接，无需进行本地化"
|
||
] );
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取保存的文件类型
|
||
* */
|
||
$url_md5 = md5( $url ); //md5图片链接
|
||
|
||
|
||
/**
|
||
* 判断文件是否已经存在
|
||
* 已经存在则直接返回数据
|
||
* */
|
||
$file_search = glob( $upload_root . '/' . $url_md5 . '.*' );
|
||
|
||
if ( ! empty( $file_search ) ) {
|
||
/**
|
||
* 判断链接是否需要添加域名
|
||
* */
|
||
$link = $this->getLink( $file_search[0] ); //获取链接
|
||
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 1,
|
||
'result' => $link
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 1,
|
||
'result' => $link
|
||
] );
|
||
}
|
||
|
||
}
|
||
|
||
|
||
/**
|
||
* 判断链接是否可以访问
|
||
* */
|
||
if ( $this->getImage( $url, 1 ) != 200 ) {
|
||
if ( $flag ) {
|
||
exit( json_encode( ( [
|
||
'code' => 0,
|
||
'result' => $url . '图片链接无法访问！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 0,
|
||
'result' => $url . '图片链接无法访问！'
|
||
] );
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 获取图片内容
|
||
* html反转义
|
||
* */
|
||
$content = @$this->getImage( $url );
|
||
|
||
/**
|
||
* 如果读取成功
|
||
* */
|
||
if ( $content ) {
|
||
|
||
$tmp = $upload_root . '/' . $url_md5; //临时文件
|
||
/**
|
||
* 写入文件
|
||
* */
|
||
if ( file_put_contents( $tmp, $content, LOCK_EX ) ) {
|
||
|
||
|
||
$filename = $tmp . "." . nicen_plugin_getType( $tmp );
|
||
rename( $tmp, $filename ); //重命名文件
|
||
$link = $this->getLink( $filename ); //获取链接
|
||
|
||
/**
|
||
* 是否需要保存到数据库
|
||
* */
|
||
if ( $this->is_sava_database ) {
|
||
$this->saveAsData( $filename );
|
||
}
|
||
|
||
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 1,
|
||
'result' => $link
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 1,
|
||
'result' => $link
|
||
] );
|
||
}
|
||
|
||
} else {
|
||
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 0,
|
||
'result' => $url . '图片保存失败！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 0,
|
||
'result' => $url . '图片保存失败！'
|
||
] );
|
||
}
|
||
|
||
}
|
||
|
||
} else {
|
||
|
||
if ( $flag ) {
|
||
exit( json_encode( $log->add( [
|
||
'code' => 0,
|
||
'result' => $url . '图片下载失败！'
|
||
] ) ) );
|
||
} else {
|
||
return $log->add( [
|
||
'code' => 0,
|
||
'result' => $url . '图片下载失败！'
|
||
] );
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取单例
|
||
* */
|
||
public static function getInstance() {
|
||
/*如果实例不存在*/
|
||
if ( ! self::$self ) {
|
||
self::$self = new self();
|
||
}
|
||
|
||
return self::$self;
|
||
}
|
||
|
||
/**
|
||
* 判断指定链接是否是白名单
|
||
*
|
||
* @param string $url
|
||
*
|
||
* @return boolean
|
||
* */
|
||
public function is_white( $url ) {
|
||
|
||
$white = explode( "\n", get_option( 'nicen_make_publish_white' ) ); //获取列表
|
||
/*判断是否为空*/
|
||
if ( empty( $white ) ) {
|
||
return false;
|
||
}
|
||
|
||
/* 移除左右特殊字符 */
|
||
foreach ( $white as $key => $host ) {
|
||
$white[ $key ] = trim( $host );
|
||
}
|
||
|
||
$link = parse_url( $url ); //解析
|
||
|
||
if ( in_array( $link['host'], $white ) ) {
|
||
return true;
|
||
} else {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @param $filename
|
||
* 获取文件链接
|
||
*/
|
||
function getLink( $filename ) {
|
||
$filename = basename( $filename ); //文件名
|
||
/**
|
||
* 判断链接是否需要添加域名
|
||
* */
|
||
if ( $this->is_add_domain ) {
|
||
$link = site_url() . $this->site_path . '/' . $filename;
|
||
} else {
|
||
$link = $this->site_path . '/' . $filename;
|
||
}
|
||
|
||
return $link;
|
||
}
|
||
|
||
/**
|
||
* 获取图片内容
|
||
*
|
||
* @param $url string,图片的链接
|
||
* */
|
||
function getImage( $url, $option = 0, $self_referer = null ) {
|
||
|
||
$link = parse_url( $url );//解析链接
|
||
|
||
/**
|
||
* 请求头模拟
|
||
* */
|
||
$headers = [
|
||
"Accept-Encoding" => "gzip, deflate",
|
||
'User-Agent' => 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/106.0.0.0 Safari/537.36',
|
||
'Referer' => $self_referer ?: $link['scheme'] . '://' . $link['host']
|
||
];
|
||
|
||
|
||
/**
|
||
* 请求数据
|
||
* */
|
||
$args = [
|
||
'headers' => $headers,
|
||
'sslverify' => false,
|
||
'timeout' => 120,
|
||
];
|
||
|
||
if ( $option === 1 ) {
|
||
$res = wp_remote_head( $url, $args );
|
||
} else {
|
||
$res = wp_remote_get( $url, $args );
|
||
}
|
||
|
||
|
||
if ( ! is_array( $res ) ) {
|
||
$this->error = $res->get_error_message();
|
||
}
|
||
|
||
$code = wp_remote_retrieve_response_code( $res );
|
||
|
||
|
||
/**
|
||
* 是否是状态码判断
|
||
* */
|
||
if ( in_array( $code, [ 301, 302 ] ) ) {
|
||
$newUrl = wp_remote_retrieve_header( $res, 'Location' );
|
||
|
||
return $this->getImage( $newUrl, $option, " " ); // 递归调用
|
||
} else {
|
||
if ( $option ) {
|
||
|
||
/* 二次确认 */
|
||
if ( $option === 1 && $code !== 200 ) {
|
||
$code = $this->getImage( $url, 2 ); // 递归调用
|
||
}
|
||
|
||
return $code;
|
||
} else {
|
||
return wp_remote_retrieve_body( $res );
|
||
}
|
||
}
|
||
|
||
|
||
}
|
||
|
||
|
||
/**
|
||
* 保存图片到数据库
|
||
*
|
||
* @param $filename string 文件名
|
||
* */
|
||
function saveAsData( $filename ) {
|
||
|
||
global $nicen_Post_ID; //正在操作的文章
|
||
|
||
/**
|
||
* 获取设置的文件保存目录
|
||
* */
|
||
$link = $this->getLink( $filename ); //访问链接
|
||
|
||
/**
|
||
* 拼接插入的数据
|
||
* */
|
||
$attachment = array(
|
||
'guid' => $link,
|
||
'post_mime_type' => wp_get_image_mime( $filename ),
|
||
'post_title' => preg_replace( '/\.[^.]+$/', '', basename( $filename ) ),
|
||
'post_content' => '',
|
||
'post_status' => 'inherit'
|
||
);
|
||
|
||
// Insert the attachment.
|
||
if ( isset( $nicen_Post_ID ) ) {
|
||
|
||
$attach_id = wp_insert_attachment( $attachment, $filename, $nicen_Post_ID );
|
||
|
||
/**
|
||
* 是否作为特色图片
|
||
* */
|
||
|
||
if ( nicen_make_config( 'nicen_make_save_as_thumb' ) ) {
|
||
set_post_thumbnail( $nicen_Post_ID, $attach_id ); //设置特色图片
|
||
}
|
||
|
||
} else {
|
||
$attach_id = wp_insert_attachment( $attachment, $filename );
|
||
}
|
||
|
||
/**
|
||
* 更新访问的URL
|
||
* */
|
||
include_once ABSPATH . WPINC . '/pluggable.php';
|
||
include_once( ABSPATH . 'wp-admin/includes/image.php' );
|
||
|
||
/* 保存到数据库不生成缩略图 */
|
||
if ( empty( $this->add_thumd ) ) {
|
||
/* 不生成缩略图 */
|
||
add_filter( 'intermediate_image_sizes_advanced', function ( $sizes ) {
|
||
return array();
|
||
} );
|
||
}
|
||
|
||
// 更新元数据
|
||
$attach_data = wp_generate_attachment_metadata( $attach_id, $filename );
|
||
wp_update_attachment_metadata( $attach_id, $attach_data );
|
||
}
|
||
|
||
/**
|
||
* 获取网站状态码
|
||
* */
|
||
function getHttpcode( $url ) {
|
||
$response = wp_remote_get( $url, [
|
||
'sslverify' => false,
|
||
'timeout' => 60,
|
||
] );
|
||
$http_code = wp_remote_retrieve_response_code( $response );
|
||
|
||
return $http_code;
|
||
}
|
||
}
|
||
|