OwlCyberSecurity - MANAGER

Edit File: class-pixelentity-theme-update.php

<?php
/**
 * This class is based on the Envato API < 3.0 which will be outdated in near future.
 * Will be replaced by Avia_Theme_Updater class that supports Envato API 3.0
 * 
 */
if ( ! defined( 'ABSPATH' ) ) {  exit;  }    // Exit if accessed directly

if (!class_exists("PixelentityThemeUpdate")) {
	class PixelentityThemeUpdate {
			
		protected $username;
		protected $apiKey;
		
		/**
		 * function wp_update_themes calls filter pre_set_site_transient_update_themes twice during theme version check.
		 * As we have a rate limiting by Envato we cache the result here
		 * 
		 * @since 4.4.3
		 * @var array 
		 */
		protected $envato_results_cache;

public function __construct($username,$apiKey,$authors) {
			// to debug
			// set_site_transient('update_themes',null);

$this->username = $username;
			$this->apiKey = $apiKey;
			$this->authors = $authors;
			
			$this->envato_results_cache = null;

add_filter("pre_set_site_transient_update_themes", array(&$this,"check"));
		}
		
		public function __destruct() {
			unset( $this->envato_results_cache );
		}

public function check($updates) {

$this->username = apply_filters("pixelentity_theme_update_username",$this->username);
			$this->apiKey = apply_filters("pixelentity_theme_update_apiKey",$this->apiKey);
			$this->authors = apply_filters("pixelentity_theme_update_authors",$this->authors);

if ($this->authors && !is_array($this->authors)) {
				$this->authors = array($this->authors);
			}

if (!$this->username || !$this->apiKey || !isset($updates->checked)) return $updates;
			
			/**
			 * If we have already cached the result from Envato we can take this and avoid multiple requests for the same data
			 */
			if( is_array( $this->envato_results_cache ) )
			{
				foreach( $this->envato_results_cache as $theme => $update ) 
				{
					$updates->response[ $theme ] = $update;
				}
				
				return $updates;
			}

if (!class_exists("Envato_Protected_API")) {
				require_once("class-envato-protected-api.php");
			}

$api = new Envato_Protected_API($this->username,$this->apiKey);
			
			add_filter("http_request_args",array(&$this,"http_timeout"),10,1);
			$purchased = $api->wp_list_themes(true);
			
			/**
			 * fixes a bug where $api->wp_list_themes returns several times the same theme/item data and breaks the envato update api 
			 * because the theme updater tries to download the same theme several times and triggers a 429 error 
			 * As Envato now also limits requests we only filter for current theme as a temp fix until we upgrade to nes API.
			 *
			 * (see https://build.envato.com/api/#rate-limit)
			 * 
			 * @since 4.4.2
			 */
			$filtered_purchased = array();
			$purchased_ids = array();
			$current_theme_name = strtolower( avia_auto_updates::get_themename() );

foreach ( $purchased as $current ) 
			{
				if ( ! in_array( $current->item_id, $purchased_ids ) ) 
				{
					if( ! isset( $current->theme_name ) || ( strtolower( $current->theme_name ) != $current_theme_name ) )
					{
						continue;
					}
					
					$filtered_purchased[] = $current;
					$purchased_ids[] = $current->item_id;
				}
			}

$purchased = $filtered_purchased;
			
			$installed = function_exists("wp_get_themes") ? wp_get_themes() : get_themes();
			$filtered = array();
			
			foreach ($installed as $theme) {
				if ($this->authors && !in_array($theme->{'Author Name'},$this->authors)) continue;
				$filtered[$theme->Name] = $theme;
			}

$this->envato_results_cache = array();
			
			foreach ($purchased as $theme) {
			
				if ( isset($theme->theme_name) && isset($filtered[$theme->theme_name])) {
					// gotcha, compare version now
					$current = $filtered[$theme->theme_name];
					if (version_compare($current->Version, $theme->version, '<')) {
						// bingo, inject the update
						if ($url = $api->wp_download($theme->item_id)) {
							if( is_string( $url ) )
							{
								$update = array(
											"url" => "http://themeforest.net/item/theme/{$theme->item_id}",
											"new_version" => $theme->version,
											"package" => $url
											);

$updates->response[$current->Stylesheet] = $update;
								$this->envato_results_cache[ $current->Stylesheet ] = $update;
							}
						}
					}
				}
			}

remove_filter("http_request_args",array(&$this,"http_timeout"));
			
			return $updates;
		}

public function http_timeout($req) {
			// increase timeout for api request
			$req["timeout"] = 300;
			return $req;
		}

public static function init($username = null,$apiKey = null,$authors = null) {
			new PixelentityThemeUpdate($username,$apiKey,$authors);
		}

}
}