Ticket #8688: 8688.patch

src/bp-groups/classes/class-bp-group-member-query.php

@@ -49 +49 @@
          */
         protected $group_member_ids;
 
+        /**
+         * Constructor.
+         *
+         * @since 10.3.0
+         *
+         * @param string|array|null $query See {@link BP_User_Query}.
+         */
+        public function __construct( $query = null ) {
+                $qv = bp_parse_args(
+                        $query,
+                        array(
+                                'count' => false, // True to perform a count query. False otherwise.
+                        )
+                );
+
+                parent::__construct( $qv );
+        }
+
         /**
          * Set up action hooks.
          *
@@ -62 +80 @@
                         $this->query_vars_raw['type'] = 'last_joined';
                 }
 
-                // Set the sort order.
-                add_action( 'bp_pre_user_query', array( $this, 'set_orderby' ) );
+                if ( ! $this->query_vars_raw['count'] ) {
+                        // Set the sort order.
+                        add_action( 'bp_pre_user_query', array( $this, 'set_orderby' ) );
 
-                // Set up our populate_extras method.
-                add_action( 'bp_user_query_populate_extras', array( $this, 'populate_group_member_extras' ), 10, 2 );
+                        // Set up our populate_extras method.
+                        add_action( 'bp_user_query_populate_extras', array( $this, 'populate_group_member_extras' ), 10, 2 );
+                } else {
+                        $this->query_vars_raw['orderby'] = 'ID';
+                }
+        }
+
+        /**
+         * Use WP_User_Query() to pull data for the user IDs retrieved in the main query.
+         *
+         * If a `count` query is performed, the function is used to validate existing users.
+         *
+         * @since 10.3.0
+         */
+        public function do_wp_user_query() {
+                if ( ! $this->query_vars_raw['count'] ) {
+                        return parent::do_wp_user_query();
+                }
+
+                /**
+                 * Filters the WP User Query arguments before passing into the class.
+                 *
+                 * @since 10.3.0
+                 *
+                 * @param array         $value Array of arguments for the user query.
+                 * @param BP_User_Query $this  Current BP_User_Query instance.
+                 */
+                $wp_user_query = new WP_User_Query(
+                        apply_filters(
+                                'bp_group_members_count_query_args',
+                                array(
+                                        // Relevant.
+                                        'fields'      => 'ID',
+                                        'include'     => $this->user_ids,
+
+                                        // Overrides
+                                        'blog_id'     => 0,    // BP does not require blog roles.
+                                        'count_total' => false // We already have a count.
+
+                                ),
+                                $this
+                        )
+                );
+
+                // Validate existing user IDs.
+                $this->user_ids = array_map( 'intval', $wp_user_query->results );
+                $this->results  = $this->user_ids;
+
+                // Set the total existing users.
+                $this->total_users = count( $this->user_ids );
         }
 
         /**
@@ -473 +540 @@
 
                 return wp_list_pluck( $group_user_ids, 'user_id' );
         }
+
+        /**
+         * Perform a database query to populate any extra metadata we might need.
+         *
+         * If a `count` query is performed, the function is used to validate active users.
+         *
+         * @since 10.3.0
+         */
+        public function populate_extras() {
+                if ( ! $this->query_vars_raw['count'] ) {
+                        return parent::populate_extras();
+                }
+
+                // Validate active users.
+                $active_users    = array_filter( BP_Core_User::get_last_activity( $this->user_ids ) );
+                $active_user_ids = array_keys( $active_users );
+                $this->results   = array_intersect( $this->user_ids, $active_user_ids );
+
+                // Set the total active users.
+                $this->total_users = count( $this->results );
+        }
 }

src/bp-groups/classes/class-bp-groups-group.php

@@ -1789 +1789 @@
          * @return int Count of confirmed members for the group.
          */
         public static function get_total_member_count( $group_id, $skip_cache = false ) {
-                $cache_key = 'total_member_count';
-                $count     = groups_get_groupmeta( $group_id, $cache_key );
+                $meta_key = 'total_member_count';
+                $count    = groups_get_groupmeta( $group_id, $meta_key );
 
                 if ( false === $count || true === $skip_cache ) {
-                        $members = groups_get_group_members(
+                        $group_members = new BP_Group_Member_Query(
                                 array(
                                         'group_id'   => $group_id,
                                         'group_role' => array( 'member', 'admin', 'mod' ),
-                                        'type'       => 'active',
+                                        'count'      => true,
                                 )
                         );
 
-                        $count = $members['count'] ? $members['count'] : 0;
-
-                        groups_update_groupmeta( $group_id, $cache_key, (int) $count );
+                        $count = $group_members->total_users;
+                        groups_update_groupmeta( $group_id, $meta_key, $count );
                 }
 
                 /**
@@ -1814 +1813 @@
                  * @param int $count    Total member count for group.
                  * @param int $group_id The ID of the group.
                  */
-                return (int) apply_filters( 'bp_groups_total_member_count', (int) $count, (int) $group_id );
+                return (int) apply_filters( 'bp_groups_total_member_count', $count, (int) $group_id );
         }
 
         /**

tests/phpunit/testcases/groups/functions.php

@@ -344 +344 @@
                 $this->assertEquals( 2, BP_Groups_Group::get_total_member_count( $g1 ) );
         }
 
+        /**
+         * @group total_member_count
+         */
+        public function test_total_member_count_groups_inactive_user() {
+                $u1 = self::factory()->user->create();
+                $u2 = wp_insert_user( array(
+                        'user_pass'  => 'foobar',
+                        'user_login' => 'foobar',
+                        'user_email' => 'foobar@buddypress.org',
+                ) );
+
+                $g1 = self::factory()->group->create( array( 'creator_id' => $u1 ) );
+
+                groups_join_group( $g1, $u2 );
+
+                $this->assertEquals( 1, groups_get_total_member_count( $g1 ) );
+        }
+
+        /**
+         * @group total_member_count
+         */
+        public function test_total_member_count_groups_spammed_user() {
+                $u1 = self::factory()->user->create();
+                $u2 = self::factory()->user->create();
+
+                $g1 = self::factory()->group->create( array( 'creator_id' => $u1 ) );
+
+                groups_join_group( $g1, $u2 );
+                bp_core_process_spammer_status( $u2, 'spam' );
+
+                $this->assertEquals( 1, groups_get_total_member_count( $g1 ) );
+        }
+
         /**
          * @group total_member_count
          * @group groups_create_group