Prevent amp-state for nav menu toggle being inserted outside of body

ampproject · May 15, 2019 · 340c53a · 340c53a
1 parent 437c335
commit 340c53a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/includes/sanitizers/class-amp-nav-menu-toggle-sanitizer.php b/includes/sanitizers/class-amp-nav-menu-toggle-sanitizer.php
@@ -99,6 +99,11 @@ public function sanitize() {
 		$script_el->appendChild( $this->dom->createTextNode( wp_json_encode( $expanded ) ) );
 		$state_el->appendChild( $script_el );
 		$nav_el->parentNode->insertBefore( $state_el, $nav_el );
+		if ( 'body' === $nav_el->nodeName ) {
+			$nav_el->insertBefore( $state_el, $nav_el->firstChild );
+		} else {
+			$nav_el->parentNode->insertBefore( $state_el, $nav_el );
+		}
 
 		$button_on = sprintf( "tap:AMP.setState({ $state_id: ! $state_id })" );
 		$button_el->setAttribute( 'on', $button_on );