added: profile field restriction

Author: achchu93

includes/wpum-fields/wpum-fields-functions.php

@@ -527,6 +527,20 @@ function wpum_is_field_required() {
 	return apply_filters( 'wpum_is_field_required', $wpum_field->is_required(), $wpum_field->get_ID() );
 }
 
+/**
+ * Verify if the field has correct user role permission.
+ *
+ * @return bool
+ */
+function wpum_field_has_user_role_permission( $field_id = null ) {
+	global $wpum_field;
+
+	$field         = $field_id ? new WPUM_Field( $field_id ) : $wpum_field;
+	$field_roles   = $field->get_meta( 'roles' );
+
+	return ! ( is_array( $field_roles ) && count( $field_roles ) && !count( array_intersect( wp_get_current_user()->roles, $field_roles ) ) );
+}
+
 /**
  * Retrieve the current field type within a loop.
  *

templates/forms/form-account.php

@@ -45,13 +45,10 @@
 			<?php foreach ( $data->fields as $key => $field ) : ?>
 
 				<?php
-					// Check if user has atleast one selected role
-					$user          = wp_get_current_user();
-					$is_renderable = ! ( is_array( $field['roles'] ) && count( $field['roles'] ) && !count( array_intersect( $user->roles, $field['roles'] ) ) );
-
-					if( ! apply_filters( 'wpum_account_form_field_can_render', $is_renderable, $field, $data ) ){
-						continue;
-					}
+				//Checks user has role perimission
+				if( ! apply_filters( 'wpum_account_form_field_can_render', wpum_field_has_user_role_permission( $field['id'] ), $field, $data ) ){
+					continue;
+				}
 				?>
 
 				<?php

templates/profiles/about.php

@@ -37,7 +37,7 @@
 				<table class="profile-fields-table">
 					<tbody>
 						<?php while ( wpum_profile_fields() ) : wpum_the_profile_field(); ?>
-							<?php if ( wpum_field_has_data() ) : ?>
+							<?php if ( wpum_field_has_data() && apply_filters( 'wpum_about_field_can_render', wpum_field_has_user_role_permission() ) ) : ?>
 								<tr class="<?php wpum_the_field_css_class(); ?>">
 									<td class="label"><?php wpum_the_field_name(); ?></td>
 									<td class="data"><?php wpum_the_field_value(); ?></td>