<?php

namespace WP_Rocket\Engine\Optimization\DelayJS;

use WP_Rocket\Admin\Options_Data;

class HTML {

	/**
	 * Plugin options instance.
	 *
	 * @since  3.7
	 *
	 * @var Options_Data
	 */
	protected $options;

	/**
	 * Allowed scripts regex.
	 *
	 * @since 3.7
	 *
	 * @var string
	 */
	private $allowed_scripts = '';

	/**
	 * Creates an instance of HTML.
	 *
	 * @since  3.7
	 *
	 * @param Options_Data $options Plugin options instance.
	 */
	public function __construct( Options_Data $options ) {
		$this->options = $options;
	}

	/**
	 * Adjust HTML to have delay js structure.
	 *
	 * @param string $html Buffer html for the page.
	 *
	 * @return string
	 */
	public function delay_js( $html ) {

		if ( ! $this->is_allowed() ) {
			return $html;
		}

		$this->allowed_scripts = $this->prepare_allowed_scripts_regex();

		if ( empty( $this->allowed_scripts ) ) {
			return $html;
		}

		return $this->parse( $html );
	}

	/**
	 * Checks if is allowed to Delay JS.
	 *
	 * @since 3.7
	 *
	 * @return bool
	 */
	public function is_allowed() {
		if ( rocket_bypass() ) {
			return false;
		}

		if ( rocket_get_constant( 'DONOTROCKETOPTIMIZE' ) ) {
			return false;
		}

		if ( is_rocket_post_excluded_option( 'delay_js' ) ) {
			return false;
		}

		return (bool) $this->options->get( 'delay_js', 0 );
	}

	/**
	 * Parse the html and add/remove attributes from specific scripts.
	 *
	 * @param string $html Buffer html for the page.
	 *
	 * @return string
	 */
	private function parse( $html ) {
		$replaced_html = preg_replace_callback( '/<\s*script\s*(?<attr>[^>]*?)?>(?<content>.*?)?<\s*\/\s*script\s*>/ims', [ $this, 'replace_scripts' ], $html );

		if ( empty( $replaced_html ) ) {
			return $html;
		}

		return $replaced_html;
	}

	/**
	 * Callback method for preg_replace_callback that is used to adjust attributes for specific scripts.
	 *
	 * @param array $matches Matches array for scripts regex.
	 *
	 * @return string
	 */
	public function replace_scripts( $matches ) {
		if (
			empty( $this->allowed_scripts )
			||
			(
				! empty( $this->allowed_scripts )
				&&
				! preg_match( '#(' . $this->allowed_scripts . ')#', $matches[0] )
			)
		) {
			return $matches[0];
		}

		$src             = '';
		$matches['attr'] = trim( $matches['attr'] );

		if ( ! empty( $matches['attr'] ) ) {
			if ( preg_match( '/src=(["\'])(.*?)\1/', $matches['attr'], $src_matches ) ) {
				$src = $src_matches[2];

				// Remove the src attribute.
				$matches['attr'] = str_replace( $src_matches[0], '', $matches['attr'] );
			}
		}

		if ( empty( $src ) ) {
			// Get the JS content.
			if ( ! empty( $matches['content'] ) ) {
				$src = 'data:text/javascript;base64,' . base64_encode( $matches['content'] );// phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions.obfuscation_base64_encode
			}
		}

		if ( empty( $src ) ) {
			return $matches[0];
		}

		return "<script data-rocketlazyloadscript='{$src}' {$matches['attr']}></script>";
	}

	/**
	 * Prepare allowed scripts to be used as regex.
	 *
	 * @return string
	 */
	private function prepare_allowed_scripts_regex() {
		$delay_js_scripts = $this->options->get( 'delay_js_scripts', [] );

		/**
		 * Filters JS files to included into delay JS.
		 *
		 * @since 3.7
		 *
		 * @param array $delay_js_scripts List of allowed JS files.
		 */
		$delay_js_scripts = (array) apply_filters( 'rocket_delay_js_scripts', $delay_js_scripts );

		if ( empty( $delay_js_scripts ) ) {
			return '';
		}

		foreach ( $delay_js_scripts as $i => $delay_js_script ) {
			$delay_js_scripts[ $i ] = preg_quote( str_replace( '#', '\#', $delay_js_script ), '#' );
		}

		return implode( '|', $delay_js_scripts );

	}

}
