WooCommerce我的账户页按指定用户角色显示自定义标签方法

问题说明 #

你已通过自定义代码在WooCommerce我的账户页面添加了全局可见的「Support」标签页，当前需要新增仅用户角色为partnersupport可见的专属VIP支持标签页，原有将函数定义包裹在current_user_can()判断块内的写法会触发站点致命错误，且存在端点冲突、无访问权限拦截的问题。

原有可正常运行的全局Support标签页代码如下：

function tb_add_support_endpoint() {
    add_rewrite_endpoint( 'support', EP_ROOT | EP_PAGES );
}
add_action( 'init', 'tb_add_support_endpoint' );

function tb_support_query_vars( $vars ) {
    $vars[] = 'support';
    return $vars;
}
add_filter( 'query_vars', 'tb_support_query_vars', 0 );

function tb_add_support_link_my_account( $items ) {
    $items['support'] = 'Support';
    return $items;
}
add_filter( 'woocommerce_account_menu_items', 'tb_add_support_link_my_account' );

function tb_support_content() {
    echo '<h3>Support</h3><p>Have a question? Use the form below to get in touch.</p>';
    echo do_shortcode( ' [wpforms id="XXX"] ' );
}
add_action( 'woocommerce_account_support_endpoint', 'tb_support_content' );

原有写法报错原因 #

• 将PHP命名函数的定义直接包裹在条件判断块中，WordPress钩子加载顺序早于当前用户身份初始化的时机，会触发「调用未定义函数」的致命错误
• 复用了已被普通Support标签页占用的support端点，会导致路由冲突
• 仅对菜单显示做了判断，未对端点直接访问做权限拦截，非授权用户输入URL即可访问到VIP支持页内容

正确实现代码 #

VIP支持页使用独立端点vip-support避免冲突，权限判断放在钩子回调函数内部执行，同时增加访问拦截逻辑。将以下代码追加到主题的functions.php或自定义功能插件中即可：

// 1. 注册VIP支持专属端点
function tbvip_add_support_endpoint() {
    add_rewrite_endpoint( 'vip-support', EP_ROOT | EP_PAGES );
}
add_action( 'init', 'tbvip_add_support_endpoint' );

// 2. 添加端点查询变量
function tbvip_support_query_vars( $vars ) {
    $vars[] = 'vip-support';
    return $vars;
}
add_filter( 'query_vars', 'tbvip_support_query_vars', 0 );

// 3. 仅对partnersupport角色显示菜单入口
function tbvip_add_support_link_my_account( $items ) {
    // 回调内部判断用户角色，符合权限才添加菜单项
    if ( current_user_can( 'partnersupport' ) ) {
        $items['vip-support'] = 'VIP Support';
    }
    return $items; 
}
add_filter( 'woocommerce_account_menu_items', 'tbvip_add_support_link_my_account' );

// 4. 渲染VIP支持页内容+访问权限拦截
function tbvip_support_content() {
    // 无权限访问时跳转回普通支持页，也可以替换为自定义无权限提示
    if ( !current_user_can( 'partnersupport' ) ) {
        wc_add_notice( '你无权访问VIP支持页面', 'error' );
        wp_safe_redirect( wc_get_page_permalink( 'myaccount' ) . 'support/' );
        exit;
    }
    echo '<h3>VIP Support</h3><p>VIP专属客服通道，请通过以下表单提交你的问题：</p>';
    // 替换XXX为你的VIP支持表单ID
    echo do_shortcode( ' [wpforms id="XXX"] ' );
}
add_action( 'woocommerce_account_vip-support_endpoint', 'tbvip_support_content' );

生效注意事项 #

• 代码添加完成后，进入WordPress后台「设置」-「固定链接」页面，无需修改任何配置，直接点击底部「保存更改」按钮，刷新WordPress重写规则即可让新端点生效
• 若current_user_can('partnersupport')判断不生效，可以替换为基于用户角色的判断逻辑，将代码中所有current_user_can( 'partnersupport' )替换为如下判断即可：

  $current_user = wp_get_current_user();
  $is_partner_support = in_array( 'partnersupport', (array) $current_user->roles );
  

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