Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\toolbar\Menu;
Chris@0: 
Chris@0: use Drupal\Core\Menu\MenuLinkTree;
Chris@0: 
Chris@0: /**
Chris@0:  * Extends MenuLinkTree to add specific theme suggestions for the toolbar.
Chris@0:  */
Chris@0: class ToolbarMenuLinkTree extends MenuLinkTree {
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function build(array $tree, $level = 0) {
Chris@0:     if ($level == 0) {
Chris@0:       if (!$tree) {
Chris@0:         return [];
Chris@0:       }
Chris@0:       $build = parent::build($tree, $level);
Chris@0: 
Chris@0:       /** @var \Drupal\Core\Menu\MenuLinkInterface $link */
Chris@0:       $first_link = reset($tree)->link;
Chris@0:       // Get the menu name of the first link.
Chris@0:       $menu_name = $first_link->getMenuName();
Chris@0:       // Add a more specific theme suggestion to differentiate this rendered
Chris@0:       // menu from others.
Chris@0:       $build['#menu_name'] = $menu_name;
Chris@0:       $build['#theme'] = 'menu__toolbar__' . strtr($menu_name, '-', '_');
Chris@0:       return $build;
Chris@0:     }
Chris@0:     else {
Chris@0:       return parent::build($tree, $level);
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0: }