firefly-iii/app/controllers/ChartController.php

<?php

use Firefly\Exception\FireflyException;
use Firefly\Helper\Preferences\PreferencesHelperInterface as PHI;
use Firefly\Helper\Toolkit\ToolkitInterface as tk;
use Firefly\Storage\Account\AccountRepositoryInterface as ARI;
use Firefly\Storage\Budget\BudgetRepositoryInterface as BRI;
use Firefly\Storage\TransactionJournal\TransactionJournalRepositoryInterface as TJRI;

/**
 * Class ChartController
 */
class ChartController extends BaseController
{

    protected $_accounts;
    protected $_journals;
    protected $_tk;
    protected $_preferences;
    protected $_budgets;


    /**
     * @param ARI  $accounts
     * @param TJRI $journals
     * @param PHI  $preferences
     * @param tk   $toolkit
     * @param BRI  $budgets
     */
    public function __construct(ARI $accounts, TJRI $journals, PHI $preferences, tk $toolkit, BRI $budgets)
    {
        $this->_accounts = $accounts;
        $this->_journals = $journals;
        $this->_preferences = $preferences;
        $this->_tk = $toolkit;
        $this->_budgets = $budgets;
    }

    /**
     * @param null $accountId
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function homeAccount($accountId = null)
    {
        list($start, $end) = $this->_tk->getDateRangeDates();
        $current = clone $start;
        $return = [];

        $account = !is_null($accountId) ? $this->_accounts->find($accountId) : null;
        $today = new Carbon\Carbon;

        if (is_null($account)) {

            $pref = $this->_preferences->get('frontpageAccounts', []);
            if ($pref->data == []) {
                $accounts = $this->_accounts->getActiveDefault();
            } else {
                $accounts = $this->_accounts->getByIds($pref->data);
            }
            foreach ($accounts as $account) {
                $return[] = ['name' => $account->name, 'id' => 'acc-' . $account->id, 'data' => []];

            }
            while ($current <= $end) {
                // loop accounts:
                foreach ($accounts as $index => $account) {
                    if ($current > $today) {
                        $return[$index]['data'][] = [$current->timestamp * 1000, $account->predict(clone $current)];
                    } else {
                        $return[$index]['data'][] = [$current->timestamp * 1000, $account->balance(clone $current)];
                    }
                }
                $current->addDay();
            }
        } else {
            $return[0] = ['name' => $account->name, 'id' => $account->id, 'data' => []];
            while ($current <= $end) {
                if ($current > $today) {
                    $return[0]['data'][] = [$current->timestamp * 1000, $account->predict(clone $current)];
                } else {
                    $return[0]['data'][] = [$current->timestamp * 1000, $account->balance(clone $current)];
                }

                $current->addDay();
            }
        }
        return Response::json($return);
    }

    /**
     * Return some beneficiary info for an account and a date.
     *
     * @param $name
     * @param $day
     * @param $month
     * @param $year
     *
     * @return $this|\Illuminate\View\View
     */
    public function homeAccountInfo($name, $day, $month, $year)
    {
        $account = $this->_accounts->findByName($name);
        $result = [];
        $sum = 0;
        if ($account) {
            $date = \Carbon\Carbon::createFromDate($year, $month, $day);
            $journals = $this->_journals->getByAccountAndDate($account, $date);

            // loop all journals:
            foreach ($journals as $journal) {
                foreach ($journal->transactions as $transaction) {
                    $name = $transaction->account->name;
                    if ($transaction->account->id != $account->id) {
                        $result[$name] = isset($result[$name]) ? $result[$name] + floatval($transaction->amount)
                            : floatval($transaction->amount);
                        $sum += floatval($transaction->amount);
                    }
                }
            }
        }
        return View::make('charts.info')->with('rows', $result)->with('sum', $sum);
    }

    /**
     * @return \Illuminate\Http\JsonResponse
     * @throws Firefly\Exception\FireflyException
     */
    public function homeCategories()
    {
        list($start, $end) = $this->_tk->getDateRangeDates();
        $result = [];
        // grab all transaction journals in this period:
        $journals = $this->_journals->getByDateRange($start, $end);

        foreach ($journals as $journal) {
            // has to be one:

            if (!isset($journal->transactions[0])) {
                throw new FireflyException('Journal #' . $journal->id . ' has ' . count($journal->transactions)
                    . ' transactions!');
            }
            $transaction = $journal->transactions[0];
            $amount = floatval($transaction->amount);

            // get budget from journal:
            $budget = $journal->categories()->first();
            $budgetName = is_null($budget) ? '(no category)' : $budget->name;

            $result[$budgetName] = isset($result[$budgetName]) ? $result[$budgetName] + floatval($amount) : $amount;

        }
        unset($journal, $transaction, $budget, $amount);

        // sort
        arsort($result);
        $chartData = [
        ];
        foreach ($result as $name => $value) {
            $chartData[] = [$name, $value];
        }


        return Response::json($chartData);

    }

    /**
     * @return \Illuminate\Http\JsonResponse
     * @throws Firefly\Exception\FireflyException
     */
    public function homeBudgets()
    {
        // grab all budgets in the time period, like the index does:
        // get the budgets for this period:
        $data = [];

        list($start) = $this->_tk->getDateRangeDates();
        $budgets = $this->_budgets->getWithRepetitionsInPeriod($start, \Session::get('range'));

        $repeatFreq = Config::get('firefly.range_to_repeat_freq.' . Session::get('range'));

        $dateFormats = Config::get('firefly.date_formats_by_period.' . $repeatFreq);
        if (is_null($dateFormats)) {
            throw new FireflyException('No date formats for ' . \Session::get('range'));
        }


        $limitInPeriod = 'Envelope for ' . $start->format($dateFormats['display_date']);
        $spentInPeriod = 'Spent in ' . $start->format($dateFormats['display_date']);

        $data['series'] = [
            [
                'name' => $limitInPeriod,
                'data' => []
            ],
            [
                'name' => $spentInPeriod,
                'data' => []
            ],
        ];


        foreach ($budgets as $budget) {
            if ($budget->count > 0) {
                $data['labels'][] = wordwrap($budget->name, 12, "<br>");
            }
            foreach ($budget->limits as $limit) {
                foreach ($limit->limitrepetitions as $rep) {
                    //0: envelope for period:
                    $data['series'][0]['data'][] = floatval($rep->amount);
                    $data['series'][1]['data'][] = $rep->spent;
                }
            }


        }
        return Response::json($data);
    }
}