Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

namespace Smush\Core\Transform;

[2] Fix | Delete

[3] Fix | Delete

use Smush\Core\Controller;

[4] Fix | Delete

use Smush\Core\Server_Utils;

[5] Fix | Delete

use Smush\Core\Url_Utils;

[6] Fix | Delete

[7] Fix | Delete

class Transformation_Controller extends Controller {

[8] Fix | Delete

[9] Fix | Delete

/**

[10] Fix | Delete

* @var Transformer

[11] Fix | Delete

[12] Fix | Delete

private $transformer;

[13] Fix | Delete

/**

[14] Fix | Delete

* @var Server_Utils

[15] Fix | Delete

[16] Fix | Delete

private $server_utils;

[17] Fix | Delete

[18] Fix | Delete

public function __construct() {

[19] Fix | Delete

$this->transformer = new Transformer();

[20] Fix | Delete

$this->server_utils = new Server_Utils();

[21] Fix | Delete

[22] Fix | Delete

$this->register_action( 'template_redirect', array( $this, 'hook_transformation_method' ), 1 );

[23] Fix | Delete

$this->register_filter( 'rest_pre_echo_response', array( $this, 'transform_rest_response' ), 10, 3 );

[24] Fix | Delete

$this->register_filter( 'wp_smush_transform_url', array( $this->transformer, 'transform_url' ) );

[25] Fix | Delete

}

[26] Fix | Delete

[27] Fix | Delete

public function should_run() {

[28] Fix | Delete

return ! is_admin() &&

[29] Fix | Delete

! wp_doing_ajax() &&

[30] Fix | Delete

! wp_doing_cron();

[31] Fix | Delete

}

[32] Fix | Delete

[33] Fix | Delete

public function hook_transformation_method() {

[34] Fix | Delete

if ( ! $this->should_transform_page() ) {

[35] Fix | Delete

return;

[36] Fix | Delete

}

[37] Fix | Delete

[38] Fix | Delete

ob_start( array( $this, 'transform_content' ) );

[39] Fix | Delete

}

[40] Fix | Delete

[41] Fix | Delete

private function should_transform_page() {

[42] Fix | Delete

$should_transform = ! is_customize_preview() &&

[43] Fix | Delete

$this->is_allowed_request_method() &&

[44] Fix | Delete

! $this->is_file_404();

[45] Fix | Delete

return apply_filters( 'wp_smush_should_transform_page', $should_transform );

[46] Fix | Delete

}

[47] Fix | Delete

[48] Fix | Delete

private function is_allowed_request_method() {

[49] Fix | Delete

$allowed = array( 'GET', 'HEAD' );

[50] Fix | Delete

[51] Fix | Delete

return in_array( $this->server_utils->get_request_method(), $allowed, true );

[52] Fix | Delete

}

[53] Fix | Delete

[54] Fix | Delete

private function is_file_404() {

[55] Fix | Delete

if ( ! is_404() ) {

[56] Fix | Delete

return false;

[57] Fix | Delete

}

[58] Fix | Delete

[59] Fix | Delete

$request_uri = $this->server_utils->get_request_uri();

[60] Fix | Delete

$extension = ( new Url_Utils() )->get_extension( $request_uri );

[61] Fix | Delete

[62] Fix | Delete

if ( empty( $extension ) ) {

[63] Fix | Delete

return false;

[64] Fix | Delete

}

[65] Fix | Delete

[66] Fix | Delete

$allowed_404_extensions = array(

[67] Fix | Delete

'html',

[68] Fix | Delete

'htm',

[69] Fix | Delete

);

[70] Fix | Delete

[71] Fix | Delete

return ! isset( $allowed_404_extensions[ $extension ] );

[72] Fix | Delete

}

[73] Fix | Delete

[74] Fix | Delete

public function transform_content( $content ) {

[75] Fix | Delete

if ( ! $this->should_parse_content( $content ) ) {

[76] Fix | Delete

return $content;

[77] Fix | Delete

}

[78] Fix | Delete

[79] Fix | Delete

return $this->transformer->transform_content(

[80] Fix | Delete

$content,

[81] Fix | Delete

$this->server_utils->get_current_url()

[82] Fix | Delete

);

[83] Fix | Delete

}

[84] Fix | Delete

[85] Fix | Delete

private function should_parse_content( $content ) {

[86] Fix | Delete

$should_parse = $this->is_html( $content );

[87] Fix | Delete

return apply_filters( 'wp_smush_should_parse_content', $should_parse, $content );

[88] Fix | Delete

}

[89] Fix | Delete

[90] Fix | Delete

private function is_html( $content ) {

[91] Fix | Delete

return (bool) preg_match( '/<\s*\/\s*html\s*>/i', $content );

[92] Fix | Delete

}

[93] Fix | Delete

[94] Fix | Delete

/**

[95] Fix | Delete

* @param $response array

[96] Fix | Delete

* @param $server \WP_REST_Server

[97] Fix | Delete

* @param $request \WP_REST_Request

[98] Fix | Delete

[99] Fix | Delete

* @return array

[100] Fix | Delete

[101] Fix | Delete

public function transform_rest_response( $response, $server, $request ) {

[102] Fix | Delete

if ( ! $this->should_transform_rest( $response, $request ) ) {

[103] Fix | Delete

return $response;

[104] Fix | Delete

}

[105] Fix | Delete

[106] Fix | Delete

return $this->transformer->transform_rest_response( $response, $request );

[107] Fix | Delete

}

[108] Fix | Delete

[109] Fix | Delete

private function should_transform_rest( $response, $request ) {

[110] Fix | Delete

$context = $request->get_param( 'context' );

[111] Fix | Delete

$referer = $request->get_header( 'referer' );

[112] Fix | Delete

$referer_from_admin = $referer && ( false !== strpos( $referer, admin_url() ) || false !== strpos( $referer, network_admin_url() ) );

[113] Fix | Delete

$should_transform_rest = 'view' === $context && ! $referer_from_admin;

[114] Fix | Delete

[115] Fix | Delete

return apply_filters( 'wp_smush_should_transform_rest', $should_transform_rest, $response, $request );

[116] Fix | Delete

}

[117] Fix | Delete

}

[118] Fix | Delete

[119] Fix | Delete