src/Wallabag/CoreBundle/Controller/ExportController.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Controller;
   4
   5 use Symfony\Bundle\FrameworkBundle\Controller\Controller;
   6 use Symfony\Component\HttpFoundation\Request;
   7 use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
   8 use Symfony\Component\Routing\Annotation\Route;
   9 use Wallabag\CoreBundle\Entity\Entry;
  10
  11 /**
  12  * The try/catch can be removed once all formats will be implemented.
  13  * Still need implementation: txt.
  14  */
  15 class ExportController extends Controller
  16 {
  17     /**
  18      * Gets one entry content.
  19      *
  20      * @param string $format
  21      *
  22      * @Route("/export/{id}.{format}", name="export_entry", requirements={
  23      *     "format": "epub|mobi|pdf|json|xml|txt|csv",
  24      *     "id": "\d+"
  25      * })
  26      *
  27      * @return \Symfony\Component\HttpFoundation\Response
  28      */
  29     public function downloadEntryAction(Entry $entry, $format)
  30     {
  31         try {
  32             return $this->get('wallabag_core.helper.entries_export')
  33                 ->setEntries($entry)
  34                 ->updateTitle('entry')
  35                 ->updateAuthor('entry')
  36                 ->exportAs($format);
  37         } catch (\InvalidArgumentException $e) {
  38             throw new NotFoundHttpException($e->getMessage());
  39         }
  40     }
  41
  42     /**
  43      * Export all entries for current user.
  44      *
  45      * @param string $format
  46      * @param string $category
  47      *
  48      * @Route("/export/{category}.{format}", name="export_entries", requirements={
  49      *     "format": "epub|mobi|pdf|json|xml|txt|csv",
  50      *     "category": "all|unread|starred|archive|tag_entries|untagged|search"
  51      * })
  52      *
  53      * @return \Symfony\Component\HttpFoundation\Response
  54      */
  55     public function downloadEntriesAction(Request $request, $format, $category)
  56     {
  57         $method = ucfirst($category);
  58         $methodBuilder = 'getBuilderFor' . $method . 'ByUser';
  59         $repository = $this->get('wallabag_core.entry_repository');
  60         $title = $method;
  61
  62         if ('tag_entries' === $category) {
  63             $tag = $this->get('wallabag_core.tag_repository')->findOneBySlug($request->query->get('tag'));
  64
  65             $entries = $repository->findAllByTagId(
  66                 $this->getUser()->getId(),
  67                 $tag->getId()
  68             );
  69
  70             $title = 'Tag ' . $tag->getLabel();
  71         } elseif ('search' === $category) {
  72             $searchTerm = (isset($request->get('search_entry')['term']) ? $request->get('search_entry')['term'] : '');
  73             $currentRoute = (null !== $request->query->get('currentRoute') ? $request->query->get('currentRoute') : '');
  74
  75             $entries = $repository->getBuilderForSearchByUser(
  76                     $this->getUser()->getId(),
  77                     $searchTerm,
  78                     $currentRoute
  79             )->getQuery()
  80              ->getResult();
  81
  82             $title = 'Search ' . $searchTerm;
  83         } else {
  84             $entries = $repository
  85                 ->$methodBuilder($this->getUser()->getId())
  86                 ->getQuery()
  87                 ->getResult();
  88         }
  89
  90         try {
  91             return $this->get('wallabag_core.helper.entries_export')
  92                 ->setEntries($entries)
  93                 ->updateTitle($title)
  94                 ->updateAuthor($method)
  95                 ->exportAs($format);
  96         } catch (\InvalidArgumentException $e) {
  97             throw new NotFoundHttpException($e->getMessage());
  98         }
  99     }
 100 }