src/Wallabag/CoreBundle/Twig/WallabagExtension.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Twig;
   4
   5 use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;
   6 use Wallabag\CoreBundle\Repository\EntryRepository;
   7
   8 class WallabagExtension extends \Twig_Extension
   9 {
  10     private $tokenStorage;
  11     private $repository;
  12
  13     public function __construct(EntryRepository $repository = null, TokenStorageInterface $tokenStorage = null)
  14     {
  15         $this->repository = $repository;
  16         $this->tokenStorage = $tokenStorage;
  17     }
  18
  19     public function getFilters()
  20     {
  21         return [
  22             new \Twig_SimpleFilter('removeWww', [$this, 'removeWww']),
  23         ];
  24     }
  25
  26     public function removeWww($url)
  27     {
  28         return preg_replace('/^www\./i', '', $url);
  29     }
  30
  31     public function getGlobals()
  32     {
  33         $user = $this->tokenStorage->getToken() ? $this->tokenStorage->getToken()->getUser() : null;
  34
  35         if (null === $user || !is_object($user)) {
  36             return array();
  37         }
  38
  39         $unreadEntries = $this->repository->getBuilderForUnreadByUser($user->getId())->getQuery()->getResult();
  40         $starredEntries = $this->repository->getBuilderForStarredByUser($user->getId())->getQuery()->getResult();
  41         $archivedEntries = $this->repository->getBuilderForArchiveByUser($user->getId())->getQuery()->getResult();
  42         $allEntries = $this->repository->getBuilderForAllByUser($user->getId())->getQuery()->getResult();
  43
  44         return array(
  45             'unreadEntries' => count($unreadEntries),
  46             'starredEntries' => count($starredEntries),
  47             'archivedEntries' => count($archivedEntries),
  48             'allEntries' => count($allEntries),
  49         );
  50     }
  51
  52     public function getName()
  53     {
  54         return 'wallabag_extension';
  55     }
  56 }