Ticket #8426: 8426.8.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 = 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, '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

@@ -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

@@ -584 +584 @@
         );
 }
 
+/**
+ * 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
+                )
+        );
+}
+
 /** 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 ];
+
+                        $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.
          *

tests/phpunit/testcases/messages/notifications.php

@@ -11 +11 @@
         public function setUp() {
                 parent::setUp();
 
+                $this->reset_user_id = get_current_user_id();
+
                 $this->filter_fired = '';
         }
 
+        public function tearDown() {
+                parent::tearDown();
+
+                $this->set_current_user( $this->reset_user_id );
+        }
+
         /**
          * @group messages_format_notifications
          */
@@ -40 +48 @@
 
         /**
          * @group messages_format_notifications
-         * @group imath
          */
         public function test_friends_format_notifications_bp_messages_single_new_message_notification_nonstring_filter() {
                 // Dummy thread ID
@@ -145 +152 @@
                 $this->assertSame( array(), $n2 );
         }
 
+        /**
+         * @ticket BP8426
+         */
+        public function test_bp_messages_mark_notification_on_mark_thread() {
+                global $wpdb;
+
+                $u1 = self::factory()->user->create();
+                $u2 = self::factory()->user->create();
+                $m1 = self::factory()->message->create_and_get( array(
+                        'sender_id'  => $u1,
+                        'recipients' => array( $u2 ),
+                        'subject'    => 'Foo',
+                ) );
+
+                self::factory()->message->create_many(
+                        9,
+                        array(
+                                'thread_id' => $m1->thread_id,
+                                'sender_id' => $u2,
+                                'recipients' => array( $u1 ),
+                                'subject' => 'Bar',
+                        )
+                );
+
+                $unreadn = wp_list_pluck(
+                        BP_Notifications_Notification::get(
+                                array(
+                                        'user_id'           => $u1,
+                                        'component_name'    => buddypress()->messages->id,
+                                        'component_action'  => 'new_message',
+                                        'is_new'            => 1,
+                                )
+                        ),
+                        'user_id',
+                        'id'
+                );
+
+                $this->set_current_user( $u1 );
+
+                // Mark a thread read.
+                bp_messages_mark_notification_on_mark_thread( $m1->thread_id, $u1, count( $unreadn ) );
+
+                $readn = wp_list_pluck(
+                        BP_Notifications_Notification::get(
+                                array(
+                                        'user_id'           => $u1,
+                                        'component_name'    => buddypress()->messages->id,
+                                        'component_action'  => 'new_message',
+                                        'is_new'            => 0,
+                                )
+                        ),
+                        'user_id',
+                        'id'
+                );
+
+                $this->assertSame( $unreadn, $readn );
+        }
+
         public function notification_filter_callback( $value ) {
                 $this->filter_fired = current_filter();
                 return $value;