From def4d1aa9f676b0a0fe6555162e28b57ada1495f Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Tue, 15 Dec 2020 08:52:29 +1100
Subject: [PATCH] Add account variables into migration

---
 .../Controllers/Migration/StepsController.php |  1 +
 app/Traits/GenerateMigrationResources.php     | 20 +++++++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/app/Http/Controllers/Migration/StepsController.php b/app/Http/Controllers/Migration/StepsController.php
index 52007b8e58..5004f212f2 100644
--- a/app/Http/Controllers/Migration/StepsController.php
+++ b/app/Http/Controllers/Migration/StepsController.php
@@ -225,6 +225,7 @@ class StepsController extends BaseController
             $fileName = "{$accountKey}-{$date}-invoiceninja";
 
             $localMigrationData['data'] = [
+                'account' => $this->getAccount(),
                 'company' => $this->getCompany(),
                 'users' => $this->getUsers(),
                 'tax_rates' => $this->getTaxRates(),
diff --git a/app/Traits/GenerateMigrationResources.php b/app/Traits/GenerateMigrationResources.php
index 78cf9a69ec..dda114273a 100644
--- a/app/Traits/GenerateMigrationResources.php
+++ b/app/Traits/GenerateMigrationResources.php
@@ -28,6 +28,26 @@ trait GenerateMigrationResources
 {
     protected $account;
 
+    protected function getAccount()
+    {
+        return [
+            'plan' => $this->account->company->plan,
+            'plan_term' =>$this->account->company->plan_term,
+            'plan_started' =>$this->account->company->plan_started,
+            'plan_paid' =>$this->account->company->plan_paid,
+            'plan_expires' =>$this->account->company->plan_expires,
+            'trial_started' =>$this->account->company->trial_started,
+            'trial_plan' =>$this->account->company->trial_plan,
+            'plan_price' =>$this->account->company->plan_price,
+            'num_users' =>$this->account->company->num_users,
+            'utm_source' =>$this->account->company->utm_source,
+            'utm_medium' =>$this->account->company->utm_medium,
+            'utm_campaign' =>$this->account->company->utm_campaign,
+            'utm_term' =>$this->account->company->utm_term,
+            'utm_content' =>$this->account->company->utm_content,
+        ];
+    }
+
     protected function getCompany()
     {
         return [