WP_Admin_Bar
概述
WP_Admin_Bar 是 WordPress 的核心类，用于实现 Toolbar API，生成登录后显示在网站顶部的工具栏。开发者可以通过 Hook 和修改该类来添加或移除工具栏中的选项。
关键要点
- WP_Admin_Bar 是 WordPress 3.3 版本后替代 Admin Bar 的 Toolbar 实现类。
- 主要通过 admin_bar_menu Hook 修改 $wp_admin_bar 对象来定制工具栏。
- 提供 add_node、remove_node、add_group 等方法管理工具栏节点和组。
- 包含 initialize、render 等内部方法用于初始化和渲染工具栏。
代码示例
add_action( 'admin_bar_menu', 'modify_admin_bar' );
function modify_admin_bar( $wp_admin_bar ){
// do something with $wp_admin_bar;
}注意事项
- 使用 add_node 方法时需确保提供有效的 ID 和 title 参数，否则可能触发警告。
- 部分方法如 recursive_render 已弃用，建议使用替代方案。
- 在修改工具栏时，注意处理多站点环境下的用户和博客数据。
Core class used to implement the Toolbar API.
Methods
| Name | Description |
|---|---|
| WP_Admin_Bar::__get | – |
| WP_Admin_Bar::_bind | – |
| WP_Admin_Bar::_get_node | – |
| WP_Admin_Bar::_get_nodes | – |
| WP_Admin_Bar::_render | – |
| WP_Admin_Bar::_render_container | – |
| WP_Admin_Bar::_render_group | – |
| WP_Admin_Bar::_render_item | – |
| WP_Admin_Bar::_set_node | – |
| WP_Admin_Bar::_unset_node | – |
| WP_Admin_Bar::add_group | Adds a group to a toolbar menu node. |
| WP_Admin_Bar::add_menu | Adds a node (menu item) to the admin bar menu. |
| WP_Admin_Bar::add_menus | Adds menus to the admin bar. |
| WP_Admin_Bar::add_node | Adds a node to the menu. |
| WP_Admin_Bar::get_node | Gets a node. |
| WP_Admin_Bar::get_nodes | – |
| WP_Admin_Bar::initialize | Initializes the admin bar. |
| WP_Admin_Bar::recursive_render | Renders toolbar items recursively. — deprecated |
| WP_Admin_Bar::remove_menu | Removes a node from the admin bar. |
| WP_Admin_Bar::remove_node | Remove a node. |
| WP_Admin_Bar::render | – |
Source
class WP_Admin_Bar {
private $nodes = array();
private $bound = false;
public $user;
/**
* Deprecated menu property.
*
* @since 3.1.0
* @deprecated 3.3.0 Modify admin bar nodes with WP_Admin_Bar::get_node(),
* WP_Admin_Bar::add_node(), and WP_Admin_Bar::remove_node().
* @var array
*/
public $menu = array();
/**
* Initializes the admin bar.
*
* @since 3.1.0
*/
public function initialize() {
$this->user = new stdClass();
if ( is_user_logged_in() ) {
/* Populate settings we need for the menu based on the current user. */
$this->user->blogs = get_blogs_of_user( get_current_user_id() );
if ( is_multisite() ) {
$this->user->active_blog = get_active_blog_for_user( get_current_user_id() );
$this->user->domain = empty( $this->user->active_blog ) ? user_admin_url() : trailingslashit( get_home_url( $this->user->active_blog->blog_id ) );
$this->user->account_domain = $this->user->domain;
} else {
$this->user->active_blog = $this->user->blogs[ get_current_blog_id() ];
$this->user->domain = trailingslashit( home_url() );
$this->user->account_domain = $this->user->domain;
}
}
add_action( 'wp_head', 'wp_admin_bar_header' );
add_action( 'admin_head', 'wp_admin_bar_header' );
if ( current_theme_supports( 'admin-bar' ) ) {
/**
* To remove the default padding styles from WordPress for the Toolbar, use the following code:
* add_theme_support( 'admin-bar', array( 'callback' => '__return_false' ) );
*/
$admin_bar_args = get_theme_support( 'admin-bar' );
$header_callback = $admin_bar_args[0]['callback'];
}
if ( empty( $header_callback ) ) {
$header_callback = '_admin_bar_bump_cb';
}
add_action( 'wp_head', $header_callback );
wp_enqueue_script( 'admin-bar' );
wp_enqueue_style( 'admin-bar' );
/**
* Fires after WP_Admin_Bar is initialized.
*
* @since 3.1.0
*/
do_action( 'admin_bar_init' );
}
/**
* Adds a node (menu item) to the admin bar menu.
*
* @since 3.3.0
*
* @param array $node The attributes that define the node.
*/
public function add_menu( $node ) {
$this->add_node( $node );
}
/**
* Removes a node from the admin bar.
*
* @since 3.1.0
*
* @param string $id The menu slug to remove.
*/
public function remove_menu( $id ) {
$this->remove_node( $id );
}
/**
* Adds a node to the menu.
*
* @since 3.1.0
* @since 4.5.0 Added the ability to pass 'lang' and 'dir' meta data.
* @since 6.5.0 Added the ability to pass 'menu_title' for an ARIA menu name.
*
* @param array $args {
* Arguments for adding a node.
*
* @type string $id ID of the item.
* @type string $title Title of the node.
* @type string $parent Optional. ID of the parent node.
* @type string $href Optional. Link for the item.
* @type bool $group Optional. Whether or not the node is a group. Default false.
* @type array $meta Meta data including the following keys: 'html', 'class', 'rel', 'lang', 'dir',
* 'onclick', 'target', 'title', 'tabindex', 'menu_title'. Default empty.
* }
*/
public function add_node( $args ) {
// Shim for old method signature: add_node( $parent_id, $menu_obj, $args ).
if ( func_num_args() >= 3 && is_string( $args ) ) {
$args = array_merge( array( 'parent' => $args ), func_get_arg( 2 ) );
}
if ( is_object( $args ) ) {
$args = get_object_vars( $args );
}
// Ensure we have a valid title.
if ( empty( $args['id'] ) ) {
if ( empty( $args['title'] ) ) {
return;
}
_doing_it_wrong( __METHOD__, __( 'The menu ID should not be empty.' ), '3.3.0' );
// Deprecated: Generate an ID from the title.
$args['id'] = esc_attr( sanitize_title( trim( $args['title'] ) ) );
}
$defaults = array(
'id' => false,
'title' => false,
'parent' => false,
'href' => false,
'group' => false,
'meta' => array(),
);
// If the node already exists, keep any data that isn't provided.
$maybe_defaults = $this->get_node( $args['id'] );
if ( $maybe_defaults ) {
$defaults = get_object_vars( $maybe_defaults );
}
// Do the same for 'meta' items.
if ( ! empty( $defaults['meta'] ) && ! empty( $args['meta'] ) ) {
$args['meta'] = wp_parse_args( $args['meta'], $defaults['meta'] );
}
$args = wp_parse_args( $args, $defaults );
$back_compat_parents = array(
'my-account-with-avatar' => array( 'my-account', '3.3' ),
'my-blogs' => array( 'my-sites', '3.3' ),
);
if ( isset( $back_compat_parents[ $args['parent'] ] ) ) {
list( $new_parent, $version ) = $back_compat_parents[ $args['parent'] ];
_deprecated_argument( __METHOD__, $version, sprintf( 'Use <code>%s
as the parent for the %s admin bar node instead of %s.', $new_parent, $args['id'], $args['parent'] ) );
$args['parent'] = $new_parent;
}
$this->_set_node( $args );
}
/**
* @since 3.3.0
*
* @param array $args
*/
final protected function _set_node( $args ) {
$this->nodes[ $args['id'] ] = (object) $args;
}
/**
* Gets a node.
*
* @since 3.3.0
*
* @param string $id
* @return object|void Node.
*/
final public function get_node( $id ) {
$node = $this->_get_node( $id );
if ( $node ) {
return clone $node;
}
}
/**
* @since 3.3.0
*
* @param string $id
* @return object|void
*/
final protected function _get_node( $id ) {
if ( $this->bound ) {
return;
}
if ( empty( $id ) ) {
$id = 'root';
}
if ( isset( $this->nodes[ $id ] ) ) {
return $this->nodes[ $id ];
}
}
/**
* @since 3.3.0
*
* @return array|void
*/
final public function get_nodes() {
$nodes = $this->_get_nodes();
if ( ! $nodes ) {
return;
}
foreach ( $nodes as &$node ) {
$node = clone $node;
}
return $nodes;
}
/**
* @since 3.3.0
*
* @return array|void
*/
final protected function _get_nodes() {
if ( $this->bound ) {
return;
}
return $this->nodes;
}
/**
* Adds a group to a toolbar menu node.
*
* Groups can be used to organize toolbar items into distinct sections of a toolbar menu.
*
* @since 3.3.0
*
* @param array $args {
* Array of arguments for adding a group.
*
* @type string $id ID of the item.
* @type string $parent Optional. ID of the parent node. Default 'root'.
* @type array $meta Meta data for the group including the following keys:
* 'class', 'onclick', 'target', and 'title'.
* }
*/
final public function add_group( $args ) {
$args['group'] = true;
$this->add_node( $args );
}
/**
* Remove a node.
*
* @since 3.1.0
*
* @param string $id The ID of the item.
*/
public function remove_node( $id ) {
$this->_unset_node( $id );
}
/**
* @since 3.3.0
*
* @param string $id
*/
final protected function _unset_node( $id ) {
unset( $this->nodes[ $id ] );
}
/**
* @since 3.1.0
*/
public function render() {
$root = $this->_bind();
if ( $root ) {
$this->_render( $root );
}
}
/**
* @since 3.3.0
*
* @return object|void
*/
final protected function _bind() {
if ( $this->bound ) {
return;
}
/*
* Add the root node.
* Clear it first, just in case. Don't mess with The Root.
*/
$this->remove_node( 'root' );
$this->add_node(
array(
'id' => 'root',
'group' => false,
)
);
// Normalize nodes: define internal 'children' and 'type' properties.
foreach ( $this->_get_nodes() as $node ) {
$node->children = array();
$node->type = ( $node->group ) ? 'group' : 'item';
unset( $node->group );
// The Root wants your orphans. No lonely items allowed.
if ( ! $node->parent ) {
$node->parent = 'root';
}
}
foreach ( $this->_get_nodes() as $node ) {
if ( 'root' === $node->id ) {
continue;
}
// Fetch the parent node. If it isn't registered, ignore the node.
$parent = $this->_get_node( $node->parent );
if ( ! $parent ) {
continue;
}
// Generate the group class (we distinguish between top level and other level groups).
$group_class = ( 'root' === $node->parent ) ? 'ab-top-menu' : 'ab-submenu';
if ( 'group' === $node->type ) {
if ( empty( $node->meta['class'] ) ) {
$node->meta['class'] = $group_class;
} else {
$node->meta['class'] .= ' ' . $group_class;
}
}
// Items in items aren't allowed. Wrap nested items in 'default' groups.
if ( 'item' === $parent->type && 'item' === $node->type ) {
$default_id = $parent->id . '-default';
$default = $this->_get_node( $default_id );
/*
* The default group is added here to allow groups that are
* added before standard menu items to render first.
*/
if ( ! $default ) {
/*
* Use _set_node because add_node can be overloaded.
* Make sure to specify default settings for all properties.
*/
$this->_set_node(
array(
'id' => $default_id,
'parent' => $parent->id,
'type' => 'group',
'children' => array(),
'meta' => array(
'class' => $group_class,
),
'title' => false,
'href' => false,
)
);
$default = $this->_get_node( $default_id );
$parent->children[] = $default;
}
$parent = $default;
/*
* Groups in groups aren't allowed. Add a special 'container' node.
* The container will invisibly wrap both groups.
*/
} elseif ( 'group' === $parent->type && 'group' === $node->type ) {
$container_id = $parent->id . '-container';
$container = $this->_get_node( $container_id );
// We need to create a container for this group, life is sad.
if ( ! $container ) {
/*
* Use _set_node because add_node can be overloaded.
* Make sure to specify default settings for all properties.
*/
$this->_set_node(
array(
'id' => $container_id,
'type' => 'container',
'children' => array( $parent ),
'parent' => false,
'title' => false,
'href' => false,
'meta' => array(),
)
);
$container = $this->_get_node( $container_id );
// Link the container node if a grandparent node exists.
$grandparent = $this->_get_node( $parent->parent );
if ( $grandparent ) {
$container->parent = $grandparent->id;
$index = array_search( $parent, $grandparent->children, true );
if ( false === $index ) {
$grandparent->children[] = $container;
} else {
array_splice( $grandparent->children, $index, 1, array( $container ) );
}
}
$parent->parent = $container->id;
}
$parent = $container;
}
// Update the parent ID (it might have changed).
$node->parent = $parent->id;
// Add the node to the tree.
$parent->children[] = $node;
}
$root = $this->_get_node( 'root' );
$this->bound = true;
return $root;
}
/**
* @since 3.3.0
*
* @param object $root
*/
final protected function _render( $root ) {
/*
* Add browser classes.
* We have to do this here since admin bar shows on the front end.
*/
$class = 'nojq nojs';
if ( wp_is_mobile() ) {
$class .= ' mobile';
}
?>
<div id="wpadminbar" class="">
<div class="quicklinks" id="wp-toolbar" role="navigation" aria-label="">
children as $group ) {
$this->_render_group( $group );
}
?>