Changeset 20739 for trunk/wp-admin/includes/user.php

Timestamp:

05/08/2012 05:01:50 PM (13 years ago)

Author:

ryan

Message:

Don't delete nav menu items when the user that owns them is deleted.

• Introduce delete_with_user flag to register_post_type
• Set delete_with_user to false for the nav_menu_item post type
• Set it to true for all other core post types
• If delete_with_user is not set, fallback to post_type_supports('author')

Props nacin
Fixes #16358

File:

• trunk/wp-admin/includes/user.php (modified) (1 diff)

trunk/wp-admin/includes/user.php

@@ -244 +244 @@
 
     if ( 'novalue' === $reassign || null === $reassign ) {
-        $post_ids = $wpdb->get_col( $wpdb->prepare("SELECT ID FROM $wpdb->posts WHERE post_author = %d", $id) );
-
+        $post_types_to_delete = array();
+        foreach ( get_post_types( array(), 'objects' ) as $post_type ) {
+            if ( $post_type->delete_with_user ) {
+                $post_types_to_delete[] = $post_type->name;
+            } elseif ( null === $post_type->delete_with_user && post_type_supports( $post_type->name, 'author' ) ) {
+                $post_types_to_delete[] = $post_type->name;
+            }
+        }
+
+        $post_types_to_delete = apply_filters( 'post_types_to_delete_with_user', $post_types_to_delete, $id );
+        $post_types_to_delete = implode( "', '", $post_types_to_delete );
+        $post_ids = $wpdb->get_col( $wpdb->prepare( "SELECT ID FROM $wpdb->posts WHERE post_author = %d AND post_type IN ('$post_types_to_delete')", $id ) );
         if ( $post_ids ) {
             foreach ( $post_ids as $post_id )
-                wp_delete_post($post_id);
+                wp_delete_post( $post_id );
         }