firefly-iii/app/lib/Firefly/Storage/Budget/EloquentBudgetRepository.php

<?php

namespace Firefly\Storage\Budget;

use Carbon\Carbon;

/**
 * Class EloquentBudgetRepository
 *
 * @package Firefly\Storage\Budget
 */
class EloquentBudgetRepository implements BudgetRepositoryInterface
{

    /**
     * @param $budgetId
     *
     * @return mixed
     */
    public function find($budgetId)
    {

        return \Auth::user()->budgets()->find($budgetId);
    }

    /**
     * @return mixed
     */
    public function get()
    {
        $set = \Auth::user()->budgets()->with(
            ['limits'                        => function ($q) {
                    $q->orderBy('limits.startdate', 'ASC');
                }, 'limits.limitrepetitions' => function ($q) {
                    $q->orderBy('limit_repetitions.startdate', 'ASC');
                }]
        )->orderBy('name', 'ASC')->get();
        foreach ($set as $budget) {
            foreach ($budget->limits as $limit) {
                foreach ($limit->limitrepetitions as $rep) {
                    $rep->left = $rep->left();
                }
            }
        }

        return $set;
    }

    /**
     * @return array|mixed
     */
    public function getAsSelectList()
    {
        $list = \Auth::user()->budgets()->with(
            ['limits', 'limits.limitrepetitions']
        )->orderBy('name', 'ASC')->get();
        $return = [];
        foreach ($list as $entry) {
            $return[intval($entry->id)] = $entry->name;
        }

        return $return;
    }

    /**
     * @param Carbon $date
     * @param        $range
     *
     * @return mixed
     */
    public function getWithRepetitionsInPeriod(Carbon $date, $range)
    {

        $set = \Auth::user()->budgets()->with(
            ['limits'                        => function ($q) use ($date) {
                    $q->orderBy('limits.startdate', 'ASC');
                }, 'limits.limitrepetitions' => function ($q) use ($date) {
                    $q->orderBy('limit_repetitions.startdate', 'ASC');
                    $q->where('startdate', $date->format('Y-m-d'));
                }]
        )->orderBy('name', 'ASC')->get();

        foreach ($set as $budget) {
            $budget->count = 0;
            foreach ($budget->limits as $limit) {
                /** @var $rep \LimitRepetition */
                foreach ($limit->limitrepetitions as $rep) {
                    $rep->left = $rep->left();
                    // overspent:
                    if ($rep->left < 0) {
                        $rep->spent = ($rep->left * -1) + $rep->amount;
                        $rep->overspent = $rep->left * -1;
                        $total = $rep->spent + $rep->overspent;
                        $rep->spent_pct = round(($rep->spent / $total) * 100);
                        $rep->overspent_pct = 100 - $rep->spent_pct;
                    } else {
                        $rep->spent = $rep->amount - $rep->left;
                        $rep->spent_pct = round(($rep->spent / $rep->amount) * 100);
                        $rep->left_pct = 100 - $rep->spent_pct;


                    }
                }
                $budget->count += count($limit->limitrepetitions);
            }
        }

        return $set;
    }

    /**
     * @param $data
     *
     * @return \Budget|mixed
     */
    public function store($data)
    {
        $budget = new \Budget;
        $budget->name = $data['name'];
        $budget->user()->associate(\Auth::user());
        $budget->save();

        // if limit, create limit (repetition itself will be picked up elsewhere).
        if ($data['amount'] > 0) {
            $limit = new \Limit;
            $limit->budget()->associate($budget);
            $startDate = new Carbon;
            switch ($data['repeat_freq']) {
                case 'daily':
                    $startDate->startOfDay();
                    break;
                case 'weekly':
                    $startDate->startOfWeek();
                    break;
                case 'monthly':
                    $startDate->startOfMonth();
                    break;
                case 'quarterly':
                    $startDate->firstOfQuarter();
                    break;
                case 'half-year':
                    $startDate->startOfYear();
                    if (intval($startDate->format('m')) >= 7) {
                        $startDate->addMonths(6);
                    }
                    break;
                case 'yearly':
                    $startDate->startOfYear();
                    break;
            }
            $limit->startdate = $startDate;
            $limit->amount = $data['amount'];
            $limit->repeats = $data['repeats'];
            $limit->repeat_freq = $data['repeat_freq'];
            $limit->save();
        }


        return $budget;
    }

}