Add customizer theme support

amp.php

@@ -87,23 +87,14 @@ function amp_after_setup_theme() {
 		define( 'AMP_QUERY_VAR', apply_filters( 'amp_query_var', 'amp' ) );
 	}
 
-	add_action( 'init', 'amp_init' );
-	add_action( 'widgets_init', 'AMP_Theme_Support::register_widgets' ); // @todo Let this be called by AMP_Theme_Support::init().
-	add_action( 'init', 'AMP_Theme_Support::setup_commenting' ); // @todo Let this be called by AMP_Theme_Support::init().
-	add_action( 'admin_init', 'AMP_Options_Manager::register_settings' );
-	add_action( 'wp_loaded', 'amp_post_meta_box' );
-	add_action( 'wp_loaded', 'amp_add_options_menu' );
-	add_action( 'parse_query', 'amp_correct_query_when_is_front_page' );
-	AMP_Post_Type_Support::add_post_type_support();
-	AMP_Validation_Utils::init();
+	add_action( 'init', 'amp_init', 0 ); // Must be 0 because widgets_init happens at init priority 1.
 }
 add_action( 'after_setup_theme', 'amp_after_setup_theme', 5 );
 
 /**
  * Init AMP.
  *
  * @since 0.1
- * @global string $pagenow
  */
 function amp_init() {
 
@@ -118,16 +109,24 @@ function amp_init() {
 
 	add_rewrite_endpoint( AMP_QUERY_VAR, EP_PERMALINK );
 
+	AMP_Theme_Support::init();
+	AMP_Post_Type_Support::add_post_type_support();
+	AMP_Validation_Utils::init();
 	add_filter( 'request', 'amp_force_query_var_value' );
-	add_action( 'wp', 'amp_maybe_add_actions' );
+	add_action( 'admin_init', 'AMP_Options_Manager::register_settings' );
+	add_action( 'wp_loaded', 'amp_post_meta_box' );
+	add_action( 'wp_loaded', 'amp_add_options_menu' );
+	add_action( 'parse_query', 'amp_correct_query_when_is_front_page' );
 
 	// Redirect the old url of amp page to the updated url.
 	add_filter( 'old_slug_redirect_url', 'amp_redirect_old_slug_to_new_url' );
 
 	if ( class_exists( 'Jetpack' ) && ! ( defined( 'IS_WPCOM' ) && IS_WPCOM ) ) {
 		require_once AMP__DIR__ . '/jetpack-helper.php';
 	}
-	amp_handle_xhr_request();
+
+	// Add actions for legacy post templates.
+	add_action( 'wp', 'amp_maybe_add_actions' );
 }
 
 // Make sure the `amp` query var has an explicit value.
@@ -150,15 +149,9 @@ function amp_force_query_var_value( $query_vars ) {
  * @return void
  */
 function amp_maybe_add_actions() {
-	$is_amp_endpoint = is_amp_endpoint();
 
-	// Add hooks for when a themes that support AMP.
+	// Short-circuit when theme supports AMP, as everything is handled by AMP_Theme_Support.
 	if ( current_theme_supports( 'amp' ) ) {
-		if ( $is_amp_endpoint ) {
-			AMP_Theme_Support::init();
-		} else {
-			amp_add_frontend_actions();
-		}
 		return;
 	}
 
@@ -168,6 +161,8 @@ function amp_maybe_add_actions() {
 		return;
 	}
 
+	$is_amp_endpoint = is_amp_endpoint();
+
 	/**
 	 * Queried post object.
 	 *

includes/amp-helper-functions.php

@@ -484,94 +484,3 @@ function amp_print_schemaorg_metadata() {
 	<script type="application/ld+json"><?php echo wp_json_encode( $metadata ); ?></script>
 	<?php
 }
-
-/**
- * Hook into a form submissions, such as comment the form or some other .
- *
- * @since 0.7.0
- * @global string $pagenow
- */
-function amp_handle_xhr_request() {
-	global $pagenow;
-	if ( ! isset( $_GET['__amp_source_origin'] ) ) { // WPCS: CSRF ok. Beware of AMP_Theme_Support::purge_amp_query_vars().
-		return;
-	}
-
-	if ( isset( $pagenow ) && 'wp-comments-post.php' === $pagenow ) {
-		// We don't need any data, so just send a success.
-		add_filter( 'comment_post_redirect', function() {
-			// We don't need any data, so just send a success.
-			wp_send_json_success();
-		}, PHP_INT_MAX, 2 );
-		amp_handle_xhr_headers_output();
-	} elseif ( isset( $_GET['_wp_amp_action_xhr_converted'] ) ) { // WPCS: CSRF ok.
-		add_filter( 'wp_redirect', 'amp_intercept_post_request_redirect', PHP_INT_MAX, 2 );
-		amp_handle_xhr_headers_output();
-	}
-}
-
-/**
- * Handle the AMP XHR headers and output errors.
- *
- * @since 0.7.0
- */
-function amp_handle_xhr_headers_output() {
-	// Add die handler for AMP error display.
-	add_filter( 'wp_die_handler', function() {
-		/**
-		 * New error handler for AMP form submission.
-		 *
-		 * @param WP_Error|string $error The error to handle.
-		 */
-		return function( $error ) {
-			status_header( 400 );
-			if ( is_wp_error( $error ) ) {
-				$error = $error->get_error_message();
-			}
-			$amp_mustache_allowed_html_tags = array( 'strong', 'b', 'em', 'i', 'u', 's', 'small', 'mark', 'del', 'ins', 'sup', 'sub' );
-			wp_send_json( array(
-				'error' => wp_kses( $error, array_fill_keys( $amp_mustache_allowed_html_tags, array() ) ),
-			) );
-		};
-	} );
-
-	// Send AMP header.
-	$origin = esc_url_raw( wp_unslash( $_GET['__amp_source_origin'] ) ); // WPCS: CSRF ok.
-	header( 'AMP-Access-Control-Allow-Source-Origin: ' . $origin, true );
-}
-
-/**
- * Intercept the response to a non-comment POST request.
- *
- * @since 0.7.0
- * @param string $location The location to redirect to.
- */
-function amp_intercept_post_request_redirect( $location ) {
-
-	// Make sure relative redirects get made absolute.
-	$parsed_location = array_merge(
-		array(
-			'scheme' => 'https',
-			'host'   => wp_parse_url( home_url(), PHP_URL_HOST ),
-			'path'   => strtok( wp_unslash( $_SERVER['REQUEST_URI'] ), '?' ),
-		),
-		wp_parse_url( $location )
-	);
-
-	$absolute_location = $parsed_location['scheme'] . '://' . $parsed_location['host'];
-	if ( isset( $parsed_location['port'] ) ) {
-		$absolute_location .= ':' . $parsed_location['port'];
-	}
-	$absolute_location .= $parsed_location['path'];
-	if ( isset( $parsed_location['query'] ) ) {
-		$absolute_location .= '?' . $parsed_location['query'];
-	}
-	if ( isset( $parsed_location['fragment'] ) ) {
-		$absolute_location .= '#' . $parsed_location['fragment'];
-	}
-
-	header( 'AMP-Redirect-To: ' . $absolute_location );
-	header( 'Access-Control-Expose-Headers: AMP-Redirect-To' );
-	// Send json success as no data is required.
-	wp_send_json_success();
-}