customize html and pdf exports with twig templates (#1582)

src/Export/Base/HtmlRenderer.php

@@ -17,6 +17,7 @@ use App\Event\ProjectMetaDisplayEvent;
 use App\Event\TimesheetMetaDisplayEvent;
 use App\Event\UserPreferenceDisplayEvent;
 use App\Export\ExportItemInterface;
+use App\Repository\ProjectRepository;
 use App\Repository\Query\CustomerQuery;
 use App\Repository\Query\TimesheetQuery;
 use Symfony\Component\EventDispatcher\EventDispatcherInterface;
@@ -35,11 +36,24 @@ class HtmlRenderer
      * @var EventDispatcherInterface
      */
     protected $dispatcher;
+    /**
+     * @var ProjectRepository
+     */
+    private $projectRepository;
+    /**
+     * @var string
+     */
+    private $id = 'html';
+    /**
+     * @var string
+     */
+    private $template = 'default.html.twig';
 
-    public function __construct(Environment $twig, EventDispatcherInterface $dispatcher)
+    public function __construct(Environment $twig, EventDispatcherInterface $dispatcher, ProjectRepository $projectRepository)
     {
         $this->twig = $twig;
         $this->dispatcher = $dispatcher;
+        $this->projectRepository = $projectRepository;
     }
 
     /**
@@ -87,10 +101,13 @@ class HtmlRenderer
         $this->dispatcher->dispatch($event);
         $userPreferences = $event->getPreferences();
 
-        $content = $this->twig->render('export/renderer/default.html.twig', array_merge([
+        $summary = $this->calculateSummary($timesheets);
+
+        $content = $this->twig->render($this->getTemplate(), array_merge([
             'entries' => $timesheets,
             'query' => $query,
-            'summaries' => $this->calculateSummary($timesheets),
+            'summaries' => $summary,
+            'budgets' => $this->calculateProjectBudget($timesheets, $query, $this->projectRepository),
             'timesheetMetaFields' => $timesheetMetaFields,
             'customerMetaFields' => $customerMetaFields,
             'projectMetaFields' => $projectMetaFields,
@@ -104,8 +121,27 @@ class HtmlRenderer
         return $response;
     }
 
+    protected function getTemplate(): string
+    {
+        return '@export/' . $this->template;
+    }
+
+    public function setTemplate(string $filename): HtmlRenderer
+    {
+        $this->template = $filename;
+
+        return $this;
+    }
+
+    public function setId(string $id): HtmlRenderer
+    {
+        $this->id = $id;
+
+        return $this;
+    }
+
     public function getId(): string
     {
-        return 'html';
+        return $this->id;
     }
 }

src/Export/Base/PDFRenderer.php

@@ -10,6 +10,7 @@
 namespace App\Export\Base;
 
 use App\Export\ExportItemInterface;
+use App\Repository\ProjectRepository;
 use App\Repository\Query\TimesheetQuery;
 use App\Timesheet\UserDateTimeFactory;
 use App\Utils\HtmlToPdfConverter;
@@ -33,17 +34,30 @@ class PDFRenderer
      * @var HtmlToPdfConverter
      */
     private $converter;
+    /**
+     * @var ProjectRepository
+     */
+    private $projectRepository;
+    /**
+     * @var string
+     */
+    private $id = 'pdf';
+    /**
+     * @var string
+     */
+    private $template = 'default.pdf.twig';
 
-    public function __construct(Environment $twig, UserDateTimeFactory $dateTime, HtmlToPdfConverter $converter)
+    public function __construct(Environment $twig, UserDateTimeFactory $dateTime, HtmlToPdfConverter $converter, ProjectRepository $projectRepository)
     {
         $this->twig = $twig;
         $this->dateTime = $dateTime;
         $this->converter = $converter;
+        $this->projectRepository = $projectRepository;
     }
 
     protected function getTemplate(): string
     {
-        return 'export/renderer/pdf.html.twig';
+        return '@export/' . $this->template;
     }
 
     protected function getOptions(TimesheetQuery $query): array
@@ -68,11 +82,13 @@ class PDFRenderer
      */
     public function render(array $timesheets, TimesheetQuery $query): Response
     {
+        $summary = $this->calculateSummary($timesheets);
         $content = $this->twig->render($this->getTemplate(), array_merge([
             'entries' => $timesheets,
             'query' => $query,
             'now' => $this->dateTime->createDateTime(),
-            'summaries' => $this->calculateSummary($timesheets),
+            'summaries' => $summary,
+            'budgets' => $this->calculateProjectBudget($timesheets, $query, $this->projectRepository),
             'decimal' => false,
         ], $this->getOptions($query)));
 
@@ -88,8 +104,22 @@ class PDFRenderer
         return $response;
     }
 
+    public function setTemplate(string $filename): PDFRenderer
+    {
+        $this->template = $filename;
+
+        return $this;
+    }
+
+    public function setId(string $id): PDFRenderer
+    {
+        $this->id = $id;
+
+        return $this;
+    }
+
     public function getId(): string
     {
-        return 'pdf';
+        return $this->id;
     }
 }

src/Export/Base/RendererTrait.php

@@ -10,6 +10,8 @@
 namespace App\Export\Base;
 
 use App\Export\ExportItemInterface;
+use App\Repository\ProjectRepository;
+use App\Repository\Query\TimesheetQuery;
 
 trait RendererTrait
 {
@@ -80,4 +82,52 @@ trait RendererTrait
 
         return $summary;
     }
+
+    /**
+     * @param ExportItemInterface[] $exportItems
+     * @param TimesheetQuery $query
+     * @param ProjectRepository $projectRepository
+     * @return array
+     */
+    protected function calculateProjectBudget(array $exportItems, TimesheetQuery $query, ProjectRepository $projectRepository)
+    {
+        $summary = [];
+
+        foreach ($exportItems as $exportItem) {
+            $customer = null;
+            $customerId = 'none';
+            $project = null;
+            $projectId = 'none';
+
+            if (null !== ($project = $exportItem->getProject())) {
+                $customer = $project->getCustomer();
+                $customerId = $customer->getId();
+                $projectId = $project->getId();
+            }
+
+            $id = $customerId . '_' . $projectId;
+
+            if (!isset($summary[$id])) {
+                $summary[$id] = [
+                    'time' => $project->getTimeBudget(),
+                    'money' => $project->getBudget(),
+                    'time_left' => null,
+                    'money_left' => null,
+                ];
+
+                if (null !== $project && ($project->getTimeBudget() > 0 || $project->getBudget() > 0)) {
+                    $projectStats = $projectRepository->getProjectStatistics($project, null, $query->getEnd());
+
+                    if ($project->getTimeBudget() > 0) {
+                        $summary[$id]['time_left'] = $project->getTimeBudget() - $projectStats->getRecordDuration();
+                    }
+                    if ($project->getBudget() > 0) {
+                        $summary[$id]['money_left'] = $project->getBudget() - $projectStats->getRecordRate();
+                    }
+                }
+            }
+        }
+
+        return $summary;
+    }
 }