improved function

[github/wallabag/wallabag.git] / src / Wallabag / CoreBundle / Helper / EntriesExport.php

diff --git a/src/Wallabag/CoreBundle/Helper/EntriesExport.php b/src/Wallabag/CoreBundle/Helper/EntriesExport.php
index 806319b1a28e24778f21d5640930ead4565b241e..e073606cf9dfe155df5f9d176f4f3f243efc6896 100644 (file)
--- a/src/Wallabag/CoreBundle/Helper/EntriesExport.php
+++ b/src/Wallabag/CoreBundle/Helper/EntriesExport.php
@@ -5,6 +5,10 @@ namespace Wallabag\CoreBundle\Helper;
 use PHPePub\Core\EPub;
 use PHPePub\Core\Structure\OPF\DublinCore;
 use Symfony\Component\HttpFoundation\Response;
+use Symfony\Component\Serializer\Serializer;
+use Symfony\Component\Serializer\Encoder\XmlEncoder;
+use Symfony\Component\Serializer\Encoder\JsonEncoder;
+use Symfony\Component\Serializer\Normalizer\ObjectNormalizer;
 
 class EntriesExport
 {
@@ -86,6 +90,12 @@ class EntriesExport
 
             case 'csv':
                 return $this->produceCSV();
+
+            case 'json':
+                return $this->produceJSON();
+
+            case 'xml':
+                return $this->produceXML();
         }
 
         throw new \InvalidArgumentException(sprintf('The format "%s" is not yet supported.', $format));
@@ -319,6 +329,47 @@ class EntriesExport
         )->send();
     }
 
+    private function produceJSON()
+    {
+        return Response::create(
+            $this->prepareSerializingContent('json'),
+            200,
+            array(
+                'Content-type' => 'application/json',
+                'Content-Disposition' => 'attachment; filename="'.$this->title.'.json"',
+                'Content-Transfer-Encoding' => 'UTF-8',
+            )
+        )->send();
+    }
+
+    private function produceXML()
+    {
+        return Response::create(
+            $this->prepareSerializingContent('xml'),
+            200,
+            array(
+                'Content-type' => 'application/xml',
+                'Content-Disposition' => 'attachment; filename="'.$this->title.'.xml"',
+                'Content-Transfer-Encoding' => 'UTF-8',
+            )
+        )->send();
+    }
+
+    /**
+     * Return a Serializer object for producing processes that need it (JSON & XML).
+     *
+     * @return Serializer
+     */
+    private function prepareSerializingContent($format)
+    {
+        $encoders = array(new XmlEncoder(), new JsonEncoder());
+        $normalizers = array(new ObjectNormalizer());
+        $normalizers[0]->setIgnoredAttributes(array('user', 'createdAt', 'updatedAt'));
+        $serializer = new Serializer($normalizers, $encoders);
+
+        return $serializer->serialize($this->entries, $format);
+    }
+
     /**
      * Return a kind of footer / information for the epub.
      *