[Summary view] [Print] [Text view]

   1  <?php
   2  class WP_Admin_Bar {
   3      private $nodes = array();
   4      private $bound = false;
   5      public $user;
   6  
   7  	public function __get( $name ) {
   8          switch ( $name ) {
   9              case 'proto' :
  10                  return is_ssl() ? 'https://' : 'http://';
  11                  break;
  12              case 'menu' :
  13                  _deprecated_argument( 'WP_Admin_Bar', '3.3', 'Modify admin bar nodes with WP_Admin_Bar::get_node(), WP_Admin_Bar::add_node(), and WP_Admin_Bar::remove_node(), not the <code>menu</code> property.' );
  14                  return array(); // Sorry, folks.
  15                  break;
  16          }
  17      }
  18  
  19  	public function initialize() {
  20          $this->user = new stdClass;
  21  
  22          if ( is_user_logged_in() ) {
  23              /* Populate settings we need for the menu based on the current user. */
  24              $this->user->blogs = get_blogs_of_user( get_current_user_id() );
  25              if ( is_multisite() ) {
  26                  $this->user->active_blog = get_active_blog_for_user( get_current_user_id() );
  27                  $this->user->domain = empty( $this->user->active_blog ) ? user_admin_url() : trailingslashit( get_home_url( $this->user->active_blog->blog_id ) );
  28                  $this->user->account_domain = $this->user->domain;
  29              } else {
  30                  $this->user->active_blog = $this->user->blogs[get_current_blog_id()];
  31                  $this->user->domain = trailingslashit( home_url() );
  32                  $this->user->account_domain = $this->user->domain;
  33              }
  34          }
  35  
  36          add_action( 'wp_head', 'wp_admin_bar_header' );
  37  
  38          add_action( 'admin_head', 'wp_admin_bar_header' );
  39  
  40          if ( current_theme_supports( 'admin-bar' ) ) {
  41              $admin_bar_args = get_theme_support( 'admin-bar' ); // add_theme_support( 'admin-bar', array( 'callback' => '__return_false') );
  42              $header_callback = $admin_bar_args[0]['callback'];
  43          }
  44  
  45          if ( empty($header_callback) )
  46              $header_callback = '_admin_bar_bump_cb';
  47  
  48          add_action('wp_head', $header_callback);
  49  
  50          wp_enqueue_script( 'admin-bar' );
  51          wp_enqueue_style( 'admin-bar' );
  52  
  53          do_action( 'admin_bar_init' );
  54      }
  55  
  56  	public function add_menu( $node ) {
  57          $this->add_node( $node );
  58      }
  59  
  60  	public function remove_menu( $id ) {
  61          $this->remove_node( $id );
  62      }
  63  
  64      /**
  65       * Add a node to the menu.
  66       *
  67       * @param array $args - The arguments for each node.
  68       * - id         - string    - The ID of the item.
  69       * - title      - string    - The title of the node.
  70       * - parent     - string    - The ID of the parent node. Optional.
  71       * - href       - string    - The link for the item. Optional.
  72       * - group      - boolean   - If the node is a group. Optional. Default false.
  73       * - meta       - array     - Meta data including the following keys: html, class, onclick, target, title, tabindex.
  74       */
  75  	public function add_node( $args ) {
  76          // Shim for old method signature: add_node( $parent_id, $menu_obj, $args )
  77          if ( func_num_args() >= 3 && is_string( func_get_arg(0) ) )
  78              $args = array_merge( array( 'parent' => func_get_arg(0) ), func_get_arg(2) );
  79  
  80          if ( is_object( $args ) )
  81              $args = get_object_vars( $args );
  82  
  83          // Ensure we have a valid title.
  84          if ( empty( $args['id'] ) ) {
  85              if ( empty( $args['title'] ) )
  86                  return;
  87  
  88              _doing_it_wrong( __METHOD__, __( 'The menu ID should not be empty.' ), '3.3' );
  89              // Deprecated: Generate an ID from the title.
  90              $args['id'] = esc_attr( sanitize_title( trim( $args['title'] ) ) );
  91          }
  92  
  93          $defaults = array(
  94              'id'     => false,
  95              'title'  => false,
  96              'parent' => false,
  97              'href'   => false,
  98              'group'  => false,
  99              'meta'   => array(),
 100          );
 101  
 102          // If the node already exists, keep any data that isn't provided.
 103          if ( $maybe_defaults = $this->get_node( $args['id'] ) )
 104              $defaults = get_object_vars( $maybe_defaults );
 105  
 106          // Do the same for 'meta' items.
 107          if ( ! empty( $defaults['meta'] ) && empty( $args['meta'] ) )
 108              $args['meta'] = wp_parse_args( $args['meta'], $defaults['meta'] );
 109  
 110          $args = wp_parse_args( $args, $defaults );
 111  
 112          $back_compat_parents = array(
 113              'my-account-with-avatar' => array( 'my-account', '3.3' ),
 114              'my-blogs'               => array( 'my-sites',   '3.3' ),
 115          );
 116  
 117          if ( isset( $back_compat_parents[ $args['parent'] ] ) ) {
 118              list( $new_parent, $version ) = $back_compat_parents[ $args['parent'] ];
 119              _deprecated_argument( __METHOD__, $version, sprintf( 'Use <code>%s</code> as the parent for the <code>%s</code> admin bar node instead of <code>%s</code>.', $new_parent, $args['id'], $args['parent'] ) );
 120              $args['parent'] = $new_parent;
 121          }
 122  
 123          $this->_set_node( $args );
 124      }
 125  
 126  	final protected function _set_node( $args ) {
 127          $this->nodes[ $args['id'] ] = (object) $args;
 128      }
 129  
 130      /**
 131       * Gets a node.
 132       *
 133       * @return object Node.
 134       */
 135  	final public function get_node( $id ) {
 136          if ( $node = $this->_get_node( $id ) )
 137              return clone $node;
 138      }
 139  
 140  	final protected function _get_node( $id ) {
 141          if ( $this->bound )
 142              return;
 143  
 144          if ( empty( $id ) )
 145              $id = 'root';
 146  
 147          if ( isset( $this->nodes[ $id ] ) )
 148              return $this->nodes[ $id ];
 149      }
 150  
 151  	final public function get_nodes() {
 152         if ( ! $nodes = $this->_get_nodes() )
 153            return;
 154  
 155         foreach ( $nodes as &$node ) {
 156             $node = clone $node;
 157         }
 158         return $nodes;
 159      }
 160  
 161  	final protected function _get_nodes() {
 162          if ( $this->bound )
 163              return;
 164  
 165          return $this->nodes;
 166      }
 167  
 168      /**
 169       * Add a group to a menu node.
 170       *
 171       * @since 3.3.0
 172       *
 173       * @param array $args - The arguments for each node.
 174       * - id         - string    - The ID of the item.
 175       * - parent     - string    - The ID of the parent node. Optional. Default root.
 176       * - meta       - array     - Meta data including the following keys: class, onclick, target, title.
 177       */
 178  	final public function add_group( $args ) {
 179          $args['group'] = true;
 180  
 181          $this->add_node( $args );
 182      }
 183  
 184      /**
 185       * Remove a node.
 186       *
 187       * @return object The removed node.
 188       */
 189  	public function remove_node( $id ) {
 190          $this->_unset_node( $id );
 191      }
 192  
 193  	final protected function _unset_node( $id ) {
 194          unset( $this->nodes[ $id ] );
 195      }
 196  
 197  	public function render() {
 198          $root = $this->_bind();
 199          if ( $root )
 200              $this->_render( $root );
 201      }
 202  
 203  	final protected function _bind() {
 204          if ( $this->bound )
 205              return;
 206  
 207          // Add the root node.
 208          // Clear it first, just in case. Don't mess with The Root.
 209          $this->remove_node( 'root' );
 210          $this->add_node( array(
 211              'id'    => 'root',
 212              'group' => false,
 213          ) );
 214  
 215          // Normalize nodes: define internal 'children' and 'type' properties.
 216          foreach ( $this->_get_nodes() as $node ) {
 217              $node->children = array();
 218              $node->type = ( $node->group ) ? 'group' : 'item';
 219              unset( $node->group );
 220  
 221              // The Root wants your orphans. No lonely items allowed.
 222              if ( ! $node->parent )
 223                  $node->parent = 'root';
 224          }
 225  
 226          foreach ( $this->_get_nodes() as $node ) {
 227              if ( 'root' == $node->id )
 228                  continue;
 229  
 230              // Fetch the parent node. If it isn't registered, ignore the node.
 231              if ( ! $parent = $this->_get_node( $node->parent ) ) {
 232                  continue;
 233              }
 234  
 235              // Generate the group class (we distinguish between top level and other level groups).
 236              $group_class = ( $node->parent == 'root' ) ? 'ab-top-menu' : 'ab-submenu';
 237  
 238              if ( $node->type == 'group' ) {
 239                  if ( empty( $node->meta['class'] ) )
 240                      $node->meta['class'] = '';
 241                  $node->meta['class'] .= ' ' . $group_class;
 242              }
 243  
 244              // Items in items aren't allowed. Wrap nested items in 'default' groups.
 245              if ( $parent->type == 'item' && $node->type == 'item' ) {
 246                  $default_id = $parent->id . '-default';
 247                  $default    = $this->_get_node( $default_id );
 248  
 249                  // The default group is added here to allow groups that are
 250                  // added before standard menu items to render first.
 251                  if ( ! $default ) {
 252                      // Use _set_node because add_node can be overloaded.
 253                      // Make sure to specify default settings for all properties.
 254                      $this->_set_node( array(
 255                          'id'        => $default_id,
 256                          'parent'    => $parent->id,
 257                          'type'      => 'group',
 258                          'children'  => array(),
 259                          'meta'      => array(
 260                              'class'     => $group_class,
 261                          ),
 262                          'title'     => false,
 263                          'href'      => false,
 264                      ) );
 265                      $default = $this->_get_node( $default_id );
 266                      $parent->children[] = $default;
 267                  }
 268                  $parent = $default;
 269  
 270              // Groups in groups aren't allowed. Add a special 'container' node.
 271              // The container will invisibly wrap both groups.
 272              } elseif ( $parent->type == 'group' && $node->type == 'group' ) {
 273                  $container_id = $parent->id . '-container';
 274                  $container    = $this->_get_node( $container_id );
 275  
 276                  // We need to create a container for this group, life is sad.
 277                  if ( ! $container ) {
 278                      // Use _set_node because add_node can be overloaded.
 279                      // Make sure to specify default settings for all properties.
 280                      $this->_set_node( array(
 281                          'id'       => $container_id,
 282                          'type'     => 'container',
 283                          'children' => array( $parent ),
 284                          'parent'   => false,
 285                          'title'    => false,
 286                          'href'     => false,
 287                          'meta'     => array(),
 288                      ) );
 289  
 290                      $container = $this->_get_node( $container_id );
 291  
 292                      // Link the container node if a grandparent node exists.
 293                      $grandparent = $this->_get_node( $parent->parent );
 294  
 295                      if ( $grandparent ) {
 296                          $container->parent = $grandparent->id;
 297  
 298                          $index = array_search( $parent, $grandparent->children, true );
 299                          if ( $index === false )
 300                              $grandparent->children[] = $container;
 301                          else
 302                              array_splice( $grandparent->children, $index, 1, array( $container ) );
 303                      }
 304  
 305                      $parent->parent = $container->id;
 306                  }
 307  
 308                  $parent = $container;
 309              }
 310  
 311              // Update the parent ID (it might have changed).
 312              $node->parent = $parent->id;
 313  
 314              // Add the node to the tree.
 315              $parent->children[] = $node;
 316          }
 317  
 318          $root = $this->_get_node( 'root' );
 319          $this->bound = true;
 320          return $root;
 321      }
 322  
 323  	final protected function _render( $root ) {
 324          global $is_IE;
 325  
 326          // Add browser classes.
 327          // We have to do this here since admin bar shows on the front end.
 328          $class = 'nojq nojs';
 329          if ( $is_IE ) {
 330              if ( strpos( $_SERVER['HTTP_USER_AGENT'], 'MSIE 7' ) )
 331                  $class .= ' ie7';
 332              elseif ( strpos( $_SERVER['HTTP_USER_AGENT'], 'MSIE 8' ) )
 333                  $class .= ' ie8';
 334              elseif ( strpos( $_SERVER['HTTP_USER_AGENT'], 'MSIE 9' ) )
 335                  $class .= ' ie9';
 336          } elseif ( wp_is_mobile() ) {
 337              $class .= ' mobile';
 338          }
 339  
 340          ?>
 341          <div id="wpadminbar" class="<?php echo $class; ?>" role="navigation">
 342              <div class="quicklinks">
 343                  <?php foreach ( $root->children as $group ) {
 344                      $this->_render_group( $group );
 345                  } ?>
 346              </div>
 347          </div>
 348  
 349          <?php
 350      }
 351  
 352  	final protected function _render_container( $node ) {
 353          if ( $node->type != 'container' || empty( $node->children ) )
 354              return;
 355  
 356          ?><div id="<?php echo esc_attr( 'wp-admin-bar-' . $node->id ); ?>" class="ab-group-container"><?php
 357              foreach ( $node->children as $group ) {
 358                  $this->_render_group( $group );
 359              }
 360          ?></div><?php
 361      }
 362  
 363  	final protected function _render_group( $node ) {
 364          if ( $node->type == 'container' )
 365              return $this->_render_container( $node );
 366  
 367          if ( $node->type != 'group' || empty( $node->children ) )
 368              return;
 369  
 370          $class = empty( $node->meta['class'] ) ? '' : $node->meta['class'];
 371  
 372          ?><ul id="<?php echo esc_attr( 'wp-admin-bar-' . $node->id ); ?>" class="<?php echo esc_attr( $class ); ?>"><?php
 373              foreach ( $node->children as $item ) {
 374                  $this->_render_item( $item );
 375              }
 376          ?></ul><?php
 377      }
 378  
 379  	final protected function _render_item( $node ) {
 380          if ( $node->type != 'item' )
 381              return;
 382  
 383          $is_parent = ! empty( $node->children );
 384          $has_link  = ! empty( $node->href );
 385  
 386          $tabindex = isset( $node->meta['tabindex'] ) ? (int) $node->meta['tabindex'] : 10;
 387  
 388          $menuclass = '';
 389          $aria_attributes = 'tabindex="' . $tabindex . '"';
 390  
 391          if ( $is_parent ) {
 392              $menuclass = 'menupop';
 393              $aria_attributes .= ' aria-haspopup="true"';
 394          }
 395  
 396          if ( ! empty( $node->meta['class'] ) )
 397              $menuclass .= ' ' . $node->meta['class'];
 398  
 399          ?>
 400  
 401          <li id="<?php echo esc_attr( 'wp-admin-bar-' . $node->id ); ?>" class="<?php echo esc_attr( $menuclass ); ?>"><?php
 402              if ( $has_link ):
 403                  ?><a class="ab-item" <?php echo $aria_attributes; ?> href="<?php echo esc_url( $node->href ) ?>"<?php
 404                      if ( ! empty( $node->meta['onclick'] ) ) :
 405                          ?> onclick="<?php echo esc_js( $node->meta['onclick'] ); ?>"<?php
 406                      endif;
 407                  if ( ! empty( $node->meta['target'] ) ) :
 408                      ?> target="<?php echo esc_attr( $node->meta['target'] ); ?>"<?php
 409                  endif;
 410                  if ( ! empty( $node->meta['title'] ) ) :
 411                      ?> title="<?php echo esc_attr( $node->meta['title'] ); ?>"<?php
 412                  endif;
 413                  ?>><?php
 414              else:
 415                  ?><div class="ab-item ab-empty-item" <?php echo $aria_attributes;
 416                  if ( ! empty( $node->meta['title'] ) ) :
 417                      ?> title="<?php echo esc_attr( $node->meta['title'] ); ?>"<?php
 418                  endif;
 419                  ?>><?php
 420              endif;
 421  
 422              echo $node->title;
 423  
 424              if ( $has_link ) :
 425                  ?></a><?php
 426              else:
 427                  ?></div><?php
 428              endif;
 429  
 430              if ( $is_parent ) :
 431                  ?><div class="ab-sub-wrapper"><?php
 432                      foreach ( $node->children as $group ) {
 433                          $this->_render_group( $group );
 434                      }
 435                  ?></div><?php
 436              endif;
 437  
 438              if ( ! empty( $node->meta['html'] ) )
 439                  echo $node->meta['html'];
 440  
 441              ?>
 442          </li><?php
 443      }
 444  
 445  	public function recursive_render( $id, $node ) {
 446          _deprecated_function( __METHOD__, '3.3', 'WP_Admin_bar::render(), WP_Admin_Bar::_render_item()' );
 447          $this->_render_item( $node );
 448      }
 449  
 450  	public function add_menus() {
 451          // User related, aligned right.
 452          add_action( 'admin_bar_menu', 'wp_admin_bar_my_account_menu', 0 );
 453          add_action( 'admin_bar_menu', 'wp_admin_bar_search_menu', 4 );
 454          add_action( 'admin_bar_menu', 'wp_admin_bar_my_account_item', 7 );
 455  
 456          // Site related.
 457          add_action( 'admin_bar_menu', 'wp_admin_bar_wp_menu', 10 );
 458          add_action( 'admin_bar_menu', 'wp_admin_bar_my_sites_menu', 20 );
 459          add_action( 'admin_bar_menu', 'wp_admin_bar_site_menu', 30 );
 460          add_action( 'admin_bar_menu', 'wp_admin_bar_updates_menu', 40 );
 461  
 462          // Content related.
 463          if ( ! is_network_admin() && ! is_user_admin() ) {
 464              add_action( 'admin_bar_menu', 'wp_admin_bar_comments_menu', 60 );
 465              add_action( 'admin_bar_menu', 'wp_admin_bar_new_content_menu', 70 );
 466          }
 467          add_action( 'admin_bar_menu', 'wp_admin_bar_edit_menu', 80 );
 468  
 469          add_action( 'admin_bar_menu', 'wp_admin_bar_add_secondary_groups', 200 );
 470  
 471          do_action( 'add_admin_bar_menus' );
 472      }
 473  }