WordPress add_action如何根据用户角色设置条件优先级

问题场景 #

开发WordPress插件向管理员顶栏（Admin Bar）添加菜单项时，需要为不同用户角色设置不同菜单加载优先级：管理员角色的菜单位置与其他用户角色不同。
原有实现触发两类报错：

• call to undefined function wp_get_user_roles
• 手动引入pluggable.php后触发undefined constant AUTH_COOKIE

原有错误代码 #

优先级判断逻辑：

set_priority() {
    $priority = 51;
    if (current_user_can('manage_options')){
    $priority = 21;
    }
    return $priority;
    }
add_action('init','set_priority');

菜单添加逻辑：

function add_context_tab($wp_admin_bar) {

if(current_user_can('manage_options')){
    $url = network_site_url(). 'wp-admin/network/admin.php?page=content-page';
}
else{
    $url = network_site_url(). 'wp-admin/admin.php?page=content-page';
}

global $wp_admin_bar;

$wp_admin_bar->add_node( array(
    'id'    => 'content-page',
    'title' => '<span class="ab-icon dashicons dashicons-welcome-add-page"></span>' . __( 'Add content' ),
    'href'  => $url,
) );
}
$priority = set_priority();
add_action ('admin_bar_menu','add_context_tab',$priority);

错误根因 #

• WordPress钩子注册时传入的优先级参数，是PHP解析插件文件阶段就固定写入的。全局作用域下直接执行$priority = set_priority()时，WordPress核心还未完成用户身份初始化流程，此时调用current_user_can()等用户权限相关函数，必然触发未定义函数、未定义常量类报错。
• 将set_priority挂载到init钩子无法解决问题：全局作用域的代码不会等待init钩子执行完成再运行，优先级参数在add_action注册时就已经固定，不会后续动态更新。
• 手动引入pluggable.php会破坏WordPress核心的正常加载顺序，会引发更多连锁加载错误，属于错误操作。

正确实现代码 #

不需要提前全局计算优先级，将所有权限判断、钩子注册逻辑放到用户身份初始化完成的钩子节点后执行即可：

/**
 * 注册自定义Admin Bar菜单
 * 触发时机选在用户身份初始化完成后，避免权限函数未加载报错
 */
function wp_register_custom_admin_bar_tab() {
    // 此时用户权限数据已完全加载，可以安全判断角色
    if ( current_user_can( 'manage_options' ) ) {
        $menu_url = network_site_url() . 'wp-admin/network/admin.php?page=content-page';
        $menu_priority = 21;
    } else {
        $menu_url = network_site_url() . 'wp-admin/admin.php?page=content-page';
        $menu_priority = 51;
    }

    // 按对应优先级挂载菜单渲染逻辑
    add_action( 'admin_bar_menu', function( $wp_admin_bar ) use ( $menu_url ) {
        $wp_admin_bar->add_node( array(
            'id'    => 'content-page',
            'title' => '<span class="ab-icon dashicons dashicons-welcome-add-page"></span>' . __( 'Add content' ),
            'href'  => $menu_url,
        ) );
    }, $menu_priority );
}
add_action( 'set_current_user', 'wp_register_custom_admin_bar_tab' );

实现说明 #

• 选择set_current_user作为注册入口，这个钩子触发时WordPress已经完成当前登录用户的身份、角色、权限初始化，所有用户相关函数、常量都已正常加载，不会再出现未定义报错。
• 去掉了冗余的$wp_admin_bar全局声明，直接使用admin_bar_menu回调传入的WP_Admin_Bar实例操作，减少全局变量依赖。
• 完全遵循WordPress核心加载流程，不需要手动引入任何核心文件，不会触发连锁加载错误。
  如果需要兼容PHP 5.2以下无闭包支持的环境，可以把菜单渲染逻辑拆成两个独立函数，分别按对应优先级挂载到admin_bar_menu即可，判断逻辑保持不变。

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