-
Notifications
You must be signed in to change notification settings - Fork 22
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
148 additions
and
109 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,97 +1,102 @@ | ||
| <?php | ||
| /* | ||
| Code to edit the BuddyPress members directory and search. | ||
| */ | ||
|
|
||
| function pmpro_bp_directory_init() { | ||
| // Don't do this if PMPro is deactivated | ||
| if( !defined( 'PMPRO_VERSION' ) ) { | ||
| return; | ||
| } | ||
|
|
||
| global $pmpro_bp_members_in_directory; | ||
| $pmpro_bp_members_in_directory = pmpro_bp_get_members_in_directory(); | ||
| add_action( 'bp_pre_user_query_construct', 'pmpro_bp_bp_pre_user_query_construct', 1, 1 ); | ||
| add_filter( 'bp_get_total_member_count', 'pmpro_bp_bp_get_total_member_count' ); | ||
| } | ||
| add_action('init', 'pmpro_bp_directory_init', 20); | ||
|
|
||
| function pmpro_bp_bp_pre_user_query_construct( $query_array ) { | ||
| // If no setting is locking the member directory, let's bail. | ||
| if( !pmpro_bp_is_member_directory_locked() ) { | ||
| return; | ||
| } | ||
|
|
||
| // Only apply this to the directory. | ||
| if ( 'members' != bp_current_component() ) { | ||
| return; | ||
| } | ||
|
|
||
| global $pmpro_bp_members_in_directory; | ||
| if( !empty( $pmpro_bp_members_in_directory ) ) { | ||
| // If an include value was already set, make sure it's in array form. | ||
| if( !empty( $query_array->query_vars['include'] ) && !is_array( $query_array->query_vars['include']) ) { | ||
| $query_array->query_vars['include'] = explode( ',', $query_array->query_vars['include'] ); | ||
| } | ||
|
|
||
| if( is_array( $query_array->query_vars['include'] ) ) { | ||
| // Compute the intersect of members and include value. | ||
| $query_array->query_vars['include'] = array_intersect( $query_array->query_vars['include'], $pmpro_bp_members_in_directory ); | ||
| } else { | ||
| // Only include members in the directory. | ||
| $query_array->query_vars['include'] = $pmpro_bp_members_in_directory; | ||
| } | ||
| } else { | ||
| // No members, block the directory. | ||
| $query_array->query_vars['include'] = array(0); | ||
| } | ||
| } | ||
|
|
||
| function pmpro_bp_bp_get_total_member_count($count) { | ||
| global $pmpro_bp_members_in_directory; | ||
|
|
||
| $count = count($pmpro_bp_members_in_directory); | ||
| return $count; | ||
| } | ||
|
|
||
| function pmpro_bp_get_members_in_directory() { | ||
| global $wpdb, $pmpro_levels; | ||
|
|
||
| if( !function_exists( 'pmpro_getAllLevels' ) ) { | ||
| return array(); | ||
| } | ||
|
|
||
| $pmpro_levels = pmpro_getAllLevels(true, true); | ||
|
|
||
| if ( empty( $pmpro_levels ) ) { | ||
| return array(); | ||
| } | ||
|
|
||
| //see if we should include them in the member directory. | ||
| $include_levels = array(); | ||
|
|
||
| foreach($pmpro_levels as $level) { | ||
| $pmpro_bp_options = pmpro_bp_get_level_options( $level->id ); | ||
|
|
||
| if( $pmpro_bp_options['pmpro_bp_member_directory'] == 1 || $pmpro_bp_options['pmpro_bp_restrictions'] == PMPROBP_GIVE_ALL_ACCESS) { | ||
| $include_levels[] = $level->id; | ||
| } | ||
| } | ||
|
|
||
| $sql = "SELECT DISTINCT user_id FROM $wpdb->pmpro_memberships_users WHERE membership_id IN (" . implode(",", array_map("intval", $include_levels)) . ") AND status = 'active'"; | ||
|
|
||
| $wpdb->flush(); | ||
| $include_users = $wpdb->get_col($sql); | ||
|
|
||
| return $include_users; | ||
| } | ||
|
|
||
| function pmpro_bp_is_member_directory_locked() { | ||
| $non_user_options = pmpro_bp_get_level_options( 0 ); | ||
|
|
||
| if( !$non_user_options['pmpro_bp_member_directory'] ) { | ||
| return true; | ||
| } else { | ||
| return false; | ||
| } | ||
| } | ||
| <?php | ||
| /* | ||
| Code to edit the BuddyPress members directory and search. | ||
| */ | ||
|
|
||
| function pmpro_bp_directory_init() { | ||
| // Don't do this if PMPro is deactivated | ||
| if( !defined( 'PMPRO_VERSION' ) ) { | ||
| return; | ||
| } | ||
|
|
||
| global $pmpro_bp_members_in_directory; | ||
| $pmpro_bp_members_in_directory = pmpro_bp_get_members_in_directory(); | ||
| add_action( 'bp_pre_user_query_construct', 'pmpro_bp_bp_pre_user_query_construct', 1, 1 ); | ||
| add_filter( 'bp_get_total_member_count', 'pmpro_bp_bp_get_total_member_count' ); | ||
| } | ||
| add_action('init', 'pmpro_bp_directory_init', 20); | ||
|
|
||
| function pmpro_bp_bp_pre_user_query_construct( $query_array ) { | ||
| // If no setting is locking the member directory, let's bail. | ||
| if( !pmpro_bp_is_member_directory_locked() ) { | ||
| return; | ||
| } | ||
|
|
||
| // Only apply this to the directory. | ||
| if ( 'members' != bp_current_component() ) { | ||
| return; | ||
| } | ||
|
|
||
| global $pmpro_bp_members_in_directory; | ||
| if( !empty( $pmpro_bp_members_in_directory ) ) { | ||
| // If an include value was already set, make sure it's in array form. | ||
| if( !empty( $query_array->query_vars['include'] ) && !is_array( $query_array->query_vars['include']) ) { | ||
| $query_array->query_vars['include'] = explode( ',', $query_array->query_vars['include'] ); | ||
| } | ||
|
|
||
| if( is_array( $query_array->query_vars['include'] ) ) { | ||
| // Compute the intersect of members and include value. | ||
| $query_array->query_vars['include'] = array_intersect( $query_array->query_vars['include'], $pmpro_bp_members_in_directory ); | ||
| } else { | ||
| // Only include members in the directory. | ||
| $query_array->query_vars['include'] = $pmpro_bp_members_in_directory; | ||
| } | ||
| } else { | ||
| // No members, block the directory. | ||
| $query_array->query_vars['include'] = array(0); | ||
| } | ||
| } | ||
|
|
||
| function pmpro_bp_bp_get_total_member_count($count) { | ||
| global $pmpro_bp_members_in_directory; | ||
|
|
||
| $count = count($pmpro_bp_members_in_directory); | ||
| return $count; | ||
| } | ||
|
|
||
| function pmpro_bp_get_members_in_directory() { | ||
| global $wpdb, $pmpro_levels; | ||
|
|
||
| if( !function_exists( 'pmpro_getAllLevels' ) ) { | ||
| return array(); | ||
| } | ||
|
|
||
| $pmpro_levels = pmpro_getAllLevels(true, true); | ||
|
|
||
| if ( empty( $pmpro_levels ) ) { | ||
| return array(); | ||
| } | ||
|
|
||
| //see if we should include them in the member directory. | ||
| $include_levels = array(); | ||
|
|
||
| foreach($pmpro_levels as $level) { | ||
| $pmpro_bp_options = pmpro_bp_get_level_options( $level->id ); | ||
|
|
||
| if( $pmpro_bp_options['pmpro_bp_member_directory'] == 1 || $pmpro_bp_options['pmpro_bp_restrictions'] == PMPROBP_GIVE_ALL_ACCESS) { | ||
| $include_levels[] = $level->id; | ||
| } | ||
| } | ||
|
|
||
| if ( empty( $include_levels ) ) { | ||
| return array(); | ||
| } | ||
|
|
||
| $sql = "SELECT DISTINCT user_id FROM $wpdb->pmpro_memberships_users WHERE membership_id IN (" . implode(",", array_map("intval", $include_levels)) . ") AND status = 'active'"; | ||
|
|
||
| $wpdb->flush(); | ||
| $include_users = $wpdb->get_col($sql); | ||
|
|
||
| return $include_users; | ||
| } | ||
|
|
||
| function pmpro_bp_is_member_directory_locked() { | ||
| $non_user_options = pmpro_bp_get_level_options( 0 ); | ||
|
|
||
| if( $non_user_options['pmpro_bp_restrictions'] != PMPROBP_GIVE_ALL_ACCESS | ||
| && ! $non_user_options['pmpro_bp_member_directory'] ) { | ||
| return true; | ||
| } else { | ||
| return false; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters