<?php

namespace Catalyst\MenuBundle\Twig;

use Twig\Extension\AbstractExtension;
use Twig\TwigFunction;

class MenuExtension extends AbstractExtension
{
    protected $menu_gen;

    public function __construct($menu_gen)
    {
        $this->menu_gen = $menu_gen;
    }

    public function getFunctions()
    {
        return [
            new TwigFunction('menu_get', [$this, 'getMenu']),
        ];
    }

    public function getMenu($menu_group)
    {
        $menu_data = $this->menu_gen->getMenu($menu_group);

        if (isset($menu_data['menu']))
            return $menu_data['menu'];

        return [];
    }
}