3 namespace Wallabag\ImportBundle\Import
;
5 use Doctrine\ORM\EntityManager
;
7 use Symfony\Component\HttpFoundation\Session\Session
;
8 use Wallabag\CoreBundle\Entity\Entry
;
9 use Wallabag\CoreBundle\Entity\Tag
;
10 use Wallabag\CoreBundle\Tools\Utils
;
12 class PocketImport
implements ImportInterface
19 public function __construct($tokenStorage, Session
$session, EntityManager
$em, $consumerKey)
21 $this->user
= $tokenStorage->getToken()->getUser();
22 $this->session
= $session;
24 $this->consumerKey
= $consumerKey;
27 public function getName()
32 public function getDescription()
34 return 'This importer will import all your <a href="https://getpocket.com">Pocket</a> data.';
38 * Create a new Client.
42 private function createClient()
47 'content-type' => 'application/json',
48 'X-Accept' => 'application/json',
55 * Returns the good title for current entry.
61 private function guessTitle($pocketEntry)
63 if (isset($pocketEntry['resolved_title']) && $pocketEntry['resolved_title'] != '') {
64 return $pocketEntry['resolved_title'];
65 } elseif (isset($pocketEntry['given_title']) && $pocketEntry['given_title'] != '') {
66 return $pocketEntry['given_title'];
72 private function assignTagsToEntry(Entry
$entry, $tags)
74 foreach ($tags as $tag) {
75 $label = trim($tag['tag']);
76 $tagEntity = $this->em
77 ->getRepository('WallabagCoreBundle:Tag')
78 ->findOneByLabelAndUserId($label, $this->user
->getId());
80 if (is_object($tagEntity)) {
81 $entry->addTag($tagEntity);
83 $newTag = new Tag($this->user
);
84 $newTag->setLabel($label);
85 $entry->addTag($newTag);
94 private function parsePocketEntries($entries)
96 foreach ($entries as $pocketEntry) {
97 $entry = new Entry($this->user
);
98 $entry->setUrl($pocketEntry['given_url']);
99 if ($pocketEntry['status'] == 1) {
100 $entry->setArchived(true);
102 if ($pocketEntry['favorite'] == 1) {
103 $entry->setStarred(true);
106 $entry->setTitle($this->guessTitle($pocketEntry));
108 if (isset($pocketEntry['excerpt'])) {
109 $entry->setContent($pocketEntry['excerpt']);
112 if (isset($pocketEntry['has_image']) && $pocketEntry['has_image'] > 0) {
113 $entry->setPreviewPicture($pocketEntry['image']['src']);
116 if (isset($pocketEntry['word_count'])) {
117 $entry->setReadingTime(Utils
::convertWordsToMinutes($pocketEntry['word_count']));
120 if (!empty($pocketEntry['tags'])) {
121 $this->assignTagsToEntry($entry, $pocketEntry['tags']);
124 $this->em
->persist($entry);
127 $this->user
->setLastPocketImport(new \
DateTime());
131 public function oAuthRequest($redirectUri, $callbackUri)
133 $client = $this->createClient();
134 $request = $client->createRequest('POST', 'https://getpocket.com/v3/oauth/request',
136 'body' => json_encode([
137 'consumer_key' => $this->consumerKey
,
138 'redirect_uri' => $redirectUri,
143 $response = $client->send($request);
144 $values = $response->json();
146 // store code in session for callback method
147 $this->session
->set('pocketCode', $values['code']);
149 return 'https://getpocket.com/auth/authorize?request_token='.$values['code'].'&redirect_uri='.$callbackUri;
152 public function oAuthAuthorize()
154 $client = $this->createClient();
156 $request = $client->createRequest('POST', 'https://getpocket.com/v3/oauth/authorize',
158 'body' => json_encode([
159 'consumer_key' => $this->consumerKey
,
160 'code' => $this->session
->get('pocketCode'),
165 $response = $client->send($request);
167 return $response->json()['access_token'];
170 public function import($accessToken)
172 $client = $this->createClient();
173 $since = (!is_null($this->user
->getLastPocketImport()) ? $this->user
->getLastPocketImport()->getTimestamp() : '');
175 $request = $client->createRequest('POST', 'https://getpocket.com/v3/get',
177 'body' => json_encode([
178 'consumer_key' => $this->consumerKey
,
179 'access_token' => $accessToken,
180 'detailType' => 'complete',
188 $response = $client->send($request);
189 $entries = $response->json();
191 $this->parsePocketEntries($entries['list']);
193 $this->session
->getFlashBag()->add(
195 count($entries['list']).' entries imported'