Untitled

add_action('b2bking_after_register_user_save_fields', function($user_id){
    $user_obj = new WP_User($user_id);
    $email = $user_obj->user_email;
    
    // Define approved domains - add your domains here
    $approved_domains = array(
        'company1.com',
        'company2.com',
        'partner-corp.com',
        'approved-business.net'
    );
    
    // Extract domain from email
    $email_domain = substr(strrchr($email, "@"), 1);
    
    // Check if domain is in approved list
    if (in_array($email_domain, $approved_domains)) {
        // Auto-approve users from approved domains
        update_user_meta($user_id, 'b2bking_account_approved', 'yes');
        update_user_meta($user_id, 'b2bking_b2buser', 'yes');
        update_user_meta($user_id, 'b2bking_customergroup', 1234); // Set your desired customer group ID here
        
    } else {
        // Auto-reject users from non-approved domains
        update_user_meta($user_id, 'b2bking_account_approved', 'no');
        update_user_meta($user_id, 'b2bking_b2buser', 'no');
        update_user_meta($user_id, 'b2bking_customergroup', 'no');
    }
    
}, 10, 1);