From c899c451e8cbe333826b816a8efcbd8b7f8fa91d Mon Sep 17 00:00:00 2001
From: thelovekesh <lovekesh.kumar@rtcamp.com>
Date: Thu, 15 Jun 2023 14:54:21 +0530
Subject: [PATCH 1/6] Update element string with Tag::P constant

---
 includes/embeds/class-amp-base-embed-handler.php | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/includes/embeds/class-amp-base-embed-handler.php b/includes/embeds/class-amp-base-embed-handler.php
index ee18eba26b8..d1811f81f3d 100644
--- a/includes/embeds/class-amp-base-embed-handler.php
+++ b/includes/embeds/class-amp-base-embed-handler.php
@@ -150,7 +150,7 @@ protected function unwrap_p_element( DOMElement $node ) {
 		if (
 			$parent_node instanceof DOMElement
 			&&
-			'p' === $parent_node->tagName
+			Tag::P === $parent_node->tagName
 			&&
 			false === $parent_node->hasAttributes()
 			&&

From 0ffcc8807ecf20509c91dbe8fac9984da0c56eb4 Mon Sep 17 00:00:00 2001
From: thelovekesh <lovekesh.kumar@rtcamp.com>
Date: Thu, 15 Jun 2023 15:06:27 +0530
Subject: [PATCH 2/6] Add `::unwrap_p_element_by_child_tag_name()` method

---
 .../embeds/class-amp-base-embed-handler.php    | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/includes/embeds/class-amp-base-embed-handler.php b/includes/embeds/class-amp-base-embed-handler.php
index d1811f81f3d..6e5768218cc 100644
--- a/includes/embeds/class-amp-base-embed-handler.php
+++ b/includes/embeds/class-amp-base-embed-handler.php
@@ -160,6 +160,24 @@ protected function unwrap_p_element( DOMElement $node ) {
 		}
 	}
 
+	/**
+	 * Unwrap `<p>` element based on passed child tag name.
+	 *
+	 * @since 2.4.2
+	 *
+	 * @param Document $dom            Document.
+	 * @param string   $child_tag_name Child tag name.
+	 */
+	protected function unwrap_p_element_by_child_tag_name( $dom, $child_tag_name ) {
+		$nodes = $dom->xpath->query( "//p/{$child_tag_name}" );
+
+		if ( $nodes->length && $nodes instanceof DOMNodeList ) {
+			foreach ( $nodes as $node ) {
+				$this->unwrap_p_element( $node );
+			}
+		}
+	}
+
 	/**
 	 * Removes the node's nearest `<script>` sibling with a `src` attribute containing the base `src` URL provided.
 	 *

From 40e9709d8049c1a086b93c8fb8dd06e5365b3da1 Mon Sep 17 00:00:00 2001
From: thelovekesh <lovekesh.kumar@rtcamp.com>
Date: Fri, 16 Jun 2023 00:02:52 +0530
Subject: [PATCH 3/6] Add `AMP_Iframe_Embed_Handler` class for handling
 amp-iframe embeds

---
 .../embeds/class-amp-iframe-embed-handler.php | 47 +++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 includes/embeds/class-amp-iframe-embed-handler.php

diff --git a/includes/embeds/class-amp-iframe-embed-handler.php b/includes/embeds/class-amp-iframe-embed-handler.php
new file mode 100644
index 00000000000..f1fb57b2844
--- /dev/null
+++ b/includes/embeds/class-amp-iframe-embed-handler.php
@@ -0,0 +1,47 @@
+<?php
+/**
+ * Class for handling `amp-iframe` embeds.
+ *
+ * @package AMP
+ * @since 2.4.1
+ */
+
+use AmpProject\Html\Tag;
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
+/**
+ * Class AMP_Iframe_Embed_Handler
+ *
+ * @since 2.4.1
+ */
+class AMP_Iframe_Embed_Handler extends AMP_Base_Embed_Handler {
+
+	/**
+	 * Register embed.
+	 */
+	public function register_embed() {
+		// Not implemented.
+	}
+
+	/**
+	 * Unregister embed.
+	 */
+	public function unregister_embed() {
+		// Not implemented.
+	}
+
+	/**
+	 * Sanitize `amp-iframe` raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		foreach ( [ Extension::IFRAME, Tag::IFRAME ] as $tag_name ) {
+			$this->unwrap_p_element_by_child_tag_name( $dom, $tag_name );
+		}
+	}
+}

From 17b2dde370b5340f90cc0cd0ed8bdfa18344c35b Mon Sep 17 00:00:00 2001
From: thelovekesh <lovekesh.kumar@rtcamp.com>
Date: Fri, 16 Jun 2023 00:03:55 +0530
Subject: [PATCH 4/6] Add `AMP_Iframe_Embed_Handler` class in content embed
 handlers

---
 includes/amp-helper-functions.php | 1 +
 1 file changed, 1 insertion(+)

diff --git a/includes/amp-helper-functions.php b/includes/amp-helper-functions.php
index 1844bb0bf06..f6d6cd8a3e5 100644
--- a/includes/amp-helper-functions.php
+++ b/includes/amp-helper-functions.php
@@ -1413,6 +1413,7 @@ function amp_get_content_embed_handlers( $post = null ) {
 			AMP_Scribd_Embed_Handler::class       => [],
 			AMP_WordPress_Embed_Handler::class    => [],
 			AMP_WordPress_TV_Embed_Handler::class => [],
+			AMP_Iframe_Embed_Handler::class       => [],
 		],
 		$post
 	);

From b42020ba584570ce06ed1fc3bf9df44edd3d3e4e Mon Sep 17 00:00:00 2001
From: thelovekesh <lovekesh.kumar@rtcamp.com>
Date: Fri, 16 Jun 2023 00:09:30 +0530
Subject: [PATCH 5/6] Update URL regex for Crowdsignal embeds

---
 includes/embeds/class-amp-crowdsignal-embed-handler.php | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/includes/embeds/class-amp-crowdsignal-embed-handler.php b/includes/embeds/class-amp-crowdsignal-embed-handler.php
index e61669ec24b..d9559427fcb 100644
--- a/includes/embeds/class-amp-crowdsignal-embed-handler.php
+++ b/includes/embeds/class-amp-crowdsignal-embed-handler.php
@@ -39,7 +39,7 @@ public function unregister_embed() {
 	 */
 	public function filter_embed_oembed_html( $cache, $url, $attr ) {
 		$parsed_url = wp_parse_url( $url );
-		if ( empty( $parsed_url['host'] ) || empty( $parsed_url['path'] ) || ! preg_match( '#(^|\.)(?P<host>polldaddy\.com|crowdsignal\.com|survey\.fm|poll\.fm)#', $parsed_url['host'], $matches ) ) {
+		if ( empty( $parsed_url['host'] ) || empty( $parsed_url['path'] ) || ! preg_match( '#(^|\.)(?P<host>polldaddy\.com|poll\.fm|crowdsignal\.com|survey\.fm|poll\.fm)#', $parsed_url['host'], $matches ) ) {
 			return $cache;
 		}
 

From c719d1c5ad94a753e275305cc7b3b926550c2b3b Mon Sep 17 00:00:00 2001
From: thelovekesh <lovekesh.kumar@rtcamp.com>
Date: Fri, 16 Jun 2023 00:10:00 +0530
Subject: [PATCH 6/6] Update embed handlers to unwrap embeds from `<p>` tag if
 embedded using shortcodes

---
 .../class-amp-dailymotion-embed-handler.php      | 15 +++++++++++++++
 .../embeds/class-amp-facebook-embed-handler.php  |  8 ++------
 .../embeds/class-amp-gfycat-embed-handler.php    | 15 +++++++++++++++
 .../embeds/class-amp-imgur-embed-handler.php     | 15 +++++++++++++++
 .../embeds/class-amp-instagram-embed-handler.php |  4 ++++
 .../embeds/class-amp-pinterest-embed-handler.php | 15 +++++++++++++++
 .../embeds/class-amp-reddit-embed-handler.php    | 16 +++++++++++++++-
 .../class-amp-soundcloud-embed-handler.php       | 15 +++++++++++++++
 .../embeds/class-amp-tiktok-embed-handler.php    |  1 +
 .../embeds/class-amp-vimeo-embed-handler.php     | 15 +++++++++++++++
 .../embeds/class-amp-youtube-embed-handler.php   |  2 ++
 11 files changed, 114 insertions(+), 7 deletions(-)

diff --git a/includes/embeds/class-amp-dailymotion-embed-handler.php b/includes/embeds/class-amp-dailymotion-embed-handler.php
index 9808cb34bf2..27c71ef88d8 100644
--- a/includes/embeds/class-amp-dailymotion-embed-handler.php
+++ b/includes/embeds/class-amp-dailymotion-embed-handler.php
@@ -5,6 +5,9 @@
  * @package AMP
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_DailyMotion_Embed_Handler
  *
@@ -117,6 +120,18 @@ public function render( $args ) {
 		);
 	}
 
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::DAILYMOTION );
+	}
+
 	/**
 	 * Determine the video ID from the URL.
 	 *
diff --git a/includes/embeds/class-amp-facebook-embed-handler.php b/includes/embeds/class-amp-facebook-embed-handler.php
index 134144855b8..6a9ad5e2094 100644
--- a/includes/embeds/class-amp-facebook-embed-handler.php
+++ b/includes/embeds/class-amp-facebook-embed-handler.php
@@ -5,6 +5,7 @@
  * @package AMP
  */
 
+use AmpProject\Extension;
 use AmpProject\Dom\Document;
 
 /**
@@ -114,12 +115,7 @@ public function render( $args ) {
 	 */
 	public function sanitize_raw_embeds( Document $dom ) {
 		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
-		$embed_nodes = $dom->xpath->query( "//p/{$this->amp_tag}" );
-		if ( $embed_nodes->length ) {
-			foreach ( $embed_nodes as $embed_node ) {
-				$this->unwrap_p_element( $embed_node );
-			}
-		}
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::FACEBOOK );
 
 		$nodes     = $dom->getElementsByTagName( $this->sanitize_tag );
 		$num_nodes = $nodes->length;
diff --git a/includes/embeds/class-amp-gfycat-embed-handler.php b/includes/embeds/class-amp-gfycat-embed-handler.php
index d2982c3b357..54f4b412612 100644
--- a/includes/embeds/class-amp-gfycat-embed-handler.php
+++ b/includes/embeds/class-amp-gfycat-embed-handler.php
@@ -6,6 +6,9 @@
  * @since 1.0
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_Gfycat_Embed_Handler
  *
@@ -79,5 +82,17 @@ public function filter_embed_oembed_html( $return, $url, $attr ) {
 		}
 		return $return;
 	}
+
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::GFYCAT );
+	}
 }
 
diff --git a/includes/embeds/class-amp-imgur-embed-handler.php b/includes/embeds/class-amp-imgur-embed-handler.php
index ad60aca4cc4..470028f85cc 100644
--- a/includes/embeds/class-amp-imgur-embed-handler.php
+++ b/includes/embeds/class-amp-imgur-embed-handler.php
@@ -6,6 +6,9 @@
  * @since 1.0
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_Imgur_Embed_Handler
  *
@@ -73,6 +76,18 @@ public function filter_embed_oembed_html( $return, $url, $attr ) {
 		return $return;
 	}
 
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::IMGUR );
+	}
+
 	/**
 	 * Get Imgur ID from URL.
 	 *
diff --git a/includes/embeds/class-amp-instagram-embed-handler.php b/includes/embeds/class-amp-instagram-embed-handler.php
index 5367a17e177..635d6a92758 100644
--- a/includes/embeds/class-amp-instagram-embed-handler.php
+++ b/includes/embeds/class-amp-instagram-embed-handler.php
@@ -8,6 +8,7 @@
 use AmpProject\Dom\Document;
 use AmpProject\Html\Attribute;
 use AmpProject\Html\Tag;
+use AmpProject\Extension;
 
 /**
  * Class AMP_Instagram_Embed_Handler
@@ -142,6 +143,9 @@ private function get_instagram_id_from_url( $url ) {
 	 * @param Document $dom DOM.
 	 */
 	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::INSTAGRAM );
+
 		/**
 		 * Node list.
 		 *
diff --git a/includes/embeds/class-amp-pinterest-embed-handler.php b/includes/embeds/class-amp-pinterest-embed-handler.php
index 5ff81a07f66..76248ae2f05 100644
--- a/includes/embeds/class-amp-pinterest-embed-handler.php
+++ b/includes/embeds/class-amp-pinterest-embed-handler.php
@@ -5,6 +5,9 @@
  * @package AMP
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_Pinterest_Embed_Handler
  *
@@ -89,4 +92,16 @@ public function render( $args ) {
 			]
 		);
 	}
+
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::PINTEREST );
+	}
 }
diff --git a/includes/embeds/class-amp-reddit-embed-handler.php b/includes/embeds/class-amp-reddit-embed-handler.php
index 922dd158c65..738d39a7b9c 100644
--- a/includes/embeds/class-amp-reddit-embed-handler.php
+++ b/includes/embeds/class-amp-reddit-embed-handler.php
@@ -6,6 +6,9 @@
  * @since 0.7
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_Reddit_Embed_Handler
  *
@@ -73,5 +76,16 @@ public function render( $args ) {
 			]
 		);
 	}
-}
 
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::EMBEDLY_CARD );
+	}
+}
diff --git a/includes/embeds/class-amp-soundcloud-embed-handler.php b/includes/embeds/class-amp-soundcloud-embed-handler.php
index 65f619bbcca..33234ad2b41 100644
--- a/includes/embeds/class-amp-soundcloud-embed-handler.php
+++ b/includes/embeds/class-amp-soundcloud-embed-handler.php
@@ -5,6 +5,9 @@
  * @package AMP
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_SoundCloud_Embed_Handler
  *
@@ -174,6 +177,18 @@ private function render_embed_fallback( $url ) {
 		);
 	}
 
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::SOUNDCLOUD );
+	}
+
 	/**
 	 * Get params from Soundcloud iframe src.
 	 *
diff --git a/includes/embeds/class-amp-tiktok-embed-handler.php b/includes/embeds/class-amp-tiktok-embed-handler.php
index 221e19d6499..4a5f6db3795 100644
--- a/includes/embeds/class-amp-tiktok-embed-handler.php
+++ b/includes/embeds/class-amp-tiktok-embed-handler.php
@@ -45,6 +45,7 @@ public function sanitize_raw_embeds( Document $dom ) {
 
 		foreach ( $nodes as $node ) {
 			$this->make_embed_amp_compatible( $node );
+			$this->unwrap_p_element( $node );
 		}
 	}
 
diff --git a/includes/embeds/class-amp-vimeo-embed-handler.php b/includes/embeds/class-amp-vimeo-embed-handler.php
index d005e960d70..9f2b803468f 100644
--- a/includes/embeds/class-amp-vimeo-embed-handler.php
+++ b/includes/embeds/class-amp-vimeo-embed-handler.php
@@ -5,6 +5,9 @@
  * @package AMP
  */
 
+use AmpProject\Extension;
+use AmpProject\Dom\Document;
+
 /**
  * Class AMP_Vimeo_Embed_Handler
  *
@@ -131,6 +134,18 @@ public function render( $args ) {
 		);
 	}
 
+	/**
+	 * Sanitize raw embeds.
+	 *
+	 * @param Document $dom Document.
+	 *
+	 * @return void
+	 */
+	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::VIMEO );
+	}
+
 	/**
 	 * Determine the video ID from the URL.
 	 *
diff --git a/includes/embeds/class-amp-youtube-embed-handler.php b/includes/embeds/class-amp-youtube-embed-handler.php
index ddcc6ae4353..82ae05f49dc 100644
--- a/includes/embeds/class-amp-youtube-embed-handler.php
+++ b/includes/embeds/class-amp-youtube-embed-handler.php
@@ -161,6 +161,8 @@ public function render( $html, $url, $video_id ) {
 	 * @return void
 	 */
 	public function sanitize_raw_embeds( Document $dom ) {
+		// If there were any previous embeds in the DOM that were wrapped by `wpautop()`, unwrap them.
+		$this->unwrap_p_element_by_child_tag_name( $dom, Extension::YOUTUBE );
 
 		$query_segments = array_map(
 			static function ( $domain ) {