1st draft for Pocket import via API

4 files changed, 174 insertions, 1 deletions

diff --git a/src/Wallabag/CoreBundle/Tools/Utils.php b/src/Wallabag/CoreBundle/Tools/Utils.php
index a16baca9..b146d98b 100644
--- a/src/Wallabag/CoreBundle/Tools/Utils.php
+++ b/src/Wallabag/CoreBundle/Tools/Utils.php
@@ -27,6 +27,15 @@ class Utils
     }
 
     /**
+     * @param $words
+     * @return float
+     */
+    public static function convertWordsToMinutes($words)
+    {
+        return floor($words / 200);
+    }
+
+    /**
      * For a given text, we calculate reading time for an article
      * based on 200 words per minute.
      *
@@ -36,6 +45,6 @@ class Utils
      */
     public static function getReadingTime($text)
     {
-        return floor(str_word_count(strip_tags($text)) / 200);
+        return self::convertWordsToMinutes(str_word_count(strip_tags($text)));
     }
 }
diff --git a/src/Wallabag/ImportBundle/Controller/PocketController.php b/src/Wallabag/ImportBundle/Controller/PocketController.php
new file mode 100644
index 00000000..ffd0c9ab
--- /dev/null
+++ b/src/Wallabag/ImportBundle/Controller/PocketController.php
@@ -0,0 +1,139 @@
+<?php
+
+namespace Wallabag\ImportBundle\Controller;
+
+use Symfony\Bundle\FrameworkBundle\Controller\Controller;
+use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
+use Symfony\Component\HttpFoundation\Request;
+use GuzzleHttp\Client;
+use Wallabag\CoreBundle\Entity\Entry;
+use Wallabag\CoreBundle\Tools\Utils;
+
+class PocketController extends Controller
+{
+    /**
+     * @Route("/import", name="import")
+     */
+    public function importAction()
+    {
+        return $this->render('WallabagImportBundle:Pocket:index.html.twig', array());
+    }
+
+    /**
+     * Create a new Client.
+     *
+     * @return Client
+     */
+    private function createClient()
+    {
+        return new Client([
+            'defaults' => [
+                'headers' => [
+                    'content-type' => 'application/json',
+                    'X-Accept' => 'application/json',
+                ],
+            ],
+        ]);
+    }
+
+    /**
+     * @Route("/auth-pocket", name="authpocket")
+     */
+    public function authAction()
+    {
+        $client = $this->createClient();
+        $request = $client->createRequest('POST', 'https://getpocket.com/v3/oauth/request',
+            [
+                'body' => json_encode([
+                    'consumer_key' => $this->container->getParameter('pocket_consumer_key'),
+                    'redirect_uri' => $this->generateUrl('import', array(), true),
+                ]),
+            ]
+        );
+
+        $response = $client->send($request);
+        $values = $response->json();
+        $code = $values['code'];
+
+        // store code in session for callback method
+        $session = $this->get('session');
+        $session->set('pocketCode',  $code);
+
+        $url = 'https://getpocket.com/auth/authorize?request_token='.$code.'&redirect_uri='.$this->generateUrl('callbackpocket', array(), true);
+
+        return $this->redirect($url, 301);
+    }
+
+    /**
+     * @Route("/callback-pocket", name="callbackpocket")
+     */
+    public function callbackAction()
+    {
+        $client = $this->createClient();
+
+        $request = $client->createRequest('POST', 'https://getpocket.com/v3/oauth/authorize',
+            [
+                'body' => json_encode([
+                    'consumer_key' => $this->container->getParameter('pocket_consumer_key'),
+                    'code' => $this->get('session')->get('pocketCode'),
+                ]),
+            ]
+        );
+
+        $response = $client->send($request);
+        $values = $response->json();
+        $accessToken = $values['access_token'];
+
+        $request = $client->createRequest('POST', 'https://getpocket.com/v3/get',
+            [
+                'body' => json_encode([
+                    'consumer_key' => $this->container->getParameter('pocket_consumer_key'),
+                    'access_token' => $accessToken,
+                    'detailType' => 'complete',
+                ]),
+            ]
+        );
+
+        $response = $client->send($request);
+        $entries = $response->json();
+
+        $this->parsePocketEntries($entries['list']);
+
+        $this->get('session')->getFlashBag()->add(
+            'notice',
+            count($entries['list']).' entries imported'
+        );
+
+        return $this->redirect($this->generateUrl('homepage'));
+    }
+
+    /**
+     * @param $entries
+     */
+    private function parsePocketEntries($entries)
+    {
+        $em = $this->getDoctrine()->getManager();
+
+        foreach ($entries as $entry) {
+            $newEntry = new Entry($this->getUser());
+            $newEntry->setUrl($entry['given_url']);
+            $newEntry->setTitle(isset($entry['resolved_title']) ? $entry['resolved_title'] : (isset($entry['given_title']) ? $entry['given_title'] : 'Untitled'));
+
+            if (isset($entry['excerpt'])) {
+                $newEntry->setContent($entry['excerpt']);
+            }
+
+            if (isset($entry['has_image']) && $entry['has_image'] > 0) {
+                $newEntry->setPreviewPicture($entry['image']['src']);
+            }
+
+            if (isset($entry['word_count'])) {
+                $newEntry->setReadingTime(Utils::convertWordsToMinutes($entry['word_count']));
+            }
+
+            $em->persist($newEntry);
+        }
+
+        $em->flush();
+    }
+}
diff --git a/src/Wallabag/ImportBundle/Resources/views/Pocket/index.html.twig b/src/Wallabag/ImportBundle/Resources/views/Pocket/index.html.twig
new file mode 100644
index 00000000..d47dd8d5
--- /dev/null
+++ b/src/Wallabag/ImportBundle/Resources/views/Pocket/index.html.twig
@@ -0,0 +1,16 @@
+{% extends "WallabagCoreBundle::layout.html.twig" %}
+{% block title %}{% trans %}import{% endtrans %}{% endblock %}
+
+{% block content %}
+
+<div class="row">
+    <div class="col s12">
+        <div class="card-panel settings">
+            {% trans %}You can import your data from your Pocket account. You just have to click on the below button and authorize the application to connect to getpocket.com.{% endtrans %}
+            <form method="post" action="{{ path('authpocket') }}">
+                <input type="submit" value="Connect to Pocket" />
+            </form>
+        </div>
+    </div>
+</div>
+{% endblock %}
diff --git a/src/Wallabag/ImportBundle/WallabagImportBundle.php b/src/Wallabag/ImportBundle/WallabagImportBundle.php
new file mode 100644
index 00000000..d00f2fe9
--- /dev/null
+++ b/src/Wallabag/ImportBundle/WallabagImportBundle.php
@@ -0,0 +1,9 @@
+<?php
+
+namespace Wallabag\ImportBundle;
+
+use Symfony\Component\HttpKernel\Bundle\Bundle;
+
+class WallabagImportBundle extends Bundle
+{
+}