Files
ultimatemember/core/um-actions-access.php
T
2016-09-07 17:07:51 +08:00

744 lines
20 KiB
PHP

<?php
/***
*** @Global Access Settings
***/
add_action('um_access_global_settings','um_access_global_settings');
function um_access_global_settings() {
global $post, $ultimatemember;
$access = um_get_option('accessible');
if ( $access == 2 && ! is_user_logged_in() ) {
$redirect = um_get_option('access_redirect');
if ( !$redirect ){
$redirect = um_get_core_page('login');
}
$redirects[] = untrailingslashit( um_get_core_page('login') );
$redirects[] = untrailingslashit( um_get_option('access_redirect') );
$exclude_uris = um_get_option('access_exclude_uris');
if ( $exclude_uris ) {
$redirects = array_merge( $redirects, $exclude_uris );
}
$redirects = array_unique( $redirects );
$current_url = $ultimatemember->permalinks->get_current_url( get_option('permalink_structure') );
$current_url = untrailingslashit( $current_url );
$current_url_slash = trailingslashit( $current_url );
if ( ( isset( $post->ID ) ) && ( in_array( $current_url, $redirects ) || in_array( $current_url_slash, $redirects ) ) ) {
// allow
}else {
$ultimatemember->access->redirect_handler = $ultimatemember->access->set_referer( $redirect, "global" );
}
}
$current_page_type = um_get_current_page_type();
do_action("um_access_post_type",$current_page_type);
do_action("um_access_post_type_{$current_page_type}");
}
/***
*** @Custom User homepage redirection
***/
add_action("um_access_user_custom_homepage","um_access_user_custom_homepage");
function um_access_user_custom_homepage(){
global $ultimatemember;
if( ! is_user_logged_in() ) return;
if( ! is_front_page() ) return;
$role_meta = $ultimatemember->query->role_data( um_user('role') );
if( isset( $role_meta['default_homepage'] ) && $role_meta['default_homepage'] == 0 ){
$redirect_to = null;
if( ! empty( $role_meta['redirect_homepage'] ) ){
$redirect_to = $role_meta['redirect_homepage'];
}else{
$redirect_to = um_get_core_page('user');
}
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "custom_homepage" );
wp_redirect( $redirect_to ); exit;
}
}
/***
*** @Front page access settings
***/
add_action('um_access_frontpage_per_role','um_access_frontpage_per_role');
function um_access_frontpage_per_role() {
global $ultimatemember, $post;
if ( is_admin() ) return;
if ( ! is_front_page() ) return;
if( is_404() ) return;
if ( ! isset( $um_post_id ) && isset( $post->ID ) ){
$um_post_id = $post->ID;
}
if( ! isset( $um_post_id ) ){
return;
}
$args = $ultimatemember->access->get_meta( $um_post_id );
extract( $args );
if ( !isset( $args['custom_access_settings'] ) || $args['custom_access_settings'] == 0 ) {
$um_post_id = apply_filters('um_access_control_for_parent_posts', $um_post_id );
$args = $ultimatemember->access->get_meta( $um_post_id );
extract( $args );
if ( !isset( $args['custom_access_settings'] ) || $args['custom_access_settings'] == 0 ) {
return;
}
}
$redirect_to = null;
if ( !isset( $accessible ) ) return;
switch( $accessible ) {
case 0:
$ultimatemember->access->allow_access = true;
$ultimatemember->access->redirect_handler = false; // open to everyone
break;
case 1:
$redirect_to = $access_redirect2;
if ( is_user_logged_in() ){
$ultimatemember->access->allow_access = false;
}
if ( ! is_user_logged_in() ){
$ultimatemember->access->allow_access = true;
}
if( ! empty( $redirect_to ) ){
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "frontpage_per_role_1a" );
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
}else{
if ( ! is_user_logged_in() ){
$redirect_to = um_get_core_page("login");
}else{
$redirect_to = um_get_core_page("user");
}
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "frontpage_per_role_1b" );
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
}
break;
case 2:
if ( ! is_user_logged_in() ){
if ( empty( $access_redirect ) ) {
$access_redirect = um_get_core_page('login');
}
$redirect_to = $access_redirect;
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "frontpage_per_role_2a" );
}
if ( is_user_logged_in() && isset( $access_roles ) && !empty( $access_roles ) ){
$access_roles = unserialize( $access_roles );
$access_roles = array_filter($access_roles);
if ( !empty( $access_roles ) && !in_array( um_user('role'), $access_roles ) ) {
if ( empty( $access_redirect ) ) {
if ( is_user_logged_in() ) {
$access_redirect = site_url();
} else {
$access_redirect = um_get_core_page('login');
}
}
$redirect_to = esc_url( $access_redirect );
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "frontpage_per_role_2b" );
}
}
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
break;
}
}
/***
*** @Posts page access settings
***/
add_action('um_access_homepage_per_role','um_access_homepage_per_role');
function um_access_homepage_per_role() {
global $ultimatemember, $post;
if ( is_admin() ) return;
if ( ! is_home() ) return;
if ( is_404() ) return;
$access = um_get_option('accessible');
$show_on_front = get_option( 'show_on_front' );
if( $show_on_front == "page" ){
$um_post_id = get_option( 'page_for_posts' );
if ( $access == 2 && ! is_user_logged_in() ) {
$ultimatemember->access->allow_access = false;
}else{
$ultimatemember->access->allow_access = true;
}
}else if( $show_on_front == "posts" ){
$ultimatemember->access->allow_access = true;
}
if ( isset( $um_post_id ) ){
$args = $ultimatemember->access->get_meta( $um_post_id );
extract( $args );
if ( !isset( $args['custom_access_settings'] ) || $args['custom_access_settings'] == 0 ) {
$um_post_id = apply_filters('um_access_control_for_parent_posts', $um_post_id );
$args = $ultimatemember->access->get_meta( $um_post_id );
extract( $args );
if ( !isset( $args['custom_access_settings'] ) || $args['custom_access_settings'] == 0 ) {
return;
}
}
$redirect_to = null;
if ( !isset( $accessible ) ) return;
switch( $accessible ) {
case 0:
$ultimatemember->access->allow_access = true;
$ultimatemember->access->redirect_handler = false; // open to everyone
break;
case 1:
$redirect_to = esc_url( $access_redirect2 );
if ( is_user_logged_in() ){
$ultimatemember->access->allow_access = false;
}
if ( ! is_user_logged_in() ){
$ultimatemember->access->allow_access = true;
}
if( ! empty( $redirect_to ) ){
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "homepage_per_role_1a" );
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
}else{
$redirect_to = null;
if ( ! is_user_logged_in() ){
$redirect_to = um_get_core_page("login");
}else{
$redirect_to = um_get_core_page("user");
}
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "homepage_per_role_1b" );
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
}
break;
case 2:
if ( ! is_user_logged_in() ){
if ( empty( $access_redirect ) ) {
$access_redirect = um_get_core_page('login');
}
$redirect_to = $access_redirect;
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "homepage_per_role_2a" );
}
if ( is_user_logged_in() && isset( $access_roles ) && !empty( $access_roles ) ){
$access_roles = unserialize( $access_roles );
$access_roles = array_filter($access_roles);
if ( !empty( $access_roles ) && !in_array( um_user('role'), $access_roles ) ) {
if ( !$access_redirect ) {
if ( is_user_logged_in() ) {
$access_redirect = site_url();
} else {
$access_redirect = um_get_core_page('login');
}
}
$redirect_to = $access_redirect;
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "homepage_per_role_2b" );
}
}
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
break;
}
}
}
/***
*** @Archieves/Taxonomies/Categories access settings
***/
add_action('um_access_category_settings','um_access_category_settings');
function um_access_category_settings() {
global $post, $wp_query, $ultimatemember;
if ( is_front_page() ||
is_home() ||
is_feed() ||
is_page() ||
is_404()
) {
return;
}
$access = um_get_option('accessible');
$current_page_type = um_get_current_page_type();
if( is_category() && ! in_array( $current_page_type , array( 'day','month','year','author','archive' ) ) ){
$um_category = get_the_category();
$um_category = current( $um_category );
$term_id = $um_category->term_id;
if( isset( $term_id ) ){
$opt = get_option("category_$term_id");
if ( isset( $opt['_um_accessible'] ) ) {
$redirect = false;
switch( $opt['_um_accessible'] ) {
case 0:
$ultimatemember->access->allow_access = true;
$ultimatemember->access->redirect_handler = ''; // open to everyone
break;
case 1:
if ( is_user_logged_in() ){
if( isset( $opt['_um_redirect'] ) && ! empty( $opt['_um_redirect'] ) ) {
$redirect = $opt['_um_redirect'];
}else{
$redirect = site_url();
}
}
$ultimatemember->access->allow_access = false;
$redirect = $ultimatemember->access->set_referer( $redirect, "category_1" );
$ultimatemember->access->redirect_handler = esc_url( $redirect );
if ( ! is_user_logged_in() && ! empty( $redirect ) ){
$ultimatemember->access->allow_access = true;
}
break;
case 2:
if ( ! is_user_logged_in() ){
if( isset( $opt['_um_redirect'] ) && ! empty( $opt['_um_redirect'] ) ){
$redirect = $opt['_um_redirect'];
}else{
$redirect = um_get_core_page('login');
}
$ultimatemember->access->allow_access = false;
$redirect = $ultimatemember->access->set_referer( $redirect, "category_2a" );
$ultimatemember->access->redirect_handler = esc_url( $redirect );
}
if ( is_user_logged_in() && isset( $opt['_um_roles'] ) && !empty( $opt['_um_roles'] ) ){
if ( ! in_array( um_user('role'), $opt['_um_roles'] ) ) {
if( isset( $opt['_um_redirect'] ) ){
$redirect = $opt['_um_redirect'];
}
$redirect = $ultimatemember->access->set_referer( $redirect, "category_2b" );
$ultimatemember->access->redirect_handler = esc_url( $redirect );
}
}
}
}
}
} else if ( $access == 2 && ! is_user_logged_in() && is_archive() ) {
$ultimatemember->access->allow_access = false;
$redirect = um_get_core_page('login');
$redirect = $ultimatemember->access->set_referer( $redirect, "category_archive" );
$ultimatemember->access->redirect_handler = $redirect;
} else if ( is_tax() && get_post_taxonomies( $post ) ) {
$taxonomies = get_post_taxonomies( $post );
$categories_ids = array();
foreach ($taxonomies as $key => $value) {
$term_list = wp_get_post_terms($post->ID, $value, array("fields" => "ids"));
foreach( $term_list as $term_id ){
array_push( $categories_ids , $term_id);
}
}
foreach( $categories_ids as $term => $term_id ) {
$opt = get_option("category_$term_id");
if ( isset( $opt['_um_accessible'] ) ) {
switch( $opt['_um_accessible'] ) {
case 0:
$ultimatemember->access->allow_access = true;
$ultimatemember->access->redirect_handler = false; // open to everyone
break;
case 1:
if ( is_user_logged_in() )
$redirect = ( isset( $opt['_um_redirect'] ) && !empty( $opt['_um_redirect'] ) ) ? $opt['_um_redirect'] : site_url();
$redirect = $ultimatemember->access->set_referer( $redirect, "categories_1" );
$ultimatemember->access->redirect_handler = $redirect;
if ( !is_user_logged_in() )
$ultimatemember->access->allow_access = true;
break;
case 2:
if ( ! is_user_logged_in() ){
$redirect = ( isset( $opt['_um_redirect'] ) && ! empty( $opt['_um_redirect'] ) ) ? $opt['_um_redirect'] : um_get_core_page('login');
$redirect = $ultimatemember->access->set_referer( $redirect, "categories_2a" );
$ultimatemember->access->redirect_handler = $redirect;
}
if ( is_user_logged_in() && isset( $opt['_um_roles'] ) && !empty( $opt['_um_roles'] ) ){
if ( !in_array( um_user('role'), $opt['_um_roles'] ) ) {
$redirect = null;
if ( is_user_logged_in() ){
$redirect = ( isset( $opt['_um_redirect'] ) ) ? $opt['_um_redirect'] : site_url();
}
if ( !is_user_logged_in() ){
$redirect = um_get_core_page('login');
}
$redirect = $ultimatemember->access->set_referer( $redirect, "categories_2b" );
$ultimatemember->access->redirect_handler = $redirect;
}
}
}
}
}
}
}
/***
*** @Posts/Page access settings
***/
add_action('um_access_post_settings','um_access_post_settings');
function um_access_post_settings() {
global $post, $ultimatemember;
// woo commerce shop ID
if( function_exists('is_shop') && is_shop() ) {
$um_post_id = get_option('woocommerce_shop_page_id');
} else if (
is_category() ||
is_archive() ||
is_search() ||
in_the_loop() ||
is_feed() ||
is_tax() ||
! get_post_type() ||
! isset( $post->ID ) ||
is_home() ||
is_front_page() ||
is_404()
) {
return;
}
if ( !isset( $um_post_id ) ){
$um_post_id = $post->ID;
}
$args = $ultimatemember->access->get_meta( $um_post_id );
extract( $args );
// Check for parent page's custom access settings
if ( !isset( $args['custom_access_settings'] ) || $args['custom_access_settings'] == 0 ) {
$um_post_id = apply_filters('um_access_control_for_parent_posts', $um_post_id );
$args = $ultimatemember->access->get_meta( $um_post_id );
extract( $args );
}
$post_type = get_post_type( $post->ID );
// post's category restriction
if ( $post_type != 'page' && isset( $args['custom_access_settings'] ) && $args['custom_access_settings'] == 0 ) {
$categories = get_the_category( $post->ID );
foreach( $categories as $cat ){
$opt = get_option("category_{$cat->term_id}");
if ( isset( $opt['_um_accessible'] ) ) {
switch( $opt['_um_accessible'] ) {
case 0: // Open to everyone
$ultimatemember->access->allow_access = true;
$ultimatemember->access->redirect_handler = false; // open to everyone
break;
case 1: // Logged out users only
if ( is_user_logged_in() ){
$redirect = ( isset( $opt['_um_redirect'] ) && ! empty( $opt['_um_redirect'] ) ) ? $opt['_um_redirect'] : site_url();
$redirect = $ultimatemember->access->set_referer( $redirect, "page_cat_1a_t{$cat->term_id}_pt{$post_type}" );
$ultimatemember->access->redirect_handler = $redirect;
$ultimatemember->access->allow_access = false;
}
if ( !is_user_logged_in() ){
$ultimatemember->access->allow_access = true;
}
break;
case 2: // Logged in users only
if ( ! is_user_logged_in() ){
$redirect = ( isset( $opt['_um_redirect'] ) && ! empty( $opt['_um_redirect'] ) ) ? $opt['_um_redirect'] : um_get_core_page('login');
$redirect = $ultimatemember->access->set_referer( $redirect, "page_cat_2a_t{$cat->term_id}_pt{$post_type}" );
$ultimatemember->access->redirect_handler = $redirect;
$ultimatemember->access->allow_access = false;
}
if ( is_user_logged_in() ){
if( isset( $opt['_um_roles'] ) && !empty( $opt['_um_roles'] ) ){
if ( in_array( um_user('role'), $opt['_um_roles'] ) ) {
$ultimatemember->access->allow_access = true;
}else{
$redirect = ( isset( $opt['_um_redirect'] ) && ! empty( $opt['_um_redirect'] ) ) ? $opt['_um_redirect'] : site_url();
$redirect = $ultimatemember->access->set_referer( $redirect, "page_cat_2b_t{$cat->term_id}_pt{$post_type}" );
$ultimatemember->access->redirect_handler = $redirect;
$ultimatemember->access->allow_access = false;
}
}else{ // if allowed all roles
$ultimatemember->access->allow_access = true;
}
}
}
} // end if isset( $opt['_um_accessible'] )
// if one of the categories has enabled restriction, apply its settings to the current post
if( $ultimatemember->access->allow_access == false ){
return;
}
} // end foreach
}
// post restriction
if ( isset( $args['custom_access_settings'] ) && $args['custom_access_settings'] == 1 ) {
$redirect_to = null;
if ( !isset( $accessible ) ) return;
switch( $accessible ) {
case 0:
$ultimatemember->access->allow_access = true;
$ultimatemember->access->redirect_handler = false; // open to everyone
break;
case 1:
$redirect_to = esc_url( $access_redirect2 );
if ( is_user_logged_in() ){
$ultimatemember->access->allow_access = false;
}
if ( ! is_user_logged_in() ){
$ultimatemember->access->allow_access = true;
}
if( ! empty( $redirect_to ) ){
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "post_1a" );
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
}else{
if ( ! is_user_logged_in() ){
$redirect_to = um_get_core_page("login");
}else{
$redirect_to = um_get_core_page("user");
}
$redirect_to = $ultimatemember->access->set_referer( $redirect_to, "post_1b" );
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
}
break;
case 2:
if ( ! is_user_logged_in() ){
if ( empty( $access_redirect ) ) {
$access_redirect = um_get_core_page('login');
}
$access_redirect = $ultimatemember->access->set_referer( $access_redirect, "post_2a" );
$redirect_to = esc_url( $access_redirect );
}
if ( is_user_logged_in() && isset( $access_roles ) && !empty( $access_roles ) ){
$access_roles = unserialize( $access_roles );
$access_roles = array_filter($access_roles);
if ( !empty( $access_roles ) && !in_array( um_user('role'), $access_roles ) ) {
if ( !$access_redirect ) {
if ( is_user_logged_in() ) {
$access_redirect = esc_url( site_url() );
} else {
$access_redirect = esc_url( um_get_core_page('login') );
}
}
$access_redirect = $ultimatemember->access->set_referer( $access_redirect, "post_2b" );
$redirect_to = esc_url( $access_redirect );
}
}
$ultimatemember->access->redirect_handler = esc_url( $redirect_to );
break;
}
}
if( um_is_core_page('user') && ! is_user_logged_in() && ! empty( $access_redirect ) ){
$ultimatemember->access->allow_access = false;
$access_redirect = $ultimatemember->access->set_referer( $access_redirect, "user_page" );
$ultimatemember->access->redirect_handler = esc_url( $access_redirect );
wp_redirect( $ultimatemember->access->redirect_handler );
exit;
}
}
/***
*** @Profile Access
***/
add_action('um_access_profile','um_access_profile');
function um_access_profile( $user_id ){
if( ! um_is_myprofile() && um_is_core_page('user') && ! current_user_can('edit_users') ){
um_fetch_user( $user_id );
if( ! in_array( um_user('account_status'), array('approved') ) ){
um_redirect_home();
}
um_reset_user();
}
}