Kunstmaan / KunstmaanBundlesCMS

src/Kunstmaan/ArticleBundle/Twig/ArticleTwigExtension.php

<?php

namespace Kunstmaan\ArticleBundle\Twig;

use Doctrine\ORM\EntityManagerInterface;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\RequestStack;
use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
use Symfony\Component\Routing\RouterInterface;
use Twig_Extension;

/**
 * Extension for article bundle.
 */
class ArticleTwigExtension extends Twig_Extension
{
    /**
     * @var EntityManagerInterface $em
     */
    private $em;

    /**
     * @var RouterInterface
     */
    private $router;

    /**
     * ArticleTwigExtension constructor.
     *
     * @param EntityManagerInterface $em
     * @param RouterInterface        $router
     */
    public function __construct(EntityManagerInterface $em, RouterInterface $router)
    {
        $this->em = $em;
        $this->router = $router;

        if (func_num_args() > 2) {
            @trigger_error(sprintf('Passing the "request_stack" service as the third argument in "%s" is deprecated in KunstmaanArticleBundle 5.1 and will be removed in KunstmaanArticleBundle 6.0. Remove the "request_stack" argument from your service definition.', __METHOD__), E_USER_DEPRECATED);

It seems like you do not handle an error condition here. This can introduce security issues, and is generally not recommended.

        }
    }

    /**
     * Returns a list of functions to add to the existing list.
     *
     * @return array An array of functions
     */
    public function getFunctions()
    {
        return array(
            new \Twig_SimpleFunction(
                'get_article_tag_path', array($this, 'getArticleTagRouterPath')
            ),
            new \Twig_SimpleFunction(
                'get_article_category_path', array($this, 'getArticleCategoryRouterPath')
            ),
            new \Twig_SimpleFunction(
                'get_article_categories', array($this, 'getCategories')
            ),
            new \Twig_SimpleFunction(
                'get_article_tags', array($this, 'getTags')
            ),
        );
    }


    /**
     * Get tags array for view.
     *
     * @param Request $request
     * @param string  $className
     *
     * @return array
     */
    public function getTags(Request $request, $className)
    {
        $context = array();

        $tagRepository = $this->em->getRepository($className);
        $context['tags'] = $tagRepository->findBy(array(), array('name' => 'ASC'));

        $searchTag = $request->get('tag') ? explode(',', $request->get('tag')) : null;
        if ($searchTag) {
            $context['activeTag'] = true;
            $context['activeTags'] = $searchTag;
        }

        return $context;
    }

    /**
     * Get categories array for view.
     *
     * @param Request $request
     * @param string  $className
     * @return array
     */
    public function getCategories(Request $request, $className)
    {
        $context = array();

        $categoryRepository = $this->em->getRepository($className);
        $context['categories'] = $categoryRepository->findBy(array(), array('name' => 'ASC'));

        $searchCategory = $request->get('category') ? explode(',', $request->get('category')) : null;
        if ($searchCategory) {
            $context['activeCategory'] = true;
            $context['activeCategories'] = $searchCategory;
        }

        return $context;
    }

    /**
     * @param string $slug
     * @param string $tag
     * @param string $locale
     * @param array  $parameters
     * @param int    $referenceType
     *
     * @return string
     */
    public function getArticleTagRouterPath($slug, $tag, $locale, $parameters = [], $referenceType = UrlGeneratorInterface::ABSOLUTE_PATH)
    {
        $routeName = sprintf('_slug_tag_%s', $locale);
        return $this->getArticleRouterPath($routeName, 'tag', $slug, $tag, $locale, $parameters, $referenceType);
    }

    /**
     * @param string $slug
     * @param string $category
     * @param string $locale
     * @param array  $parameters
     * @param int    $referenceType
     *
     * @return string
     */
    public function getArticleCategoryRouterPath($slug, $category, $locale, $parameters = [], $referenceType = UrlGeneratorInterface::ABSOLUTE_PATH)
    {
        $routeName = sprintf('_slug_category_%s', $locale);
        return $this->getArticleRouterPath($routeName, 'category', $slug, $category, $locale, $parameters, $referenceType);
    }

    /**
     * @return string
     */
    public function getName()
    {
        return 'article_twig_extension';
    }

    /**
     * @param string $routeName
     * @param string $type
     * @param string $slug
     * @param string $tagOrCategory
     * @param string $locale
     * @param array  $parameters
     * @param int    $referenceType
     *
     * @return string
     */
    protected function getArticleRouterPath($routeName, $type, $slug, $tagOrCategory, $locale, $parameters = [], $referenceType = UrlGeneratorInterface::ABSOLUTE_PATH)
    {
        if (!$this->articleRouteExists($type, $locale)) {
            $routeName = '_slug';
        }
        if (!isset($parameters[$type])) {
            $parameters[$type] = $tagOrCategory;
        }
        if (!isset($parameters['url'])) {
            $parameters['url'] = $slug;
        }
        if (!isset($parameters['_locale'])) {
            $parameters['_locale'] = $locale;
        }
        return $this->router->generate($routeName, $parameters, $referenceType);
    }

    /**
     * @param string $type
     * @param string $locale
     * @return bool
     */
    protected function articleRouteExists($type, $locale)
    {
        $routeName = sprintf('_slug_%s_%s', $type, $locale);

        try {
            return !is_null($this->router->getRouteCollection()->get($routeName));
        } catch (\Exception $e) {
            return false;
        }
    }
}