]>
Commit | Line | Data |
---|---|---|
d5fbb570 TC |
1 | <?php |
2 | ||
3 | namespace Wallabag\CoreBundle\Command; | |
4 | ||
5 | use Doctrine\ORM\NoResultException; | |
6 | use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand; | |
7 | use Symfony\Component\Console\Input\InputArgument; | |
8 | use Symfony\Component\Console\Input\InputInterface; | |
9 | use Symfony\Component\Console\Output\OutputInterface; | |
10 | use Wallabag\UserBundle\Entity\User; | |
11 | ||
12 | class 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( | |
24 | 'username', | |
25 | InputArgument::OPTIONAL, | |
26 | 'User to process entries' | |
27 | ); | |
28 | } | |
29 | ||
30 | protected function execute(InputInterface $input, OutputInterface $output) | |
31 | { | |
32 | $this->output = $output; | |
33 | ||
34 | $username = $input->getArgument('username'); | |
35 | ||
36 | if ($username) { | |
37 | try { | |
38 | $user = $this->getUser($username); | |
39 | $this->generateHashedUrls($user); | |
40 | } catch (NoResultException $e) { | |
41 | $output->writeln(sprintf('<error>User "%s" not found.</error>', $username)); | |
42 | ||
43 | return 1; | |
44 | } | |
45 | } else { | |
46 | $users = $this->getDoctrine()->getRepository('WallabagUserBundle:User')->findAll(); | |
47 | ||
48 | $output->writeln(sprintf('Generating hashed urls for the %d user account entries', count($users))); | |
49 | ||
50 | foreach ($users as $user) { | |
51 | $output->writeln(sprintf('Processing user %s', $user->getUsername())); | |
52 | $this->generateHashedUrls($user); | |
53 | } | |
54 | $output->writeln(sprintf('Finished generated hashed urls')); | |
55 | } | |
56 | ||
57 | return 0; | |
58 | } | |
59 | ||
60 | /** | |
61 | * @param User $user | |
62 | */ | |
63 | private function generateHashedUrls(User $user) | |
64 | { | |
65 | $em = $this->getContainer()->get('doctrine.orm.entity_manager'); | |
66 | $repo = $this->getDoctrine()->getRepository('WallabagCoreBundle:Entry'); | |
67 | ||
68 | $entries = $repo->findByUser($user->getId()); | |
69 | ||
70 | foreach ($entries as $entry) { | |
71 | $entry->setHashedUrl(hash('sha512', $entry->getUrl())); | |
72 | $em->persist($entry); | |
73 | $em->flush(); | |
74 | } | |
75 | ||
76 | $this->output->writeln(sprintf('Generated hashed urls for user %s', $user->getUserName())); | |
77 | } | |
78 | ||
79 | /** | |
80 | * Fetches a user from its username. | |
81 | * | |
82 | * @param string $username | |
83 | * | |
84 | * @return \Wallabag\UserBundle\Entity\User | |
85 | */ | |
86 | private function getUser($username) | |
87 | { | |
88 | return $this->getDoctrine()->getRepository('WallabagUserBundle:User')->findOneByUserName($username); | |
89 | } | |
90 | ||
91 | private function getDoctrine() | |
92 | { | |
93 | return $this->getContainer()->get('doctrine'); | |
94 | } | |
95 | } |