<?php

/**

 * Toolbar API: WP_Admin_Bar class

 *

 * @package WordPress

 * @subpackage Toolbar

 * @since 3.1.0

 */

/**

 * Core class used to implement the Toolbar API.

 *

 * @since 3.1.0

 */

class WP_Admin_Bar {

	private $nodes = array();

	private $bound = false;

	public $user;

	/**

	 * @param string $name

	 * @return string|array|void

	 */

	public function __get( $name ) {

		switch ( $name ) {

			case 'proto':

				return is_ssl() ? 'https://' : 'http://';

			case 'menu':

				_deprecated_argument( 'WP_Admin_Bar', '3.3.0', '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.' );

				return array(); // Sorry, folks.

		}

	}

	/**

	 */

	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' );

	}

	/**

	 *

	 * @since 3.3.0

	 *

	 * @param array $node The attributes that define the node.

	 */

	public function add_menu( $node ) {

		$this->add_node( $node );

	}

	/**

	 *

	 * @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.

	 *

	 * @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'. 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</code> as the parent for the <code>%s</code> admin bar node instead of <code>%s</code>.', $new_parent, $args['id'], $args['parent'] ) );

			$args['parent'] = $new_parent;

		}

		$this->_set_node( $args );

	}

	/**

	 * @param array $args

	 */

	final protected function _set_node( $args ) {

		$this->nodes[ $args['id'] ] = (object) $args;

	}

	/**

	 * Gets a node.

	 *

	 * @param string $id

	 * @return object|void Node.

	 */

	final public function get_node( $id ) {

		$node = $this->_get_node( $id );

		if ( $node ) {

			return clone $node;

		}

	}

	/**

	 * @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 ];

		}

	}

	/**

	 * @return array|void

	 */

	final public function get_nodes() {

		$nodes = $this->_get_nodes();

		if ( ! $nodes ) {

			return;

		}

		foreach ( $nodes as &$node ) {

			$node = clone $node;

		}

		return $nodes;

	}

	/**

	 * @return array|void

	 */

	final protected function _get_nodes() {

		if ( $this->bound ) {

			return;

		}

		return $this->nodes;

	}

	/**

	 *

	 * 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.

	 *

	 * @param string $id The ID of the item.

	 */

	public function remove_node( $id ) {

		$this->_unset_node( $id );

	}

	/**

	 * @param string $id

	 */

	final protected function _unset_node( $id ) {

		unset( $this->nodes[ $id ] );

	}

	/**

	 */

	public function render() {

		$root = $this->_bind();

		if ( $root ) {

			$this->_render( $root );

		}

	}

	/**

	 * @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(),

						)

					);