From 66fe5aa4ee877528307ebc39b66250b3ab838dac Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Tue, 23 Jun 2020 07:56:32 +1000
Subject: [PATCH] Add documents to transformers

---
 app/Transformers/PaymentTransformer.php |  8 ++++++++
 app/Transformers/ProductTransformer.php | 11 ++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/app/Transformers/PaymentTransformer.php b/app/Transformers/PaymentTransformer.php
index 1347a760cf3a..c07d98d62672 100644
--- a/app/Transformers/PaymentTransformer.php
+++ b/app/Transformers/PaymentTransformer.php
@@ -12,9 +12,11 @@
 namespace App\Transformers;
 
 use App\Models\Client;
+use App\Models\Document;
 use App\Models\Invoice;
 use App\Models\Payment;
 use App\Models\Paymentable;
+use App\Transformers\DocumentTransformer;
 use App\Utils\Traits\MakesHash;
 
 class PaymentTransformer extends EntityTransformer
@@ -29,6 +31,7 @@ class PaymentTransformer extends EntityTransformer
          'client',
          'invoices',
          'paymentables'
+         'documents'
     ];
 
     public function __construct($serializer = null)
@@ -59,6 +62,11 @@ class PaymentTransformer extends EntityTransformer
         return $this->includeCollection($payment->paymentables, $transformer, Paymentable::class);
     }
 
+    public function includeDocuments(Payment $payment)
+    {
+        $transformer = new DocumentTransformer($this->serializer);
+        return $this->includeCollection($payment->documents, $transformer, Document::class);
+    }
 
     public function transform(Payment $payment)
     {
diff --git a/app/Transformers/ProductTransformer.php b/app/Transformers/ProductTransformer.php
index 65f4ba966139..8c907a6cfbfd 100644
--- a/app/Transformers/ProductTransformer.php
+++ b/app/Transformers/ProductTransformer.php
@@ -12,8 +12,10 @@
 namespace App\Transformers;
 
 use App\Models\Company;
+use App\Models\Document;
 use App\Models\Product;
 use App\Models\User;
+use App\Transformers\DocumentTransformer;
 use App\Utils\Traits\MakesHash;
 
 class ProductTransformer extends EntityTransformer
@@ -28,7 +30,8 @@ class ProductTransformer extends EntityTransformer
      */
     protected $availableIncludes = [
         'company',
-        'user'
+        'user',
+        'documents',
     ];
 
 
@@ -56,6 +59,12 @@ class ProductTransformer extends EntityTransformer
         return $this->includeItem($product->company, $transformer, Company::class);
     }
 
+    public function includeDocuments(Product $product)
+    {
+        $transformer = new DocumentTransformer($this->serializer);
+        return $this->includeCollection($product->documents, $transformer, Document::class);
+    }
+
     public function transform(Product $product)
     {
         return [