Ticket #8426: 8426.9.patch

src/bp-messages/bp-messages-notifications.php

@@ -245 +245 @@
 
         // Get the unread message ids for this thread only.
         $message_ids = array_intersect( $unread_message_ids, wp_list_pluck( $thread_template->thread->messages, 'id' ) );
-
-        // Mark each notification for each PM message as read.
-        foreach ( $message_ids as $message_id ) {
-                bp_notifications_mark_notifications_by_item_id( bp_loggedin_user_id(), (int) $message_id, buddypress()->messages->id, 'new_message' );
-        }
+        bp_notifications_mark_notifications_by_item_ids( bp_loggedin_user_id(), $message_ids, 'messages', 'new_message', false );
 }
 add_action( 'thread_loop_start', 'bp_messages_screen_conversation_mark_notifications', 10 );
 
@@ -261 +257 @@
  * @since 3.0.0
  *
  * @param int $thread_id ID of the thread being marked as read.
+ * @param int $user_id   ID of the user thread will be marked as unread.
+ * @param int $num_rows  The number of affected rows by the "mark read" update query.
  */
-function bp_messages_mark_notification_on_mark_thread( $thread_id ) {
+function bp_messages_mark_notification_on_mark_thread( $thread_id, $user_id = 0, $num_rows = 0 ) {
+        if ( ! $num_rows ) {
+                return;
+        }
+
         $thread_messages = BP_Messages_Thread::get_messages( $thread_id );
+        $message_ids     = wp_list_pluck( $thread_messages, 'id' );
 
-        foreach ( $thread_messages as $thread_message ) {
-                bp_notifications_mark_notifications_by_item_id( bp_loggedin_user_id(), $thread_message->id, buddypress()->messages->id, 'new_message' );
-        }
+        bp_notifications_mark_notifications_by_item_ids( $user_id, $message_ids, buddypress()->messages->id, 'new_message', false );
 }
-add_action( 'messages_thread_mark_as_read', 'bp_messages_mark_notification_on_mark_thread' );
+add_action( 'messages_thread_mark_as_read', 'bp_messages_mark_notification_on_mark_thread', 10, 3 );
 
 /**
  * When a message is deleted, delete corresponding notifications.
@@ -282 +283 @@
 function bp_messages_message_delete_notifications( $thread_id, $message_ids ) {
         // For each recipient, delete notifications corresponding to each message.
         $thread = new BP_Messages_Thread( $thread_id );
+
         foreach ( $thread->get_recipients() as $recipient ) {
-                foreach ( $message_ids as $message_id ) {
-                        if ( ! empty( $recipient->user_id ) ) {
-                                bp_notifications_delete_notifications_by_item_id(
-                                        $recipient->user_id,
-                                        (int) $message_id,
-                                        buddypress()->messages->id,
-                                        'new_message'
-                                );
-                        }
+                if ( true === empty( $recipient->user_id ) ) {
+                        continue;
                 }
+
+                bp_notifications_delete_notifications_by_item_ids( $recipient->user_id, $message_ids, buddypress()->messages->id, 'new_message' );
         }
 }
 add_action( 'bp_messages_thread_after_delete', 'bp_messages_message_delete_notifications', 10, 2 );

src/bp-messages/classes/class-bp-messages-thread.php

@@ -766 +766 @@
                                 bp_loggedin_user_id();
                 }
 
-                $bp     = buddypress();
-                $retval = $wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = 0 WHERE user_id = %d AND thread_id = %d", $user_id, $thread_id ) );
+                $bp       = buddypress();
+                $num_rows = $wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = 0 WHERE user_id = %d AND thread_id = %d", $user_id, $thread_id ) );
 
                 wp_cache_delete( 'thread_recipients_' . $thread_id, 'bp_messages' );
                 wp_cache_delete( $user_id, 'bp_messages_unread_count' );
@@ -777 +777 @@
                  *
                  * @since 2.8.0
                  * @since 9.0.0 Added the `user_id` parameter.
+                 * @since 10.0.0 Added the `$num_rows` parameter.
                  *
                  * @param int $thread_id The message thread ID.
                  * @param int $user_id   The user the thread will be marked as read.
+                 * @param bool|int $num_rows    Number of threads marked as unread or false on error.
                  */
-                do_action( 'messages_thread_mark_as_read', $thread_id, $user_id );
+                do_action( 'messages_thread_mark_as_read', $thread_id, $user_id, $num_rows );
 
-                return $retval;
+                return $num_rows;
         }
 
         /**
@@ -810 +812 @@
                                 bp_loggedin_user_id();
                 }
 
-                $bp     = buddypress();
-                $retval = $wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = 1 WHERE user_id = %d AND thread_id = %d", $user_id, $thread_id ) );
+                $bp       = buddypress();
+                $num_rows = $wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = 1 WHERE user_id = %d AND thread_id = %d", $user_id, $thread_id ) );
 
                 wp_cache_delete( 'thread_recipients_' . $thread_id, 'bp_messages' );
                 wp_cache_delete( $user_id, 'bp_messages_unread_count' );
@@ -821 +823 @@
                  *
                  * @since 2.8.0
                  * @since 9.0.0  Added the `$user_id` parameter.
-                 * @since 10.0.0 Added the `$retval` parameter.
+                 * @since 10.0.0 Added the `$num_rows` parameter.
                  *
                  * @param int      $thread_id The message thread ID.
                  * @param int      $user_id   The user the thread will be marked as unread.
-                 * @param bool|int $retval     =Number of threads marked as unread or false on error.
+                 * @param bool|int $num_rows  Number of threads marked as unread or false on error.
                  */
-                do_action( 'messages_thread_mark_as_unread', $thread_id, $user_id, $retval );
+                do_action( 'messages_thread_mark_as_unread', $thread_id, $user_id, $num_rows );
 
-                return $retval;
+                return $num_rows;
         }
 
         /**

src/bp-notifications/bp-notifications-functions.php

@@ -384 +384 @@
         ) );
 }
 
+/**
+ * Delete notifications by notification ids and user.
+ *
+ * @since 10.0.0
+ *
+ * @param  int       $user_id          ID of the user whose notifications are being deleted.
+ * @param  array     $ids              IDs of the associated notifications.
+ * @param  string    $component_name   Name of the associated component.
+ * @param  string    $component_action Name of the associated action.
+ * @return int|false                   The number of rows updated. False on error.
+ */
+function bp_notifications_delete_notifications_by_ids( $user_id, $ids, $component_name, $component_action ) {
+        return BP_Notifications_Notification::delete_by_id_list(
+                'id',
+                $ids,
+                array(
+                        'user_id'          => $user_id,
+                        'component_name'   => $component_name,
+                        'component_action' => $component_action
+                )
+        );
+}
+
+/**
+ * Delete notifications by item ids and user.
+ *
+ * @since 10.0.0
+ *
+ * @param  int       $user_id          ID of the user whose notifications are being deleted.
+ * @param  array     $item_ids         ID of the associated item.
+ * @param  string    $component_name   Name of the associated component.
+ * @param  string    $component_action Name of the associated action.
+ * @return int|false                   The number of rows updated. False on error.
+ */
+function bp_notifications_delete_notifications_by_item_ids( $user_id, $item_ids, $component_name, $component_action ) {
+        return BP_Notifications_Notification::delete_by_id_list(
+                'item_id',
+                $item_ids,
+                array(
+                        'user_id'          => $user_id,
+                        'component_name'   => $component_name,
+                        'component_action' => $component_action
+                )
+        );
+}
+
 /**
  * Delete all notifications by type.
  *
@@ -584 +630 @@
         );
 }
 
+/**
+ * Mark notifications read/unread by item ids and user.
+ *
+ * @since 10.0.0
+ *
+ * @param  int       $user_id          ID of the user whose notifications are being deleted.
+ * @param  array     $item_ids         ID of the associated item.
+ * @param  string    $component_name   Name of the associated component.
+ * @param  string    $component_action Name of the associated action.
+ * @param  int|false $is_new           0 for read, 1 for unread.
+ * @return int|false                   The number of rows updated. False on error.
+ */
+function bp_notifications_mark_notifications_by_item_ids( $user_id, $item_ids, $component_name, $component_action, $is_new = false ) {
+        return BP_Notifications_Notification::update_id_list(
+                'item_id',
+                $item_ids,
+                array(
+                        'is_new' => $is_new,
+                ),
+                array(
+                        'user_id'          => $user_id,
+                        'component_name'   => $component_name,
+                        'component_action' => $component_action
+                )
+        );
+}
+
+/**
+ * Mark notifications read/unread by item ids.
+ *
+ * @since 10.0.0
+ *
+ * @param  array     $ids              IDs of the associated notification items.
+ * @param  string    $component_name   Name of the associated component.
+ * @param  string    $component_action Name of the associated action.
+ * @param  int|false $is_new           0 for read, 1 for unread.
+ * @return int|false                   The number of rows updated. False on error.
+ */
+function bp_notifications_mark_notifications_by_ids( $ids, $component_name, $component_action, $is_new = false ) {
+        return BP_Notifications_Notification::update_id_list(
+                'id',
+                $ids,
+                array(
+                        'is_new' => $is_new,
+                ),
+                array(
+                        'component_name'   => $component_name,
+                        'component_action' => $component_action
+                )
+        );
+}
+
 /** Helpers *******************************************************************/
 
 /**

src/bp-notifications/classes/class-bp-notifications-notification.php

@@ -910 +910 @@
                 );
         }
 
+        /**
+         * Update notifications using a list of ids/items_ids.
+         *
+         * @since 10.0.0
+         *
+         * @param string $field The name of the db field of the items to update.
+         *                      Possible values are `id` or `item_id`.
+         * @param int[]  $items The list of items to update.
+         * @param array  $data  Array of notification data to update, passed to
+         *                      {@link wpdb::update()}. Accepts any property of a
+         *                      BP_Notification_Notification object.
+         * @param array  $where The WHERE params as passed to wpdb::update().
+         *                      Typically consists of array( 'ID' => $id ) to specify the ID
+         *                      of the item being updated. See {@link wpdb::update()}.
+         * @return int|false    The number of rows updated. False on error.
+         */
+        public static function update_id_list( $field, $items = array(), $data = array(), $where = array() ) {
+                global $wpdb;
+                $bp = buddypress();
+
+                $supported_fields = array( 'id', 'item_id' );
+
+                if ( false === in_array( $field, $supported_fields, true ) ) {
+                        return false;
+                }
+
+                if ( ! is_array( $items ) || ! is_array( $data ) || ! is_array( $where ) ) {
+                        return false;
+                }
+
+                $data  = self::get_query_clauses( $data );
+                $where = self::get_query_clauses( $where );
+
+                $fields     = array();
+                $conditions = array();
+                $values     = array();
+
+                $_items       = implode( ',', wp_parse_id_list( $items ) );
+                $conditions[] = "{$field} IN ({$_items})";
+
+                foreach ( $data['data'] as $field => $value ) {
+                        $index  = array_search( $field, array_keys( $data['data'] ) );
+                        $format = $data['format'][ $index ];
+
+                        $conditions[] = "{$field} = {$format}";
+                        $values[]     = $value;
+                }
+
+                foreach ( $where['data'] as $field => $value ) {
+                        $index  = array_search( $field, array_keys( $where['data'] ) );
+                        $format = $where['format'][ $index ];
+
+                        $conditions[] = "{$field} = {$format}";
+                        $values[]     = $value;
+                }
+
+                $fields     = implode( ', ', $fields );
+                $conditions = implode( ' AND ', $conditions );
+
+                $sql = "UPDATE {$bp->notifications->table_name} SET {$fields} WHERE {$conditions}";
+
+                return $wpdb->query( $wpdb->prepare( $sql, $values ) );
+        }
+
         /**
          * Delete notifications.
          *
@@ -940 +1004 @@
                 return self::_delete( $where['data'], $where['format'] );
         }
 
+        /**
+         * Delete notifications using a list of ids/items_ids.
+         *
+         * @since 10.0.0
+         *
+         * @param string $field The name of the db field of the items to update.
+         *                      Possible values are `id` or `item_id`.
+         * @param int[]  $items The list of items to update.
+         * @param array  $where The WHERE params as passed to wpdb::update().
+         *                      Typically consists of array( 'ID' => $id ) to specify the ID
+         *                      of the item being updated. See {@link wpdb::update()}.
+         * @return int|false    The number of rows updated. False on error.
+         */
+        public static function delete_by_id_list( $field, $items = array(), $where = array() ) {
+                global $wpdb;
+                $bp = buddypress();
+
+                $supported_fields = array( 'id', 'item_id' );
+
+                if ( false === in_array( $field, $supported_fields, true ) ) {
+                        return false;
+                }
+
+                if ( ! is_array( $items ) || ! is_array( $where ) ) {
+                        return false;
+                }
+
+                $where = self::get_query_clauses( $where );
+
+                $conditions = array();
+                $values     = array();
+
+                $_items       = implode( ',', wp_parse_id_list( $items ) );
+                $conditions[] = "{$field} IN ({$_items})";
+
+                foreach ( $where['data'] as $field => $value ) {
+                        $index  = array_search( $field, array_keys( $where['data'] ) );
+                        $format = $where['format'][ $index ];
+
+                        $conditions[] = "{$field} = {$format}";
+                        $values[]     = $value;
+                }
+
+                $conditions = implode( ' AND ', $conditions );
+
+                $sql = "DELETE FROM {$bp->notifications->table_name} WHERE {$conditions}";
+
+                return $wpdb->query( $wpdb->prepare( $sql, $values ) );
+        }
+
         /** Convenience methods ***************************************************/
 
         /**