class-multiple-user-roles.php — Admin and Site Enhancements (ASE) 7.6.10

admin-site-enhancements / classes / class-multiple-user-roles.php

admin-site-enhancements / classes Last commit date

class-multiple-user-roles.php

134 lines

1	<?php
2
3	namespace ASENHA\Classes;
4
5	/**
6	* Class for Multiple User Roles module
7	*
8	* @since 6.9.5
9	*/
10	class Multiple_User_Roles {
11
12	/**
13	* Add UI to enable multiple user roles selection
14	*
15	* @since 4.8.0
16	*/
17	public function add_multiple_roles_ui( $user ) {
18
19	// Get user roles that the current user is allowed to edit
20	$roles = get_editable_roles();
21
22	// Get the roles of the user being shown / edited / created
23	if ( ! empty( $user->roles ) ) {
24	$user_roles = array_intersect( array_values( $user->roles ), array_keys( $roles ) ); // indexed array of role slugs
25	} else {
26	$user_roles = array();
27	}
28
29	// Only show roles checkboxes for users that can assign roles to other users
30	if ( current_user_can( 'promote_users', get_current_user_id() ) ) {
31
32	?>
33	<div class="asenha-roles-temporary-container">
34	<table class="form-table">
35	<tr>
36	<th>
37	<label>Roles</label>
38	</th>
39	<td>
40	<?php
41	foreach ( $roles as $role_slug => $role_info ) {
42
43	$checkbox_id = $role_slug . '_role';
44	$role_name = translate_user_role( $role_info['name'] );
45	if ( ! empty( $user_roles ) && in_array( $role_slug, $user_roles ) ) {
46	$checked = 'checked="checked"';
47	} else {
48	$checked = '';
49	}
50
51	// Output roles checkboxes
52	?>
53	<label for="<?php esc_attr_e( $checkbox_id ); ?>"><input type="checkbox" id="<?php esc_attr_e( $checkbox_id ); ?>" value="<?php esc_attr_e( $role_slug ); ?>" name="asenha_assigned_roles[]" <?php esc_attr_e( $checked ) ?> /> <?php esc_html_e( $role_name ); ?></label><br />
54	<?php
55
56	}
57
58	wp_nonce_field( 'asenha_set_multiple_roles', 'asenha_multiple_roles_nonce' );
59
60	?>
61	</td>
62	</tr>
63	</table>
64	</div>
65	<?php
66
67	}
68
69	}
70
71	/**
72	* Save changes in roles assignment
73	*
74	* @since 4.8.0
75	*/
76	public function save_roles_assignment( $user_id ) {
77
78	if ( ! isset( $_POST['asenha_multiple_roles_nonce'] ) ) {
79	return;
80	}
81
82	if ( ! current_user_can( 'promote_users', get_current_user_id() ) \|\| ! wp_verify_nonce( $_POST['asenha_multiple_roles_nonce'], 'asenha_set_multiple_roles' ) ) {
83	return;
84	}
85
86	// Get user roles that the current user is allowed to edit
87	$roles = get_editable_roles();
88
89	// Get the roles of the user being shown / edited / created
90	$user = get_user_by( 'id', (int) $user_id ); // WP_User object
91	$user_roles = array_intersect( array_values( $user->roles ), array_keys( $roles ) ); // Current/existing roles
92
93	if ( ! empty( $_POST['asenha_assigned_roles'] ) ) {
94
95	$assigned_roles = array_map( 'sanitize_text_field', $_POST['asenha_assigned_roles'] );
96
97	// Make sure only valid roles are processed
98	$assigned_roles = array_intersect( $assigned_roles, array_keys( $roles ) );
99
100	$roles_to_remove = array();
101	$roles_to_add = array();
102
103	if ( empty( $assigned_roles ) ) {
104
105	// Remove all current/existing roles
106	$roles_to_remove = $user_roles;
107
108	} else {
109
110	// Identify and remove roles not present in the newly assigned roles
111	$roles_to_remove = array_diff( $user_roles, $assigned_roles );
112
113	if ( ! empty( $roles_to_remove ) ) {
114	foreach ( $roles_to_remove as $role_to_remove ) {
115	$user->remove_role( $role_to_remove );
116	}
117	}
118
119	// Identify and add roles not present in the existing roles
120	$roles_to_add = array_diff( $assigned_roles, $user_roles );
121
122	if ( ! empty( $roles_to_add ) ) {
123	foreach ( $roles_to_add as $role_to_add ) {
124	$user->add_role( $role_to_add );
125	}
126	}
127
128	}
129
130	}
131
132	}
133
134	}