MenuBuilder.php

<?php

/**
 * (c) FSi sp. z o.o. <info@fsi.pl>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace FSi\Bundle\AdminBundle\Menu;

use FSi\Bundle\AdminBundle\Admin\ElementInterface;
use FSi\Bundle\AdminBundle\Admin\Manager;
use Knp\Menu\FactoryInterface;
use Knp\Menu\MenuItem;
use Symfony\Component\HttpFoundation\Request;

class MenuBuilder
{
    /**
     * @var \Knp\Menu\FactoryInterface
     */
    protected $factory;

    /**
     * @var \FSi\Bundle\AdminBundle\Admin\Manager
     */
    protected $manager;

    /**
     * @var \Symfony\Component\HttpFoundation\Request
     */
    protected $request;

    /**
     * @param \Knp\Menu\FactoryInterface $factory
     * @param \FSi\Bundle\AdminBundle\Admin\Manager $manager
     */
    public function __construct(FactoryInterface $factory, Manager $manager)
    {
        $this->factory = $factory;
        $this->manager = $manager;
    }

    /**
     * @param \Symfony\Component\HttpFoundation\Request $request
     */
    public function setRequest(Request $request = null)
    {
        $this->request = $request;
    }

    /**
     * @return \Knp\Menu\ItemInterface
     */
    public function createMenu()
    {
        $menu = $this->createMenuRoot();

        if (isset($this->request)) {
            $menu->setCurrentUri($this->request->getRequestUri());
        }

        $this->populateWithElementsWithoutGroup($menu);
        $this->populateWithElementsWithGroups($menu);

        return $menu;
    }

    /**
     * @return \Knp\Menu\ItemInterface
     */
    protected function createMenuRoot()
    {
        $menu = $this->factory->createItem('root');
        $menu->setChildrenAttribute('class', 'nav navbar-nav');
        $menu->setChildrenAttribute('id', 'top-menu');

        return $menu;
    }

    /**
     * @param $menu
     * @return ElementInterface
     */
    protected function populateWithElementsWithoutGroup($menu)
    {
        foreach ($this->manager->getElementsWithoutGroup() as $element) {
            $this->addElementToMenu($menu, $element);
        }
    }

    /**
     * @param $menu
     */
    protected function populateWithElementsWithGroups($menu)
    {
        foreach ($this->manager->getGroups() as $group) {
            $menu->addChild($group, array('uri' => '#'))
                ->setAttribute('dropdown', true);

            foreach ($this->manager->getElementsByGroup($group) as $element) {
                $this->addElementToMenu($menu[$group], $element);
            }
        }
    }

    /**
     * @param MenuItem $menu
     * @param ElementInterface $element
     */
    protected function addElementToMenu(MenuItem $menu, Elementinterface $element)
    {
        if ($element->hasOption('menu') && $element->getOption('menu') == true) {
            $menu->addChild($element->getName(), array(
                'route' => $element->getRoute(),
                'routeParameters' => $element->getRouteParameters(),
            ));
            $menu[$element->getName()]->setAttribute('class', 'admin-element');
        }
    }
}