Checker.php
1 year ago
PostContent.php
9 months ago
Redirect.php
1 year ago
RestrictionShortcode.php
3 months ago
SearchAndAPI.php
1 year ago
index.php
5 years ago
restricted-template.php
1 year ago
RestrictionShortcode.php
105 lines
| 1 | <?php |
| 2 | |
| 3 | namespace ProfilePress\Core\ContentProtection\Frontend; |
| 4 | |
| 5 | use ProfilePress\Core\Membership\Models\Customer\CustomerFactory; |
| 6 | |
| 7 | class RestrictionShortcode |
| 8 | { |
| 9 | public function __construct() |
| 10 | { |
| 11 | add_shortcode('pp-restrict-content', [$this, 'shortcode_handler']); |
| 12 | } |
| 13 | |
| 14 | public function shortcode_handler($atts, $content = null) |
| 15 | { |
| 16 | $atts = shortcode_atts(array( |
| 17 | 'roles' => '', |
| 18 | 'users' => '', |
| 19 | 'plans' => '', |
| 20 | 'status' => '', |
| 21 | 'action' => 'show' // value can be "show" or "hide" |
| 22 | ), $atts); |
| 23 | |
| 24 | if ($atts['action'] == 'hide') { |
| 25 | return $this->rule_matches($atts['roles'], $atts['users'], $atts['plans'], $atts['status']) ? '' : \do_shortcode($content); |
| 26 | } else { |
| 27 | return $this->rule_matches($atts['roles'], $atts['users'], $atts['plans'], $atts['status']) ? \do_shortcode($content) : ''; |
| 28 | } |
| 29 | } |
| 30 | |
| 31 | public function rule_matches($roles = '', $user_ids = '', $plans = '', $sub_status = false) |
| 32 | { |
| 33 | if (is_user_logged_in()) { |
| 34 | |
| 35 | $current_user_id = get_current_user_id(); |
| 36 | |
| 37 | if ( ! empty($plans)) { |
| 38 | |
| 39 | $plans = array_map('absint', explode(',', $plans)); |
| 40 | |
| 41 | $customer = CustomerFactory::fromUserId($current_user_id); |
| 42 | |
| 43 | foreach ($plans as $plan_id) { |
| 44 | |
| 45 | if ( ! empty($sub_status)) { |
| 46 | |
| 47 | $latestOnly = apply_filters('ppress_restriction_shortcode_rule_latest_only', false); |
| 48 | |
| 49 | if ($latestOnly) { |
| 50 | $subs = $customer->get_subscriptions([], ['number' => 1]); |
| 51 | if ( ! empty($subs) && $subs[0]->status == $sub_status) return true; |
| 52 | } else { |
| 53 | if ($customer->has_any_status_subscription($plan_id, [$sub_status])) return true; |
| 54 | } |
| 55 | } else { |
| 56 | if ($customer->has_active_subscription($plan_id)) return true; |
| 57 | } |
| 58 | } |
| 59 | } |
| 60 | |
| 61 | if ( ! empty($roles)) { |
| 62 | |
| 63 | $roles = array_map('sanitize_text_field', explode(',', $roles)); |
| 64 | |
| 65 | $user_roles = wp_get_current_user()->roles; |
| 66 | |
| 67 | if ( ! empty(array_intersect($roles, $user_roles))) return true; |
| 68 | } |
| 69 | |
| 70 | if ( ! empty($user_ids)) { |
| 71 | |
| 72 | $user_ids = array_map('sanitize_text_field', explode(',', $user_ids)); |
| 73 | |
| 74 | foreach ($user_ids as $user_id) { |
| 75 | |
| 76 | if (is_numeric($user_id) && $current_user_id == absint($user_id)) { |
| 77 | return true; |
| 78 | } else { |
| 79 | $user = get_user_by('login', $user_id); |
| 80 | if ( ! $user) { |
| 81 | $user = get_user_by('email', $user_id); |
| 82 | } |
| 83 | |
| 84 | if ($user instanceof \WP_User && $current_user_id == absint($user->ID)) { |
| 85 | return true; |
| 86 | } |
| 87 | } |
| 88 | } |
| 89 | } |
| 90 | } |
| 91 | |
| 92 | return false; |
| 93 | } |
| 94 | |
| 95 | public static function get_instance() |
| 96 | { |
| 97 | static $instance = null; |
| 98 | |
| 99 | if (is_null($instance)) { |
| 100 | $instance = new self(); |
| 101 | } |
| 102 | |
| 103 | return $instance; |
| 104 | } |
| 105 | } |