src/Wallabag/ImportBundle/Import/PocketImport.php

   1 <?php
   2
   3 namespace Wallabag\ImportBundle\Import;
   4
   5 use Psr\Log\NullLogger;
   6 use Doctrine\ORM\EntityManager;
   7 use GuzzleHttp\Client;
   8 use GuzzleHttp\Exception\RequestException;
   9 use Wallabag\CoreBundle\Entity\Entry;
  10 use Wallabag\CoreBundle\Helper\ContentProxy;
  11 use Craue\ConfigBundle\Util\Config;
  12
  13 class PocketImport extends AbstractImport
  14 {
  15     private $client;
  16     private $consumerKey;
  17     private $skippedEntries = 0;
  18     private $importedEntries = 0;
  19     protected $accessToken;
  20
  21     public function __construct(EntityManager $em, ContentProxy $contentProxy, Config $craueConfig)
  22     {
  23         $this->em = $em;
  24         $this->contentProxy = $contentProxy;
  25         $this->consumerKey = $craueConfig->get('pocket_consumer_key');
  26         $this->logger = new NullLogger();
  27     }
  28
  29     /**
  30      * {@inheritdoc}
  31      */
  32     public function getName()
  33     {
  34         return 'Pocket';
  35     }
  36
  37     /**
  38      * {@inheritdoc}
  39      */
  40     public function getUrl()
  41     {
  42         return 'import_pocket';
  43     }
  44
  45     /**
  46      * {@inheritdoc}
  47      */
  48     public function getDescription()
  49     {
  50         return 'import.pocket.description';
  51     }
  52
  53     /**
  54      * Return the oauth url to authenticate the client.
  55      *
  56      * @param string $redirectUri Redirect url in case of error
  57      *
  58      * @return string|false request_token for callback method
  59      */
  60     public function getRequestToken($redirectUri)
  61     {
  62         $request = $this->client->createRequest('POST', 'https://getpocket.com/v3/oauth/request',
  63             [
  64                 'body' => json_encode([
  65                     'consumer_key' => $this->consumerKey,
  66                     'redirect_uri' => $redirectUri,
  67                 ]),
  68             ]
  69         );
  70
  71         try {
  72             $response = $this->client->send($request);
  73         } catch (RequestException $e) {
  74             $this->logger->error(sprintf('PocketImport: Failed to request token: %s', $e->getMessage()), ['exception' => $e]);
  75
  76             return false;
  77         }
  78
  79         return $response->json()['code'];
  80     }
  81
  82     /**
  83      * Usually called by the previous callback to authorize the client.
  84      * Then it return a token that can be used for next requests.
  85      *
  86      * @param string $code request_token from getRequestToken
  87      *
  88      * @return bool
  89      */
  90     public function authorize($code)
  91     {
  92         $request = $this->client->createRequest('POST', 'https://getpocket.com/v3/oauth/authorize',
  93             [
  94                 'body' => json_encode([
  95                     'consumer_key' => $this->consumerKey,
  96                     'code' => $code,
  97                 ]),
  98             ]
  99         );
 100
 101         try {
 102             $response = $this->client->send($request);
 103         } catch (RequestException $e) {
 104             $this->logger->error(sprintf('PocketImport: Failed to authorize client: %s', $e->getMessage()), ['exception' => $e]);
 105
 106             return false;
 107         }
 108
 109         $this->accessToken = $response->json()['access_token'];
 110
 111         return true;
 112     }
 113
 114     /**
 115      * {@inheritdoc}
 116      */
 117     public function import()
 118     {
 119         $request = $this->client->createRequest('POST', 'https://getpocket.com/v3/get',
 120             [
 121                 'body' => json_encode([
 122                     'consumer_key' => $this->consumerKey,
 123                     'access_token' => $this->accessToken,
 124                     'detailType' => 'complete',
 125                     'state' => 'all',
 126                     'sort' => 'oldest',
 127                 ]),
 128             ]
 129         );
 130
 131         try {
 132             $response = $this->client->send($request);
 133         } catch (RequestException $e) {
 134             $this->logger->error(sprintf('PocketImport: Failed to import: %s', $e->getMessage()), ['exception' => $e]);
 135
 136             return false;
 137         }
 138
 139         $entries = $response->json();
 140
 141         if ($this->producer) {
 142             $this->parseEntriesForProducer($entries['list']);
 143
 144             return true;
 145         }
 146
 147         $this->parseEntries($entries['list']);
 148
 149         return true;
 150     }
 151
 152     /**
 153      * {@inheritdoc}
 154      */
 155     public function getSummary()
 156     {
 157         return [
 158             'skipped' => $this->skippedEntries,
 159             'imported' => $this->importedEntries,
 160         ];
 161     }
 162
 163     /**
 164      * Set the Guzzle client.
 165      *
 166      * @param Client $client
 167      */
 168     public function setClient(Client $client)
 169     {
 170         $this->client = $client;
 171     }
 172
 173     public function parseEntry(array $importedEntry)
 174     {
 175         $url = isset($importedEntry['resolved_url']) && $importedEntry['resolved_url'] != '' ? $importedEntry['resolved_url'] : $importedEntry['given_url'];
 176
 177         $existingEntry = $this->em
 178             ->getRepository('WallabagCoreBundle:Entry')
 179             ->findByUrlAndUserId($url, $this->user->getId());
 180
 181         if (false !== $existingEntry) {
 182             ++$this->skippedEntries;
 183
 184             return;
 185         }
 186
 187         $entry = new Entry($this->user);
 188         $entry = $this->fetchContent($entry, $url);
 189
 190         // jump to next entry in case of problem while getting content
 191         if (false === $entry) {
 192             ++$this->skippedEntries;
 193
 194             return;
 195         }
 196
 197         // 0, 1, 2 - 1 if the item is archived - 2 if the item should be deleted
 198         if ($importedEntry['status'] == 1 || $this->markAsRead) {
 199             $entry->setArchived(true);
 200         }
 201
 202         // 0 or 1 - 1 If the item is starred
 203         if ($importedEntry['favorite'] == 1) {
 204             $entry->setStarred(true);
 205         }
 206
 207         $title = 'Untitled';
 208         if (isset($importedEntry['resolved_title']) && $importedEntry['resolved_title'] != '') {
 209             $title = $importedEntry['resolved_title'];
 210         } elseif (isset($importedEntry['given_title']) && $importedEntry['given_title'] != '') {
 211             $title = $importedEntry['given_title'];
 212         }
 213
 214         $entry->setTitle($title);
 215         $entry->setUrl($url);
 216
 217         // 0, 1, or 2 - 1 if the item has images in it - 2 if the item is an image
 218         if (isset($importedEntry['has_image']) && $importedEntry['has_image'] > 0 && isset($importedEntry['images'][1])) {
 219             $entry->setPreviewPicture($importedEntry['images'][1]['src']);
 220         }
 221
 222         if (isset($importedEntry['tags']) && !empty($importedEntry['tags'])) {
 223             $this->contentProxy->assignTagsToEntry(
 224                 $entry,
 225                 array_keys($importedEntry['tags'])
 226             );
 227         }
 228
 229         $this->em->persist($entry);
 230         ++$this->importedEntries;
 231
 232         return $entry;
 233     }
 234
 235     /**
 236      * Faster parse entries for Producer.
 237      * We don't care to make check at this time. They'll be done by the consumer.
 238      *
 239      * @param array $entries
 240      */
 241     public function parseEntriesForProducer($entries)
 242     {
 243         foreach ($entries as $importedEntry) {
 244             // set userId for the producer (it won't know which user is connected)
 245             $importedEntry['userId'] = $this->user->getId();
 246
 247             if ($this->markAsRead) {
 248                 $importedEntry['status'] = 1;
 249             }
 250
 251             ++$this->importedEntries;
 252
 253             $this->producer->publish(json_encode($importedEntry));
 254         }
 255     }
 256 }