Made Twig download extension compatible with the file id

Author: benatespina

src/BenGorFile/FileBundle/BenGorFileBundle.php

@@ -17,6 +17,7 @@
 use BenGorFile\FileBundle\DependencyInjection\Compiler\ApplicationQueriesPass;
 use BenGorFile\FileBundle\DependencyInjection\Compiler\DomainServicesPass;
 use BenGorFile\FileBundle\DependencyInjection\Compiler\RoutesPass;
+use BenGorFile\FileBundle\DependencyInjection\Compiler\TwigPass;
 use Symfony\Component\DependencyInjection\Compiler\PassConfig;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
@@ -38,6 +39,7 @@ public function build(ContainerBuilder $container)
         $container->addCompilerPass(new ApplicationDataTransformersPass(), PassConfig::TYPE_OPTIMIZE);
         $container->addCompilerPass(new ApplicationQueriesPass(), PassConfig::TYPE_OPTIMIZE);
         $container->addCompilerPass(new RoutesPass(), PassConfig::TYPE_OPTIMIZE);
+        $container->addCompilerPass(new TwigPass(), PassConfig::TYPE_OPTIMIZE);
 
         $this->buildLoadableBundles($container);
     }

src/BenGorFile/FileBundle/DependencyInjection/Compiler/TwigPass.php

@@ -0,0 +1,46 @@
+<?php
+
+/*
+ * This file is part of the BenGorFile package.
+ *
+ * (c) Beñat Espiña <benatespina@gmail.com>
+ * (c) Gorka Laucirica <gorka.lauzirika@gmail.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace BenGorFile\FileBundle\DependencyInjection\Compiler;
+
+use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+
+/**
+ * Register Twig services compiler pass.
+ *
+ * @author Beñat Espiña <benatespina@gmail.com>
+ */
+class TwigPass implements CompilerPassInterface
+{
+    /**
+     * {@inheritdoc}
+     */
+    public function process(ContainerBuilder $container)
+    {
+        if (!$container->hasDefinition('bengor_file.file_bundle.twig.download_extension')) {
+            return;
+        }
+
+        $config = $container->getParameter('bengor_file.config');
+
+        $handlers = [];
+        foreach ($config['file_class'] as $key => $file) {
+            $handlers[$key] = $container->getDefinition('bengor.file.application.query.' . $key . '_of_id');
+        }
+
+        $container->getDefinition('bengor_file.file_bundle.twig.download_extension')->replaceArgument(
+            1,
+            $handlers
+        );
+    }
+}

src/BenGorFile/FileBundle/Resources/config/services/twig.yml

@@ -11,6 +11,7 @@ services:
         class: BenGorFile\FileBundle\Twig\DownloadExtension
         arguments:
             - '@router'
+            - []
         public: false
         tags:
             - { name: twig.extension }

src/BenGorFile/FileBundle/Twig/DownloadExtension.php

@@ -12,10 +12,12 @@
 
 namespace BenGorFile\FileBundle\Twig;
 
+use BenGorFile\File\Application\Query\FileOfIdQuery;
+use BenGorFile\File\Domain\Model\FileDoesNotExistException;
 use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
 
 /**
- * File download Twig function.
+ * File download by id Twig function.
  *
  * @author Beñat Espiña <benatespina@gmail.com>
  */
@@ -28,14 +30,23 @@ class DownloadExtension extends \Twig_Extension
      */
     private $urlGenerator;
 
+    /**
+     * The file handlers.
+     *
+     * @var array
+     */
+    private $handlers;
+
     /**
      * Constructor.
      *
      * @param UrlGeneratorInterface $anUrlGenerator The URL generator
+     * @param array                 $handlers       The file handlers
      */
-    public function __construct(UrlGeneratorInterface $anUrlGenerator)
+    public function __construct(UrlGeneratorInterface $anUrlGenerator, array $handlers)
     {
         $this->urlGenerator = $anUrlGenerator;
+        $this->handlers = $handlers;
     }
 
     /**
@@ -49,18 +60,25 @@ public function getFunctions()
     }
 
     /**
-     * Generates the url that returns the file of given file type and name.
+     * Generates the url that returns the file of given file type and file.
      *
      * @param string $fileClass The file type type
-     * @param string $name      The file name
+     * @param string $q         The file query
      *
      * @return string
      */
-    public function download($fileClass, $name)
+    public function download($fileClass, $q)
     {
+        try {
+            $file = $this->handlers[$fileClass]->__invoke(new FileOfIdQuery($q));
+            $filename = $file['file_name'];
+        } catch (FileDoesNotExistException $exception) {
+            $filename = $q;
+        }
+
         return $this->urlGenerator->generate(
             'bengor_file_' . $fileClass . '_download',
-            ['filename' => $name],
+            ['filename' => $filename],
             UrlGeneratorInterface::ABSOLUTE_URL
         );
     }