/pmpro-bbpress.php

  1. <?php 
  2. /** 
  3.  * Plugin Name: Paid Memberships Pro - bbPress Add On 
  4.  * Plugin URI: http://www.paidmembershipspro.com/pmpro-bbpress/ 
  5.  * Description: Allow individual forums to be locked down for members. 
  6.  * Version: 1.5.2 
  7.  * Author: Stranger Studios, Scott Sousa 
  8.  * Author URI: http://www.strangerstudios.com 
  9.  */ 
  10.  
  11.  //includes 
  12.  define('PMPROBB_DIR', dirname(__FILE__)); 
  13.  require_once(PMPROBB_DIR . '/includes/functions.php'); 
  14.  require_once(PMPROBB_DIR . '/includes/options.php');  
  15.  require_once(PMPROBB_DIR . '/includes/options-membership-levels.php'); 
  16.  require_once(PMPROBB_DIR . '/includes/shortcodes.php');  
  17.   
  18. /** 
  19.  * Admin init 
  20.  */ 
  21. function pmprobb_admin_init() { 
  22.   //if on the edit level page, enqueue color picker 
  23.   if(!empty($_REQUEST['page']) && $_REQUEST['page'] == 'pmpro-membershiplevels' && isset($_REQUEST['edit'])) { 
  24.       wp_enqueue_script( 'wp-color-picker' ); 
  25.       wp_enqueue_style( 'wp-color-picker' ); 
  26.   }     
  27. } 
  28. add_action('admin_init', 'pmprobb_admin_init'); 
  29.  
  30. /** 
  31.  * These next two functions work together to lock down bbPress forums based on PMPro membership level. 
  32.  */     
  33.  
  34. /** 
  35.  * Check that the current user has access to this forum. 
  36.  */ 
  37. add_action( 'template_redirect', 'pmprobbp_check_forum' ); 
  38. function pmprobbp_check_forum() { 
  39.   //make sure pmpro and bbpress are active 
  40.   if ( !defined('PMPRO_VERSION') || !class_exists('bbPress') ) 
  41.       return; 
  42.    
  43.   global $current_user; 
  44.  
  45.   $forum_id = bbp_get_forum_id(); 
  46.    
  47.   // Is this even a forum page at all? 
  48.   if( ! bbp_is_forum_archive() && ! empty( $forum_id ) && pmpro_bbp_is_forum() ) { 
  49.       // The current user does not have access to this forum, re-direct them away 
  50.       if( ! pmpro_has_membership_access( $forum_id ) ) { 
  51.            
  52.           // save to session in case we want to redirect later on 
  53.           $_SESSION['pmpro_bbp_redirected_from'] = $_SERVER['REQUEST_URI']; 
  54.           wp_redirect( add_query_arg( 'noaccess', 1, get_post_type_archive_link( 'forum' ) ) ); 
  55.           exit; 
  56.       } 
  57.   } 
  58. } 
  59.  
  60. /** 
  61.  * Function to tell if the current forum, topic, or reply is a subpost of the forum_id passed. 
  62.  * If no forum_id is passed, it will return true if it is any forum, topic, or reply. 
  63.  */ 
  64. function pmpro_bbp_is_forum( $forum_id = NULL ) { 
  65.   global $post; 
  66.    
  67.   //avoid notices on 404 pages 
  68.   if(!$post) 
  69.       return false; 
  70.        
  71.   if(bbp_is_forum($post->ID)) 
  72.   {         
  73.       if(!empty($forum_id) && $post->ID == $forum_id) 
  74.           return true; 
  75.       elseif(empty($forum_id)) 
  76.           return true; 
  77.       else 
  78.           return false; 
  79.   } 
  80.   elseif(bbp_is_topic($post->ID)) 
  81.   {         
  82.       if(!empty($forum_id) && $post->post_parent == $forum_id) 
  83.           return true; 
  84.       elseif(empty($forum_id)) 
  85.           return true; 
  86.       else 
  87.           return false; 
  88.   } 
  89.   elseif(bbp_is_reply($post->ID)) 
  90.   {         
  91.       if(!empty($forum_id) && in_array($forum_id, $post->ancestors)) 
  92.           return true; 
  93.       elseif(empty($forum_id)) 
  94.           return true; 
  95.       else 
  96.           return false; 
  97.   } 
  98.   else 
  99.       return false; 
  100. } 
  101.  
  102. /** Add membership level required message if user does not have access */ 
  103. function pmpro_bbp_membership_msg()  
  104. { 
  105.   $pmpro_bbp_error_msg = apply_filters('pmpro_bbp_error_msg', 'You do not have the required membership level to access that forum.'); // error message to display 
  106.  
  107.   if (bbp_is_forum_archive() && !empty($_REQUEST['noaccess']))  
  108.   { 
  109.       echo '<p class="pmpro_bbp_membership_msg">' . $pmpro_bbp_error_msg . '</p>'; 
  110.   } 
  111. } 
  112. add_action('bbp_template_before_forums_index', 'pmpro_bbp_membership_msg'); 
  113.  
  114. /** 
  115.  * Add topics and forums to pmpro_search_query 
  116.  */ 
  117. function pmprobb_pre_get_posts($query) { 
  118.  
  119.   global $wpdb; 
  120.        
  121.   //only filter front end searches 
  122.   if(is_admin() || !$query->is_search || bbp_is_single_topic()) 
  123.       return $query; 
  124.    
  125.   //get all member forums 
  126.   $sqlQuery = "SELECT ID FROM $wpdb->posts WHERE post_type LIKE 'forum'"; 
  127.   $all_forums = $wpdb->get_col($sqlQuery); 
  128.    
  129.   //no forums? 
  130.   if(empty($all_forums)) 
  131.       return $query; 
  132.    
  133.   //add restricted forums to array 
  134.   $restricted_forum_ids = array(); 
  135.   foreach($all_forums as $forum_id) { 
  136.       if(!pmpro_has_membership_access($forum_id)) 
  137.           $restricted_forum_ids[] = $forum_id; 
  138.   } 
  139.  
  140.   //if there are restricted forums, find topics and exclude them all from searches 
  141.   if(!empty($restricted_forum_ids)) 
  142.   {     
  143.       //get topics belonging to restricted forums 
  144.       $sqlQuery = "SELECT post_id FROM $wpdb->postmeta WHERE meta_key LIKE '_bbp_forum_id' AND meta_value IN(" . implode(', ', $restricted_forum_ids) . ")"; 
  145.       $restricted_topic_ids = $wpdb->get_col($sqlQuery); 
  146.  
  147.       //exclude restricted topics and posts 
  148.       $query->set('post__not_in', array_merge($query->get('post__not_in'), $restricted_topic_ids, $restricted_forum_ids));         
  149.   } 
  150.    
  151.   return $query; 
  152. } 
  153.  
  154. /** 
  155. Function to add links to the plugin row meta 
  156. */ 
  157. function pmprobb_plugin_row_meta($links, $file) { 
  158.   if(strpos($file, 'pmpro-bbpress.php') !== false) 
  159.   { 
  160.       $new_links = array( 
  161.           '<a href="' . esc_url('http://www.paidmembershipspro.com/add-ons/plugins-wordpress-repository/pmpro-bbpress/')  . '" title="' . esc_attr( __( 'View Documentation', 'pmpro' ) ) . '">' . __( 'Docs', 'pmpro' ) . '</a>',  
  162.           '<a href="' . esc_url('http://paidmembershipspro.com/support/') . '" title="' . esc_attr( __( 'Visit Customer Support Forum', 'pmpro' ) ) . '">' . __( 'Support', 'pmpro' ) . '</a>',  
  163.  ); 
  164.       $links = array_merge($links, $new_links); 
  165.   } 
  166.   return $links; 
  167. } 
  168. add_filter('plugin_row_meta', 'pmprobb_plugin_row_meta', 10, 2); 
  169.  
  170. /** 
  171.   Adds "pmpro-level-ID" to the forum topic replies post class where ID 
  172.   is the membership level of the reply author. Useful for styling 
  173.   forum replies based on membership level. 
  174.    
  175.   Style the replies using this post class in the following format: 
  176.   #bbpress-forums li.bbp-body div.pmpro-level-1 { }  
  177.   #bbpress-forums li.bbp-body div.pmpro-level-2 { } 
  178.   Add this code to your active theme's functions.php or a custom plugin. 
  179. */ 
  180. function pmprobb_pmpro_reply_post_class($classes) { 
  181.   global $reply_id; 
  182.   $reply_id = bbp_get_reply_id( $reply_id ); 
  183.   $reply_author_id = bbp_get_reply_author_id( $reply_id ); 
  184.   $reply_author_membership_level = pmpro_getMembershipLevelForUser($reply_author_id); 
  185.   if(!empty($reply_author_membership_level)) { 
  186.       $classes[] = 'pmpro-level-' . $reply_author_membership_level->id; 
  187.   } 
  188.   return $classes; 
  189. } 
  190. add_filter( 'bbp_get_reply_class', 'pmprobb_pmpro_reply_post_class'); 
  191.  
  192. /** 
  193.   Generates CSS to color member posts. 
  194. */ 
  195. function pmprobb_forum_color_css() { 
  196.   //only on forum pages 
  197.   if(!pmpro_bbp_is_forum()) 
  198.       return; 
  199.    
  200.   //get color options and build rules 
  201.   $options = pmprobb_getOptions();     
  202.   $rule = array(); 
  203.   if(!empty($options['levels'])) { 
  204.       foreach($options['levels'] as $level) { 
  205.           if(!empty($level['color'])) 
  206.               $rules[] = ".topic.pmpro-level-1, .reply.pmpro-level-1 {background-color: " . $level['color'] . " !important; }"; 
  207.       } 
  208.   } 
  209.    
  210.   //no rules? 
  211.   if(empty($rules)) 
  212.       return false; 
  213.    
  214.   //show rules 
  215.   ?> 
  216. <style type="text/css" media="screen"> 
  217.   <?php echo implode("\n", $rules) . "\n";?> 
  218. </style> 
  219.   <?php 
  220. } 
  221. add_action('wp_head', 'pmprobb_forum_color_css'); 
  222.  
  223. /** 
  224.   Add links to the top of the member links 
  225. */ 
  226. function pmprobb_pmpro_member_links_top() { 
  227.   $options = pmprobb_getOptions(); 
  228.   if(empty($options['member_links'])) 
  229.       return; 
  230.    
  231.   $forums = get_posts(array('post_type'=>'forum', 'post_status'=>'publish'));     
  232.   foreach($forums as $forum) { 
  233.       //show in member links?     
  234.       if(pmpro_has_membership_access($forum->ID)) { 
  235.       ?> 
  236.       <li><a href="<?php echo get_permalink($forum->ID);?>"><?php echo $forum->post_title;?></a></li> 
  237.       <?php 
  238.       } 
  239.   } 
  240. } 
  241. add_filter('pmpro_member_links_top', 'pmprobb_pmpro_member_links_top'); 
  242.  
  243. /** 
  244.   Hide forums from list and search results 
  245. */ 
  246. function pmprobb_pmpro_search_filter_post_types($post_types) 
  247. { 
  248.   $options = pmprobb_getOptions(); 
  249.   if(!empty($options['hide_member_forums'])) { 
  250.       $post_types[] = 'forum'; 
  251.       array_unique($post_types);     
  252.   } 
  253.   return $post_types; 
  254. } 
  255. $options = pmprobb_getOptions(); 
  256. if(!empty($options['hide_member_forums'])) { 
  257.   add_filter( 'pre_get_posts', 'pmpro_search_filter' );     
  258.   add_filter( 'pmpro_search_filter_post_types', 'pmprobb_pmpro_search_filter_post_types' ); 
  259. } 
  260.  
  261. /** 
  262.  * Change error message for PMPro bbPress     
  263.  */ 
  264. function pmprobb_pmpro_bbp_error_msg() 
  265. { 
  266.   $options = pmprobb_getOptions(); 
  267.   return $options['error_message']; 
  268. } 
  269. add_filter('pmpro_bbp_error_msg', 'pmprobb_pmpro_bbp_error_msg'); 
  270.  
  271. /** 
  272.   Hide the forum role from the bbPress forums replies author link.     
  273. */     
  274. function pmprobb_pmpro_hide_role($args) { 
  275.   $options = pmprobb_getOptions(); 
  276.   if(!empty($options['hide_forum_roles'])) 
  277.       $args['show_role'] = false; 
  278.   return $args; 
  279. } 
  280. add_filter ('bbp_before_get_reply_author_link_parse_args', 'pmprobb_pmpro_hide_role' ); 
  281.  
  282. /** 
  283.   Adds a Section "Membership Level" and displays the user's level 
  284.   on the bbPress User Profile page.     
  285. */ 
  286. function pmprobb_pmpro_bbp_template_before_user_profile()  
  287. { 
  288.   $options = pmprobb_getOptions(); 
  289.   if(empty($options['show_membership_levels'])) 
  290.       return; 
  291.    
  292.   $profile_user = new stdClass(); 
  293.   $profile_user->membership_level = pmpro_getMembershipLevelForUser(bbp_get_user_id( 0, true, false )); 
  294.   if(!empty($profile_user->membership_level)) 
  295.   { 
  296.       ?> 
  297.       <div id="bbp-user-profile" class="bbp-user-profile"> 
  298.           <h2 class="entry-title"><?php _e('Membership Level', 'pmpro');?></h2> 
  299.           <div class="bbp-user-section"> 
  300.               <?php echo $profile_user->membership_level->name; ?> 
  301.           </div> 
  302.       </div> 
  303.       <?php 
  304.   } 
  305. }; 
  306. add_action( 'bbp_template_before_user_profile', 'pmprobb_pmpro_bbp_template_before_user_profile', 10, 0 ); 
  307.  
  308. /** 
  309.   Display the Membership Level of the reply author  
  310.   in your bbPress forum replies.     
  311. */ 
  312. function pmprobb_pmpro_bbp_theme_after_reply_author_details()  
  313. { 
  314.   $options = pmprobb_getOptions(); 
  315.   if(empty($options['show_membership_levels'])) 
  316.       return; 
  317.        
  318.   $displayed_user = bbp_get_reply_author_id(bbp_get_reply_id()); 
  319.   $membership_level = pmpro_getMembershipLevelForUser($displayed_user); 
  320.   if(!empty($membership_level)) 
  321.   { 
  322.     echo '<br /><div class="bbp-author-role">' . $membership_level->name . '</div>'; 
  323.   } 
  324. } 
  325. add_action('bbp_theme_after_reply_author_details', 'pmprobb_pmpro_bbp_theme_after_reply_author_details', 10, 0); 
.