6 files changed, 361 insertions, 4 deletions

diff --git a/src/Wallabag/CoreBundle/Controller/ExportController.php b/src/Wallabag/CoreBundle/Controller/ExportController.php
new file mode 100644
index 00000000..123e491a
--- /dev/null
+++ b/src/Wallabag/CoreBundle/Controller/ExportController.php
@@ -0,0 +1,65 @@
+<?php
+
+namespace Wallabag\CoreBundle\Controller;
+
+use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
+use Symfony\Bundle\FrameworkBundle\Controller\Controller;
+use Wallabag\CoreBundle\Entity\Entry;
+use Wallabag\CoreBundle\Helper\EntriesExport;
+
+class ExportController extends Controller
+{
+    /**
+     * Gets all entries for current user.
+     *
+     * @Route("/export/{category}.{format}", name="ebook", requirements={
+     *     "_format": "epub|mobi|pdf|json|xml|txt|csv"
+     * })
+     */
+    public function getEntriesAction($format, $category)
+    {
+        $repository = $this->getDoctrine()->getRepository('WallabagCoreBundle:Entry');
+        switch ($category) {
+            case 'all':
+                $method = 'All';
+                break;
+
+            case 'unread':
+                $method = 'Unread';
+                break;
+
+            case 'starred':
+                $method = 'Starred';
+                break;
+
+            case 'archive':
+                $method = 'Archive';
+                break;
+
+            default:
+                break;
+        }
+
+        $methodBuilder = 'getBuilderFor'.$method.'ByUser';
+        $qb = $repository->$methodBuilder($this->getUser()->getId());
+        $entries = $qb->getQuery()->getResult();
+
+        $export = new EntriesExport($entries);
+        $export->setMethod($method);
+        $export->exportAs($format);
+    }
+
+    /**
+     * Gets one entry content.
+     *
+     * @param Entry $entry
+     *
+     * @Route("/export/id/{id}.{format}", requirements={"id" = "\d+"}, name="ebook_entry")
+     */
+    public function getEntryAction(Entry $entry, $format)
+    {
+        $export = new EntriesExport(array($entry));
+        $export->setMethod('entry');
+        $export->exportAs($format);
+    }
+}
diff --git a/src/Wallabag/CoreBundle/Helper/EntriesExport.php b/src/Wallabag/CoreBundle/Helper/EntriesExport.php
new file mode 100644
index 00000000..fad0bb97
--- /dev/null
+++ b/src/Wallabag/CoreBundle/Helper/EntriesExport.php
@@ -0,0 +1,263 @@
+<?php
+
+namespace Wallabag\CoreBundle\Helper;
+
+use PHPePub\Core\EPub;
+use PHPePub\Core\Structure\OPF\DublinCore;
+
+class EntriesExport
+{
+    private $format;
+    private $method;
+    private $title;
+    private $entries;
+    private $authors = array('wallabag');
+    private $language;
+    private $tags;
+
+    public function __construct($entries)
+    {
+        $this->entries = $entries;
+
+        foreach ($entries as $entry) {
+            $this->tags[] = $entry->getTags();
+        }
+        if (count($entries) === 1) {
+            $this->language = $entries[0]->getLanguage();
+        }
+    }
+
+    /**
+     * Sets the category of which we want to get articles, or just one entry.
+     *
+     * @param string $method Method to get articles
+     */
+    public function setMethod($method)
+    {
+        $this->method = $method;
+
+        switch ($this->method) {
+            case 'All':
+                $this->title = 'All Articles';
+                break;
+            case 'Unread':
+                $this->title = 'Unread articles';
+                break;
+            case 'Starred':
+                $this->title = 'Starred articles';
+                break;
+            case 'Archive':
+                $this->title = 'Archived articles';
+                break;
+            case 'entry':
+                $this->title = $this->entries[0]->getTitle();
+                break;
+            default:
+                break;
+        }
+    }
+
+    /**
+     * Sets the output format.
+     *
+     * @param string $format
+     */
+    public function exportAs($format)
+    {
+        $this->format = $format;
+
+        switch ($this->format) {
+            case 'epub':
+                $this->produceEpub();
+                break;
+
+            case 'mobi':
+                $this->produceMobi();
+                break;
+
+            case 'pdf':
+                $this->producePDF();
+                break;
+
+            case 'csv':
+                $this->produceCSV();
+                break;
+
+            default:
+                break;
+        }
+    }
+
+    private function produceEpub()
+    {
+        /*
+         * Start and End of the book
+         */
+        $content_start =
+            "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
+            ."<html xmlns=\"http://www.w3.org/1999/xhtml\" xmlns:epub=\"http://www.idpf.org/2007/ops\">\n"
+            .'<head>'
+            ."<meta http-equiv=\"Default-Style\" content=\"text/html; charset=utf-8\" />\n"
+            .'<title>'._('wallabag articles book')."</title>\n"
+            ."</head>\n"
+            ."<body>\n";
+
+        $bookEnd = "</body>\n</html>\n";
+
+        $book = new EPub(EPub::BOOK_VERSION_EPUB3);
+
+        /*
+         * Book metadata
+         */
+
+        $book->setTitle($this->title);
+        $book->setIdentifier($this->title, EPub::IDENTIFIER_URI); // Could also be the ISBN number, prefered for published books, or a UUID.
+        $book->setLanguage($this->language); // Not needed, but included for the example, Language is mandatory, but EPub defaults to "en". Use RFC3066 Language codes, such as "en", "da", "fr" etc.
+        $book->setDescription(_('Some articles saved on my wallabag'));
+
+        foreach ($this->authors as $author) {
+            $book->setAuthor($author, $author);
+        }
+
+        $book->setPublisher('wallabag', 'wallabag'); // I hope this is a non existant address :)
+        $book->setDate(time()); // Strictly not needed as the book date defaults to time().
+        $book->setSourceURL("http://$_SERVER[HTTP_HOST]");
+
+        $book->addDublinCoreMetadata(DublinCore::CONTRIBUTOR, 'PHP');
+        $book->addDublinCoreMetadata(DublinCore::CONTRIBUTOR, 'wallabag');
+
+        /*
+         * Front page
+         */
+
+        $book->setCoverImage('Cover.png', file_get_contents('themes/_global/img/appicon/apple-touch-icon-152.png'), 'image/png');
+
+        $cover = $content_start.'<div style="text-align:center;"><p>'._('Produced by wallabag with PHPePub').'</p><p>'._('Please open <a href="https://github.com/wallabag/wallabag/issues" >an issue</a> if you have trouble with the display of this E-Book on your device.').'</p></div>'.$bookEnd;
+
+        $book->addChapter('Notices', 'Cover2.html', $cover);
+
+        $book->buildTOC();
+
+        /*
+         * Adding actual entries
+         */
+
+        foreach ($this->entries as $entry) { //set tags as subjects
+                foreach ($this->tags as $tag) {
+                    $book->setSubject($tag['value']);
+                }
+
+            $chapter = $content_start.$entry->getContent().$bookEnd;
+            $book->addChapter($entry->getTitle(), htmlspecialchars($entry->getTitle()).'.html', $chapter, true, EPub::EXTERNAL_REF_ADD);
+        }
+        $book->finalize();
+        $book->sendBook($this->title);
+    }
+
+    private function produceMobi()
+    {
+        $mobi = new \MOBI();
+        $content = new \MOBIFile();
+
+        /*
+         * Book metadata
+         */
+
+        $content->set('title', $this->title);
+        $content->set('author', implode($this->authors));
+        $content->set('subject', $this->title);
+
+        /*
+         * Front page
+         */
+
+        $content->appendParagraph('<div style="text-align:center;" ><p>'._('Produced by wallabag with PHPMobi').'</p><p>'._('Please open <a href="https://github.com/wallabag/wallabag/issues" >an issue</a> if you have trouble with the display of this E-Book on your device.').'</p></div>');
+        $content->appendImage(imagecreatefrompng('themes/_global/img/appicon/apple-touch-icon-152.png'));
+        $content->appendPageBreak();
+
+        /*
+         * Adding actual entries
+         */
+
+        foreach ($this->entries as $entry) {
+            $content->appendChapterTitle($entry->getTitle());
+            $content->appendParagraph($entry->getContent());
+            $content->appendPageBreak();
+        }
+        $mobi->setContentProvider($content);
+
+        // the browser inside Kindle Devices doesn't likes special caracters either, we limit to A-z/0-9
+        $this->title = preg_replace('/[^A-Za-z0-9\-]/', '', $this->title);
+
+        // we offer file to download
+        $mobi->download($this->title.'.mobi');
+    }
+
+    private function producePDF()
+    {
+        $pdf = new \TCPDF(PDF_PAGE_ORIENTATION, PDF_UNIT, PDF_PAGE_FORMAT, true, 'UTF-8', false);
+
+        /*
+         * Book metadata
+         */
+
+        $pdf->SetCreator(PDF_CREATOR);
+        $pdf->SetAuthor('wallabag');
+        $pdf->SetTitle($this->title);
+        $pdf->SetSubject('Articles via wallabag');
+        $pdf->SetKeywords('wallabag');
+
+        /*
+         * Front page
+         */
+
+        $pdf->AddPage();
+        $intro = '<h1>'.$this->title.'</h1><div style="text-align:center;" >
+        <p>'._('Produced by wallabag with tcpdf').'</p>
+        <p>'._('Please open <a href="https://github.com/wallabag/wallabag/issues" >an issue</a> if you have trouble with the display of this E-Book on your device.').'</p>
+        <img src="themes/_global/img/appicon/apple-touch-icon-152.png" /></div>';
+
+        $pdf->writeHTMLCell(0, 0, '', '', $intro, 0, 1, 0, true, '', true);
+
+        /*
+         * Adding actual entries
+         */
+
+        foreach ($this->entries as $entry) {
+            foreach ($this->tags as $tag) {
+                $pdf->SetKeywords($tag['value']);
+            }
+
+            $pdf->AddPage();
+            $html = '<h1>'.$entry->getTitle().'</h1>';
+            $html .= $entry->getContent();
+            $pdf->writeHTMLCell(0, 0, '', '', $html, 0, 1, 0, true, '', true);
+        }
+
+        // set image scale factor
+        $pdf->setImageScale(PDF_IMAGE_SCALE_RATIO);
+
+        $pdf->Output($this->title.'.pdf', 'D');
+    }
+
+    private function produceCSV()
+    {
+        header('Content-type: application/csv');
+        header('Content-Disposition: attachment; filename="'.$this->title.'.csv"');
+        header('Content-Transfer-Encoding: UTF-8');
+
+        $output = fopen('php://output', 'a');
+
+        fputcsv($output, array('Title', 'URL', 'Content', 'Tags', 'MIME Type', 'Language'));
+        foreach ($this->entries as $entry) {
+            fputcsv($output, array($entry->getTitle(),
+                                   $entry->getURL(),
+                                   $entry->getContent(),
+                                   implode(', ', $entry->getTags()->toArray()),
+                                   $entry->getMimetype(),
+                                   $entry->getLanguage(), ));
+        }
+        fclose($output);
+        exit();
+    }
+}
diff --git a/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entries.html.twig b/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entries.html.twig
index 668824bc..5a231c86 100644
--- a/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entries.html.twig
+++ b/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entries.html.twig
@@ -91,6 +91,24 @@
         {% endfor %}
     </ul>
 
+    <!-- Export -->
+    <div id="export" class="side-nav fixed right-aligned">
+    {% set currentRoute = app.request.attributes.get('_route') %}
+    {% if currentRoute == 'homepage' %}
+        {% set currentRoute = 'unread' %}
+    {% endif %}
+        <h4 class="center">{% trans %}Export{% endtrans %}</h4>
+        <ul>
+            <li class="bold"><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'epub' }) }}">{% trans %}EPUB{% endtrans %}</a></li>
+            <li class="bold"><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'mobi' }) }}">{% trans %}MOBI{% endtrans %}</a></li>
+            <li class="bold"><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'pdf' }) }}">{% trans %}PDF{% endtrans %}</a></li>
+            <li class="bold"><del><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'xml' }) }}">{% trans %}XML{% endtrans %}</a></del></li>
+            <li class="bold"><del><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'json' }) }}">{% trans %}JSON{% endtrans %}</a></del></li>
+            <li class="bold"><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'csv' }) }}">{% trans %}CSV{% endtrans %}</a></li>
+            <li class="bold"><del><a class="waves-effect" href="{{ path('ebook', { 'category': currentRoute, 'format': 'txt' }) }}">{% trans %}TXT{% endtrans %}</a></del></li>
+        </ul>
+    </div>
+
     <!-- Filters -->
     <div id="filters" class="side-nav fixed right-aligned">
         <form action="{{ path('all') }}">
diff --git a/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entry.html.twig b/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entry.html.twig
index 7230506c..bece099a 100644
--- a/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entry.html.twig
+++ b/src/Wallabag/CoreBundle/Resources/views/themes/material/Entry/entry.html.twig
@@ -102,13 +102,14 @@
         <li class="bold">
             <a class="waves-effect collapsible-header">
                 <i class="mdi-file-file-download small"></i>
-                <span><del>{% trans %}Download{% endtrans %}</del></span>
+                <span>{% trans %}Download{% endtrans %}</span>
             </a>
             <div class="collapsible-body">
                 <ul>
-                    {% if export_epub %}<li><del><a href="?epub&amp;method=id&amp;value={{ entry.id }}" title="Generate ePub file">EPUB</a></del></li>{% endif %}
-                    {% if export_mobi %}<li><del><a href="?mobi&amp;method=id&amp;value={{ entry.id }}" title="Generate Mobi file">MOBI</a></del></li>{% endif %}
-                    {% if export_pdf %}<li><del><a href="?pdf&amp;method=id&amp;value={{ entry.id }}" title="Generate PDF file">PDF</a></del> </li>{% endif %}
+                    {% if export_epub %}<li><a href="{{ path('ebook_entry', { 'id': entry.id, 'format': 'epub' }) }}" title="Generate ePub file">EPUB</a></li>{% endif %}
+                    {% if export_mobi %}<li><a href="{{ path('ebook_entry', { 'id': entry.id, 'format': 'mobi' }) }}" title="Generate Mobi file">MOBI</a></li>{% endif %}
+                    {% if export_pdf %}<li><a href="{{ path('ebook_entry', { 'id': entry.id, 'format': 'pdf' }) }}" title="Generate PDF file">PDF</a></li>{% endif %}
+                    <li><a href="{{ path('ebook_entry', { 'id': entry.id, 'format': 'csv' }) }}" title="Generate CSV file">CSV</a></li>
                 </ul>
             </div>
         </li>
diff --git a/src/Wallabag/CoreBundle/Resources/views/themes/material/layout.html.twig b/src/Wallabag/CoreBundle/Resources/views/themes/material/layout.html.twig
index 95b3977c..f426e25b 100644
--- a/src/Wallabag/CoreBundle/Resources/views/themes/material/layout.html.twig
+++ b/src/Wallabag/CoreBundle/Resources/views/themes/material/layout.html.twig
@@ -59,6 +59,7 @@
                     <li class="bold"><a title="{% trans %}Add a new entry{% endtrans %}" class="waves-effect" href="{{ path('new') }}" id="nav-btn-add"><i class="mdi-content-add"></i></a></li>
                     <li><a title="{% trans %}Search{% endtrans %}" class="waves-effect" href="javascript: void(null);" id="nav-btn-search"><i class="mdi-action-search"></i></a>
                     <li id="button_filters"><a title="{% trans %}Filter entries{% endtrans %}" href="#" data-activates="filters" class="nav-panel-menu button-collapse-right"><i class="mdi-content-filter-list"></i></a></li>
+                    <li id="button_export"><a title="{% trans %}Export{% endtrans %}" class="nav-panel-menu button-collapse-right" href="#" data-activates="export" class="nav-panel-menu button-collapse-right"><i class="mdi-file-file-download"></i></a></li>
                 </ul>
             </div>
             <form method="get" action="index.php">
diff --git a/src/Wallabag/CoreBundle/Resources/views/themes/material/public/js/init.js b/src/Wallabag/CoreBundle/Resources/views/themes/material/public/js/init.js
index edfdee82..491a7916 100755
--- a/src/Wallabag/CoreBundle/Resources/views/themes/material/public/js/init.js
+++ b/src/Wallabag/CoreBundle/Resources/views/themes/material/public/js/init.js
@@ -11,6 +11,14 @@ function init_filters() {
     }
 }
 
+function init_export() {
+    // no display if export not aviable
+    if ($("div").is("#export")) {
+        $('#button_export').show();
+        $('.button-collapse-right').sideNav({ edge: 'right' });
+    }
+}
+
 $(document).ready(function(){
     // sideNav
     $('.button-collapse').sideNav();
@@ -26,6 +34,7 @@ $(document).ready(function(){
         format: 'dd/mm/yyyy',
     });
     init_filters();
+    init_export();
 
     $('#nav-btn-add-tag').on('click', function(){
        $(".nav-panel-add-tag").toggle(100);