184 lines
		
	
	
		
			5.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			184 lines
		
	
	
		
			5.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?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);
 | 
						|
		  }
 | 
						|
}
 |