Ticket #8426: 8426.7.patch

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

@@ -261 +261 @@
  * @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, $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, 'messages', '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.

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

@@ -763 +763 @@
                                 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' );
@@ -774 +774 @@
                  *
                  * @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;
         }
 
         /**
@@ -807 +809 @@
                                 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' );
@@ -818 +820 @@
                  *
                  * @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

@@ -580 +580 @@
         );
 }
 
+/**
+ * Mark notifications read/unread by item ids and user.
+ *
+ * @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|bool  $is_new  0 for read, 1 for unread.
+ *
+ * @return int|false True on success, false on failure.
+ */
+function bp_notifications_mark_notifications_by_item_ids( $user_id, $item_ids, $component_name, $component_action, $is_new = false ) {
+        return BP_Notifications_Notification::update_many(
+                'item_id',
+                $item_ids,
+                array(
+                        'is_new' => $is_new,
+                ),
+                array(
+                        'user_id'          => $user_id,
+                        'component_name'   => $component_name,
+                        'component_action' => $component_action
+                )
+        );
+}
+
 /** Helpers *******************************************************************/
 
 /**

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

@@ -907 +907 @@
                 );
         }
 
+        /**
+         * Update notifications.
+         *
+         * @since 10.0.0
+         *
+         * @param  string $field
+         * @param  array  $items
+         * @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, or false on error.
+         */
+        public static function update_many( $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 ];
+
+                        $fields[] = "`$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.
          *