Use WP_HTML_Tag_Processor in auto_sizes_update_content_img_tag

mukeshpanchal27 · mukeshpanchal27 · commit 32b9abc15b84 · 2024-08-14T16:22:03.000+05:30
diff --git a/plugins/auto-sizes/hooks.php b/plugins/auto-sizes/hooks.php
@@ -57,24 +57,32 @@ function auto_sizes_update_content_img_tag( $html ): string {
 		$html = '';
 	}
 
+	$processor = new WP_HTML_Tag_Processor( $html );
+
+	// Bail if there is no IMG tag.
+	if ( ! $processor->next_tag( array( 'tag_name' => 'IMG' ) ) ) {
+		return $html;
+	}
+
 	// Bail early if the image is not lazy-loaded.
-	if ( false === strpos( $html, 'loading="lazy"' ) ) {
+	if ( 'lazy' !== $processor->get_attribute( 'loading' ) ) {
 		return $html;
 	}
 
+	$sizes = $processor->get_attribute( 'sizes' );
+
 	// Bail early if the image is not responsive.
-	if ( 1 !== preg_match( '/sizes="([^"]+)"/', $html, $match ) ) {
+	if ( ! is_string( $sizes ) ) {
 		return $html;
 	}
 
 	// Don't add 'auto' to the sizes attribute if it already exists.
-	if ( auto_sizes_attribute_includes_valid_auto( $match[1] ) ) {
+	if ( auto_sizes_attribute_includes_valid_auto( $sizes ) ) {
 		return $html;
 	}
 
-	$html = str_replace( 'sizes="', 'sizes="auto, ', $html );
-
-	return $html;
+	$processor->set_attribute( 'sizes', "auto, $sizes" );
+	return $processor->get_updated_html();
 }
 add_filter( 'wp_content_img_tag', 'auto_sizes_update_content_img_tag' );
 
