aibishal

✏️ Editing: class-urcr-shortcodes.php

<?php
/**
 * User Registration Content Restriction Shortcodes.
 *
 * @class    URCR_Shortcodes
 * @version  4.0
 * @package  UserRegistrationContentRestriction/Classes
 * @category Class
 * @author   WPEverest
 */

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

/**
 * URCR_Shortcodes Class
 */

class URCR_Shortcodes {

public function __construct() {
		add_shortcode( 'urcr_restrict', array( $this, 'urcr_restrict_shortcode' ) );
	}

public function urcr_restrict_shortcode( $atts, $content = null ) {
		global $post;

if ( ! is_object( $post ) ) {
			return;
		}

$enable_disable = ur_string_to_bool( get_option( 'user_registration_content_restriction_enable', true ) );

if ( $enable_disable ) {
			$override_global_settings = get_post_meta( $post->ID, 'urcr_meta_override_global_settings', $single = true );

$allowed_roles = get_option( 'user_registration_content_restriction_allow_to_roles', 'administrator' );
			$allowed_memberships = get_option( 'user_registration_content_restriction_allow_to_memberships');

$current_user_role = is_user_logged_in() ? wp_get_current_user()->roles[0] : 'guest';
			$get_meta_data_roles = get_post_meta( $post->ID, 'urcr_meta_roles', $single = true );
			$get_meta_data_memberships = get_post_meta( $post->ID, 'urcr_meta_memberships', true );

$roles = isset( $atts['access_role'] ) ? trim( $atts['access_role'] ) : '';
			$is_membership_active = ur_check_module_activation('membership');

if( $is_membership_active ) {
				$members_subscription = new \WPEverest\URMembership\Admin\Repositories\MembersSubscriptionRepository();
				$subscription = $members_subscription->get_member_subscription( wp_get_current_user()->ID);

$current_user_membership = ( !empty ( $subscription ) ) ? $subscription['item_id'] : array();
			}

if ( empty( $roles ) ) {
				$override_global_settings = get_post_meta( $post->ID, 'urcr_meta_override_global_settings', $single = true );

if ( $override_global_settings !== 'on' ) {
					if ( '0' == get_option( 'user_registration_content_restriction_allow_access_to', '0' ) ) {
						if ( is_user_logged_in() ) {
							return do_shortcode( $content );
						}
					} elseif ( '1' == get_option( 'user_registration_content_restriction_allow_access_to' ) ) {
						if ( is_array( $allowed_roles ) && in_array( $current_user_role, $allowed_roles ) ) {
							return do_shortcode( $content );
						}
					} elseif ( '2' === get_option( 'user_registration_content_restriction_allow_access_to' ) ) {
						if ( ! is_user_logged_in() ) {
							return do_shortcode( $content );
						}
					}
					elseif ( '3' === get_option( 'user_registration_content_restriction_allow_access_to' ) ) {
						if ( is_array( $allowed_memberships ) && in_array( $current_user_membership, $allowed_memberships ) ) {
							return do_shortcode( $content );
						}
					}
				} else {
					$get_meta_data_allow_to = get_post_meta( $post->ID, 'urcr_allow_to', $single = true );

switch ( $get_meta_data_allow_to ) {
						case '0':
							if ( is_user_logged_in() ) {
								return do_shortcode( $content );
							}
							break;
						case '1':
							if ( isset( $get_meta_data_roles ) && ! empty( $get_meta_data_roles ) ) {
								if ( is_array( $get_meta_data_roles ) && in_array( $current_user_role, $get_meta_data_roles ) ) {
									return do_shortcode( $content );
								}
							}
							break;
						case '2':
							if ( ! is_user_logged_in() ) {
								return do_shortcode( $content );
							}
							break;
						case '3':
							if ( ! empty( $get_meta_data_memberships ) ) {
								if ( is_array( $get_meta_data_memberships ) && in_array( $current_user_membership, $get_meta_data_memberships ) ) {
									return do_shortcode( $content );
								}
							}
							break;
					}
				}
			}

$roles = explode( ',', $roles );

$roles = array_map( 'trim', $roles );

$message = get_option( 'user_registration_content_restriction_message' );

$message = empty( $message ) ? __( 'This content is restricted!', 'user-registration' ) : $message;

$message = apply_filters( 'user_registration_process_smart_tags', $message );

$message = do_shortcode( $message );

if ( in_array( 'all_logged_in_users', $roles ) ) {
				if ( is_user_logged_in() ) {
					return do_shortcode( $content );
				}
			} elseif ( in_array( 'guest', $roles ) ) {
				if ( ! is_user_logged_in() ) {
					return do_shortcode( $content );
				}
			} elseif ( in_array( $current_user_role, $roles ) ) {
				return do_shortcode( $content );
			}

return '<span class="urcr-restrict-message">' . $message . '</span>';
		}
			return do_shortcode( $content );
	}
}
return new URCR_Shortcodes();

Gifari Industries - BD Cyber Security Team