mirror of
https://github.com/10h30/ultimatemember.git
synced 2026-06-05 15:09:37 +09:00
Champ Camba
339 lines
9.7 KiB
PHP
339 lines
9.7 KiB
PHP
<?php
|
|
namespace um\core;
|
|
|
|
// Exit if accessed directly
|
|
if ( ! defined( 'ABSPATH' ) ) exit;
|
|
|
|
|
|
if ( ! class_exists( 'Roles_Capabilities' ) ) {
|
|
class Roles_Capabilities {
|
|
|
|
function __construct() {
|
|
|
|
add_action( 'wp_roles_init', array( &$this, 'um_roles_init' ), 99999 );
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
* Loop through dynamic roles and add them to the $wp_roles array
|
|
*
|
|
* @param null|object $wp_roles
|
|
* @return null
|
|
*/
|
|
function um_roles_init( $wp_roles = null ) {
|
|
|
|
//Add UM role data to WP Roles
|
|
foreach ( $wp_roles->roles as $roleID => $role_data ) {
|
|
$role_meta = get_option( "um_role_{$roleID}_meta" );
|
|
|
|
if ( ! empty( $role_meta ) )
|
|
$wp_roles->roles[$roleID] = array_merge( $role_data, $role_meta );
|
|
}
|
|
|
|
|
|
//Add custom UM roles
|
|
$roles = array();
|
|
|
|
$role_keys = get_option( 'um_roles' );
|
|
|
|
if ( $role_keys ) {
|
|
|
|
foreach ( $role_keys as $role_key ) {
|
|
$role_meta = get_option( "um_role_{$role_key}_meta" );
|
|
if ( $role_meta ) {
|
|
//$role_meta['name'] = 'UM ' . $role_meta['name'];
|
|
$roles['um_' . $role_key] = $role_meta;
|
|
}
|
|
}
|
|
|
|
foreach ( $roles as $role_id => $details ) {
|
|
$capabilities = ! empty( $details['wp_capabilities'] ) ? array_keys( $details['wp_capabilities'] ) : array();
|
|
$details['capabilities'] = array_fill_keys( array_values( $capabilities ), true );
|
|
unset( $details['wp_capabilities'] );
|
|
$wp_roles->roles[$role_id] = $details;
|
|
$wp_roles->role_objects[$role_id] = new \WP_Role( $role_id, $capabilities );
|
|
$wp_roles->role_names[$role_id] = $details['name'];
|
|
}
|
|
|
|
}
|
|
|
|
// Return the modified $wp_roles array
|
|
return $wp_roles;
|
|
}
|
|
|
|
|
|
/**
|
|
* Check if role is custom
|
|
*
|
|
* @param $role
|
|
* @return bool
|
|
*/
|
|
function is_role_custom( $role ) {
|
|
// User has roles so look for a UM Role one
|
|
$role_keys = get_option( 'um_roles' );
|
|
|
|
if ( empty( $role_keys ) )
|
|
return false;
|
|
|
|
$role_keys = array_map( function( $item ) {
|
|
return 'um_' . $item;
|
|
}, $role_keys );
|
|
|
|
return in_array( $role, $role_keys );
|
|
}
|
|
|
|
|
|
/**
|
|
* Return a user's main role
|
|
*
|
|
* @param int $user_id
|
|
* @param string $new_role
|
|
* @uses get_userdata() To get the user data
|
|
* @uses apply_filters() Calls 'um_set_user_role' with the role and user id
|
|
* @return string
|
|
*/
|
|
function set_role( $user_id, $new_role = '' ) {
|
|
// Validate user id
|
|
$user = get_userdata( $user_id );
|
|
|
|
// User exists
|
|
if ( ! empty( $user ) ) {
|
|
// Get users old UM role
|
|
$role = $this->um_get_user_role( $user_id );
|
|
|
|
// User already has this role so no new role is set
|
|
if ( $new_role === $role || ( ! $this->is_role_custom( $new_role ) && user_can( $user, $new_role ) ) ) {
|
|
$new_role = false;
|
|
} else {
|
|
// Users role is different than the new role
|
|
|
|
// Remove the old UM role
|
|
if ( ! empty( $role ) && $this->is_role_custom( $role ) ) {
|
|
$user->remove_role( $role );
|
|
}
|
|
|
|
// Add the new role
|
|
if ( ! empty( $new_role ) ) {
|
|
$user->add_role( $new_role );
|
|
}
|
|
|
|
do_action( 'um_when_role_is_set', $user_id );
|
|
|
|
do_action('um_before_user_role_is_changed');
|
|
|
|
UM()->user()->profile['role'] = $new_role;
|
|
|
|
do_action( 'um_member_role_upgrade', $role, UM()->user()->profile['role'] );
|
|
|
|
UM()->user()->update_usermeta_info( 'role' );
|
|
|
|
do_action('um_after_user_role_is_changed');
|
|
|
|
do_action('um_after_user_role_is_updated', $user_id, $role );
|
|
}
|
|
} else {
|
|
// User does don exist so return false
|
|
$new_role = false;
|
|
}
|
|
|
|
return apply_filters( 'um_set_user_role', $new_role, $user_id, $user );
|
|
}
|
|
|
|
/**
|
|
* Remove user role
|
|
*
|
|
* @param $user_id
|
|
* @param $role
|
|
*/
|
|
function remove_role( $user_id, $role ) {
|
|
// Validate user id
|
|
$user = get_userdata( $user_id );
|
|
|
|
// User exists
|
|
if ( ! empty( $user ) ) {
|
|
// Remove role
|
|
$user->remove_role( $role );
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Set roles to user (remove all previous roles)
|
|
* make user only with $roles roles
|
|
*
|
|
* @param int $user_id
|
|
* @param string|array $roles
|
|
*/
|
|
function set_roles( $user_id, $roles ) {
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
* Get user one of UM roles if it has it
|
|
*
|
|
* @param int $user_id
|
|
* @return bool|mixed
|
|
*/
|
|
function um_get_user_role( $user_id ) {
|
|
$user = get_userdata( $user_id );
|
|
|
|
if ( empty( $user->roles ) )
|
|
return false;
|
|
|
|
// User has roles so look for a UM Role one
|
|
$role_keys = get_option( 'um_roles' );
|
|
|
|
if ( empty( $role_keys ) )
|
|
return array_shift( $user->roles );
|
|
|
|
$role_keys = array_map( function( $item ) {
|
|
return 'um_' . $item;
|
|
}, $role_keys );
|
|
|
|
$roles = array_intersect( array_values( $user->roles ), $role_keys );
|
|
if ( ! empty( $roles ) ) {
|
|
return array_shift( $roles );
|
|
} else {
|
|
return array_shift( $user->roles );
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Get role name by roleID
|
|
*
|
|
* @param $slug
|
|
* @return bool|string
|
|
*/
|
|
function get_role_name( $slug ) {
|
|
$roledata = $this->role_data( $slug );
|
|
|
|
if ( empty( $roledata['name'] ) ) {
|
|
global $wp_roles;
|
|
|
|
if ( empty( $wp_roles->roles[$slug] ) )
|
|
return false;
|
|
else
|
|
return $wp_roles->roles[$slug]['name'];
|
|
}
|
|
|
|
|
|
return $roledata['name'];
|
|
}
|
|
|
|
|
|
/**
|
|
* Get role data
|
|
*
|
|
* @param int $roleID Role ID
|
|
* @return mixed|void
|
|
*/
|
|
function role_data( $roleID ) {
|
|
if ( strpos( $roleID, 'um_' ) === 0 )
|
|
$roleID = substr( $roleID, 3 );
|
|
|
|
$role_data = get_option( "um_role_{$roleID}_meta" );
|
|
|
|
if ( ! $role_data )
|
|
return array();
|
|
|
|
$temp = array();
|
|
foreach ( $role_data as $key=>$value ) {
|
|
if ( strpos( $key, '_um_' ) === 0 )
|
|
$key = str_replace( '_um_', '', $key );
|
|
$temp[$key] = $value;
|
|
}
|
|
return $temp;
|
|
}
|
|
|
|
|
|
/***
|
|
*** @Query for UM roles
|
|
***/
|
|
function get_roles( $add_default = false, $exclude = null ){
|
|
global $wp_roles;
|
|
|
|
if ( empty( $wp_roles ) ) {
|
|
return array();
|
|
}
|
|
|
|
$roles = $wp_roles->role_names;
|
|
|
|
if ( $add_default ) {
|
|
$roles[0] = $add_default;
|
|
}
|
|
|
|
if ( $exclude ) {
|
|
foreach( $exclude as $role ) {
|
|
unset( $roles[$role] );
|
|
}
|
|
}
|
|
|
|
return $roles;
|
|
}
|
|
|
|
|
|
/***
|
|
*** @Current user can
|
|
***/
|
|
function um_current_user_can( $cap, $user_id ) {
|
|
if ( ! is_user_logged_in() )
|
|
return false;
|
|
|
|
$return = 1;
|
|
|
|
um_fetch_user( get_current_user_id() );
|
|
|
|
switch( $cap ) {
|
|
case 'edit':
|
|
if ( get_current_user_id() == $user_id && um_user( 'can_edit_profile' ) )
|
|
$return = 1;
|
|
elseif ( ! um_user( 'can_edit_everyone' ) )
|
|
$return = 0;
|
|
elseif ( get_current_user_id() == $user_id && ! um_user( 'can_edit_profile') )
|
|
$return = 0;
|
|
elseif ( um_user( 'can_edit_roles' ) && ! in_array( UM()->roles()->um_get_user_role( $user_id ), um_user( 'can_edit_roles' ) ) )
|
|
$return = 0;
|
|
break;
|
|
|
|
case 'delete':
|
|
if ( ! um_user( 'can_delete_everyone' ) ) $return = 0;
|
|
elseif ( um_user( 'can_delete_roles' ) && ! in_array( UM()->roles()->um_get_user_role( $user_id ), um_user( 'can_delete_roles' ) ) ) $return = 0;
|
|
break;
|
|
|
|
}
|
|
|
|
um_fetch_user( $user_id );
|
|
|
|
return $return;
|
|
}
|
|
|
|
|
|
/**
|
|
* User can ( role settings )
|
|
*
|
|
* @param $permission
|
|
* @return bool|mixed
|
|
*/
|
|
function um_user_can( $permission ) {
|
|
if ( ! is_user_logged_in() )
|
|
return false;
|
|
|
|
$user_id = get_current_user_id();
|
|
$role = $this->um_get_user_role( $user_id );
|
|
|
|
$permissions = $this->role_data( $role );
|
|
$permissions = apply_filters( 'um_user_permissions_filter', $permissions, $user_id );
|
|
|
|
if ( isset( $permissions[ $permission ] ) && is_serialized( $permissions[ $permission ] ) )
|
|
return unserialize( $permissions[ $permission ] );
|
|
|
|
if ( isset( $permissions[ $permission ] ) && $permissions[ $permission ] == 1 )
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
}
|
|
} |