Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

namespace Smush\Core\Webp;

[2] Fix | Delete

[3] Fix | Delete

use Smush\Core\File_System;

[4] Fix | Delete

use Smush\Core\Media\Media_Item;

[5] Fix | Delete

use Smush\Core\Media\Media_Item_Optimization;

[6] Fix | Delete

use Smush\Core\Media\Media_Item_Stats;

[7] Fix | Delete

use Smush\Core\Settings;

[8] Fix | Delete

[9] Fix | Delete

/**

[10] Fix | Delete

* TODO: the response from the API has webp: false and mime_content_type of the written file is not webp, investigate

[11] Fix | Delete

[12] Fix | Delete

class Webp_Optimization extends Media_Item_Optimization {

[13] Fix | Delete

const OPTIMIZATION_KEY = 'webp_optimization';

[14] Fix | Delete

private $webp_dir;

[15] Fix | Delete

/**

[16] Fix | Delete

* @var Media_Item

[17] Fix | Delete

[18] Fix | Delete

private $media_item;

[19] Fix | Delete

/**

[20] Fix | Delete

* @var Webp_Helper

[21] Fix | Delete

[22] Fix | Delete

private $webp_helper;

[23] Fix | Delete

/**

[24] Fix | Delete

* @var Settings|null

[25] Fix | Delete

[26] Fix | Delete

private $settings;

[27] Fix | Delete

/**

[28] Fix | Delete

* @var Webp_Converter

[29] Fix | Delete

[30] Fix | Delete

private $converter;

[31] Fix | Delete

/**

[32] Fix | Delete

* @var File_System

[33] Fix | Delete

[34] Fix | Delete

private $fs;

[35] Fix | Delete

[36] Fix | Delete

public function __construct( $media_item ) {

[37] Fix | Delete

$this->webp_dir = new Webp_Dir();

[38] Fix | Delete

$this->webp_helper = new Webp_Helper();

[39] Fix | Delete

$this->media_item = $media_item;

[40] Fix | Delete

$this->settings = Settings::get_instance();

[41] Fix | Delete

$this->converter = new Webp_Converter();

[42] Fix | Delete

$this->fs = new File_System();

[43] Fix | Delete

}

[44] Fix | Delete

[45] Fix | Delete

public function get_key() {

[46] Fix | Delete

return self::OPTIMIZATION_KEY;

[47] Fix | Delete

}

[48] Fix | Delete

[49] Fix | Delete

public function is_optimized() {

[50] Fix | Delete

$attachment_id = $this->media_item->get_id();

[51] Fix | Delete

$webp_flag = $this->webp_helper->get_webp_flag( $attachment_id );

[52] Fix | Delete

if ( empty( $webp_flag ) ) {

[53] Fix | Delete

return false;

[54] Fix | Delete

}

[55] Fix | Delete

[56] Fix | Delete

$webp_file_path = trailingslashit( $this->webp_dir->get_webp_path() ) . ltrim( $webp_flag, '/' );

[57] Fix | Delete

[58] Fix | Delete

return $this->fs->file_exists( $webp_file_path );

[59] Fix | Delete

}

[60] Fix | Delete

[61] Fix | Delete

public function should_optimize() {

[62] Fix | Delete

if (

[63] Fix | Delete

$this->media_item->is_skipped()

[64] Fix | Delete

|| $this->media_item->has_errors()

[65] Fix | Delete

|| ! $this->settings->is_webp_module_active()

[66] Fix | Delete

) {

[67] Fix | Delete

return false;

[68] Fix | Delete

}

[69] Fix | Delete

[70] Fix | Delete

return in_array(

[71] Fix | Delete

$this->media_item->get_mime_type(),

[72] Fix | Delete

$this->webp_helper->supported_mime_types(),

[73] Fix | Delete

true

[74] Fix | Delete

);

[75] Fix | Delete

}

[76] Fix | Delete

[77] Fix | Delete

public function should_reoptimize() {

[78] Fix | Delete

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

[79] Fix | Delete

return false;

[80] Fix | Delete

}

[81] Fix | Delete

[82] Fix | Delete

$smushable_sizes = $this->media_item->get_smushable_sizes();

[83] Fix | Delete

foreach ( $smushable_sizes as $size ) {

[84] Fix | Delete

$webp_file_path = $this->webp_helper->get_webp_file_path( $size->get_file_path() );

[85] Fix | Delete

if ( ! $this->fs->file_exists( $webp_file_path ) ) {

[86] Fix | Delete

return true;

[87] Fix | Delete

}

[88] Fix | Delete

}

[89] Fix | Delete

[90] Fix | Delete

return false;

[91] Fix | Delete

}

[92] Fix | Delete

[93] Fix | Delete

public function save() {

[94] Fix | Delete

$webp_file_path = $this->webp_helper->get_webp_file_path( $this->media_item->get_main_size()->get_file_path() );

[95] Fix | Delete

if ( $this->fs->file_exists( $webp_file_path ) ) {

[96] Fix | Delete

$relative_path = substr( $webp_file_path, strlen( $this->webp_dir->get_webp_path() . '/' ) );

[97] Fix | Delete

$this->webp_helper->update_webp_flag( $this->media_item->get_id(), $relative_path );

[98] Fix | Delete

}

[99] Fix | Delete

}

[100] Fix | Delete

[101] Fix | Delete

public function get_stats() {

[102] Fix | Delete

// Empty stats for now since we don't store webp savings

[103] Fix | Delete

return new Media_Item_Stats();

[104] Fix | Delete

}

[105] Fix | Delete

[106] Fix | Delete

public function get_size_stats( $size_key ) {

[107] Fix | Delete

// Empty stats for now since we don't store webp savings

[108] Fix | Delete

return new Media_Item_Stats();

[109] Fix | Delete

}

[110] Fix | Delete

[111] Fix | Delete

public function should_optimize_size( $size ) {

[112] Fix | Delete

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

[113] Fix | Delete

return false;

[114] Fix | Delete

}

[115] Fix | Delete

[116] Fix | Delete

return array_key_exists(

[117] Fix | Delete

$size->get_key(),

[118] Fix | Delete

$this->media_item->get_smushable_sizes()

[119] Fix | Delete

);

[120] Fix | Delete

}

[121] Fix | Delete

[122] Fix | Delete

public function delete_data() {

[123] Fix | Delete

$this->webp_helper->unset_webp_flag( $this->media_item->get_id() );

[124] Fix | Delete

}

[125] Fix | Delete

[126] Fix | Delete

public function optimize() {

[127] Fix | Delete

$media_item = $this->media_item;

[128] Fix | Delete

$file_paths = array_map( function ( $size ) {

[129] Fix | Delete

return $size->get_file_path();

[130] Fix | Delete

}, $media_item->get_smushable_sizes() );

[131] Fix | Delete

$responses = $this->converter->smush( $file_paths, ! $media_item->is_large() );

[132] Fix | Delete

$success_responses = array_filter( $responses );

[133] Fix | Delete

if ( count( $success_responses ) !== count( $responses ) ) {

[134] Fix | Delete

return false;

[135] Fix | Delete

}

[136] Fix | Delete

$this->save();

[137] Fix | Delete

[138] Fix | Delete

return true;

[139] Fix | Delete

}

[140] Fix | Delete

[141] Fix | Delete

public function get_errors() {

[142] Fix | Delete

return $this->converter->get_errors();

[143] Fix | Delete

}

[144] Fix | Delete

[145] Fix | Delete

public function get_optimized_sizes_count() {

[146] Fix | Delete

// We don't keep per-size stats

[147] Fix | Delete

return 0;

[148] Fix | Delete

}

[149] Fix | Delete

}

[150] Fix | Delete

[151] Fix | Delete