firefly-iii/app/Http/Controllers/Chart/ReportController.php

<?php
declare(strict_types = 1);

namespace FireflyIII\Http\Controllers\Chart;


use Carbon\Carbon;
use FireflyIII\Generator\Chart\Report\ReportChartGeneratorInterface;
use FireflyIII\Http\Controllers\Controller;
use FireflyIII\Repositories\Account\AccountRepositoryInterface;
use FireflyIII\Support\CacheProperties;
use Illuminate\Support\Collection;
use Navigation;
use Response;
use Steam;

/**
 * Class ReportController
 *
 * @package FireflyIII\Http\Controllers\Chart
 */
class ReportController extends Controller
{

    /** @var ReportChartGeneratorInterface */
    protected $generator;

    /**
     *
     */
    public function __construct()
    {
        parent::__construct();
        // create chart generator:
        $this->generator = app(ReportChartGeneratorInterface::class);
    }

    /**
     * This chart, by default, is shown on the multi-year and year report pages,
     * which means that giving it a 2 week "period" should be enough granularity.
     *
     * @param string     $reportType
     * @param Carbon     $start
     * @param Carbon     $end
     * @param Collection $accounts
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function netWorth(string $reportType, Carbon $start, Carbon $end, Collection $accounts)
    {
        // chart properties for cache:
        $cache = new CacheProperties;
        $cache->addProperty('netWorth');
        $cache->addProperty($start);
        $cache->addProperty($reportType);
        $cache->addProperty($accounts);
        $cache->addProperty($end);
        if ($cache->has()) {
            return Response::json($cache->get());
        }
        $ids     = $accounts->pluck('id')->toArray();
        $current = clone $start;
        $entries = new Collection;
        while ($current < $end) {
            $balances = Steam::balancesById($ids, $current);
            $sum      = $this->arraySum($balances);
            $entries->push(
                [
                    'date'      => clone $current,
                    'net-worth' => $sum,
                ]
            );

            $current->addDays(7);
        }
        $data = $this->generator->netWorth($entries);

        $cache->store($data);

        return Response::json($data);
    }


    /**
     * @param AccountRepositoryInterface $repository
     * @param string                     $reportType
     * @param Carbon                     $start
     * @param Carbon                     $end
     * @param Collection                 $accounts
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function yearInOut(AccountRepositoryInterface $repository, string $reportType, Carbon $start, Carbon $end, Collection $accounts)
    {
        // chart properties for cache:
        $cache = new CacheProperties;
        $cache->addProperty('yearInOut');
        $cache->addProperty($start);
        $cache->addProperty($reportType);
        $cache->addProperty($accounts);
        $cache->addProperty($end);
        if ($cache->has()) {
             return Response::json($cache->get());
        }

        // always per month.
        $currentStart = clone $start;
        $spentArray   = [];
        $earnedArray  = [];
        while ($currentStart <= $end) {
            $currentEnd         = Navigation::endOfPeriod($currentStart, '1M');
            $date               = $currentStart->format('Y-m');
            $spent              = $repository->spentInPeriod($accounts, $currentStart, $currentEnd);
            $earned             = $repository->earnedInPeriod($accounts, $currentStart, $currentEnd);
            $spentArray[$date]  = $spent;
            $earnedArray[$date] = $earned;
            $currentStart       = Navigation::addPeriod($currentStart, '1M', 0);
        }

        if ($start->diffInMonths($end) > 12) {
            // data = method X
            $data = $this->multiYearInOut($earnedArray, $spentArray, $start, $end);
        } else {
            // data = method Y
            $data = $this->singleYearInOut($earnedArray, $spentArray, $start, $end);
        }

        $cache->store($data);

        return Response::json($data);

    }

    /**
     * @param AccountRepositoryInterface $repository
     * @param string                     $reportType
     * @param Carbon                     $start
     * @param Carbon                     $end
     * @param Collection                 $accounts
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function yearInOutSummarized(AccountRepositoryInterface $repository, string $reportType, Carbon $start, Carbon $end, Collection $accounts)
    {

        // chart properties for cache:
        $cache = new CacheProperties;
        $cache->addProperty('yearInOutSummarized');
        $cache->addProperty($start);
        $cache->addProperty($end);
        $cache->addProperty($reportType);
        $cache->addProperty($accounts);
        if ($cache->has()) {
            return Response::json($cache->get());
        }

        // always per month.
        $currentStart = clone $start;
        $spentArray   = [];
        $earnedArray  = [];
        while ($currentStart <= $end) {
            $currentEnd         = Navigation::endOfPeriod($currentStart, '1M');
            $date               = $currentStart->format('Y-m');
            $spent              = $repository->spentInPeriod($accounts, $currentStart, $currentEnd);
            $earned             = $repository->earnedInPeriod($accounts, $currentStart, $currentEnd);
            $spentArray[$date]  = $spent;
            $earnedArray[$date] = $earned;
            $currentStart       = Navigation::addPeriod($currentStart, '1M', 0);
        }


        if ($start->diffInMonths($end) > 12) {
            // per year
            $data = $this->multiYearInOutSummarized($earnedArray, $spentArray, $start, $end);
        } else {
            // per month!
            $data = $this->singleYearInOutSummarized($earnedArray, $spentArray, $start, $end);
        }
        $cache->store($data);

        return Response::json($data);
    }

    /**
     * @param array  $earned
     * @param array  $spent
     * @param Carbon $start
     * @param Carbon $end
     *
     * @return array
     */
    protected function multiYearInOut(array $earned, array $spent, Carbon $start, Carbon $end)
    {
        $entries = new Collection;
        while ($start < $end) {

            $incomeSum  = $this->pluckFromArray($start->year, $earned);
            $expenseSum = $this->pluckFromArray($start->year, $spent) * -1;

            $entries->push([clone $start, $incomeSum, $expenseSum]);
            $start->addYear();
        }

        $data = $this->generator->multiYearInOut($entries);

        return $data;
    }

    /**
     * @param array  $earned
     * @param array  $spent
     * @param Carbon $start
     * @param Carbon $end
     *
     * @return array
     */
    protected function multiYearInOutSummarized(array $earned, array $spent, Carbon $start, Carbon $end)
    {
        $income  = '0';
        $expense = '0';
        $count   = 0;
        while ($start < $end) {

            $currentIncome  = $this->pluckFromArray($start->year, $earned);
            $currentExpense = bcmul($this->pluckFromArray($start->year, $spent), '-1');
            $income         = bcadd($income, $currentIncome);
            $expense        = bcadd($expense, $currentExpense);

            $count++;
            $start->addYear();
        }

        $data = $this->generator->multiYearInOutSummarized($income, $expense, $count);

        return $data;
    }

    /**
     * @param int   $year
     * @param array $set
     *
     * @return string
     */
    protected function pluckFromArray($year, array $set)
    {
        $sum = '0';
        foreach ($set as $date => $amount) {
            if (substr($date, 0, 4) == $year) {
                $sum = bcadd($sum, $amount);
            }
        }

        return $sum;

    }

    /**
     * @param array  $earned
     * @param array  $spent
     * @param Carbon $start
     * @param Carbon $end
     *
     * @return array
     */
    protected function singleYearInOut(array $earned, array $spent, Carbon $start, Carbon $end)
    {
        // per month? simply use each month.

        $entries = new Collection;
        while ($start < $end) {
            // total income and total expenses:
            $date       = $start->format('Y-m');
            $incomeSum  = isset($earned[$date]) ? $earned[$date] * -1 : 0;
            $expenseSum = isset($spent[$date]) ? $spent[$date] * -1 : 0;

            $entries->push([clone $start, $incomeSum, $expenseSum]);
            $start->addMonth();
        }

        $data = $this->generator->yearInOut($entries);

        return $data;
    }

    /**
     * @param array  $earned
     * @param array  $spent
     * @param Carbon $start
     * @param Carbon $end
     *
     * @return array
     */
    protected function singleYearInOutSummarized(array $earned, array $spent, Carbon $start, Carbon $end)
    {
        $income  = '0';
        $expense = '0';
        $count   = 0;
        while ($start < $end) {
            $date           = $start->format('Y-m');
            $currentIncome  = $earned[$date] ?? '0';
            $currentExpense = $spent[$date] ??'0';
            $income         = bcadd($income, $currentIncome);
            $expense        = bcadd($expense, $currentExpense);

            $count++;
            $start->addMonth();
        }

        $data = $this->generator->yearInOutSummarized($income, $expense, $count);

        return $data;
    }

    /**
     * @param $array
     *
     * @return string
     */
    private function arraySum($array) : string
    {
        $sum = '0';
        foreach ($array as $entry) {
            $sum = bcadd($sum, $entry);
        }

        return $sum;
    }
}