src/Wallabag/CoreBundle/Command/GenerateUrlHashesCommand.php

   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 }