aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/Wallabag/CoreBundle/Command
diff options
context:
space:
mode:
authorJérémy Benoist <j0k3r@users.noreply.github.com>2019-04-25 13:28:09 +0200
committerGitHub <noreply@github.com>2019-04-25 13:28:09 +0200
commit522e37ad274361dde697da13a92ff3f846599822 (patch)
treea2b9302d885d886e013a6c33e800f5b39293e861 /src/Wallabag/CoreBundle/Command
parent3620dae1e6b3fab5a4ba4001b4581ce7ed795996 (diff)
parent76bc05ebc02408b213b536fec44e94b092889118 (diff)
downloadwallabag-522e37ad274361dde697da13a92ff3f846599822.tar.gz
wallabag-522e37ad274361dde697da13a92ff3f846599822.tar.zst
wallabag-522e37ad274361dde697da13a92ff3f846599822.zip
Merge pull request #3158 from wallabag/hash-exist-url
Hash exist url
Diffstat (limited to 'src/Wallabag/CoreBundle/Command')
-rw-r--r--src/Wallabag/CoreBundle/Command/GenerateUrlHashesCommand.php98
1 files changed, 98 insertions, 0 deletions
diff --git a/src/Wallabag/CoreBundle/Command/GenerateUrlHashesCommand.php b/src/Wallabag/CoreBundle/Command/GenerateUrlHashesCommand.php
new file mode 100644
index 00000000..45bd8c5f
--- /dev/null
+++ b/src/Wallabag/CoreBundle/Command/GenerateUrlHashesCommand.php
@@ -0,0 +1,98 @@
1<?php
2
3namespace Wallabag\CoreBundle\Command;
4
5use Doctrine\ORM\NoResultException;
6use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
7use Symfony\Component\Console\Input\InputArgument;
8use Symfony\Component\Console\Input\InputInterface;
9use Symfony\Component\Console\Output\OutputInterface;
10use Wallabag\UserBundle\Entity\User;
11
12class GenerateUrlHashesCommand extends ContainerAwareCommand
13{
14 /** @var OutputInterface */
15 protected $output;
16
17 protected function configure()
18 {
19 $this
20 ->setName('wallabag:generate-hashed-urls')
21 ->setDescription('Generates hashed urls for each entry')
22 ->setHelp('This command helps you to generates hashes of the url of each entry, to check through API if an URL is already saved')
23 ->addArgument('username', InputArgument::OPTIONAL, 'User to process entries');
24 }
25
26 protected function execute(InputInterface $input, OutputInterface $output)
27 {
28 $this->output = $output;
29
30 $username = (string) $input->getArgument('username');
31
32 if ($username) {
33 try {
34 $user = $this->getUser($username);
35 $this->generateHashedUrls($user);
36 } catch (NoResultException $e) {
37 $output->writeln(sprintf('<error>User "%s" not found.</error>', $username));
38
39 return 1;
40 }
41 } else {
42 $users = $this->getDoctrine()->getRepository('WallabagUserBundle:User')->findAll();
43
44 $output->writeln(sprintf('Generating hashed urls for "%d" users', \count($users)));
45
46 foreach ($users as $user) {
47 $output->writeln(sprintf('Processing user: %s', $user->getUsername()));
48 $this->generateHashedUrls($user);
49 }
50 $output->writeln('Finished generated hashed urls');
51 }
52
53 return 0;
54 }
55
56 /**
57 * @param User $user
58 */
59 private function generateHashedUrls(User $user)
60 {
61 $em = $this->getContainer()->get('doctrine.orm.entity_manager');
62 $repo = $this->getDoctrine()->getRepository('WallabagCoreBundle:Entry');
63
64 $entries = $repo->findByUser($user->getId());
65
66 $i = 1;
67 foreach ($entries as $entry) {
68 $entry->setHashedUrl(hash('sha1', $entry->getUrl()));
69 $em->persist($entry);
70
71 if (0 === ($i % 20)) {
72 $em->flush();
73 }
74 ++$i;
75 }
76
77 $em->flush();
78
79 $this->output->writeln(sprintf('Generated hashed urls for user: %s', $user->getUserName()));
80 }
81
82 /**
83 * Fetches a user from its username.
84 *
85 * @param string $username
86 *
87 * @return \Wallabag\UserBundle\Entity\User
88 */
89 private function getUser($username)
90 {
91 return $this->getDoctrine()->getRepository('WallabagUserBundle:User')->findOneByUserName($username);
92 }
93
94 private function getDoctrine()
95 {
96 return $this->getContainer()->get('doctrine');
97 }
98}