src/Wallabag/ImportBundle/Import/ReadabilityImport.php

   1 <?php
   2
   3 namespace Wallabag\ImportBundle\Import;
   4
   5 use Wallabag\CoreBundle\Entity\Entry;
   6 use Wallabag\UserBundle\Entity\User;
   7
   8 class ReadabilityImport extends AbstractImport
   9 {
  10     private $user;
  11     private $skippedEntries = 0;
  12     private $importedEntries = 0;
  13     private $filepath;
  14     private $markAsRead;
  15
  16     /**
  17      * We define the user in a custom call because on the import command there is no logged in user.
  18      * So we can't retrieve user from the `security.token_storage` service.
  19      *
  20      * @param User $user
  21      */
  22     public function setUser(User $user)
  23     {
  24         $this->user = $user;
  25
  26         return $this;
  27     }
  28
  29     /**
  30      * {@inheritdoc}
  31      */
  32     public function getName()
  33     {
  34         return 'Readability';
  35     }
  36
  37     /**
  38      * {@inheritdoc}
  39      */
  40     public function getUrl()
  41     {
  42         return 'import_readability';
  43     }
  44
  45     /**
  46      * {@inheritdoc}
  47      */
  48     public function getDescription()
  49     {
  50         return 'import.readability.description';
  51     }
  52
  53     /**
  54      * Set file path to the json file.
  55      *
  56      * @param string $filepath
  57      */
  58     public function setFilepath($filepath)
  59     {
  60         $this->filepath = $filepath;
  61
  62         return $this;
  63     }
  64
  65     /**
  66      * Set whether articles must be all marked as read.
  67      *
  68      * @param bool $markAsRead
  69      */
  70     public function setMarkAsRead($markAsRead)
  71     {
  72         $this->markAsRead = $markAsRead;
  73
  74         return $this;
  75     }
  76
  77     /**
  78      * Get whether articles must be all marked as read.
  79      */
  80     public function getMarkAsRead()
  81     {
  82         return $this->markAsRead;
  83     }
  84
  85     /**
  86      * {@inheritdoc}
  87      */
  88     public function getSummary()
  89     {
  90         return [
  91             'skipped' => $this->skippedEntries,
  92             'imported' => $this->importedEntries,
  93         ];
  94     }
  95
  96     /**
  97      * {@inheritdoc}
  98      */
  99     public function import()
 100     {
 101         if (!$this->user) {
 102             $this->logger->error('ReadabilityImport: user is not defined');
 103
 104             return false;
 105         }
 106
 107         if (!file_exists($this->filepath) || !is_readable($this->filepath)) {
 108             $this->logger->error('ReadabilityImport: unable to read file', ['filepath' => $this->filepath]);
 109
 110             return false;
 111         }
 112
 113         $data = json_decode(file_get_contents($this->filepath), true);
 114
 115         if (empty($data) || empty($data['bookmarks'])) {
 116             return false;
 117         }
 118
 119         $this->parseEntries($data['bookmarks']);
 120
 121         return true;
 122     }
 123
 124     /**
 125      * Parse and insert all given entries.
 126      *
 127      * @param $entries
 128      */
 129     protected function parseEntries($entries)
 130     {
 131         $i = 1;
 132
 133         foreach ($entries as $importedEntry) {
 134             $existingEntry = $this->em
 135                 ->getRepository('WallabagCoreBundle:Entry')
 136                 ->findByUrlAndUserId($importedEntry['article__url'], $this->user->getId());
 137
 138             if (false !== $existingEntry) {
 139                 ++$this->skippedEntries;
 140                 continue;
 141             }
 142
 143             $data = [
 144                 'title' => $importedEntry['article__title'],
 145                 'url' => $importedEntry['article__url'],
 146                 'content_type' => '',
 147                 'language' => '',
 148                 'is_archived' => $importedEntry['archive'] || $this->markAsRead,
 149                 'is_starred' => $importedEntry['favorite'],
 150             ];
 151
 152             $entry = $this->fetchContent(
 153                 new Entry($this->user),
 154                 $data['url'],
 155                 $data
 156             );
 157
 158             // jump to next entry in case of problem while getting content
 159             if (false === $entry) {
 160                 ++$this->skippedEntries;
 161                 continue;
 162             }
 163             $entry->setArchived($data['is_archived']);
 164             $entry->setStarred($data['is_starred']);
 165
 166             $this->em->persist($entry);
 167             ++$this->importedEntries;
 168
 169             // flush every 20 entries
 170             if (($i % 20) === 0) {
 171                 $this->em->flush();
 172                 $this->em->clear($entry);
 173             }
 174             ++$i;
 175         }
 176
 177         $this->em->flush();
 178     }
 179 }