Ticket #33920: 33920.3.diff

src/wp-includes/query.php

@@ -4725 +4725 @@
  *
  * @since 2.1.0
  *
- * @global WP_Query $wp_query Global WP_Query instance.
- * @global wpdb     $wpdb     WordPress database abstraction object.
+ * @global WP_Query   $wp_query   Global WP_Query instance.
+ * @global wpdb       $wpdb       WordPress database abstraction object.
+ * @global WP_Rewrite $wp_rewrite WordPress rewrite component.
  */
 function wp_old_slug_redirect() {
-        global $wp_query;
-        if ( is_404() && '' != $wp_query->query_vars['name'] ) :
+        global $wp_query, $wp_rewrite;
+
+        if ( '' !== $wp_query->query_vars['name'] ) :
                 global $wpdb;
 
                 // Guess the current post_type based on the query vars.
@@ -4767 +4769 @@
                 if ( ! $id )
                         return;
 
-                $link = get_permalink($id);
+                $link = get_permalink( $id );
+
+                if ( is_404() ) {
+                        // Add rewrite endpoints if necessary.
+                        foreach ( $wp_rewrite->endpoints as $endpoint ) {
+                                if ( $endpoint[2] && false !== get_query_var( $endpoint[2], false ) ) {
+                                        $link = user_trailingslashit( trailingslashit( $link ) . $endpoint[2] );
+                                }
+                        }
+                } elseif ( is_feed() ) {
+                        $link = user_trailingslashit( trailingslashit( $link ) . 'feed' );
+                }
 
-                if ( !$link )
+                /**
+                 * Filter the old slug redirect URL.
+                 *
+                 * @since 4.4.0
+                 *
+                 * @param string $link The redirect URL.
+                 */
+                $link = apply_filters( 'old_slug_redirect_url', $link );
+
+                if ( ! $link ) {
                         return;
+                }
 
                 wp_redirect( $link, 301 ); // Permanent redirect
                 exit;

new file tests/phpunit/tests/rewrite/oldSlugRedirect.php

@@ +1 @@
+<?php
+
+/**
+ * @group rewrite
+ * @ticket 33920
+ */
+class Tests_Rewrite_OldSlugRedirect extends WP_UnitTestCase {
+        protected $old_slug_redirect_url;
+
+        protected $post_id;
+
+        public static function setUpBeforeClass() {
+                WP_UnitTestCase::setUpBeforeClass();
+
+                global $wp_rewrite;
+
+                $wp_rewrite->init();
+                $wp_rewrite->set_permalink_structure( '/%postname%/' );
+                add_rewrite_endpoint( 'custom-endpoint', EP_PERMALINK );
+                $wp_rewrite->flush_rules();
+        }
+
+        public static function tearDownAfterClass() {
+                WP_UnitTestCase::tearDownAfterClass();
+
+                global $wp_rewrite;
+
+                $wp_rewrite->set_permalink_structure( '' );
+                $wp_rewrite->init();
+        }
+
+        public function setUp() {
+                parent::setUp();
+
+                $this->post_id = $this->factory->post->create( array(
+                        'post_title'   => 'Foo Bar',
+                        'post_name'   => 'foo-bar',
+                ) );
+
+                add_filter( 'old_slug_redirect_url', array( $this, 'filter_old_slug_redirect_url' ), 10, 1 );
+        }
+
+        public function tearDown() {
+                parent::tearDown();
+
+                $this->old_slug_redirect_url = null;
+
+                remove_filter( 'old_slug_redirect_url', array( $this, 'filter_old_slug_redirect_url' ), 10 );
+        }
+
+        public function test_old_slug_redirect() {
+                $old_permalink = user_trailingslashit( get_permalink( $this->post_id ) );
+
+                wp_update_post( array(
+                        'ID' => $this->post_id,
+                        'post_name' => 'bar-baz',
+                ) );
+
+                $permalink = user_trailingslashit( get_permalink( $this->post_id ) );
+
+                $this->go_to( $old_permalink );
+                $this->assertQueryTrue( 'is_404' );
+
+                wp_old_slug_redirect();
+                $this->assertEquals( $permalink, $this->old_slug_redirect_url );
+        }
+
+        public function test_old_slug_redirect_endpoint() {
+                $old_permalink = user_trailingslashit( trailingslashit( get_permalink( $this->post_id ) ) . 'custom-endpoint' );
+
+                wp_update_post( array(
+                        'ID' => $this->post_id,
+                        'post_name' => 'bar-baz',
+                ) );
+
+                $permalink = user_trailingslashit( trailingslashit( get_permalink( $this->post_id ) ) . 'custom-endpoint' );
+
+                $this->go_to( $old_permalink );
+                $GLOBALS['wp_query']->query_vars['custom-endpoint'] = true;
+                $this->assertQueryTrue( 'is_404' );
+
+                wp_old_slug_redirect();
+                $this->assertEquals( $permalink, $this->old_slug_redirect_url );
+        }
+
+        public function test_old_slug_redirect_feed() {
+                $old_permalink = user_trailingslashit( trailingslashit( get_permalink( $this->post_id ) ) . 'feed' );
+
+                wp_update_post( array(
+                        'ID' => $this->post_id,
+                        'post_name' => 'bar-baz',
+                ) );
+
+                $permalink = user_trailingslashit( trailingslashit( get_permalink( $this->post_id ) ) . 'feed' );
+
+                $this->go_to( $old_permalink );
+
+                wp_old_slug_redirect();
+                $this->assertEquals( $permalink, $this->old_slug_redirect_url );
+        }
+
+        public function filter_old_slug_redirect_url( $url ) {
+                $this->old_slug_redirect_url = $url;
+                return false;
+        }
+}