Spaces:
No application file
No application file
namespace Mautic\NotificationBundle\Entity; | |
use Doctrine\ORM\Query; | |
use Doctrine\ORM\Tools\Pagination\Paginator; | |
use Mautic\CoreBundle\Entity\CommonRepository; | |
/** | |
* @extends CommonRepository<Notification> | |
*/ | |
class NotificationRepository extends CommonRepository | |
{ | |
/** | |
* Get a list of entities. | |
* | |
* @return Paginator | |
*/ | |
public function getEntities(array $args = []) | |
{ | |
$q = $this->_em | |
->createQueryBuilder() | |
->select('e') | |
->from(Notification::class, 'e', 'e.id'); | |
if (empty($args['iterator_mode']) && empty($args['iterable_mode'])) { | |
$q->leftJoin('e.category', 'c'); | |
} | |
$args['qb'] = $q; | |
return parent::getEntities($args); | |
} | |
/** | |
* Get amounts of sent and read notifications. | |
* | |
* @return array | |
*/ | |
public function getSentReadCount() | |
{ | |
$q = $this->_em->createQueryBuilder(); | |
$q->select('SUM(e.sentCount) as sent_count, SUM(e.readCount) as read_count') | |
->from(Notification::class, 'e'); | |
$results = $q->getQuery()->getSingleResult(Query::HYDRATE_ARRAY); | |
if (!isset($results['sent_count'])) { | |
$results['sent_count'] = 0; | |
} | |
if (!isset($results['read_count'])) { | |
$results['read_count'] = 0; | |
} | |
return $results; | |
} | |
/** | |
* @param \Doctrine\ORM\QueryBuilder|\Doctrine\DBAL\Query\QueryBuilder $q | |
*/ | |
protected function addSearchCommandWhereClause($q, $filter): array | |
{ | |
[$expr, $parameters] = $this->addStandardSearchCommandWhereClause($q, $filter); | |
if ($expr) { | |
return [$expr, $parameters]; | |
} | |
$command = $filter->command; | |
$unique = $this->generateRandomParameterName(); | |
$returnParameter = false; // returning a parameter that is not used will lead to a Doctrine error | |
switch ($command) { | |
case $this->translator->trans('mautic.core.searchcommand.lang'): | |
case $this->translator->trans('mautic.core.searchcommand.lang', [], null, 'en_US'): | |
$langUnique = $this->generateRandomParameterName(); | |
$langValue = $filter->string.'_%'; | |
$forceParameters = [ | |
$langUnique => $langValue, | |
$unique => $filter->string, | |
]; | |
$expr = $q->expr()->orX( | |
$q->expr()->eq('e.language', ":$unique"), | |
$q->expr()->like('e.language', ":$langUnique") | |
); | |
$returnParameter = true; | |
break; | |
} | |
if ($expr && $filter->not) { | |
$expr = $q->expr()->not($expr); | |
} | |
if (!empty($forceParameters)) { | |
$parameters = $forceParameters; | |
} elseif ($returnParameter) { | |
$string = ($filter->strict) ? $filter->string : "%{$filter->string}%"; | |
$parameters = ["$unique" => $string]; | |
} | |
return [$expr, $parameters]; | |
} | |
/** | |
* @return string[] | |
*/ | |
public function getSearchCommands(): array | |
{ | |
$commands = [ | |
'mautic.core.searchcommand.ispublished', | |
'mautic.core.searchcommand.isunpublished', | |
'mautic.core.searchcommand.isuncategorized', | |
'mautic.core.searchcommand.ismine', | |
'mautic.core.searchcommand.category', | |
'mautic.core.searchcommand.lang', | |
]; | |
return array_merge($commands, parent::getSearchCommands()); | |
} | |
/** | |
* @return array<array<string>> | |
*/ | |
protected function getDefaultOrder(): array | |
{ | |
return [ | |
['e.name', 'ASC'], | |
]; | |
} | |
public function getTableAlias(): string | |
{ | |
return 'e'; | |
} | |
/** | |
* Up the click/sent counts. | |
* | |
* @param string $type | |
* @param int $increaseBy | |
*/ | |
public function upCount($id, $type = 'sent', $increaseBy = 1): void | |
{ | |
try { | |
$q = $this->_em->getConnection()->createQueryBuilder(); | |
$q->update(MAUTIC_TABLE_PREFIX.'push_notifications') | |
->set($type.'_count', $type.'_count + '.(int) $increaseBy) | |
->where('id = '.(int) $id); | |
$q->executeStatement(); | |
} catch (\Exception) { | |
// not important | |
} | |
} | |
/** | |
* @param string $search | |
* @param int $limit | |
* @param int $start | |
* @param bool $viewOther | |
* @param string $notificationType | |
* | |
* @return array | |
*/ | |
public function getNotificationList($search = '', $limit = 10, $start = 0, $viewOther = false, $notificationType = null) | |
{ | |
$q = $this->createQueryBuilder('e'); | |
$q->select('partial e.{id, name, language}'); | |
if (!empty($search)) { | |
if (is_array($search)) { | |
$search = array_map('intval', $search); | |
$q->andWhere($q->expr()->in('e.id', ':search')) | |
->setParameter('search', $search); | |
} else { | |
$q->andWhere($q->expr()->like('e.name', ':search')) | |
->setParameter('search', "%{$search}%"); | |
} | |
} | |
if (!$viewOther) { | |
$q->andWhere($q->expr()->eq('e.createdBy', ':id')) | |
->setParameter('id', $this->currentUser->getId()); | |
} | |
if (!empty($notificationType)) { | |
$q->andWhere( | |
$q->expr()->eq('e.notificationType', $q->expr()->literal($notificationType)) | |
); | |
} | |
$q->andWhere('e.mobile != 1'); | |
$q->orderBy('e.name'); | |
if (!empty($limit)) { | |
$q->setFirstResult($start) | |
->setMaxResults($limit); | |
} | |
return $q->getQuery()->getArrayResult(); | |
} | |
/** | |
* @param string $search | |
* @param int $limit | |
* @param int $start | |
* @param bool $viewOther | |
* @param string $notificationType | |
* | |
* @return array | |
*/ | |
public function getMobileNotificationList($search = '', $limit = 10, $start = 0, $viewOther = false, $notificationType = null) | |
{ | |
$q = $this->createQueryBuilder('e'); | |
$q->select('partial e.{id, name, language}'); | |
if (!empty($search)) { | |
if (is_array($search)) { | |
$search = array_map('intval', $search); | |
$q->andWhere($q->expr()->in('e.id', ':search')) | |
->setParameter('search', $search); | |
} else { | |
$q->andWhere($q->expr()->like('e.name', ':search')) | |
->setParameter('search', "%{$search}%"); | |
} | |
} | |
if (!$viewOther) { | |
$q->andWhere($q->expr()->eq('e.createdBy', ':id')) | |
->setParameter('id', $this->currentUser->getId()); | |
} | |
if (!empty($notificationType)) { | |
$q->andWhere( | |
$q->expr()->eq('e.notificationType', $q->expr()->literal($notificationType)) | |
); | |
} | |
$q->andWhere('e.mobile = 1'); | |
$q->orderBy('e.name'); | |
if (!empty($limit)) { | |
$q->setFirstResult($start) | |
->setMaxResults($limit); | |
} | |
return $q->getQuery()->getArrayResult(); | |
} | |
} | |