Ticket #38121: 38121.2.diff

src/wp-includes/general-template.php

@@ -2812 +2812 @@
                 'preconnect'   => array(),
                 'prefetch'     => array(),
                 'prerender'    => array(),
+                'preload'      => array(),
         );
 
         /*
@@ -2822 +2823 @@
         $hints['dns-prefetch'][] = apply_filters( 'emoji_svg_url', 'https://s.w.org/images/core/emoji/2.2.1/svg/' );
 
         foreach ( $hints as $relation_type => $urls ) {
+                $unique_urls = array();
+
                 /**
                  * Filters domains and URLs for resource hints of relation type.
                  *
@@ -2833 +2836 @@
                 $urls = apply_filters( 'wp_resource_hints', $urls, $relation_type );
 
                 foreach ( $urls as $key => $url ) {
+                        $atts = array();
+
+                        if ( is_array( $url ) ) {
+                                $atts = $url[ key( $url ) ];
+                                $url  = key( $url );
+                        }
+
                         $url = esc_url( $url, array( 'http', 'https' ) );
+
                         if ( ! $url ) {
-                                unset( $urls[ $key ] );
+                                continue;
+                        }
+
+                        if ( isset( $unique_urls[ $url ] ) ) {
                                 continue;
                         }
 
                         if ( in_array( $relation_type, array( 'preconnect', 'dns-prefetch' ) ) ) {
                                 $parsed = wp_parse_url( $url );
+
                                 if ( empty( $parsed['host'] ) ) {
-                                        unset( $urls[ $key ] );
                                         continue;
                                 }
 
@@ -2854 +2868 @@
                                 }
                         }
 
-                        $urls[ $key ] = $url;
+                        $atts['rel'] = $relation_type;
+                        $atts['href'] = $url;
+
+                        $unique_urls[ $url ] = $atts;
                 }
 
-                $urls = array_unique( $urls );
+                foreach ( $unique_urls as $atts ) {
+                        $html = '';
+
+                        foreach ( $atts as $attr => $value ) {
+                                if ( is_array( $value ) || ! in_array( $attr, array( 'as', 'crossorigin', 'href', 'pr', 'rel', 'type' ), true ) ) {
+                                        continue;
+                                }
+
+                                $attr  = esc_attr( $attr );
+                                $value = ( 'href' === $attr ) ? esc_url( $value ) : esc_attr( $value );
+
+                                if ( ! is_string( $attr ) ) {
+                                        $html .= " $value";
+                                } else {
+                                        $html .= " $attr='$value'";
+                                }
+                        }
+
+                        $html = trim( $html );
 
-                foreach ( $urls as $url ) {
-                        printf( "<link rel='%s' href='%s' />\n", $relation_type, $url );
+                        echo "<link $html />\n";
                 }
         }
 }

tests/phpunit/tests/general/resourceHints.php

@@ -1 +1 @@
 <?php
 
 /**
- * @group  template
+ * @group template
  * @ticket 34292
  */
 class Tests_WP_Resource_Hints extends WP_UnitTestCase {
@@ -242 +242 @@
 
                 return $hints;
         }
+
+        /**
+         * @group 38121
+         */
+        function test_preload() {
+                $expected = "<link rel='dns-prefetch' href='//s.w.org' />\n" .
+                            "<link rel='preload' href='https://make.wordpress.org/great-again' />\n" .
+                            "<link rel='preload' href='http://jobs.wordpress.net' />\n" .
+                            "<link rel='preload' href='//core.trac.wordpress.org' />\n";
+
+                add_filter( 'wp_resource_hints', array( $this, '_add_preload_urls' ), 10, 2 );
+
+                $actual = get_echo( 'wp_resource_hints' );
+
+                remove_filter( 'wp_resource_hints', array( $this, '_add_preload_urls' ) );
+
+                $this->assertEquals( $expected, $actual );
+        }
+
+        function _add_preload_urls( $hints, $method ) {
+                if ( 'preload' === $method ) {
+                        $hints[] = 'https://make.wordpress.org/great-again';
+                        $hints[] = 'http://jobs.wordpress.net';
+                        $hints[] = '//core.trac.wordpress.org';
+                        $hints[] = 'htps://wordpress.org'; // Invalid URLs should be skipped.
+                }
+
+                return $hints;
+        }
+
+        function test_custom_attributes() {
+                $expected = "<link rel='dns-prefetch' href='//s.w.org' />\n" .
+                            "<link rel='preconnect' href='https://make.wordpress.org' />\n" .
+                            "<link pr='0.5' rel='prefetch' href='http://wordpress.org' />\n" .
+                            "<link rel='prerender' href='http://wordpress.org' />\n" .
+                            "<link as='font' type='font/woff2' rel='preload' href='https://example.com/font.woff2' />\n";
+
+                add_filter( 'wp_resource_hints', array( $this, '_add_url_with_attributes' ), 10, 2 );
+
+                $actual = get_echo( 'wp_resource_hints' );
+
+                remove_filter( 'wp_resource_hints', array( $this, '_add_url_with_attributes' ) );
+
+                $this->assertEquals( $expected, $actual );
+        }
+
+        function _add_url_with_attributes( $hints, $method ) {
+
+                if ( 'preconnect' === $method ) {
+                        // Should ignore rel and href attributes.
+                        $hints[] = array(
+                                'https://make.wordpress.org/great-again' => array(
+                                        'rel'  => 'foo',
+                                        'href' => 'bar',
+                                ),
+                        );
+                } elseif ( 'preload' === $method ) {
+                        // Should support typical preload argument
+                        $hints[] = array(
+                                'https://example.com/font.woff2' => array(
+                                        'crossorigin',
+                                        'as'   => 'font',
+                                        'type' => 'font/woff2',
+                                ),
+                        );
+                } elseif ( 'preload' === $method ) {
+                        // Should support typical preload argument
+                        $hints[] = array(
+                                'https://example.com/font.woff2' => array(
+                                        'crossorigin',
+                                        'as'   => 'font',
+                                        'type' => 'font/woff2',
+                                ),
+                        );
+                        $hints[] = array(
+                                'https://example.com/foo.css' => array(
+                                        'crossorigin' => 'use-credentials',
+                                        'as'          => 'style',
+                                ),
+                        );
+                } elseif ( 'prefetch' === $method ) {
+                        $hints[] = array(
+                                'http://wordpress.org' => array(
+                                        'pr' => 0.5,
+                                ),
+                        );
+                } elseif ( 'prerender' === $method ) {
+                        $hints[] = array(
+                                'http://wordpress.org' => array(
+                                        'foo' => 'bar',
+                                        'bar' => 'baz',
+                                ),
+                        );
+                }
+
+                return $hints;
+        }
 }