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('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 }