Integrate RewriteAmpUrls transformer

Author: schlessera

composer.json

@@ -13,7 +13,7 @@
     "ext-json": "*",
     "ext-libxml": "*",
     "ext-spl": "*",
-    "ampproject/amp-toolbox": "^0.3",
+    "ampproject/amp-toolbox": "dev-add/20-rewrite-amp-urls-transformer",
     "cweagans/composer-patches": "1.7.0",
     "fasterimage/fasterimage": "1.5.0",
     "sabberworm/php-css-parser": "dev-master#bfdd976"

composer.lock

@@ -1401,6 +1401,7 @@ public static function ensure_required_markup( Document $dom, $script_handles =
 				),
 			],
 		];
+
 		$link_elements = $dom->head->getElementsByTagName( Tag::LINK );
 		/**
 		 * Link element.
@@ -1500,32 +1501,16 @@ public static function ensure_required_markup( Document $dom, $script_handles =
 			}
 		}
 
-		/* phpcs:ignore Squiz.PHP.CommentedOutCode.Found
-		 *
-		 * "2. Next, preload the AMP runtime v0.js <script> tag with <link as=script href=https://cdn.ampproject.org/v0.js rel=preload>.
-		 * The AMP runtime should start downloading as soon as possible because the AMP boilerplate hides the document via body { visibility:hidden }
-		 * until the AMP runtime has loaded. Preloading the AMP runtime tells the browser to download the script with a higher priority."
-		 * {@link https://amp.dev/documentation/guides-and-tutorials/optimize-and-measure/optimize_amp/ Optimize the AMP Runtime loading}
+		/*
+		 * "3. If your page includes render-delaying extensions (e.g., amp-experiment, amp-dynamic-css-classes, amp-story),
+		 * preload those extensions as they're required by the AMP runtime for rendering the page."
+		 * @TODO: Move into RewriteAmpUrls transformer, as that will support self-hosting as well.
 		 */
 		$prioritized_preloads = [];
 		if ( ! isset( $links[ Attribute::REL_PRELOAD ] ) ) {
 			$links[ Attribute::REL_PRELOAD ] = [];
 		}
 
-		$prioritized_preloads[] = AMP_DOM_Utils::create_node(
-			$dom,
-			Tag::LINK,
-			[
-				Attribute::REL  => Attribute::REL_PRELOAD,
-				Attribute::AS_  => RequestDestination::SCRIPT,
-				Attribute::HREF => $runtime_src,
-			]
-		);
-
-		/*
-		 * "3. If your page includes render-delaying extensions (e.g., amp-experiment, amp-dynamic-css-classes, amp-story),
-		 * preload those extensions as they're required by the AMP runtime for rendering the page."
-		 */
 		$amp_script_handles = array_keys( $amp_scripts );
 		foreach ( array_intersect( Amp::RENDER_DELAYING_EXTENSIONS, $amp_script_handles ) as $script_handle ) {
 			if ( ! in_array( $script_handle, Amp::RENDER_DELAYING_EXTENSIONS, true ) ) {
@@ -1975,6 +1960,8 @@ public static function prepare_response( $response, $args = [] ) {
 			}
 		}
 
+		self::ensure_required_markup( $dom, array_keys( $amp_scripts ) );
+
 		$enable_optimizer = array_key_exists( ConfigurationArgument::ENABLE_OPTIMIZER, $args )
 			? $args[ ConfigurationArgument::ENABLE_OPTIMIZER ]
 			: true;
@@ -2033,8 +2020,6 @@ static function( Optimizer\Error $error ) {
 			do_action( 'amp_server_timing_stop', 'amp_optimizer' );
 		}
 
-		self::ensure_required_markup( $dom, array_keys( $amp_scripts ) );
-
 		$can_serve = AMP_Validation_Manager::finalize_validation( $dom );
 
 		// Redirect to the non-AMP version if not on an AMP-first site.
@@ -2094,6 +2079,10 @@ private static function get_optimizer( $args ) {
 	private static function get_optimizer_configuration( $args ) {
 		$transformers = Optimizer\Configuration::DEFAULT_TRANSFORMERS;
 
+		$enable_esm = array_key_exists( ConfigurationArgument::ENABLE_ESM, $args )
+				? $args[ ConfigurationArgument::ENABLE_ESM ]
+				: true;
+
 		$enable_ssr = array_key_exists( ConfigurationArgument::ENABLE_SSR, $args )
 			? $args[ ConfigurationArgument::ENABLE_SSR ]
 			: true;
@@ -2107,6 +2096,15 @@ private static function get_optimizer_configuration( $args ) {
 		 */
 		$enable_ssr = apply_filters( 'amp_enable_ssr', $enable_ssr );
 
+		/**
+		 * Filter whether the AMP Optimizer should use ES modules for the runtime and extensions.
+		 *
+		 * @since 2.1.0
+		 *
+		 * @param bool $enable_esm Whether the AMP Optimizer should use ES modules for the runtime and extensions.
+		 */
+		$enable_esm = apply_filters( 'amp_enable_esm', $enable_esm );
+
 		// In debugging mode, we don't use server-side rendering, as it further obfuscates the HTML markup.
 		if ( ! $enable_ssr ) {
 			$transformers = array_diff(
@@ -2137,6 +2135,9 @@ private static function get_optimizer_configuration( $args ) {
 					Optimizer\Transformer\PreloadHeroImage::class => [
 						Optimizer\Configuration\PreloadHeroImageConfiguration::INLINE_STYLE_BACKUP_ATTRIBUTE => 'data-amp-original-style',
 					],
+					Optimizer\Transformer\RewriteAmpUrls::class => [
+						Optimizer\Configuration\RewriteAmpUrlsConfiguration::ESM_MODULES_ENABLED => $enable_esm,
+					],
 				],
 				$args
 			)

includes/class-amp-theme-support.php

@@ -16,6 +16,7 @@
  */
 interface ConfigurationArgument {
 
+	const ENABLE_ESM       = 'enable_esm';
 	const ENABLE_OPTIMIZER = 'enable_optimizer';
 	const ENABLE_SSR       = 'enable_ssr';
 }