在WooCommerce注册表单添加角色选择器遇白屏/致命错误求助

原代码触发白屏/致命错误的核心原因：woocommerce_register_form钩子传递的参数是HTML字符串，而非数组。你直接尝试修改数组结构，会导致PHP报错，进而引发白屏。

以下是修正后的完整实现代码，包含表单字段添加、验证、角色分配三个关键环节：

1. 添加角色选择字段到注册表单 #

// 在WooCommerce注册表单添加角色选择器
add_action('woocommerce_register_form', 'add_vendor_role_to_registration_form');
function add_vendor_role_to_registration_form() {
    ?>
    <p class="form-row form-row-wide">
        <label for="account-role"><?php esc_html_e('Choose Your Account Type', 'woocommerce'); ?> <span class="required">*</span></label>
        <select name="account-role" id="account-role" class="select" required>
            <option value="customer"><?php esc_html_e('Customer (Only Shopping)', 'woocommerce'); ?></option>
            <option value="vendor"><?php esc_html_e('Vendor (Can Add Products, Cannot Publish)', 'woocommerce'); ?></option>
        </select>
    </p>
    <?php
}

2. 验证表单字段合法性 #

// 验证角色选择字段
add_filter('woocommerce_registration_errors', 'validate_vendor_role_registration_field', 10, 3);
function validate_vendor_role_registration_field($errors, $username, $email) {
    if (!isset($_POST['account-role']) || !in_array($_POST['account-role'], ['customer', 'vendor'])) {
        $errors->add('invalid_role', __('Please select a valid account type.', 'woocommerce'));
    }
    return $errors;
}

3. 根据选择设置用户角色 #

// 用户注册后分配对应角色
add_action('woocommerce_created_customer', 'set_user_role_after_registration');
function set_user_role_after_registration($customer_id) {
    if (isset($_POST['account-role']) && in_array($_POST['account-role'], ['customer', 'vendor'])) {
        $role = sanitize_text_field($_POST['account-role']);
        $user = new WP_User($customer_id);
        
        // 清除默认角色，设置选中的角色
        $user->remove_all_roles();
        $user->add_role($role);
    }
}

额外说明 #

• 确保vendor角色已在你的WordPress站点中存在（可通过用户角色管理插件或代码预先创建）
• 若需限制Vendor角色的商品发布权限，可通过角色编辑工具，移除publish_products权限，保留edit_products权限

内容的提问来源于stack exchange，提问作者Learning WooStoreFront