first

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

@@ -0,0 +1,183 @@
+<?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);
+		  }
+}