Select role from register page

My current code is:

add_filter(‘hook’, ‘assign_user_role_on_registration’);

function assign_user_role_on_registration( $data ) {

global $post;
$post_id = $post->ID;

if ($post_id) {
$role = geodir_get_post_meta($post_id, ‘role’);

$current_user = wp_get_current_user();
$current_user_roles = $current_user->roles;

if (in_array(‘subscriber’, $current_user_roles)){

$current_user->set_role( $role );

}
}
return $data;
}

Where hook is the relevant hook (not sure what this is). I’m also not sure if geodir_get_post_meta is the right way to access the form field.

Thanks Patrik

It’s now working 🙂 In case anyone else on here is interested in doing this, I’ve written my code below. The html variable name for the role field is ‘role’ which must be prefixed by ‘uwp_account_’. The two options it can take on my form are ‘Worker’ and ‘Employer’ which correspond to my two roles ‘um_worker’ and ’employer’.

add_action('uwp_after_custom_fields_save', 'assign_user_role_on_registration', 10, 4);

function assign_user_role_on_registration( $form_type, $data, $result, $user_id ) {
  
  if ($form_type == 'register') {
	
        $role = $data['uwp_account_role'];	
		$current_user = get_userdata($user_id);
	    
	    if ($current_user) {
	
			$current_user_roles = $current_user->roles;
	
			if (in_array('subscriber', $current_user_roles)) {
		  
		  		if ($role == 'Worker') {
	  
	  				$current_user->set_role( 'um_worker' );	
		    	}
		  
		  		if ($role == 'Employer') {
		   
		    		$current_user->set_role( 'employer' );			  
		    	}
  	  		 }
	   }
  }
}