Kunstmaan / KunstmaanBundlesCMS

src/Kunstmaan/SitemapBundle/Twig/SitemapTwigExtension.php

<?php

namespace Kunstmaan\SitemapBundle\Twig;

use Kunstmaan\NodeBundle\Helper\NodeMenuItem;

class SitemapTwigExtension extends \Twig_Extension
{
    /**
     * Returns a list of functions to add to the existing list.
     *
     * @return array An array of functions

Consider making the return type a bit more specific; maybe use \Twig_SimpleFunction[].

     */
    public function getFunctions()
    {
        return array(
            new \Twig_SimpleFunction('hide_from_sitemap', array($this, 'isHiddenFromSitemap')),
            new \Twig_SimpleFunction('hide_children_from_sitemap', array($this, 'isHiddenChildrenFromSitemap')),
        );
    }

    /**
     * Returns true when the item should be hidden from the sitemap
     *
     * @param NodeMenuItem $item
     *
     * @return \Kunstmaan\NodeBundle\Helper\NodeMenuItem
     */
    public function isHiddenFromSitemap(NodeMenuItem $item)
    {
        if (is_subclass_of($item->getNode()->getRefEntityName(), 'Kunstmaan\\SitemapBundle\\Helper\\HiddenFromSitemapInterface')) {
            $page = $item->getPage();

            return $page->isHiddenFromSitemap();
        }

        return false;
    }

    /**
     * Returns true when the children of the item should be hidden from the sitemap
     *
     * @param NodeMenuItem $item
     *
     * @return bool
     */
    public function isHiddenChildrenFromSitemap(NodeMenuItem $item)
    {
        if (is_subclass_of($item->getNode()->getRefEntityName(), 'Kunstmaan\\SitemapBundle\\Helper\\HiddenFromSitemapInterface')) {
            $page = $item->getPage();

            return $page->isChildrenHiddenFromSitemap();
        }

        return false;
    }
}