能否为未登录访客分配WooCommerce用户角色实现差异化定价？

原代码问题分析 #

• 逻辑判断错误：!is_user_logged_in() == false 等价于 is_user_logged_in()，只会对已登录用户生效，完全不符合需求。
• 语法错误：if 语句的闭合括号误用了 }，导致代码无法正常解析执行。
• 对象操作无效：未登录时 wp_get_current_user() 返回的是ID为0的匿名用户对象，set_role() 方法仅对已注册（有有效ID）的用户生效，无法给访客设置角色。

正确解决方案 #

不需要给访客实际分配数据库中的用户角色，只需通过钩子模拟角色判断，让B2B插件将访客识别为guest角色即可。

步骤1：创建guest角色（如果未存在） #

先确保你的WordPress中有guest这个角色，可将以下代码添加到主题functions.php或自定义插件中：

// 激活时创建guest角色
function create_guest_role() {
    add_role(
        'guest',
        '访客',
        array(
            'read' => true, // 基础阅读权限，可按需调整
        )
    );
}
register_activation_hook(__FILE__, 'create_guest_role');

步骤2：修改访客的角色识别逻辑 #

添加以下代码，让WordPress在处理未登录用户时，将其角色识别为guest：

// 覆盖匿名用户的角色信息
function override_guest_user_role($user) {
    if (!is_user_logged_in() && $user->ID === 0) {
        // 为匿名用户设置guest角色
        $user->roles = array('guest');
        // 同步该角色的权限
        $guest_role = get_role('guest');
        if ($guest_role) {
            $user->allcaps = $guest_role->capabilities;
            $user->caps = $user->allcaps;
        }
    }
    return $user;
}
add_filter('wp_get_current_user', 'override_guest_user_role');

// 确保角色检查函数能识别guest角色
function add_guest_role_to_check($roles) {
    if (!is_user_logged_in()) {
        $roles['guest'] = true;
    }
    return $roles;
}
add_filter('wp_roles', 'add_guest_role_to_check');

补充提示 #

如果B2B for Woocommerce插件自带访客价格设置功能，优先使用插件原生选项，无需自定义代码。只有当插件无此功能时，再采用上述代码方案。

内容的提问来源于stack exchange，提问作者Jakub Potocki