Closes #6518: Inject beacon for saas visit

inc/Engine/Media/AboveTheFold/Frontend/Controller.php

@@ -93,6 +93,10 @@ public function lcp( $html ): string {
 	 * @return string
 	 */
 	private function preload_lcp( $html, $row ) {
+		if ( rocket_bypass() ) { // Bail out if rocket_bypass() returns true
+			return $html;
+		}
+
 		if ( ! preg_match( '#</title\s*>#', $html, $matches ) ) {
 			return $html;
 		}

inc/Engine/Media/AboveTheFold/Frontend/Subscriber.php

@@ -29,8 +29,9 @@ public function __construct( Controller $controller ) {
 	 */
 	public static function get_subscribed_events() {
 		return [
-			'rocket_buffer'                => [ 'lcp', 17 ],
-			'rocket_lazyload_excluded_src' => 'add_exclusions',
+			'rocket_buffer'                           => [ 'lcp', 17 ],
+			'rocket_lazyload_excluded_src'            => 'add_exclusions',
+			'rocket_critical_image_saas_visit_buffer' => [ 'lcp', 17 ],
 		];
 	}
 

inc/Engine/Media/ImageDimensions/Subscriber.php

@@ -64,6 +64,23 @@ public function specify_image_dimensions( $buffer ) {
 		return $this->dimensions->specify_image_dimensions( $buffer );
 	}
 
+	/**
+	 * Update images that have no width/height with real dimentions for the SaaS
+	 *
+	 * @param string $buffer Page HTML content.
+	 *
+	 * @return string Page HTML content after update.
+	 */
+	public function prepare_critical_image_saas_visit( $buffer ) {
+		if ( ! isset( $_GET['wpr_imagedimensions'] ) ) { // Bail out if wpr_imagedimensions is not set
+			return $buffer;
+		}
+
+		do_action( 'rocket_critical_image_saas_visit_buffer', $buffer ); // Fire an action
+
+		return $this->dimensions->specify_image_dimensions( $buffer );
+	}
+
 	/**
 	 * Start image dimensions buffer to add
 	 *
@@ -80,6 +97,6 @@ public function start_image_dimensions_buffer() {
 
 		add_filter( 'rocket_specify_image_dimensions', '__return_true' );
 
-		ob_start( [ $this, 'specify_image_dimensions' ] );
+		ob_start( [ $this, 'prepare_critical_image_saas_visit' ] );
 	}
 }