强制客户在注册时选择 WooCommerce 角色

Question

我读过这篇文章，也看到其他人问同样的问题，但无论我做什么或做过什么，似乎都没有按照我需要的方式工作。

我已经删除了所有其他角色（编辑、订阅者、贡献者、商店经理等），现在只剩下管理员和客户了。

因此，我需要所有新用户在填写 WooCommerce 注册表时，在“

Buyer

”和“

Seller

”之间进行选择。

这是我正在使用的代码：

add_action( 'woocommerce_register_form', 'user_role_dropdown' );
function user_role_dropdown() {

    $wp_roles = new WP_Roles();
    $wp_roles->use_db = true;
    echo '<select id="role" name="role">';
    
        foreach ( $wp_roles->roles as $key=>$value ) {
        if ( ( $value['name'] !== 'Buyer') && ( $value['name'] !== 'Seller' ) ) {
    
        echo '<option value="' . $key . '">' . $value['name'] . '</option>';
        }
    }
    echo '</select>';
}

add_action( 'personal_options_update', 'save_role_selection_field' );
add_action( 'edit_user_profile_update', 'save_role_selection_field' );
function save_role_selection_field( $user_id ) {

    update_user_meta( $user_id, 'role', $_POST['role'] );
    $user = new WP_User( $user_id );
    $current_user_role = get_current_user_role();
    $user->remove_role( $current_user_role );
    $user->add_role( $_POST['role'] );
}

function get_current_user_role() {
    global $current_user;
    get_currentuserinfo();
    $user_roles = $current_user->roles;
    $user_role = array_shift( $user_roles );
    return $user_role;
};


add_action( 'user_register', 'register_role' );
function register_role( $user_id, $password="", $meta=array() ) {

    $userdata = array();
    $userdata[ 'ID' ] = $user_id;
    $userdata[ 'role' ] = $_POST[ 'role' ];
    if ( $userdata[ 'role' ] ) {
    wp_update_user( $userdata );
    }
}

add_filter( 'registration_errors', 'new_usersroles_registration_errors', 10, 3 );
function new_usersroles_registration_errors( $errors, $sanitized_user_login, $user_email ) {

    if ( empty( $_POST['role'] ) || ! empty( $_POST['role'] ) && trim( $_POST['role'] ) == '' ) {

        $errors->add( 'role_error', __( '<strong>FYI:</strong> You must choose a role for your account. Buyers can not sell. Sellers cannot buy.', 'woocommerce' ) );
    }

    return $errors;
}

add_action( 'user_register', 'new_usersroles_user_register' );
function new_usersroles_user_register( $user_id ) {
$user_id = wp_update_user( array( 'ID' => $user_id, 'role' => $_POST[ 'role' ] ) );
}

Answer 1

对所需的钩子有一些混乱和一些小错误。使用以下命令向 WooCommerce 注册表单显示自定义选择字段（验证并保存数据）：

add_action( 'woocommerce_register_form', 'display_user_role_registration_field' );
function display_user_role_registration_field() {

    $wp_roles = new WP_Roles();
    $wp_roles->use_db = true;

    echo '<select id="role" name="role">
        <option value="">' . __("Select a role") . '</option>';

    foreach ( $wp_roles->roles as $role_key => $role_data ) {
        if ( in_array($role_data['name'], ['Buyer', 'Seller']) ) {
            echo '<option value="' . $role_key . '">' . $role_data['name'] . '</option>';
        }
    }
    echo '</select>';
}

// Validate WooCommerce registration select role fields.
add_action( 'woocommerce_register_post', 'validate_user_role_registration_field', 10, 3 );
function validate_user_role_registration_field($username, $email, $validation_errors) {
    if (isset($_POST['role']) && empty($_POST['role']) ) {
        $message = __('<strong>FYI:</strong> You must choose a role for your account. Buyers can not sell. Sellers cannot buy.', 'woocommerce');
        $validation_errors->add('role_error', $message);
    }
    return $validation_errors;
}

// Save WooCommerce registration select role fields.
add_action( 'woocommerce_created_customer', 'save_user_role_registration_field' );
function save_user_role_registration_field( $customer_id ) {
    if ( isset($_POST['role']) && ! empty($_POST['role']) ) {
        $user = new WP_User($customer_id);
        $user->set_role($_POST['role']);
    }
}

代码位于活动子主题（或活动主题）的 function.php 文件中。已测试并有效。

相关：Woocommerce 注册中的附加用户角色选择字段