250 lines
7.3 KiB
PHP
250 lines
7.3 KiB
PHP
<?php
|
|
|
|
trait Brizy_Editor_Asset_StaticFileTrait
|
|
{
|
|
|
|
public static function get_asset_content($asset_source)
|
|
{
|
|
$http = new WP_Http();
|
|
$wp_response = null;
|
|
if (is_string($asset_source)) {
|
|
$wp_response = $http->request($asset_source, array('timeout' => 30));
|
|
} else {
|
|
foreach ($asset_source as $url) {
|
|
$wp_response = $http->request($url, array('timeout' => 30));
|
|
|
|
if (is_wp_error($wp_response)) {
|
|
Brizy_Logger::instance()->error('Unable to get media content', array('exception' => $wp_response));
|
|
continue;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
$code = wp_remote_retrieve_response_code($wp_response);
|
|
|
|
if (is_wp_error($wp_response) || !($code >= 200 && $code < 300)) {
|
|
Brizy_Logger::instance()->error('Unable to get media content', array('exception' => $wp_response));
|
|
|
|
return false;
|
|
}
|
|
|
|
$content = wp_remote_retrieve_body($wp_response);
|
|
|
|
return $content;
|
|
}
|
|
|
|
/**
|
|
* @param $asset_source
|
|
* @param $asset_path
|
|
*
|
|
* @return bool
|
|
*/
|
|
protected function store_file($asset_source, $asset_path)
|
|
{
|
|
|
|
if (file_exists($asset_path)) {
|
|
return true;
|
|
}
|
|
|
|
try {
|
|
// check destination dir
|
|
$dir_path = dirname($asset_path);
|
|
|
|
if (!file_exists($dir_path)) {
|
|
if (!file_exists($dir_path) && !mkdir($dir_path, 0755, true) && !is_dir($dir_path)) {
|
|
throw new \RuntimeException(sprintf('Directory "%s" was not created', $dir_path));
|
|
}
|
|
}
|
|
|
|
$content = self::get_asset_content($asset_source);
|
|
|
|
if ($content !== false) {
|
|
file_put_contents($asset_path, $content);
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
} catch (Exception $e) {
|
|
// clean up
|
|
if ($asset_path) {
|
|
@unlink($asset_path);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
protected function create_attachment($madia_name, $absolute_asset_path, $relative_asset_path, $post_id = null, $uid = null)
|
|
{
|
|
return self::createMediaAttachment($absolute_asset_path, $relative_asset_path, $post_id, $uid);
|
|
}
|
|
|
|
public static function createMediaAttachment($absolute_asset_path, $relative_asset_path, $post_id = null, $uid = null)
|
|
{
|
|
$filetype = wp_check_filetype($absolute_asset_path);
|
|
|
|
$upload_path = wp_upload_dir();
|
|
|
|
$attachment = array(
|
|
'guid' => $upload_path['baseurl'] . "/" . $relative_asset_path,
|
|
'post_mime_type' => $filetype['type'],
|
|
'post_title' => basename($absolute_asset_path),
|
|
'post_content' => '',
|
|
'post_status' => 'inherit'
|
|
);
|
|
|
|
$attachment_id = wp_insert_attachment($attachment, $relative_asset_path, $post_id);
|
|
|
|
if (is_wp_error($attachment_id) || $attachment_id === 0) {
|
|
return false;
|
|
}
|
|
|
|
update_post_meta($attachment_id, 'brizy_attachment_uid', $uid ? $uid : md5($attachment_id . time()));
|
|
|
|
if (!function_exists('wp_generate_attachment_metadata')) {
|
|
include_once ABSPATH . "/wp-admin/includes/image.php";
|
|
}
|
|
|
|
$attach_data = wp_generate_attachment_metadata($attachment_id, $absolute_asset_path);
|
|
wp_update_attachment_metadata($attachment_id, $attach_data);
|
|
|
|
return $attachment_id;
|
|
}
|
|
|
|
/**
|
|
* @param $filename
|
|
* @param array $headers
|
|
*/
|
|
public function send_file($filename, $headers = array())
|
|
{
|
|
if (file_exists($filename)) {
|
|
|
|
$defaultHeaders = array(
|
|
'Content-Type' => self::get_mime($filename, 1),
|
|
'Cache-Control' => 'max-age=600'
|
|
);
|
|
|
|
$content = file_get_contents($filename);
|
|
|
|
// send headers
|
|
$headers = array_merge($defaultHeaders, $headers);
|
|
|
|
foreach ($headers as $key => $val) {
|
|
if (is_array($val)) {
|
|
$val = implode(', ', $val);
|
|
}
|
|
|
|
header("{$key}: {$val}");
|
|
}
|
|
// send file content
|
|
echo $content;
|
|
exit;
|
|
} else {
|
|
global $wp_query;
|
|
$wp_query->set_404();
|
|
|
|
return;
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* @param $filename
|
|
* @param int $mode
|
|
*
|
|
* @return mixed|string
|
|
*/
|
|
public static function get_mime($filename, $mode = 0)
|
|
{
|
|
|
|
// mode 0 = full check
|
|
// mode 1 = extension check only
|
|
|
|
$mime_types = array(
|
|
|
|
'txt' => 'text/plain',
|
|
'htm' => 'text/html',
|
|
'html' => 'text/html',
|
|
'php' => 'text/html',
|
|
'css' => 'text/css',
|
|
'js' => 'application/javascript',
|
|
'json' => 'application/json',
|
|
'xml' => 'application/xml',
|
|
'swf' => 'application/x-shockwave-flash',
|
|
'flv' => 'video/x-flv',
|
|
|
|
// images
|
|
'png' => 'image/png',
|
|
'jpe' => 'image/jpeg',
|
|
'jpeg' => 'image/jpeg',
|
|
'jpg' => 'image/jpeg',
|
|
'gif' => 'image/gif',
|
|
'bmp' => 'image/bmp',
|
|
'ico' => 'image/vnd.microsoft.icon',
|
|
'tiff' => 'image/tiff',
|
|
'tif' => 'image/tiff',
|
|
'svg' => 'image/svg+xml',
|
|
'svgz' => 'image/svg+xml',
|
|
'webp' => 'image/webp',
|
|
|
|
// archives
|
|
'zip' => 'application/zip',
|
|
'rar' => 'application/x-rar-compressed',
|
|
'exe' => 'application/x-msdownload',
|
|
'msi' => 'application/x-msdownload',
|
|
'cab' => 'application/vnd.ms-cab-compressed',
|
|
|
|
// audio/video
|
|
'mp3' => 'audio/mpeg',
|
|
'qt' => 'video/quicktime',
|
|
'mov' => 'video/quicktime',
|
|
|
|
// adobe
|
|
'pdf' => 'application/pdf',
|
|
'psd' => 'image/vnd.adobe.photoshop',
|
|
'ai' => 'application/postscript',
|
|
'eps' => 'application/postscript',
|
|
'ps' => 'application/postscript',
|
|
|
|
// ms office
|
|
'doc' => 'application/msword',
|
|
'rtf' => 'application/rtf',
|
|
'xls' => 'application/vnd.ms-excel',
|
|
'ppt' => 'application/vnd.ms-powerpoint',
|
|
'docx' => 'application/msword',
|
|
'xlsx' => 'application/vnd.ms-excel',
|
|
'pptx' => 'application/vnd.ms-powerpoint',
|
|
|
|
|
|
// open office
|
|
'odt' => 'application/vnd.oasis.opendocument.text',
|
|
'ods' => 'application/vnd.oasis.opendocument.spreadsheet',
|
|
);
|
|
|
|
$array = explode('.', $filename);
|
|
$str = end($array);
|
|
$ext = strtolower($str);
|
|
|
|
if (function_exists('mime_content_type') && $mode == 0) {
|
|
$mimetype = mime_content_type($filename);
|
|
|
|
return $mimetype;
|
|
|
|
} elseif (function_exists('finfo_open') && $mode == 0) {
|
|
$finfo = finfo_open(FILEINFO_MIME);
|
|
$mimetype = finfo_file($finfo, $filename);
|
|
finfo_close($finfo);
|
|
|
|
return $mimetype;
|
|
} elseif (array_key_exists($ext, $mime_types)) {
|
|
return $mime_types[$ext];
|
|
} else {
|
|
return 'application/octet-stream';
|
|
}
|
|
}
|
|
}
|