From b67951d1286a5d60decb6fdd2778ab15aaa121a8 Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Wed, 29 Nov 2023 08:57:43 +1100
Subject: [PATCH] catches for company set

---
 app/Services/Pdf/PdfBuilder.php    |  1 +
 app/Services/PdfMaker/PdfMaker.php | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/app/Services/Pdf/PdfBuilder.php b/app/Services/Pdf/PdfBuilder.php
index cb269d85b55f..789a34ec97f4 100644
--- a/app/Services/Pdf/PdfBuilder.php
+++ b/app/Services/Pdf/PdfBuilder.php
@@ -113,6 +113,7 @@ class PdfBuilder
         $contents = $this->document->getElementsByTagName('ninja');
                 
         $template_service = new TemplateService();
+        $template_service->setCompany($this->service->company);
         $data = $template_service->processData($this->service->options)->getData();
 
         $twig = $template_service->twig;
diff --git a/app/Services/PdfMaker/PdfMaker.php b/app/Services/PdfMaker/PdfMaker.php
index 680bf7ef8146..c13f1a974d85 100644
--- a/app/Services/PdfMaker/PdfMaker.php
+++ b/app/Services/PdfMaker/PdfMaker.php
@@ -81,6 +81,17 @@ class PdfMaker
             $contents = $this->document->getElementsByTagName('ninja');
             
             $ts = new TemplateService();
+
+            if(isset($this->data['template']['entity'])){
+                try{
+                    $entity = $this->data['template']['entity'];
+                    $ts->setCompany($entity->company);
+                }
+                catch(\Exception $e){
+                 
+                }
+            }
+
             $data = $ts->processData($this->options)->getData();
             $twig = $ts->twig;