Ticket #21112: 21112.3.diff

src/wp-includes/post.php

@@ -3999 +3999 @@
                 return $slug;
         }
 
+        /**
+         * Filter to allow setting the unique post slug before it is generated.
+         *
+         * Returning a non-empty value will preempt unique slug detection and
+         * generation and will result in that value being used as the post's slug.
+         *
+         * @since 4.1
+         *
+         * @param string $default_slug  The default post slug. Default empty.
+         * @param string $slug          The desired slug (post_name).
+         * @param int    $post_ID       Post ID.
+         * @param string $post_status   The post status.
+         * @param string $post_type     Post type.
+         * @param int    $post_parent   Post parent ID
+         */
+        $override_slug = apply_filters( 'pre_wp_unique_post_slug', '', $slug, $post_ID, $post_status, $post_type, $post_parent );
+        if ( $override_slug ) {
+                return $override_slug;
+        }
+
         global $wpdb, $wp_rewrite;
 
         $original_slug = $slug;

tests/phpunit/tests/post.php

@@ -1354 +1354 @@
                 $this->assertEquals( $changeset_data, json_decode( get_post( $post_id )->post_content, true ) );
         }
 
+        /**
+         * Test ensuring that the post_slug can be filtered with a custom value short circuiting the built in
+         * function that tries to create a unique name based on the post name.
+         *
+         * @see wp_unique_post_slug()
+         * @ticket 21112
+         */
+        function test_pre_wp_unique_post_slug_filter() {
+                add_filter( 'pre_wp_unique_post_slug', array( $this, 'filter_pre_wp_unique_post_slug' ), 10, 6 );
+
+                $post_id = $this->factory->post->create( array( 'title' => 'An example', 'post_status' => 'publish', 'post_type' => 'page' ) );
+                $post = get_post( $post_id );
+                $this->assertEquals( 'override-slug-' . $post->post_type, $post->post_name );
+
+                remove_filter( 'pre_wp_unique_post_slug', array( $this, 'filter_pre_wp_unique_post_slug' ), 10, 6 );
+        }
+
+        function filter_pre_wp_unique_post_slug( $default, $slug, $post_ID, $post_status, $post_type, $post_parent ) {
+                return 'override-slug-' . $post_type;
+        }
 }