From 4b4e023cb345482fae765fef077cf5a9218a4d6d Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Mon, 21 Nov 2022 21:49:33 +1100
Subject: [PATCH] Add rules to company transformer

---
 app/Models/Company.php                  |  5 +++++
 app/Transformers/CompanyTransformer.php | 10 ++++++++++
 2 files changed, 15 insertions(+)

diff --git a/app/Models/Company.php b/app/Models/Company.php
index 641add772796..bb705a82841d 100644
--- a/app/Models/Company.php
+++ b/app/Models/Company.php
@@ -189,6 +189,11 @@ class Company extends BaseModel
         return $this->hasMany(BankTransaction::class);
     }
 
+    public function bank_transaction_rules()
+    {
+        return $this->hasMany(BankTransactionRule::class);
+    }
+
     public function getCompanyIdAttribute()
     {
         return $this->encodePrimaryKey($this->id);
diff --git a/app/Transformers/CompanyTransformer.php b/app/Transformers/CompanyTransformer.php
index 45ef466f87d2..a38aaed32194 100644
--- a/app/Transformers/CompanyTransformer.php
+++ b/app/Transformers/CompanyTransformer.php
@@ -43,6 +43,7 @@ use App\Models\TaxRate;
 use App\Models\User;
 use App\Models\Webhook;
 use App\Transformers\BankIntegrationTransformer;
+use App\Transformers\BankTransactionRuleTransformer;
 use App\Transformers\BankTransactionTransformer;
 use App\Transformers\PurchaseOrderTransformer;
 use App\Transformers\RecurringExpenseTransformer;
@@ -104,6 +105,7 @@ class CompanyTransformer extends EntityTransformer
         'purchase_orders',
         'bank_integrations',
         'bank_transactions',
+        'bank_transaction_rules',
     ];
 
     /**
@@ -231,6 +233,14 @@ class CompanyTransformer extends EntityTransformer
         return $this->includeCollection($company->bank_transactions, $transformer, BankTransaction::class);
     }
 
+
+    public function includeBankTransactionRules(Company $company)
+    {
+        $transformer = new BankTransactionRuleTransformer($this->serializer);
+
+        return $this->includeCollection($company->bank_transaction_rules, $transformer, BankTransactionRule::class);
+    }
+
     public function includeBankIntegrations(Company $company)
     {
         $transformer = new BankIntegrationTransformer($this->serializer);