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\Smush\Smush_Optimization;

[6] Fix | Delete

[7] Fix | Delete

class Webp_Helper {

[8] Fix | Delete

const WEBP_FLAG = 'webp_flag';

[9] Fix | Delete

/**

[10] Fix | Delete

* @var Webp_Dir

[11] Fix | Delete

[12] Fix | Delete

private $webp_dir;

[13] Fix | Delete

/**

[14] Fix | Delete

* @var File_System

[15] Fix | Delete

[16] Fix | Delete

private $fs;

[17] Fix | Delete

[18] Fix | Delete

public function __construct() {

[19] Fix | Delete

$this->webp_dir = new Webp_Dir();

[20] Fix | Delete

$this->fs = new File_System();

[21] Fix | Delete

}

[22] Fix | Delete

[23] Fix | Delete

/**

[24] Fix | Delete

* Converts the URL of a media item/media item size to a webp URL.

[25] Fix | Delete

[26] Fix | Delete

* @param $url string Original URL.

[27] Fix | Delete

[28] Fix | Delete

* @return false|string

[29] Fix | Delete

[30] Fix | Delete

public function get_webp_file_url( $url ) {

[31] Fix | Delete

$upload_dir_url = $this->webp_dir->get_upload_url();

[32] Fix | Delete

$upload_dir_path = $this->webp_dir->get_upload_path();

[33] Fix | Delete

$is_ssl = str_starts_with( $url, 'https:' );

[34] Fix | Delete

[35] Fix | Delete

// Temporarily add the same scheme to both URLs

[36] Fix | Delete

$url = set_url_scheme( $url, 'http' );

[37] Fix | Delete

$upload_dir_url = set_url_scheme( $upload_dir_url, 'http' );

[38] Fix | Delete

[39] Fix | Delete

$is_media_lib_file = strpos( $url, $upload_dir_url ) !== false;

[40] Fix | Delete

if ( ! $is_media_lib_file ) {

[41] Fix | Delete

return false;

[42] Fix | Delete

}

[43] Fix | Delete

[44] Fix | Delete

$file_path = str_replace( $upload_dir_url, $upload_dir_path, $url );

[45] Fix | Delete

if ( ! file_exists( $file_path ) ) {

[46] Fix | Delete

return false;

[47] Fix | Delete

}

[48] Fix | Delete

[49] Fix | Delete

$webp_path = $this->get_webp_file_path( $file_path );

[50] Fix | Delete

if ( ! file_exists( $webp_path ) ) {

[51] Fix | Delete

return false;

[52] Fix | Delete

}

[53] Fix | Delete

[54] Fix | Delete

$webp_url = str_replace(

[55] Fix | Delete

$this->webp_dir->get_webp_path(),

[56] Fix | Delete

$this->webp_dir->get_webp_url(),

[57] Fix | Delete

$webp_path

[58] Fix | Delete

);

[59] Fix | Delete

[60] Fix | Delete

return set_url_scheme( $webp_url, $is_ssl ? 'https' : 'http' );

[61] Fix | Delete

}

[62] Fix | Delete

[63] Fix | Delete

public function get_webp_file_path( $file_path, $make = false ) {

[64] Fix | Delete

$file_rel_path = substr( $file_path, strlen( $this->webp_dir->get_upload_path() ) );

[65] Fix | Delete

$webp_file_path = $this->webp_dir->get_webp_path() . $file_rel_path . '.webp';

[66] Fix | Delete

[67] Fix | Delete

if ( $make ) {

[68] Fix | Delete

$webp_file_dir = dirname( $webp_file_path );

[69] Fix | Delete

if ( ! $this->fs->is_dir( $webp_file_dir ) ) {

[70] Fix | Delete

wp_mkdir_p( $webp_file_dir );

[71] Fix | Delete

}

[72] Fix | Delete

}

[73] Fix | Delete

[74] Fix | Delete

return $webp_file_path;

[75] Fix | Delete

}

[76] Fix | Delete

[77] Fix | Delete

public function supported_mime_types() {

[78] Fix | Delete

return array(

[79] Fix | Delete

'image/jpg',

[80] Fix | Delete

'image/jpeg',

[81] Fix | Delete

'image/x-citrix-jpeg',

[82] Fix | Delete

'image/png',

[83] Fix | Delete

'image/x-png',

[84] Fix | Delete

);

[85] Fix | Delete

}

[86] Fix | Delete

[87] Fix | Delete

public function get_webp_flag( $attachment_id ) {

[88] Fix | Delete

$meta = $this->get_smush_meta( $attachment_id );

[89] Fix | Delete

[90] Fix | Delete

return empty( $meta[ self::WEBP_FLAG ] ) ? '' : $meta[ self::WEBP_FLAG ];

[91] Fix | Delete

}

[92] Fix | Delete

[93] Fix | Delete

public function update_webp_flag( $attachment_id, $value ) {

[94] Fix | Delete

$meta = $this->get_smush_meta( $attachment_id );

[95] Fix | Delete

if ( empty( $value ) ) {

[96] Fix | Delete

unset( $meta[ self::WEBP_FLAG ] );

[97] Fix | Delete

} else {

[98] Fix | Delete

$meta[ self::WEBP_FLAG ] = $value;

[99] Fix | Delete

}

[100] Fix | Delete

update_post_meta( $attachment_id, Smush_Optimization::SMUSH_META_KEY, $meta );

[101] Fix | Delete

}

[102] Fix | Delete

[103] Fix | Delete

public function unset_webp_flag( $attachment_id ) {

[104] Fix | Delete

$this->update_webp_flag( $attachment_id, false );

[105] Fix | Delete

}

[106] Fix | Delete

[107] Fix | Delete

/**

[108] Fix | Delete

* @return array|mixed

[109] Fix | Delete

[110] Fix | Delete

private function get_smush_meta( $attachment_id ) {

[111] Fix | Delete

$meta = get_post_meta( $attachment_id, Smush_Optimization::SMUSH_META_KEY, true );

[112] Fix | Delete

[113] Fix | Delete

return empty( $meta ) ? array() : $meta;

[114] Fix | Delete

}

[115] Fix | Delete

[116] Fix | Delete

/**

[117] Fix | Delete

* @param $media_item Media_Item

[118] Fix | Delete

[119] Fix | Delete

* @return void

[120] Fix | Delete

[121] Fix | Delete

public function delete_media_item_webp_versions( $media_item ) {

[122] Fix | Delete

foreach ( $media_item->get_sizes() as $size ) {

[123] Fix | Delete

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

[124] Fix | Delete

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

[125] Fix | Delete

$this->fs->unlink( $webp_file_path );

[126] Fix | Delete

}

[127] Fix | Delete

}

[128] Fix | Delete

}

[129] Fix | Delete

[130] Fix | Delete

public function delete_all_webp_files() {

[131] Fix | Delete

$webp_path = $this->webp_dir->get_webp_path();

[132] Fix | Delete

[133] Fix | Delete

// Delete the whole webp directory only when on single install or network admin.

[134] Fix | Delete

$this->fs->get_wp_filesystem()->delete( $webp_path, true );

[135] Fix | Delete

[136] Fix | Delete

do_action( 'wp_smush_after_delete_all_webp_files' );

[137] Fix | Delete

}

[138] Fix | Delete

}

[139] Fix | Delete

[140] Fix | Delete