JezK

Edit File: manager.php

<?php
namespace Elementor\Data;

use Elementor\Core\Base\Module as BaseModule;
use Elementor\Data\Base\Processor;

if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly.
}

class Manager extends BaseModule {

const ROOT_NAMESPACE = 'elementor';

const REST_BASE = '';

const VERSION = '1';

/**
	 * @var \WP_REST_Server
	 */
	private $server;

/**
	 * @var boolean
	 */
	private $is_internal = false;

/**
	 * @var array
	 */
	private $cache = [];

/**
	 * Loaded controllers.
	 *
	 * @var \Elementor\Data\Base\Controller[]
	 */
	public $controllers = [];

/**
	 * Loaded command(s) format.
	 *
	 * @var string[]
	 */
	public $command_formats = [];

/**
	 * Fix issue with 'Potentially polymorphic call. The code may be inoperable depending on the actual class instance passed as the argument.'.
	 *
	 * @return \Elementor\Core\Base\Module|\Elementor\Data\Manager
	 */
	public static function instance() {
		return ( parent::instance() );
	}

public function __construct() {
		add_action( 'rest_api_init', [ $this, 'register_rest_error_handler' ] );
	}

public function get_name() {
		return 'data-manager';
	}

/**
	 * @return \Elementor\Data\Base\Controller[]
	 */
	public function get_controllers() {
		return $this->controllers;
	}

private function get_cache( $key ) {
		return self::get_items( $this->cache, $key );
	}

private function set_cache( $key, $value ) {
		$this->cache[ $key ] = $value;
	}

/**
	 * Register controller.
	 *
	 * @param string $controller_class_name
	 *
	 * @return \Elementor\Data\Base\Controller
	 */
	public function register_controller( $controller_class_name ) {
		$controller_instance = new $controller_class_name();

return $this->register_controller_instance( $controller_instance );
	}

/**
	 * Register controller instance.
	 *
	 * @param \Elementor\Data\Base\Controller $controller_instance
	 *
	 * @return \Elementor\Data\Base\Controller
	 */
	public function register_controller_instance( $controller_instance ) {
		// TODO: Validate instance.

$this->controllers[ $controller_instance->get_name() ] = $controller_instance;

return $controller_instance;
	}

/**
	 * Register endpoint format.
	 *
	 * @param string $command
	 * @param string $format
	 */
	public function register_endpoint_format( $command, $format ) {
		$this->command_formats[ $command ] = rtrim( $format, '/' );
	}

public function register_rest_error_handler() {
		if ( ! $this->is_internal() ) {
			$logger_manager = \Elementor\Core\Logger\Manager::instance();
			$logger_manager->register_error_handler();
		}
	}

/**
	 * Find controller instance.
	 *
	 * By given command name.
	 *
	 * @param string $command
	 *
	 * @return false|\Elementor\Data\Base\Controller
	 */
	public function find_controller_instance( $command ) {
		$command_parts = explode( '/', $command );
		$assumed_command_parts = [];

foreach ( $command_parts as $command_part ) {
			$assumed_command_parts [] = $command_part;

foreach ( $this->controllers as $controller_name => $controller ) {
				$assumed_command = implode( '/', $assumed_command_parts );

if ( $assumed_command === $controller_name ) {
					return $controller;
				}
			}
		}

return false;
	}

/**
	 * Command extract args.
	 *
	 * @param string $command
	 * @param array  $args
	 *
	 * @return \stdClass
	 */
	public function command_extract_args( $command, $args = [] ) {
		$result = new \stdClass();
		$result->command = $command;
		$result->args = $args;

if ( false !== strpos( $command, '?' ) ) {
			$command_parts = explode( '?', $command );
			$pure_command = $command_parts[0];
			$query_string = $command_parts[1];

parse_str( $query_string, $temp );

$result->command = rtrim( $pure_command, '/' );
			$result->args = array_merge( $args, $temp );
		}

return $result;
	}

/**
	 * Command to endpoint.
	 *
	 * Format is required otherwise $command will returned.
	 *
	 * @param string $command
	 * @param string $format
	 * @param array  $args
	 *
	 * @return string endpoint
	 */
	public function command_to_endpoint( $command, $format, $args ) {
		$endpoint = $command;

if ( $format ) {
			$formatted = $format;

array_walk( $args, function ( $val, $key ) use ( &$formatted ) {
				$formatted = str_replace( '{' . $key . '}', $val, $formatted );
			} );

// Remove remaining format if not requested via `$args`.
			if ( strstr( $formatted, '/{' ) ) {
				/**
				 * Example:
				 * $command = 'example/documents';
				 * $format = 'example/documents/{document_id}/elements/{element_id}';
				 * $formatted = 'example/documents/1618/elements/{element_id}';
				 * Result:
				 * $formatted = 'example/documents/1618/elements';
				 */
				$formatted = substr( $formatted, 0, strpos( $formatted, '/{' ) );
			}

$endpoint = $formatted;
		}

return $endpoint;
	}

/**
	 * Run server.
	 *
	 * Init WordPress reset api.
	 *
	 * @return \WP_REST_Server
	 */
	public function run_server() {
		/**
		 * If run_server() called means, that rest api is simulated from the backend.
		 */
		$this->is_internal = true;

if ( ! $this->server ) {
			// Remove all 'rest_api_init' actions.
			remove_all_actions( 'rest_api_init' );

// Call custom reset api loader.
			do_action( 'elementor_rest_api_before_init' );

$this->server = rest_get_server(); // Init API.
		}

return $this->server;
	}

/**
	 * Kill server.
	 *
	 * Free server and controllers.
	 */
	public function kill_server() {
		global $wp_rest_server;

$this->controllers = [];
		$this->command_formats = [];
		$this->server = false;
		$this->is_internal = false;
		$this->cache = [];
		$wp_rest_server = false;
	}

/**
	 * Run processor.
	 *
	 * @param Processor $processor
	 * @param array     $data
	 *
	 * @return mixed
	 */
	public function run_processor( $processor, $data ) {
		if ( call_user_func_array( [ $processor, 'get_conditions' ], $data ) ) {
			return call_user_func_array( [ $processor, 'apply' ], $data );
		}

return null;
	}

/**
	 * Run processors.
	 *
	 * Filter them by class.
	 *
	 * @param Processor[] $processors
	 * @param string      $filter_by_class
	 * @param array       $data
	 *
	 * @return false|array
	 */
	public function run_processors( $processors, $filter_by_class, $data ) {
		foreach ( $processors as $processor ) {
			if ( $processor instanceof $filter_by_class ) {
				if ( Processor\Before::class === $filter_by_class ) {
					$this->run_processor( $processor, $data );
				} elseif ( Processor\After::class === $filter_by_class ) {
					$result = $this->run_processor( $processor, $data );
					if ( $result ) {
						$data[1] = $result;
					}
				} else {
					// TODO: error
					break;
				}
			}
		}

return isset( $data[1] ) ? $data[1] : false;
	}

/**
	 * Run request.
	 *
	 * Simulate rest API from within the backend.
	 * Use args as query.
	 *
	 * @param string $endpoint
	 * @param array  $args
	 * @param string $method
	 *
	 * @return \WP_REST_Response
	 */
	private function run_request( $endpoint, $args, $method ) {
		$this->run_server();

$endpoint = '/' . self::ROOT_NAMESPACE . '/v' . self::VERSION . '/' . $endpoint;

// Run reset api.
		$request = new \WP_REST_Request( $method, $endpoint );

if ( 'GET' === $method ) {
			$request->set_query_params( $args );
		} else {
			$request->set_body_params( $args );
		}

return rest_do_request( $request );
	}

/**
	 * Run endpoint.
	 *
	 * Wrapper for `$this->run_request` return `$response->getData()` instead of `$response`.
	 *
	 * @param string $endpoint
	 * @param array  $args
	 * @param string $method
	 *
	 * @return array
	 */
	public function run_endpoint( $endpoint, $args = [], $method = 'GET' ) {
		$response = $this->run_request( $endpoint, $args, $method );

return $response->get_data();
	}

/**
	 * Run ( simulated reset api ).
	 *
	 * Do:
	 * Init reset server.
	 * Run before processors.
	 * Run command as reset api endpoint from internal.
	 * Run after processors.
	 *
	 * @param string $command
	 * @param array  $args
	 * @param string $method
	 *
	 * @return array|false processed result
	 */
	public function run( $command, $args = [], $method = 'GET' ) {
		$key = crc32( $command . '-' . wp_json_encode( $args ) . '-' . $method );
		$cache = $this->get_cache( $key );

if ( $cache ) {
			return $cache;
		}

$this->run_server();

$controller_instance = $this->find_controller_instance( $command );

if ( ! $controller_instance ) {
			$this->set_cache( $key, [] );
			return [];
		}

$extracted_command = $this->command_extract_args( $command, $args );
		$command = $extracted_command->command;
		$args = $extracted_command->args;

$format = isset( $this->command_formats[ $command ] ) ? $this->command_formats[ $command ] : false;

$command_processors = $controller_instance->get_processors( $command );

$endpoint = $this->command_to_endpoint( $command, $format, $args );

$this->run_processors( $command_processors, Processor\Before::class, [ $args ] );

$response = $this->run_request( $endpoint, $args, $method );
		$result = $response->get_data();

if ( $response->is_error() ) {
			$this->set_cache( $key, [] );
			return [];
		}

$result = $this->run_processors( $command_processors, Processor\After::class, [ $args, $result ] );

$this->set_cache( $key, $result );

return $result;
	}

public function is_internal() {
		return $this->is_internal;
	}
}