diff options
author | Nicolas LÅ“uillet <nicolas@loeuillet.org> | 2016-04-21 22:30:50 +0200 |
---|---|---|
committer | Jeremy Benoist <jeremy.benoist@gmail.com> | 2016-08-23 07:47:55 +0200 |
commit | 8315130a75c8f411f76134b6205a017409583d50 (patch) | |
tree | d6ef8c9725809df2885dfd1351a7ec188fd51c44 /src/Wallabag/CoreBundle/Twig | |
parent | 79efca1e6ff28362d4bd2713f68205294cdd07de (diff) | |
download | wallabag-8315130a75c8f411f76134b6205a017409583d50.tar.gz wallabag-8315130a75c8f411f76134b6205a017409583d50.tar.zst wallabag-8315130a75c8f411f76134b6205a017409583d50.zip |
Display entries number for each category
Diffstat (limited to 'src/Wallabag/CoreBundle/Twig')
-rw-r--r-- | src/Wallabag/CoreBundle/Twig/WallabagExtension.php | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/Wallabag/CoreBundle/Twig/WallabagExtension.php b/src/Wallabag/CoreBundle/Twig/WallabagExtension.php index 1a308070..c116248f 100644 --- a/src/Wallabag/CoreBundle/Twig/WallabagExtension.php +++ b/src/Wallabag/CoreBundle/Twig/WallabagExtension.php | |||
@@ -2,8 +2,20 @@ | |||
2 | 2 | ||
3 | namespace Wallabag\CoreBundle\Twig; | 3 | namespace Wallabag\CoreBundle\Twig; |
4 | 4 | ||
5 | use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface; | ||
6 | use Wallabag\CoreBundle\Repository\EntryRepository; | ||
7 | |||
5 | class WallabagExtension extends \Twig_Extension | 8 | class WallabagExtension extends \Twig_Extension |
6 | { | 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 | |||
7 | public function getFilters() | 19 | public function getFilters() |
8 | { | 20 | { |
9 | return [ | 21 | return [ |
@@ -16,6 +28,27 @@ class WallabagExtension extends \Twig_Extension | |||
16 | return preg_replace('/^www\./i', '', $url); | 28 | return preg_replace('/^www\./i', '', $url); |
17 | } | 29 | } |
18 | 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 | |||
19 | public function getName() | 52 | public function getName() |
20 | { | 53 | { |
21 | return 'wallabag_extension'; | 54 | return 'wallabag_extension'; |