WIP

Signed-off-by: Thomas Citharel <tcit@tcit.fr>

2 files changed, 81 insertions, 0 deletions

diff --git a/src/Wallabag/CoreBundle/Controller/EntryController.php b/src/Wallabag/CoreBundle/Controller/EntryController.php
index fafa49f1..ba3db806 100644
--- a/src/Wallabag/CoreBundle/Controller/EntryController.php
+++ b/src/Wallabag/CoreBundle/Controller/EntryController.php
@@ -6,7 +6,9 @@ use Pagerfanta\Adapter\DoctrineORMAdapter;
 use Pagerfanta\Exception\OutOfRangeCurrentPageException;
 use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
 use Symfony\Bundle\FrameworkBundle\Controller\Controller;
+use Symfony\Component\HttpFoundation\JsonResponse;
 use Symfony\Component\HttpFoundation\Request;
+use Symfony\Component\HttpFoundation\Response;
 use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
 use Wallabag\CoreBundle\Entity\Entry;
 use Wallabag\CoreBundle\Form\Type\EntryFilterType;
@@ -604,4 +606,50 @@ class EntryController extends Controller
     {
         return $this->showEntries('untagged', $request, $page);
     }
+
+    /**
+     * Get the progress of an entry.
+     *
+     * @param Entry $entry
+     *
+     * @Route("/progress/{entry}", name="get_progress")
+     *
+     * @return JsonResponse
+     */
+    public function getEntriesProgressAction(Entry $entry)
+    {
+        $this->checkUserAction($entry);
+
+        $json = $this->get('serializer')->serialize($entry->getProgress(), 'json');
+
+        return (new JsonResponse())->setJson($json);
+    }
+
+    /**
+     * Set the progress of an entry.
+     *
+     * @param Entry $entry
+     * @param int   $progress
+     *
+     * @Route("/set-progress/{entry}", name="set_progress")
+     *
+     * @return JsonResponse
+     */
+    public function setEntriesProgressAction(Entry $entry, $progress)
+    {
+        $this->checkUserAction($entry);
+        $response = new JsonResponse();
+
+        if (is_null($progress)) {
+            $response->setStatusCode(Response::HTTP_BAD_REQUEST);
+        } else {
+            $progress = (int) $progress;
+            if ($progress >= 0 && $progress <= 100) {
+                $entry->setProgress($progress);
+                $response->setStatusCode(Response::HTTP_OK);
+            }
+        }
+
+        return $response;
+    }
 }
diff --git a/src/Wallabag/CoreBundle/Entity/Entry.php b/src/Wallabag/CoreBundle/Entity/Entry.php
index a0503c39..639af6aa 100644
--- a/src/Wallabag/CoreBundle/Entity/Entry.php
+++ b/src/Wallabag/CoreBundle/Entity/Entry.php
@@ -211,6 +211,15 @@ class Entry
     private $headers;
 
     /**
+     * @var int
+     *
+     * @ORM\Column(name="progress", type="integer", nullable=true)
+     *
+     * @Groups({"entries_for_user", "export_all"})
+     */
+    private $progress = 0;
+
+    /**
      * @Exclude
      *
      * @ORM\ManyToOne(targetEntity="Wallabag\UserBundle\Entity\User", inversedBy="entries")
@@ -760,6 +769,20 @@ class Entry
     }
 
     /**
+     * Set progress.
+     *
+     * @param int $progress
+     *
+     * @return Entry
+     */
+    public function setProgress($progress)
+    {
+        $this->progress = $progress;
+
+        return $this;
+    }
+
+    /**
      * @return array
      */
     public function getHeaders()
@@ -778,4 +801,14 @@ class Entry
 
         return $this;
     }
+
+    /**
+     * Get progress.
+     *
+     * @return int
+     */
+    public function getProgress()
+    {
+        return $this->progress;
+    }
 }