<?php

namespace Smush\Core\Parser;

use Smush\Core\Array_Utils;

class Parser {
	/**
	 * @var Array_Utils
	 */
	private $array_utils;

	public function __construct() {
		$this->array_utils = new Array_Utils();
	}

	public function get_base_url( $markup ) {
		$pattern = "/<base[^>]*?href\s*=\s*['\"](.*?)['\"]\s*\/\s*>/is";
		if ( preg_match( $pattern, $markup, $matches ) && ! empty( $matches[1] ) ) {
			return $matches[1];
		}

		return '';
	}

	public function get_inline_style_blocks( $markup ) {
		$pattern = '/<style\b[^>]*>(.*?)<\/style>/msi';
		if ( ! preg_match_all( $pattern, $markup, $matches, PREG_OFFSET_CAPTURE ) ) {
			return array();
		}
		return $matches[1];
	}

	/**
	 * @param $markup
	 * @param $base_url
	 *
	 * @return Style[]
	 */
	public function get_inline_styles( $markup, $base_url ) {
		$styles              = array();
		$inline_style_blocks = $this->get_inline_style_blocks( $markup );
		foreach ( $inline_style_blocks as $style_block ) {
			if ( empty( $style_block ) || ! is_array( $style_block ) || count( $style_block ) < 2 ) {
				continue;
			}

			list( $inline_style_block, $inline_style_block_position ) = $style_block;

			$image_urls = $this->get_image_urls( $inline_style_block, $base_url );
			$style      = new Style( $inline_style_block, $image_urls, $inline_style_block_position );
			$styles[]   = $style;
		}
		return $styles;
	}

	/**
	 * @param $markup string
	 * @param $base_url
	 *
	 * @return Image_URL[]
	 */
	public function get_image_urls( $markup, $base_url ) {
		// IMPORTANT: the following regex is a copy of the one in the JS function SmushLCPDetector.getBackgroundDataForPropertyValue(). Remember to keep them synced.
		$pattern = '@(?<src>(?:https?:/|\.+)?/[^\'",\s\(\)]+\.(?<ext>jpe?g|png|gif|webp|svg|avif)(?:\?[^\s\'",?)]+)?)\b@is';
		$pattern = apply_filters( 'wp_smush_image_urls_regex', $pattern );
		if ( ! preg_match_all( $pattern, $markup, $matches, PREG_SET_ORDER ) ) {
			return array();
		}

		$image_urls = array();
		foreach ( $matches as $match ) {
			if ( ! isset( $match['src'], $match['ext'] ) ) {
				continue;
			}
			$src                = $match['src'];
			$image_urls[ $src ] = new Image_URL(
				$this->remove_quote_entities( $src ),
				$match['ext'],
				$base_url
			);
		}

		return array_values( $image_urls );
	}

	public function get_tags( $markup, $tags ) {
		$tags_string = join( '|', $tags );

		$matches = array();
		if ( preg_match_all( '/<(' . $tags_string . ').*?\/\1>/s', $markup, $matches, PREG_PATTERN_ORDER ) ) {
			return $matches[0];
		}

		return array();
	}

	public function get_block_by_tag( $markup, $tag ) {
		$pattern = "/<$tag\b([^>]*>.*)<\/$tag>/is";
		if ( ! preg_match( $pattern, $markup, $matches ) ) {
			return $markup;
		}

		return $matches[1];
	}

	/**
	 * @param $markup
	 * @param $base_url
	 * @param $tag_names
	 * @param $lcp_location
	 *
	 * @return Composite_Element[]
	 */
	public function get_composite_elements( $markup, $base_url, $tag_names, $lcp_location = - 1 ) {
		$composite_elements = array();
		foreach ( $tag_names as $tag_name ) {
			$match_found = preg_match_all( '/<(' . $tag_name . ').*?\/\1>/s', $markup, $matches, PREG_PATTERN_ORDER | PREG_OFFSET_CAPTURE );
			if ( ! $match_found ) {
				continue;
			}

			$html_elements = empty( $matches[0] ) ? array() : $matches[0];
			foreach ( $html_elements as $html_element ) {
				$composite_element_markup   = $html_element[0];
				$composite_element_position = $html_element[1];
				$elements                   = $this->get_elements_with_image_attributes( $composite_element_markup, $base_url );

				if ( ! empty( $elements ) ) {
					$has_lcp_element = false;
					foreach ( $elements as $element ) {
						$actual_position = $composite_element_position + $element->get_position();
						$element->set_position( $actual_position );
						$element->set_lcp( $lcp_location === $actual_position );

						if ( $element->is_lcp() ) {
							$has_lcp_element = true;
						}
					}

					$composite_elements[] = new Composite_Element( $composite_element_markup, $tag_name, $elements, $composite_element_position, $has_lcp_element );
				}
			}
		}
		return $composite_elements;
	}

	/**
	 * @param $markup
	 * @param $base_url
	 * @param int $lcp_position
	 *
	 * @return Element[]
	 */
	public function get_elements_with_image_attributes( $markup, $base_url, $lcp_position = - 1 ) {
		$pattern = '@(?<element><(?:(?<img>img)\b[^>]+|(?<tag>[a-zA-Z]+)\b[^>]+\.(?:jpe?g|png|gif|webp|svg|avif)[^>]+)>)@is';
		$pattern = apply_filters( 'wp_smush_images_from_content_regex', $pattern );

		return $this->get_elements_matching_pattern( $pattern, $markup, $base_url, $lcp_position );
	}

	public function get_elements_with_id_attribute( $markup, $id, $base_url ) {
		$pattern = '@(?<element><(?<tag>[a-zA-Z]+)\b[^>]+(?<!\S)id=(["\'])' . preg_quote( $id ) . '\3[^>]+>)@is';

		return $this->get_elements_matching_pattern( $pattern, $markup, $base_url );
	}

	public function get_elements_with_class_attribute( $markup, $class, $base_url ) {
		$pattern = '@(?<element><(?<tag>[a-zA-Z]+)\b[^>]+(?<!\S)class=(["\'])' . preg_quote( $class ) . '\3[^>]+>)@is';

		return $this->get_elements_matching_pattern( $pattern, $markup, $base_url );
	}

	/**
	 * @param $markup
	 * @param $image_url
	 * @param $base_url
	 *
	 * @return Element[]
	 */
	public function get_elements_with_image_url( $markup, $image_url, $base_url ) {
		$pattern = '@(?<element><(?<tag>[a-zA-Z]+)\b[^>]+' . preg_quote( $image_url ) . '[^>]+>)@is';

		return $this->get_elements_matching_pattern( $pattern, $markup, $base_url );
	}

	private function get_elements_matching_pattern( $pattern, $markup, $base_url, $lcp_position = - 1 ) {
		if ( ! preg_match_all( $pattern, $markup, $matches, PREG_SET_ORDER | PREG_OFFSET_CAPTURE ) ) {
			return array();
		}

		$elements = array();
		foreach ( $matches as $item ) {
			$element          = (string) $this->array_utils->get_array_value( $item, [ 'element', 0 ] );
			$element_position = (int) $this->array_utils->get_array_value( $item, [ 'element', 1 ] );
			if ( empty( $element ) ) {
				continue;
			}

			$img_tag_name   = (string) $this->array_utils->get_array_value( $item, [ 'img', 0 ] );
			$tag_name       = (string) $this->array_utils->get_array_value( $item, [ 'tag', 0 ] );
			$tag_name       = ! empty( $img_tag_name ) ? $img_tag_name : $tag_name;
			$attributes     = $this->get_element_attributes( $element, $base_url );
			$background     = $this->get_element_background_image( $element, $base_url );
			$css_properties = $background ? array( $background ) : array();

			// TODO: Support CSS variable with images.
			if ( empty( $attributes ) && empty( $css_properties ) ) {
				continue;
			}

			$is_lcp_element = $lcp_position === $element_position;
			$elements[]     = new Element( $element, $tag_name, $attributes, $css_properties, $element_position, $is_lcp_element );
		}

		return array_values( $elements );
	}

	/**
	 * @param $element
	 * @param $base_url
	 *
	 * @return Element_Attribute[]
	 */
	public function get_element_attributes( $element, $base_url ) {
		$image_attributes = array();

		$pattern = '#\b(?<name>(?:data-(?:[a-z0-9_-]+-)?)?(?:[a-z0-9_-]+))\s*=\s*(["\'])(?<value>[^\'"]+)\2#is';
		$pattern = apply_filters( 'wp_smush_image_attributes_regex', $pattern );

		if ( ! preg_match_all( $pattern, $element, $matches, PREG_SET_ORDER ) ) {
			return $image_attributes;
		}

		foreach ( $matches as $attr_data ) {
			if ( ! isset( $attr_data['name'], $attr_data['value'] ) ) {
				continue;
			}
			$attr_name  = $attr_data['name'];
			$attr_value = trim( $attr_data['value'] );
			if ( ! $this->is_safe( $attr_name ) || ! $this->is_safe( $attr_value ) ) {
				continue;
			}
			$image_urls = $this->get_image_urls( $attr_value, $base_url );

			$image_attributes[ $attr_name ] = new Element_Attribute(
				$attr_name,
				$attr_value,
				$attr_data[0],
				$image_urls
			);
		}

		return $image_attributes;
	}

	public function get_element_background_image( $element, $base_url ) {
		if ( ! strpos( $element, 'background' ) ) {
			return null;
		}

		$style = $this->get_element_attribute_value( $element, 'style' );
		if ( empty( $style ) ) {
			return null;
		}

		/**
		 * Background image regex supports:
		 *
		 * 1. background or background-image.
		 * 2. Multiple background images.
		 *
		 * background: url(img_flwr.gif) right bottom no-repeat, url(paper.gif) left top repeat;
		 * background-image: url(&quot;image1.png&quot;), url(https://sample.com/image2.png?lossy=2&amp;strip=1&amp;webp=1), linear-gradient(to right, rgba(30, 75, 115, 1), rgba(255, 255, 255, 0));
		 */

		// Regex rule to get all inline style after background(-image) property.
		$pattern = '#(?<!-)\b(?<property>background(?:-image)?):(?<value>[^;:]*?url\s*\([^>]+\)[^=:;]*);{0,1}#is';
		$pattern = apply_filters( 'wp_smush_background_images_regex', $pattern );

		if ( ! preg_match_all( $pattern, $style, $matches, PREG_SET_ORDER ) ) {
			return null;
		}

		$bg_property = null;
		foreach ( $matches as $bg_image_data ) {
			if ( ! isset( $bg_image_data['property'], $bg_image_data['value'] ) ) {
				continue;
			}
			$bg_image_property = $bg_image_data['property'];
			$bg_image_value    = $bg_image_data['value'];
			if ( ! $this->is_safe( $bg_image_property ) || ! $this->is_safe( $bg_image_value ) ) {
				continue;
			}
			$image_urls = $this->get_image_urls( $bg_image_value, $base_url );
			if ( empty( $image_urls ) ) {
				continue;
			}

			// background-image:url(&quot;image1.png&quot;); background-repeat: no-repeat.
			if ( substr_count( $bg_image_value, ';' ) ) {
				$bg_image_value = $this->extract_background_image( $bg_image_value );
			}

			// An element only has one background, so let try to get the latest one.
			$bg_property = new Element_CSS_Property(
				$bg_image_data[0],
				$bg_image_property,
				$bg_image_value,
				$image_urls
			);
		}

		return $bg_property;
	}

	/**
	 * Extract background image value from the inline style.
	 *
	 * Input: linear-gradient(to right, rgba(30, 75, 115, 1), rgba(255, 255, 255, 0)), url(&quot;image1.png&quot;) right bottom no-repeat,
	 *                          url(https://sample.com/image2.png?lossy=2&amp;strip=1&amp;webp=1) left top repeat; background-color: #fff;
	 * Output: linear-gradient(to right, rgba(30, 75, 115, 1), rgba(255, 255, 255, 0)), url(&quot;image1.png&quot;) right bottom no-repeat,
	 *                          url(https://sample.com/image2.png?lossy=2&amp;strip=1&amp;webp=1) left top repeat;
	 */
	public function extract_background_image( $bg_image_value ) {
		$urls                       = array();
		$bg_image_value_without_url = preg_replace_callback(
			'#url\s*\(([^\)]+)\)#is',
			function ( $matches ) use ( &$urls ) {
				if ( ! empty( $matches[1] ) ) {
					$url        = $matches[1];
					$urls[]     = $url;
					$count      = count( $urls );
					$matches[0] = str_replace( $url, "[URL{$count}]", $matches[0] );
				}
				return $matches[0];
			},
			$bg_image_value
		);

		if ( ! empty( $urls ) && preg_match( '/[^;]+/is', $bg_image_value_without_url, $matches ) ) {
			$bg_image_value = $matches[0];
			foreach ( $urls as $index => $url ) {
				$index ++;
				$bg_image_value = str_replace( "[URL{$index}]", $url, $bg_image_value );
			}
		}

		return $bg_image_value;
	}

	private function remove_quote_entities( $image ) {
		// Quote entities.
		$quotes = apply_filters( 'wp_smush_background_image_quotes', array( '&quot;', '&#034;', '&#039;', '&apos;' ) );

		$image = trim( $image );
		if ( empty( $image ) || strlen( $image ) < 6 ) {
			return $image;
		}

		// Remove the starting quotes.
		if ( in_array( substr( $image, 0, 6 ), $quotes, true ) ) {
			$image = substr( $image, 6 );
		}

		// Remove the ending quotes.
		if ( in_array( substr( $image, - 6 ), $quotes, true ) ) {
			$image = substr( $image, 0, - 6 );
		}

		return $image;
	}

	public function add_attribute_to_element( $element_markup, $tag_name, $attribute_name, $attribute_value = null ) {
		$pattern = '~<' . $tag_name . '\b[^>]*>~';
		if ( preg_match_all( $pattern, $element_markup, $matches, PREG_PATTERN_ORDER | PREG_OFFSET_CAPTURE ) ) {
			$starting_tag          = $this->array_utils->get_array_value( $matches, [ 0, 0, 0 ] );
			$starting_tag_position = $this->array_utils->get_array_value( $matches, [ 0, 0, 1 ] );
			$updated_starting_tag  = $this->add_attribute_to_self_closing_element( $starting_tag, $attribute_name, $attribute_value );
			if ( $starting_tag !== $updated_starting_tag ) {
				$before         = substr( $element_markup, 0, $starting_tag_position );
				$after          = substr( $element_markup, $starting_tag_position + strlen( $starting_tag ) );
				$element_markup = $before . $updated_starting_tag . $after;
			}
		}

		return $element_markup;
	}

	public function add_element_attribute( $element, $name, $value = null ) {
		_deprecated_function( __METHOD__, '3.19.1', '\Smush\Core\Parser\Parser::add_attribute_to_element()' );

		return $this->add_attribute_to_self_closing_element( $element, $name, $value );
	}

	private function add_attribute_to_self_closing_element( $element, $name, $value = null ) {
		$closing = false === strpos( $element, '/>' ) ? '>' : ' />';
		$quotes  = false === strpos( $element, '"' ) ? '\'' : '"';

		if ( ! is_null( $value ) ) {
			$element = rtrim( $element, $closing ) . " {$name}={$quotes}{$value}{$quotes}{$closing}";
		} else {
			$element = rtrim( $element, $closing ) . " {$name}{$closing}";
		}

		return $element;
	}

	public function remove_element_attribute( $element, $attribute ) {
		return preg_replace( '/\s' . $attribute . '=[\'"](.*?)[\'"]/i', '', $element );
	}

	/**
	 * @param $markup
	 * @param $base_url
	 *
	 * @return Element[]
	 */
	public function get_iframe_elements( $markup, $base_url ) {
		if ( strpos( $markup, '<iframe' ) === false ) {
			return array();
		}

		// Iframe tag has srcdocs attribute which might contains HTML code.
		$pattern = '#<iframe\b[^>]*\s(?<attr>src\s*=\s*(\'|")(?<src>[^\'"]+)\2)[^>]*>(.*?)</iframe>#is';
		$pattern = apply_filters( 'wp_smush_iframes_regex', $pattern );
		$iframes = array();

		if ( ! preg_match_all( $pattern, $markup, $matches, PREG_SET_ORDER | PREG_OFFSET_CAPTURE ) ) {
			return $iframes;
		}

		foreach ( $matches as $iframe_data ) {
			if ( empty( $iframe_data[0][0] ) || ! isset( $iframe_data[0][1] ) || empty( $iframe_data['attr'] ) || empty( $iframe_data['src'] ) ) {
				continue;
			}

			$iframe_markup   = $iframe_data[0][0];
			$iframe_position = $iframe_data[0][1];
			$attributes      = $this->get_element_attributes( $iframe_markup, $base_url );
			$iframe_element  = new Element( $iframe_markup, 'iframe', $attributes, array(), $iframe_position );

			$iframes[] = $iframe_element;
		}

		return $iframes;
	}

	public function get_element_attribute_value( $element_markup, $attribute_name ) {
		if ( strpos( $element_markup, $attribute_name ) === false ) {
			return '';
		}

		$pattern = '#' . $attribute_name . '\s*=\s*([\'|"])(?<value>(?:(?!\1).)*)\1#is';
		if ( ! preg_match_all( $pattern, $element_markup, $matches, PREG_SET_ORDER ) ) {
			return '';
		}

		return empty( $matches[0]['value'] )
			? ''
			: $matches[0]['value'];
	}

	public function markup_contains_noscript( $markup ) {
		return false !== strpos( $markup, '<noscript>' );
	}

	private function is_safe( $str ) {
		$str = trim( $str );
		return $this->sanitize_value( $str ) === $str;
	}

	/**
	 * This is almost the same as {@see _sanitize_text_fields()} but it doesn't remove percent encoded values because they are valid.
	 */
	private function sanitize_value( $str ) {
		if ( is_object( $str ) || is_array( $str ) ) {
			return '';
		}

		$str = (string) $str;

		$filtered = wp_check_invalid_utf8( $str );

		if ( str_contains( $filtered, '<' ) ) {
			$filtered = wp_pre_kses_less_than( $filtered );
			// This will strip extra whitespace for us.
			$filtered = wp_strip_all_tags( $filtered );

			/*
			 * Use HTML entities in a special case to make sure that
			 * later newline stripping stages cannot lead to a functional tag.
			 */
			$filtered = str_replace( "<\n", "&lt;\n", $filtered );
		}
		/**
		 * Skip removal of percent encoded values {@see _sanitize_text_fields}
		 */
		return trim( $filtered );
	}

	public function get_self_closing_element_and_position( $tag, $markup, $index ) {
		$html_element_markup   = null;
		$html_element_position = null;
		$pattern               = '/<(' . $tag . ')[^>]+>/is';
		$tags_found            = preg_match_all( $pattern, $markup, $matches, PREG_PATTERN_ORDER | PREG_OFFSET_CAPTURE );
		if ( $tags_found ) {
			$html_element_markup   = $this->array_utils->get_array_value( $matches, [ 0, $index, 0 ] );
			$html_element_position = $this->array_utils->get_array_value( $matches, [ 0, $index, 1 ] );
		}
		return [ $html_element_markup, $html_element_position ];
	}

	public function get_top_level_element_and_position( $tag, $markup, $index ) {
		$html_element_markup         = null;
		$html_element_position       = null;
		$html_element_inner_markup   = null;
		$html_element_inner_position = null;
		if ( substr_count( $markup, '<' . $tag ) < 2 ) {
			$pattern = '~<(?<tag>' . $tag . ')\b[^>]*>(.*?)</' . $tag . '>~is';
		} else {
			$pattern = '~<(?<tag>[a-zA-Z][a-zA-Z0-9]*)\b[^>]*>((?>(?:[^<]+|<(?!/?\1\b[^>]*>)+|(?R))*))</\1>~is';
		}
		$tags_found = preg_match_all( $pattern, $markup, $matches, PREG_PATTERN_ORDER | PREG_OFFSET_CAPTURE );
		if ( $tags_found ) {
			$matches = $this->filter_out_irrelevant_elements( $matches, $tag );
			if ( $matches ) {
				$html_element_markup         = $this->array_utils->get_array_value( $matches, [ 0, $index, 0 ] );
				$html_element_position       = $this->array_utils->get_array_value( $matches, [ 0, $index, 1 ] );
				$html_element_inner_markup   = $this->array_utils->get_array_value( $matches, [ 1, $index, 0 ] );
				$html_element_inner_position = $this->array_utils->get_array_value( $matches, [ 1, $index, 1 ] );
			}
		}

		return [ $html_element_markup, $html_element_position, $html_element_inner_markup, $html_element_inner_position ];
	}

	private function filter_out_irrelevant_elements( $matches, $target_tag ) {
		$outer          = $matches[0];
		$tags           = $matches[1];
		$inner          = $matches[2];
		$filtered_outer = array();
		$filtered_inner = array();
		foreach ( $outer as $index => $outer_element ) {
			$tag = $tags[ $index ][0];
			if ( $tag !== $target_tag ) {
				continue;
			}

			$filtered_outer[] = $outer_element;
			$filtered_inner[] = $inner[ $index ];
		}

		return array( $filtered_outer, $filtered_inner );
	}
}
