wp_back/wp-content/plugins/shortpixel-image-optimiser/class/Helper/DownloadHelper.php

184 lines
5.3 KiB
PHP
Raw Normal View History

2024-05-20 15:37:46 +03:00
<?php
namespace ShortPixel\Helper;
if ( ! defined( 'ABSPATH' ) ) {
exit; // Exit if accessed directly.
}
use ShortPixel\ShortPixelLogger\ShortPixelLogger as Log;
use ShortPixel\Controller\ResponseController as ResponseController;
class DownloadHelper
{
private static $instance;
public function __construct()
{
$this->checkEnv();
}
public static function getInstance()
{
if (is_null(self::$instance))
{
self::$instance = new DownloadHelper();
}
return self::$instance;
}
protected function checkEnv()
{
if ( ! function_exists( 'download_url' ) ) {
require_once ABSPATH . 'wp-admin/includes/file.php';
}
}
public function downloadFile($url, $args = array())
{
$defaults = array(
'expectedSize' => null,
'destinationPath' => null,
);
$args = wp_parse_args($args, $defaults);
$success = false;
Log::addDebug('Downloading file :' . $url, $args);
$methods = array(
"download_url" => array(array($this, 'downloadURLMethod'), $url, false),
"download_url_force" => array(array($this, 'downloadURLMethod'), true),
"remote_get" => array(array($this, 'remoteGetMethod'), $url)
);
foreach($methods as $name => $data)
{
$function = $data[0];
if (is_callable($function))
{
$result = call_user_func_array($function, array_slice($data, 1) );
if (false !== $result)
{
$tempFile = $result;
$success = true;
break;
}
}
}
if (false === $success)
{
Log::addError('Failed to download File', $result);
ResponseController::addData('is_error', true);
//Responsecontroller::addData('message', $tempFile->get_error_message());
return false;
}
$fs = \wpSPIO()->filesystem();
$file = $fs->getFile($tempFile);
if (! is_null($args['destinationPath']))
{
$result = $this->moveDownload($file, $args['destinationPath']);
if (false === $result)
{
Log::addError('Failed to move Download', $args);
ResponseController::addData('is_error', true);
Responsecontroller::addData('message', __('Failed to move download to destination!', 'shortpixel-image-optimiser'));
return false;
}
else {
$file = $result;
}
}
return $file;
}
protected function moveDownload($fileObj, $destinationPath)
{
$fs = \wpSPIO()->filesystem();
$destinationFile = $fs->getFile($destinationPath);
// If file is non-existing, check directory and write-permissions.
if (false == $destinationFile->exists())
{
$dirObj = $destinationFile->getFileDir();
$dirObj->check(true);
}
$result = $fileObj->copy($destinationFile);
if ($result === false)
return false;
return $destinationFile;
}
private function downloadURLMethod($url, $force = false)
{
$executionTime = ini_get('max_execution_time');
if (! is_numeric($executionTime)) // edge case
{
$executionTime = 0;
}
$downloadTimeout = max($executionTime - 10, 15);
$url = $this->setPreferredProtocol(urldecode($url), $force);
$tempFile = \download_url($url, $downloadTimeout);
if (is_wp_error($tempFile))
{
Log::addError('Failed to Download File ', $tempFile);
Responsecontroller::addData('message', $tempFile->get_error_message());
return false;
}
return $tempFile;
}
private function remoteGetMethod($url)
{
//get_temp_dir
$tmpfname = tempnam(get_temp_dir(), 'spiotmp');
$downloadTimeout = max(ini_get('max_execution_time') - 10, 15);
$args_for_get = array(
'stream' => true,
'filename' => $tmpfname,
'timeout' => $downloadTimeout,
);
$response = wp_remote_get( $url, $args_for_get );
if (wp_remote_retrieve_response_code($response) == 200 && isset($response['filename']))
{
$filepath = $response['filename'];
return $filepath; // body is the full image is all went well.
}
else {
Log::addError('Wp Remote Get failed', $response);
}
return false;
}
private function setPreferredProtocol($url, $reset = false) {
//switch protocol based on the formerly detected working protocol
$settings = \wpSPIO()->settings();
if($settings->downloadProto == '' || $reset) {
//make a test to see if the http is working
$testURL = 'http://' . SHORTPIXEL_API . '/img/connection-test-image.png';
$result = download_url($testURL, 10);
$settings->downloadProto = is_wp_error( $result ) ? 'https' : 'http';
}
return $settings->downloadProto == 'http' ?
str_replace('https://', 'http://', $url) :
str_replace('http://', 'https://', $url);
}
}