1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
<?php
namespace Wallabag\CoreBundle\Command;
use Doctrine\ORM\NoResultException;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Wallabag\CoreBundle\Entity\Notification;
use Wallabag\CoreBundle\Notifications\InfoAction;
use Wallabag\UserBundle\Entity\User;
class AdminNotificationCommand extends AbstractNotificationCommand
{
protected function configure()
{
$this
->setName('wallabag:notification:send')
->setDescription('Emits a notification to all users')
->setHelp('This command helps you send notifications to all of the users instance, or just for one user.')
->addArgument(
'title',
InputArgument::REQUIRED,
'Title of your notification. This is required if if the type of notification is an admin one.'
)
->addArgument(
'message',
InputArgument::REQUIRED,
'Message of your notification. This is required if the type of notification is an admin one.'
)
->addOption(
'link',
'l',
InputOption::VALUE_REQUIRED,
'A link to display with the notification'
)
;
parent::configure();
}
protected function execute(InputInterface $input, OutputInterface $output)
{
$this->output = $output;
$username = $input->getArgument('username');
$message = $input->getArgument('message');
$title = $input->getArgument('title');
$link = $input->getOption('link');
if ($username) {
try {
$user = $this->getUser($username);
$this->sendNotification($user, $title, $message, $link);
} catch (NoResultException $e) {
$output->writeln(sprintf('<error>User "%s" not found.</error>', $username));
return 1;
}
} else {
$users = $this->getDoctrine()->getRepository('WallabagUserBundle:User')->findAll();
$output->writeln(sprintf('Sending notifications to %d user accounts. This can take some time.', count($users)));
foreach ($users as $user) {
$output->writeln(sprintf('Processing user %s', $user->getUsername()));
$this->sendNotification($user, $title, $message, $link);
}
$output->writeln('Finished sending notifications.');
}
return 0;
}
/**
* @param User $user
* @param $title
* @param $message
* @param null $link
*/
private function sendNotification(User $user, $title, $message, $link = null)
{
$em = $this->getContainer()->get('doctrine.orm.entity_manager');
$notification = new Notification($user);
$notification->setTitle($title)
->setDescription($message)
->setType(Notification::TYPE_ADMIN);
if ($link) {
$action = new InfoAction($link);
$notification->addAction($action);
}
$em->persist($notification);
$em->flush();
$this->output->writeln(sprintf('Sent notification for user %s', $user->getUserName()));
}
}
|
