Fix reports and charts for #508

app/Helpers/Collection/BalanceLine.php

@@ -14,6 +14,7 @@ namespace FireflyIII\Helpers\Collection;
 
 use Carbon\Carbon;
 use FireflyIII\Models\Budget as BudgetModel;
+use FireflyIII\Models\BudgetLimit;
 use Illuminate\Support\Collection;
 
 /**
@@ -34,12 +35,27 @@ class BalanceLine
 
     /** @var BudgetModel */
     protected $budget;
-    /** @var  Carbon */
-    protected $endDate;
     /** @var int */
     protected $role = self::ROLE_DEFAULTROLE;
-    /** @var  Carbon */
-    protected $startDate;
+
+    /** @var  BudgetLimit */
+    protected $budgetLimit;
+
+    /**
+     * @return BudgetLimit
+     */
+    public function getBudgetLimit(): BudgetLimit
+    {
+        return $this->budgetLimit;
+    }
+
+    /**
+     * @param BudgetLimit $budgetLimit
+     */
+    public function setBudgetLimit(BudgetLimit $budgetLimit)
+    {
+        $this->budgetLimit = $budgetLimit;
+    }
 
     /**
      *
@@ -95,15 +111,7 @@ class BalanceLine
      */
     public function getEndDate()
     {
-        return $this->endDate;
-    }
-
-    /**
-     * @param Carbon $endDate
-     */
-    public function setEndDate($endDate)
-    {
-        $this->endDate = $endDate;
+        return $this->budgetLimit->end_date ?? new Carbon;
     }
 
     /**
@@ -127,15 +135,7 @@ class BalanceLine
      */
     public function getStartDate()
     {
-        return $this->startDate;
-    }
-
-    /**
-     * @param Carbon $startDate
-     */
-    public function setStartDate($startDate)
-    {
-        $this->startDate = $startDate;
+        return $this->budgetLimit->start_date ?? new Carbon;
     }
 
     /**
@@ -170,7 +170,7 @@ class BalanceLine
      */
     public function leftOfRepetition(): string
     {
-        $start = $this->budget->amount ?? '0';
+        $start = $this->budgetLimit->amount ?? '0';
         /** @var BalanceEntry $balanceEntry */
         foreach ($this->getBalanceEntries() as $balanceEntry) {
             $start = bcadd($balanceEntry->getSpent(), $start);

app/Helpers/Collection/BudgetLine.php

@@ -13,7 +13,7 @@ declare(strict_types = 1);
 namespace FireflyIII\Helpers\Collection;
 
 use FireflyIII\Models\Budget as BudgetModel;
-use FireflyIII\Models\LimitRepetition;
+use FireflyIII\Models\BudgetLimit;
 
 /**
  *
@@ -26,14 +26,14 @@ class BudgetLine
 
     /** @var  BudgetModel */
     protected $budget;
+    /** @var  BudgetLimit */
+    protected $budgetLimit;
     /** @var string */
     protected $budgeted = '0';
     /** @var string */
     protected $left = '0';
     /** @var string */
     protected $overspent = '0';
-    /** @var  LimitRepetition */
-    protected $repetition;
     /** @var string */
     protected $spent = '0';
 
@@ -57,6 +57,26 @@ class BudgetLine
         return $this;
     }
 
+    /**
+     * @return BudgetLimit
+     */
+    public function getBudgetLimit(): BudgetLimit
+    {
+        return $this->budgetLimit ?? new BudgetLimit;
+    }
+
+    /**
+     * @param BudgetLimit $budgetLimit
+     *
+     * @return BudgetLimit
+     */
+    public function setBudgetLimit(BudgetLimit $budgetLimit): BudgetLine
+    {
+        $this->budgetLimit = $budgetLimit;
+
+        return $this;
+    }
+
     /**
      * @return string
      */
@@ -117,26 +137,6 @@ class BudgetLine
         return $this;
     }
 
-    /**
-     * @return LimitRepetition
-     */
-    public function getRepetition(): LimitRepetition
-    {
-        return $this->repetition ?? new LimitRepetition;
-    }
-
-    /**
-     * @param LimitRepetition $repetition
-     *
-     * @return BudgetLine
-     */
-    public function setRepetition(LimitRepetition $repetition): BudgetLine
-    {
-        $this->repetition = $repetition;
-
-        return $this;
-    }
-
     /**
      * @return string
      */