Ticket #9726: 9726.2.diff

wp-includes/post.php

@@ -1736 +1736 @@
  * @param integer $post_parent
  * @return string unique slug for the post, based on $post_name (with a -1, -2, etc. suffix)
  */
-function wp_unique_post_slug($slug, $post_ID, $post_status, $post_type, $post_parent) {
+function wp_unique_post_slug($slug, $post_ID, $post_status, $post_type = '', $post_parent = '') {
         global $wpdb, $wp_rewrite;
-        if ( !in_array( $post_status, array( 'draft', 'pending' ) ) ) {
-                $hierarchical_post_types = apply_filters('hierarchical_post_types', array('page', 'attachment'));
-                if ( in_array($post_type, $hierarchical_post_types) ) {
-                        $check_sql = "SELECT post_name FROM $wpdb->posts WHERE post_name = %s AND post_type IN ( '" . implode("', '", $wpdb->escape($hierarchical_post_types)) . "' ) AND ID != %d AND post_parent = %d LIMIT 1";
-                        $post_name_check = $wpdb->get_var($wpdb->prepare($check_sql, $slug, $post_ID, $post_parent));
-                } else {
-                        $check_sql = "SELECT post_name FROM $wpdb->posts WHERE post_name = %s AND post_type = %s AND ID != %d AND post_parent = %d LIMIT 1";
-                        $post_name_check = $wpdb->get_var($wpdb->prepare($check_sql, $slug, $post_type, $post_ID, $post_parent));
-                }
 
-                if ( $post_name_check || in_array($slug, $wp_rewrite->feeds) ) {
-                        $suffix = 2;
-                        do {
-                                $alt_post_name = substr($slug, 0, 200-(strlen($suffix)+1)). "-$suffix";
-                                $post_name_check = $wpdb->get_var($wpdb->prepare($check_sql, $alt_post_name, $post_type, $post_ID, $post_parent));
-                                $suffix++;
-                        } while ($post_name_check);
-                        $slug = $alt_post_name;
-                }
+        if ( in_array( $post_status, array( 'draft', 'pending' ) ) )
+                return $slug;
+        
+        $check_sql = "SELECT post_name FROM $wpdb->posts WHERE post_name = %s AND post_type != 'revision' AND post_status != 'draft' AND ID != %d LIMIT 1";
+        $post_name_check = $wpdb->get_var($wpdb->prepare($check_sql, $slug, $post_ID));
+
+        if ( $post_name_check || in_array($slug, $wp_rewrite->feeds) ) {
+                $suffix = 2;
+                do {
+                        $alt_post_name = substr($slug, 0, 200-(strlen($suffix)+1)). "-$suffix";
+                        $post_name_check = $wpdb->get_var($wpdb->prepare($check_sql, $alt_post_name, $post_ID));
+                        $suffix++;
+                } while ($post_name_check);
+                $slug = $alt_post_name;
         }
+
         return $slug;
 }
 
@@ -2452 +2449 @@
         else
                 $post_name = sanitize_title($post_name);
 
-        // expected_slashed ($post_name)
-        $post_name_check = $wpdb->get_var( $wpdb->prepare( "SELECT post_name FROM $wpdb->posts WHERE post_name = %s AND post_status = 'inherit' AND ID != %d LIMIT 1", $post_name, $post_ID ) );
+        $post_name = wp_unique_post_slug($post_name, $post_ID, $post_status, $post_type, $post_parent);
 
-        if ( $post_name_check ) {
-                $suffix = 2;
-                while ( $post_name_check ) {
-                        $alt_post_name = $post_name . "-$suffix";
-                        // expected_slashed ($alt_post_name, $post_name)
-                        $post_name_check = $wpdb->get_var( $wpdb->prepare( "SELECT post_name FROM $wpdb->posts WHERE post_name = %s AND post_status = 'inherit' AND ID != %d LIMIT 1", $alt_post_name, $post_ID ) );
-                        $suffix++;
-                }
-                $post_name = $alt_post_name;
-        }
-
         if ( empty($post_date) )
                 $post_date = current_time('mysql');
         if ( empty($post_date_gmt) )
@@ -3776 +3761 @@
 
                 add_filter('the_preview', '_set_preview');
         }
-}
+}
+ No newline at end of file