HOME

<?php
/**
 * @category    Fishpig
 * @package     Fishpig_Wordpress
 * @license     http://fishpig.co.uk/license.txt
 * @author      Ben Tideswell <help@fishpig.co.uk>
 */

namespace FishPig\WordPress\Block\Sidebar\Widget;

class NavMenu extends AbstractWidget
{
	/**
	 * Get the menu model for the current widget
	 *
	 * @return Fishpig_Wordpress_Model_Menu
	 */
	public function getMenu()
	{
		if (!$this->hasMenu()) {
			$this->setMenu(false);

			$menu = $this->_factory->getFactory('Menu')->create()->load($this->_getData('nav_menu'));
			if ($menu) {
				$this->setMenu($menu);
			}
		}

		return $this->_getData('menu');
	}

	/**
	 * Retrieve the default title
	 *
	 * @return string
	 */
	public function getDefaultTitle()
	{
		if ($this->getMenu()) {
			return $this->getMenu()->getName();
		}

		return __('Menu');
	}

	protected function _beforeToHtml()
	{
		if (!$this->getTemplate()) {
			$this->setTemplate('sidebar/widget/navmenu.phtml');
		}

		return parent::_beforeToHtml();
	}

	/**
	 * Recursively uild and return tree html
	 *
	 * @return string
	 */
	public function getTreeHtml()
	{
		return $this->_getTreeHtmlLevel(0, $this->getMenu()->getMenuTreeObjects());
	}

	/**
	 * Build and return a single level of tree html and recurse to render sub items
	 *
	 * @param int $level Menu level (0-index)
	 * @param FishPig\WordPress\Model\Menu\Item[] $menuTreeObjects Collection of menu items
	 * @return string
	 */
	protected function _getTreeHtmlLevel($level, $menuTreeObjects)
	{
		$indentString = str_repeat("\t", $level);

		$html = '';

		foreach ($menuTreeObjects as $current) {
			$classes = [
				'menu-item',
				'menu-item-' . $current->getId(),
				'menu-item-type-' . $current->getItemType(),
				'menu-item-object-' . $current->getObjectType(),
			];

			if (count($current->getChildrenItems())) {
				$classes[] = 'menu-item-has-children';
			}

			$html .= $indentString . '<li id="menu-item-' . $current->getId() . '" class="' . implode(' ', $classes) . '">' . PHP_EOL;
			$html .= $indentString . "\t" . '<a href="' . $this->escapeHtml($current->getUrl()) . '" title="' . $this->escapeHtml($current->getLabel()) . '">';
			$html .= $this->escapeHtml($current->getLabel()) . '</a>' . PHP_EOL;

			if (count($current->getChildrenItems())) {
				$html .= $indentString . "\t" . '<ul class="sub-menu">' . PHP_EOL;
				$html .= $this->_getTreeHtmlLevel($level + 1, $current->getChildrenItems()->getItems());
				$html .= $indentString . "\t" . '</ul>' . PHP_EOL;
			}

			$html .= $indentString . '</li>' . PHP_EOL;
		}

		return $html;
	}
}