From aed7e6d28917cfe96ef094f9dee9868b954509a7 Mon Sep 17 00:00:00 2001
From: James Cole <thegrumpydictator@gmail.com>
Date: Sun, 22 Jul 2018 08:27:18 +0200
Subject: [PATCH] Fix #1549

---
 app/Repositories/Budget/BudgetRepository.php | 42 +++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/app/Repositories/Budget/BudgetRepository.php b/app/Repositories/Budget/BudgetRepository.php
index 8d1277f6f6..d9464e73a2 100644
--- a/app/Repositories/Budget/BudgetRepository.php
+++ b/app/Repositories/Budget/BudgetRepository.php
@@ -30,6 +30,8 @@ use FireflyIII\Models\AccountType;
 use FireflyIII\Models\AvailableBudget;
 use FireflyIII\Models\Budget;
 use FireflyIII\Models\BudgetLimit;
+use FireflyIII\Models\RuleAction;
+use FireflyIII\Models\RuleTrigger;
 use FireflyIII\Models\Transaction;
 use FireflyIII\Models\TransactionCurrency;
 use FireflyIII\Models\TransactionJournal;
@@ -815,11 +817,49 @@ class BudgetRepository implements BudgetRepositoryInterface
      */
     public function update(Budget $budget, array $data): Budget
     {
-        // update the account:
+        $oldName = $budget->name;
+        // update the budget:
         $budget->name   = $data['name'];
         $budget->active = $data['active'];
         $budget->save();
 
+        // find any rule triggers related to budgets, with this budget name, and update them accordingly.
+        $types    = [
+            'budget_is',
+        ];
+        $triggers = RuleTrigger::leftJoin('rules', 'rules.id', '=', 'rule_triggers.rule_id')
+                               ->where('rules.user_id', $this->user->id)
+                               ->whereIn('rule_triggers.trigger_type', $types)
+                               ->where('rule_triggers.trigger_value', $oldName)
+                               ->get(['rule_triggers.*']);
+        Log::debug(sprintf('Found %d triggers to update.', $triggers->count()));
+        /** @var RuleTrigger $trigger */
+        foreach ($triggers as $trigger) {
+            $trigger->trigger_value = $data['name'];
+            $trigger->save();
+            Log::debug(sprintf('Updated trigger %d: %s', $trigger->id, $trigger->trigger_value));
+        }
+
+        
+        // find any rule actions related to budgets, with this budget name, and update them accordingly.
+        $types    = [
+            'set_budget',
+        ];
+        $actions = RuleAction::leftJoin('rules', 'rules.id', '=', 'rule_actions.rule_id')
+                               ->where('rules.user_id', $this->user->id)
+                               ->whereIn('rule_actions.action_type', $types)
+                               ->where('rule_actions.action_value', $oldName)
+                               ->get(['rule_actions.*']);
+        Log::debug(sprintf('Found %d actions to update.', $actions->count()));
+        /** @var RuleAction $action */
+        foreach($actions as $action) {
+            $action->action_value = $data['name'];
+            $action->save();
+            Log::debug(sprintf('Updated action %d: %s', $action->id, $action->action_value));
+        }
+
+        app('preferences')->mark();
+
         return $budget;
     }