Move any content output during shutdown to be injected before closing…

… body tag
ampproject · Apr 29, 2018 · 2f2a315 · 2f2a315
1 parent 1e2cd22
commit 2f2a315
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/includes/class-amp-theme-support.php b/includes/class-amp-theme-support.php
@@ -1043,6 +1043,10 @@ public static function prepare_response( $response, $args = array() ) {
 				1
 			);
 		}
+
+		// Move anything after </html>, such as Query Monitor output added at shutdown, to be moved before </body>.
+		$response = preg_replace( '#(</body>.*</html>)(.+)#s', '$2$1', $response );
+
 		$dom = AMP_DOM_Utils::get_dom( $response );
 
 		$xpath = new DOMXPath( $dom );