1:  2:  3:  4:  5:  6:  7:  8:  9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 38: 39: 40: 41: 42: 43: 44: 45: 46: 47: 48: 49: 50: 51: 52: 53: 54: 55: 56: 57: 58: 59: 60: 61: 62: 63: 64: 65: 66: 67: 68: 69: 70: 71: 72: 73: 74: 75: 76: 77: 78: 79: 80: 81: 82: 83: 84: 85: 86: 87: 

<?php

    namespace Inteve\Navigation;

    use Nette\Utils\Strings;


    class MenuControl extends BaseControl
    {
        const TEMPLATE_DEFAULT = 'default';

        /** @var Navigation */
        private $navigation;

        /** @var string|NULL */
        private $subTree;

        /** @var string */
        private $templateFile;


        public function __construct(Navigation $navigation)
        {
            $this->navigation = $navigation;
            $this->setTemplateFile(self::TEMPLATE_DEFAULT);
        }


        /**
         * @param  string  file path or template name
         * @return self
         */
        public function setTemplateFile($file)
        {
            if ($file === self::TEMPLATE_DEFAULT) {
                $file = __DIR__ . '/templates/menu/' . $file . '.latte';
            }
            $this->templateFile = $file;
            return $this;
        }


        /**
         * @deprecated  use setSubTree()
         */
        public function setParentPage($parentPage)
        {
            return $this->setSubTree($parentPage);
        }


        /**
         * @param  string|NULL
         */
        public function setSubTree($subTree)
        {
            $this->subTree = Navigation::formatPageId($subTree);
            return $this;
        }


        /**
         * @return void
         */
        public function render()
        {
            $items = array();
            $subTree = $this->subTree . '/';
            $subTreeLevel = substr_count($subTree, '/');

            foreach ($this->navigation->getPages() as $pageId => $page) {
                $pageLevel = substr_count($pageId, '/');

                if (Strings::startsWith($pageId, $subTree) && $subTreeLevel === $pageLevel) {
                    $items[] = array(
                        'page' => $page,
                        'level' => 0,
                    );
                }
            }

            $template = $this->createTemplate();
            $template->items = $items;
            $template->render($this->templateFile);
        }
    }