WordPress.org

Make WordPress Core

Ticket #50910: 50910.2.diff

File 50910.2.diff, 9.4 KB (added by pbiron, 6 months ago)
  • src/wp-includes/canonical.php

    From 845cdc30c6c1b7c3e77f40cfdebb099b85b0a3a1 Mon Sep 17 00:00:00 2001
    From: Paul Biron <paul@sparrowhawkcomputing.com>
    Date: Tue, 25 Aug 2020 15:08:38 -0600
    Subject: [PATCH] nsure sitemap URLs with paged > 1 work as expected.
    
    Introduces the get_sitemap_url() function which is sort of equivalent to get_permalink() but for sitemaps.  That new function is used in redirect_canonical().
    ---
     src/wp-includes/canonical.php              | 15 ++--
     src/wp-includes/sitemaps.php               | 36 +++++++++
     tests/phpunit/tests/canonical/sitemaps.php | 37 +++++++++
     tests/phpunit/tests/sitemaps/functions.php | 88 ++++++++++++++++++++++
     4 files changed, 169 insertions(+), 7 deletions(-)
    
    diff --git a/src/wp-includes/canonical.php b/src/wp-includes/canonical.php
    index 5a63de3481..31f128e033 100644
    a b function redirect_canonical( $requested_url = null, $do_redirect = true ) { 
    410410                        $redirect['query'] = remove_query_arg( 'page', $redirect['query'] );
    411411                }
    412412
    413                 // Paging and feeds.
    414                 if ( get_query_var( 'paged' ) || is_feed() || get_query_var( 'cpage' ) ) {
     413                if ( get_query_var( 'sitemap' ) ) {
     414                        $redirect_url      = get_sitemap_url( get_query_var( 'sitemap' ), get_query_var( 'sitemap-subtype' ), get_query_var( 'paged' ) );
     415                        $redirect['query'] = remove_query_arg( array( 'sitemap', 'sitemap-subtype', 'paged' ), $redirect['query'] );
     416                } elseif ( get_query_var( 'paged' ) || is_feed() || get_query_var( 'cpage' ) ) {
     417                        // Paging and feeds.
    415418                        $paged = get_query_var( 'paged' );
    416419                        $feed  = get_query_var( 'feed' );
    417420                        $cpage = get_query_var( 'cpage' );
    function redirect_canonical( $requested_url = null, $do_redirect = true ) { 
    508511                                $redirect['path'] = trailingslashit( $redirect['path'] ) . $addl_path;
    509512                        }
    510513
    511                         // Remove trailing slash for sitemaps requests.
    512                         if ( ! empty( get_query_var( 'sitemap' ) ) ) {
    513                                 $redirect['path'] = untrailingslashit( $redirect['path'] );
    514                         }
    515 
    516514                        $redirect_url = $redirect['scheme'] . '://' . $redirect['host'] . $redirect['path'];
     515                } elseif ( get_query_var( 'sitemap' ) ) {
     516                        $redirect_url      = get_sitemap_url( get_query_var( 'sitemap' ), get_query_var( 'sitemap-subtype' ), get_query_var( 'paged' ) );
     517                        $redirect['query'] = remove_query_arg( array( 'sitemap', 'sitemap-subtype', 'paged' ), $redirect['query'] );
    517518                }
    518519
    519520                if ( 'wp-register.php' === basename( $redirect['path'] ) ) {
  • src/wp-includes/sitemaps.php

    diff --git a/src/wp-includes/sitemaps.php b/src/wp-includes/sitemaps.php
    index 62752dada9..1dfcd1d69c 100644
    a b function wp_sitemaps_get_max_urls( $object_type ) { 
    8787         */
    8888        return apply_filters( 'wp_sitemaps_max_urls', 2000, $object_type );
    8989}
     90
     91/**
     92 * Retrieves the full URL for a sitemap.
     93 *
     94 * @since 5.5.1
     95 *
     96 * @param string $name         The sitemap name.
     97 * @param string $subtype_name The sitemap subtype name.  Default empty string.
     98 * @param int    $page         The page of the sitemap.  Default 1.
     99 * @return string|false The sitemap URL or false if the sitemap doesn't exist.
     100 */
     101function get_sitemap_url( $name, $subtype_name = '', $page = 1 ) {
     102        $sitemaps = wp_sitemaps_get_server();
     103        if ( ! $sitemaps ) {
     104                return false;
     105        }
     106
     107        if ( 'index' === $name ) {
     108                return $sitemaps->index->get_index_url();
     109        }
     110
     111        $provider = $sitemaps->registry->get_provider( $name );
     112        if ( ! $provider ) {
     113                return false;
     114        }
     115
     116        if ( $subtype_name && ! in_array( $subtype_name, array_keys( $provider->get_object_subtypes() ), true ) ) {
     117                return false;
     118        }
     119
     120        $page = absint( $page );
     121        if ( 0 >= $page ) {
     122                $page = 1;
     123        }
     124        return $provider->get_sitemap_url( $subtype_name, $page );
     125}
  • tests/phpunit/tests/canonical/sitemaps.php

    diff --git a/tests/phpunit/tests/canonical/sitemaps.php b/tests/phpunit/tests/canonical/sitemaps.php
    index dd0bdfbdb6..1e0c86fad4 100644
    a b class Tests_Canonical_Sitemaps extends WP_Canonical_UnitTestCase { 
    4040                $this->assertCanonical( '/wp-sitemap.xsl/', '/wp-sitemap.xsl' );
    4141        }
    4242
     43        /**
     44         * Ensure sitemaps redirects work as expected.
     45         *
     46         * @dataProvider sitemaps_canonical_redirects_provider
     47         * @ticket 50910
     48         */
     49        public function test_sitemaps_canonical_redirects( $test_url, $expected ) {
     50                $this->assertCanonical( $test_url, $expected, 50910 );
     51        }
     52
     53        /**
     54         * Data provider for test_sitemaps_canonical_redirects.
     55         *
     56         * @return array[] {
     57         *     Data to test with.
     58         *
     59         *     @type string $0 The test URL.
     60         *     @type string $1 The expected canonical URL.
     61         * }
     62         */
     63        public function sitemaps_canonical_redirects_provider() {
     64                return array(
     65                        // Ugly/incorrect versions redirect correctly.
     66                        array( '/?sitemap=index', '/wp-sitemap.xml' ),
     67                        array( '/wp-sitemap.xml/', '/wp-sitemap.xml' ),
     68                        array( '/?sitemap=posts&sitemap-subtype=post', '/wp-sitemap-posts-post-1.xml' ),
     69                        array( '/?sitemap=posts&sitemap-subtype=post&paged=2', '/wp-sitemap-posts-post-2.xml' ),
     70                        array( '/?sitemap=taxonomies&sitemap-subtype=category', '/wp-sitemap-taxonomies-category-1.xml' ),
     71                        array( '/?sitemap=taxonomies&sitemap-subtype=category&paged=2', '/wp-sitemap-taxonomies-category-2.xml' ),
     72
     73                        // Pretty versions don't redirect incorrectly.
     74                        array( '/wp-sitemap-posts-post-1.xml', '/wp-sitemap-posts-post-1.xml' ),
     75                        array( '/wp-sitemap-posts-post-2.xml', '/wp-sitemap-posts-post-2.xml' ),
     76                        array( '/wp-sitemap-taxonomies-category-1.xml', '/wp-sitemap-taxonomies-category-1.xml' ),
     77                        array( '/wp-sitemap-taxonomies-category-2.xml', '/wp-sitemap-taxonomies-category-2.xml' ),
     78                );
     79        }
    4380}
  • tests/phpunit/tests/sitemaps/functions.php

    diff --git a/tests/phpunit/tests/sitemaps/functions.php b/tests/phpunit/tests/sitemaps/functions.php
    index 3fe904a6ae..7ab5c5490f 100644
    a b class Test_Sitemaps_Functions extends WP_UnitTestCase { 
    5858                        $this->assertTrue( is_a( $sitemaps[ $name ], $provider ), "Default $name sitemap is not a $provider object." );
    5959                }
    6060        }
     61
     62        /**
     63         * Test get_sitemap_url() with ugly permalinks.
     64         *
     65         * @dataProvider ugly_permalinks_provider
     66         */
     67        public function test_get_sitemap_url_ugly_permalinks( $name, $subtype_name, $page, $expected ) {
     68                $actual = get_sitemap_url( $name, $subtype_name, $page );
     69
     70                $this->assertSame( $expected, $actual );
     71        }
     72
     73        /**
     74         * Test get_sitemap_url() with pretty permalinks.
     75         *
     76         * @dataProvider pretty_permalinks_provider
     77         */
     78        public function test_get_sitemap_url_pretty_permalinks( $name, $subtype_name, $page, $expected ) {
     79                $this->set_permalink_structure( '/%postname%/' );
     80
     81                $actual = get_sitemap_url( $name, $subtype_name, $page );
     82
     83                $this->assertSame( $expected, $actual );
     84        }
     85
     86        /**
     87         * Data provider for test_get_sitemap_url_ugly_permalinks.
     88         *
     89         * @return array[] {
     90         *     Data to test with.
     91         *
     92         *     @type string       $0 Sitemap name.
     93         *     @type string       $1 Sitemap subtype name.
     94         *     @type int          $3 Sitemap page.
     95         *     @type string|false $4 Sitemap URL.
     96         * }
     97         */
     98        function ugly_permalinks_provider() {
     99                return array(
     100                        array( 'posts', 'post', 1, home_url( '/?sitemap=posts&sitemap-subtype=post&paged=1' ) ),
     101                        array( 'posts', 'post', 0, home_url( '/?sitemap=posts&sitemap-subtype=post&paged=1' ) ),
     102                        array( 'posts', 'page', 1, home_url( '/?sitemap=posts&sitemap-subtype=page&paged=1' ) ),
     103                        array( 'posts', 'page', 5, home_url( '/?sitemap=posts&sitemap-subtype=page&paged=5' ) ),
     104                        // post_type doesn't exist.
     105                        array( 'posts', 'foo', 5, false ),
     106                        array( 'taxonomies', 'category', 1, home_url( '/?sitemap=taxonomies&sitemap-subtype=category&paged=1' ) ),
     107                        array( 'taxonomies', 'post_tag', 1, home_url( '/?sitemap=taxonomies&sitemap-subtype=post_tag&paged=1' ) ),
     108                        array( 'taxonomies', 'post_tag', -1, home_url( '/?sitemap=taxonomies&sitemap-subtype=post_tag&paged=1' ) ),
     109                        // negative paged, gets converted to it's absolute value.
     110                        array( 'users', '', 4, home_url( '/?sitemap=users&paged=4' ) ),
     111                        // users provider doesn't allow subtypes.
     112                        array( 'users', 'foo', 4, false ),
     113                        // provider doesn't exist.
     114                        array( 'foo', '', 4, false ),
     115                );
     116        }
     117
     118        /**
     119         * Data provider for test_get_sitemap_url_pretty_permalinks
     120         *
     121         * @return array[] {
     122         *     Data to test with.
     123         *
     124         *     @type string       $0 Sitemap name.
     125         *     @type string       $1 Sitemap subtype name.
     126         *     @type int          $3 Sitemap page.
     127         *     @type string|false $4 Sitemap URL.
     128         * }
     129         */
     130        function pretty_permalinks_provider() {
     131                return array(
     132                        array( 'posts', 'post', 1, home_url( '/wp-sitemap-posts-post-1.xml' ) ),
     133                        array( 'posts', 'post', 0, home_url( '/wp-sitemap-posts-post-1.xml' ) ),
     134                        array( 'posts', 'page', 1, home_url( '/wp-sitemap-posts-page-1.xml' ) ),
     135                        array( 'posts', 'page', 5, home_url( '/wp-sitemap-posts-page-5.xml' ) ),
     136                        // post_type doesn't exist.
     137                        array( 'posts', 'foo', 5, false ),
     138                        array( 'taxonomies', 'category', 1, home_url( '/wp-sitemap-taxonomies-category-1.xml' ) ),
     139                        array( 'taxonomies', 'post_tag', 1, home_url( '/wp-sitemap-taxonomies-post_tag-1.xml' ) ),
     140                        // negative paged, gets converted to it's absolute value.
     141                        array( 'taxonomies', 'post_tag', -1, home_url( '/wp-sitemap-taxonomies-post_tag-1.xml' ) ),
     142                        array( 'users', '', 4, home_url( '/wp-sitemap-users-4.xml' ) ),
     143                        // users provider doesn't allow subtypes.
     144                        array( 'users', 'foo', 4, false ),
     145                        // provider doesn't exist.
     146                        array( 'foo', '', 4, false ),
     147                );
     148        }
    61149}