Ticket #21112: 21112.2.diff

src/wp-includes/post.php

@@ -3665 +3665 @@
         if ( in_array( $post_status, array( 'draft', 'pending', 'auto-draft' ) ) || ( 'inherit' == $post_status && 'revision' == $post_type ) )
                 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

@@ -1016 +1016 @@
                 _unregister_post_type( 'post-type-1' );
                 _unregister_post_type( 'post-type-2' );
         }
+
+        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;
+        }
+
 }