137 lines
6.2 KiB
PHP
137 lines
6.2 KiB
PHP
<?php
|
|
/* handle field output */
|
|
function wppb_user_role_handler( $output, $form_location, $field, $user_id, $field_check_errors, $request_data ){
|
|
if ( $field['field'] == 'Select (User Role)' ){
|
|
|
|
$user_role = '';
|
|
$user_can_manage_options = false;
|
|
|
|
// Get user data, set user's role and check to see if user can manage options
|
|
if( $user_id != 0 ) {
|
|
$user_data = get_userdata( $user_id );
|
|
$user_role = $user_data->roles[0];
|
|
|
|
if( isset( $user_data->allcaps['manage_options'] ) && $user_data->allcaps['manage_options'] == 1 ) {
|
|
$user_can_manage_options = true;
|
|
}
|
|
}
|
|
|
|
$input_value = isset( $request_data['custom_field_user_role'] ) ? $request_data['custom_field_user_role'] : $user_role;
|
|
|
|
$item_title = apply_filters( 'wppb_'.$form_location.'_user_role_custom_field_'.$field['id'].'_item_title', wppb_icl_t( 'plugin profile-builder-pro', 'custom_field_'.$field['id'].'_title_translation', $field['field-title'] ) );
|
|
$item_description = wppb_icl_t( 'plugin profile-builder-pro', 'custom_field_'.$field['id'].'_description_translation', $field['description'] );
|
|
|
|
//get user roles
|
|
if( !empty( $field['user-roles'] ) ) {
|
|
global $wp_roles;
|
|
|
|
$available_user_roles = explode( ', ', $field['user-roles'] );
|
|
|
|
foreach( $available_user_roles as $key => $role_slug ) {
|
|
if( isset( $wp_roles->roles[$role_slug]['name'] ) ) {
|
|
$available_user_roles[$key] = array(
|
|
'slug' => $role_slug,
|
|
'name' => $wp_roles->roles[$role_slug]['name']
|
|
);
|
|
} else {
|
|
unset( $available_user_roles[$key] );
|
|
}
|
|
}
|
|
}
|
|
|
|
$extra_attr = apply_filters( 'wppb_extra_attribute', '', $field, $form_location );
|
|
|
|
if( $form_location == 'register' || ( $form_location == 'edit_profile' && current_user_can('manage_options') && $user_can_manage_options == false ) ) {
|
|
$error_mark = ( ( $field['required'] == 'Yes' ) ? '<span class="wppb-required" title="'.wppb_required_field_error($field["field-title"]).'">*</span>' : '' );
|
|
|
|
if ( array_key_exists( $field['id'], $field_check_errors ) )
|
|
$error_mark = '<img src="'.WPPB_PLUGIN_URL.'assets/images/pencil_delete.png" title="'.wppb_required_field_error($field["field-title"]).'"/>';
|
|
|
|
$output = '
|
|
<label for="custom_field_user_role">'.$item_title.$error_mark.'</label>
|
|
<select name="custom_field_user_role" id="'.$field['meta-name'].'" class="custom_field_user_role '. apply_filters( 'wppb_fields_extra_css_class', '', $field ) .'" '. $extra_attr .'>';
|
|
|
|
$extra_select_option = apply_filters( 'wppb_extra_select_option', '', $field, $item_title );
|
|
if( ! empty( $extra_select_option ) ) {
|
|
$output .= $extra_select_option;
|
|
}
|
|
|
|
if( !empty( $available_user_roles ) ) {
|
|
foreach( $available_user_roles as $user_role ){
|
|
$output .= '<option value="'. $user_role['slug'] .'"';
|
|
|
|
$output .= selected( $input_value, $user_role['slug'], false );
|
|
|
|
$output .= '>'. $user_role['name'] .'</option>';
|
|
}
|
|
}
|
|
|
|
$output .= '</select>';
|
|
|
|
if( $form_location == 'edit_profile' )
|
|
$output .= '<span class="wppb-description-delimiter">'. __( 'Only administrators can see this field on edit profile forms.', 'profile-builder' ) .'</span>';
|
|
|
|
if( !empty( $item_description ) )
|
|
$output .= '<span class="wppb-description-delimiter">'.$item_description.'</span>';
|
|
|
|
} elseif( $form_location == 'edit_profile' && current_user_can('manage_options') && $user_can_manage_options == true ) {
|
|
|
|
$output = '
|
|
<label for="custom_field_user_role">'.$item_title.'</label>
|
|
<p>' . __( 'As an administrator you cannot change your role.', 'profile-builder' ) . '</p>';
|
|
|
|
$output .= '</select>';
|
|
|
|
$output .= '<span class="wppb-description-delimiter">'. __( 'Only administrators can see this field on edit profile forms.', 'profile-builder' ) .'</span>';
|
|
|
|
if( !empty( $item_description ) )
|
|
$output .= '<span class="wppb-description-delimiter">'.$item_description.'</span>';
|
|
|
|
}
|
|
|
|
return apply_filters( 'wppb_'.$form_location.'_user_role_custom_field_'.$field['id'], $output, $form_location, $field, $user_id, $field_check_errors, $request_data, $input_value );
|
|
}
|
|
}
|
|
add_filter( 'wppb_output_form_field_select-user-role', 'wppb_user_role_handler', 10, 6 );
|
|
|
|
|
|
/* handle field validation */
|
|
function wppb_check_user_role_value( $message, $field, $request_data, $form_location ) {
|
|
|
|
if( $form_location == 'back_end' )
|
|
return $message;
|
|
|
|
$field['meta-name'] = 'custom_field_user_role';
|
|
|
|
if( $field['field'] == 'Select (User Role)' ){
|
|
|
|
if( $form_location == 'register' && $field['required'] == 'Yes' && current_user_can( 'manage_options' ) === false ) {
|
|
if( ( isset( $request_data[wppb_handle_meta_name( $field['meta-name'] )] ) && ( trim( $request_data[wppb_handle_meta_name( $field['meta-name'] )] ) == '' ) ) || !isset( $request_data[wppb_handle_meta_name( $field['meta-name'] )] ) ){
|
|
return wppb_required_field_error($field["field-title"]);
|
|
}
|
|
}
|
|
|
|
if( isset( $field['user-roles'] ) && isset( $request_data['custom_field_user_role'] ) ) {
|
|
$available_user_roles = explode(', ', $field['user-roles'] );
|
|
|
|
if( !in_array( $request_data['custom_field_user_role'], $available_user_roles ) ) {
|
|
return __( 'You cannot register this user role', 'profile-builder');
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
return $message;
|
|
}
|
|
add_filter( 'wppb_check_form_field_select-user-role', 'wppb_check_user_role_value', 10, 4 );
|
|
|
|
|
|
/* handle field save */
|
|
function wppb_userdata_add_user_role( $userdata, $global_request ){
|
|
|
|
if ( isset( $global_request['custom_field_user_role'] ) )
|
|
$userdata['role'] = sanitize_text_field( trim( $global_request['custom_field_user_role'] ) );
|
|
|
|
return $userdata;
|
|
}
|
|
add_filter( 'wppb_build_userdata', 'wppb_userdata_add_user_role', 10, 2 ); |