3 namespace Wallabag\CoreBundle\Twig
;
5 use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface
;
6 use Symfony\Component\Translation\TranslatorInterface
;
7 use Wallabag\CoreBundle\Repository\EntryRepository
;
8 use Wallabag\CoreBundle\Repository\TagRepository
;
10 class WallabagExtension
extends \Twig_Extension
implements \Twig_Extension_GlobalsInterface
12 private $tokenStorage;
13 private $entryRepository;
14 private $tagRepository;
18 public function __construct(EntryRepository
$entryRepository, TagRepository
$tagRepository, TokenStorageInterface
$tokenStorage, $lifeTime, TranslatorInterface
$translator)
20 $this->entryRepository
= $entryRepository;
21 $this->tagRepository
= $tagRepository;
22 $this->tokenStorage
= $tokenStorage;
23 $this->lifeTime
= $lifeTime;
24 $this->translator
= $translator;
27 public function getFilters()
30 new \
Twig_SimpleFilter('removeWww', [$this, 'removeWww']),
34 public function getFunctions()
37 new \
Twig_SimpleFunction('count_entries', [$this, 'countEntries']),
38 new \
Twig_SimpleFunction('count_tags', [$this, 'countTags']),
39 new \
Twig_SimpleFunction('display_stats', [$this, 'displayStats']),
43 public function removeWww($url)
45 return preg_replace('/^www\./i', '', $url);
49 * Return number of entries depending of the type (unread, archive, starred or all).
51 * @param string $type Type of entries to count
55 public function countEntries($type)
57 $user = $this->tokenStorage
->getToken() ? $this->tokenStorage
->getToken()->getUser() : null;
59 if (null === $user || !is_object($user)) {
65 $qb = $this->entryRepository
->getBuilderForStarredByUser($user->getId());
68 $qb = $this->entryRepository
->getBuilderForArchiveByUser($user->getId());
71 $qb = $this->entryRepository
->getBuilderForUnreadByUser($user->getId());
74 $qb = $this->entryRepository
->getBuilderForAllByUser($user->getId());
77 throw new \
InvalidArgumentException(sprintf('Type "%s" is not implemented.', $type));
80 // THANKS to PostgreSQL we CAN'T make a DEAD SIMPLE count(e.id)
81 // ERROR: column "e0_.id" must appear in the GROUP BY clause or be used in an aggregate function
87 $query->useQueryCache(true);
88 $query->useResultCache(true);
89 $query->setResultCacheLifetime($this->lifeTime
);
91 return count($query->getArrayResult());
95 * Return number of tags.
99 public function countTags()
101 $user = $this->tokenStorage
->getToken() ? $this->tokenStorage
->getToken()->getUser() : null;
103 if (null === $user || !is_object($user)) {
107 return $this->tagRepository
->countAllTags($user->getId());
111 * Display a single line about reading stats.
115 public function displayStats()
117 $user = $this->tokenStorage
->getToken() ? $this->tokenStorage
->getToken()->getUser() : null;
119 if (null === $user || !is_object($user)) {
123 $query = $this->entryRepository
->getBuilderForArchiveByUser($user->getId())
128 $query->useQueryCache(true);
129 $query->useResultCache(true);
130 $query->setResultCacheLifetime($this->lifeTime
);
132 $nbArchives = count($query->getArrayResult());
134 $interval = $user->getCreatedAt()->diff(new \
DateTime('now'));
135 $nbDays = (int) $interval->format('%a') ?: 1;
137 // force setlocale for date translation
138 setlocale(LC_TIME
, strtolower($user->getConfig()->getLanguage()) . '_' . strtoupper(strtolower($user->getConfig()->getLanguage())));
140 return $this->translator
->trans('footer.stats', [
141 '%user_creation%' => strftime('%e %B %Y', $user->getCreatedAt()->getTimestamp()),
142 '%nb_archives%' => $nbArchives,
143 '%per_day%' => round($nbArchives / $nbDays, 2),
147 public function getName()
149 return 'wallabag_extension';