firefly-iii/app/controllers/PiggybankController.php

<?php

use Firefly\Exception\FireflyException;
use Firefly\Storage\Account\AccountRepositoryInterface as ARI;
use Firefly\Storage\Piggybank\PiggybankRepositoryInterface as PRI;

/**
 * Class PiggybankController
 *
 */
class PiggybankController extends BaseController
{

    protected $_repository;
    protected $_accounts;

    /**
     * @param PRI $repository
     * @param ARI $accounts
     */
    public function __construct(PRI $repository, ARI $accounts)
    {
        $this->_repository = $repository;
        $this->_accounts = $accounts;
    }

    /**
     * @param Piggybank $piggyBank
     */
    public function addMoney(Piggybank $piggyBank)
    {
        $what = 'add';
        $maxAdd = $this->_repository->leftOnAccount($piggyBank->account);
        $maxRemove = null;

        return View::make('piggybanks.modifyAmount')->with('what', $what)->with('maxAdd', $maxAdd)->with(
            'maxRemove', $maxRemove
        )->with('piggybank', $piggyBank);
    }

    /**
     * @return $this
     */
    public function createPiggybank()
    {
        $periods = Config::get('firefly.piggybank_periods');
        $accounts = $this->_accounts->getActiveDefaultAsSelectList();

        return View::make('piggybanks.create-piggybank')->with('accounts', $accounts)->with('periods', $periods);
    }

    /**
     * @return $this
     */
    public function createRepeated()
    {
        $periods = Config::get('firefly.piggybank_periods');
        $accounts = $this->_accounts->getActiveDefaultAsSelectList();

        return View::make('piggybanks.create-repeated')->with('accounts', $accounts)->with('periods', $periods);
    }

    /**
     * @param Piggybank $piggyBank
     *
     * @return $this
     */
    public function delete(Piggybank $piggyBank)
    {
        return View::make('piggybanks.delete')->with('piggybank', $piggyBank);
    }

    /**
     * @param Piggybank $piggyBank
     *
     * @return \Illuminate\Http\RedirectResponse
     */
    public function destroy(Piggybank $piggyBank)
    {
        $this->_repository->destroy($piggyBank);
        Event::fire('piggybanks.change');
        Session::flash('success', 'Piggy bank deleted.');

        return Redirect::route('piggybanks.index');
    }

    /**
     * @param Piggybank $piggyBank
     *
     * @return $this
     */
    public function edit(Piggybank $piggyBank)
    {
        $accounts = $this->_accounts->getActiveDefaultAsSelectList();
        $periods = Config::get('firefly.piggybank_periods');
        if ($piggyBank->repeats == 1) {
            return View::make('piggybanks.edit-repeated')->with('piggybank', $piggyBank)->with('accounts', $accounts)
                ->with('periods', $periods);
        } else {
            return View::make('piggybanks.edit-piggybank')->with('piggybank', $piggyBank)->with('accounts', $accounts)
                ->with('periods', $periods);
        }


    }

    /**
     * @return $this
     */
    public function index()
    {
        Event::fire('piggybanks.change');
        $countRepeating = $this->_repository->countRepeating();
        $countNonRepeating = $this->_repository->countNonrepeating();

        $piggybanks = $this->_repository->get();

        // get the accounts with each piggy bank and check their balance; we might need to
        // show the user a correction.

        $accounts = [];
        /** @var \Piggybank $piggybank */
        foreach($piggybanks as $piggybank) {
            $account = $piggybank->account;
            $id = $account->id;
            if(!isset($accounts[$id])) {
                $accounts[$id] = ['account' => $account, 'left' => $this->_repository->leftOnAccount($account)];
            }
        }

        return View::make('piggybanks.index')->with('piggybanks', $piggybanks)
            ->with('countRepeating', $countRepeating)
            ->with('countNonRepeating', $countNonRepeating)
            ->with('accounts',$accounts);
    }

    /**
     * @param Piggybank $piggyBank
     *
     * @return \Illuminate\Http\RedirectResponse
     * @throws Firefly\Exception\FireflyException
     */
    public function modMoney(Piggybank $piggyBank)
    {
        $amount = floatval(Input::get('amount'));
        switch (Input::get('what')) {
            default:
                throw new FireflyException('No such action');
                break;
            case 'add':
                $maxAdd = $this->_repository->leftOnAccount($piggyBank->account);
                if (round($amount, 2) <= round(min($maxAdd, $piggyBank->targetamount), 2)) {
                    Session::flash('success', 'Amount updated!');
                    $this->_repository->modifyAmount($piggyBank, $amount);
                } else {
                    Session::flash('warning', 'Could not!');
                }
                break;
            case 'remove':
                $rep = $piggyBank->currentRelevantRep();
                $maxRemove = $rep->currentamount;
                if (round($amount, 2) <= round($maxRemove, 2)) {
                    Session::flash('success', 'Amount updated!');
                    $this->_repository->modifyAmount($piggyBank, ($amount * -1));
                } else {
                    Session::flash('warning', 'Could not!');
                }
                break;
        }

        return Redirect::route('piggybanks.index');
    }

    /**
     * @param Piggybank $piggyBank
     */
    public function removeMoney(Piggybank $piggyBank)
    {
        $what = 'remove';
        $maxAdd = $this->_repository->leftOnAccount($piggyBank->account);
        $maxRemove = $piggyBank->currentRelevantRep()->currentamount;

        return View::make('piggybanks.modifyAmount')->with('what', $what)->with('maxAdd', $maxAdd)->with(
            'maxRemove', $maxRemove
        )->with('piggybank', $piggyBank);
    }

    /**
     *
     */
    public function show(Piggybank $piggyBank)
    {
        return View::make('piggybanks.show')->with('piggyBank', $piggyBank);
    }

    /**
     * @return $this|\Illuminate\Http\RedirectResponse
     */
    public function storePiggybank()
    {
        $data = Input::all();
        unset($data['_token']);

        // extend the data array with the settings needed to create a piggy bank:
        $data['repeats'] = 0;
        $data['rep_times'] = 1;
        $data['rep_every'] = 1;
        $data['order'] = 0;

        $piggyBank = $this->_repository->store($data);
        if (!is_null($piggyBank->id)) {
            Session::flash('success', 'New piggy bank "' . $piggyBank->name . '" created!');
            Event::fire('piggybanks.change');

            return Redirect::route('piggybanks.index');


        } else {
            Session::flash('error', 'Could not save piggy bank: ' . $piggyBank->errors()->first());

            return Redirect::route('piggybanks.create.piggybank')->withInput()->withErrors($piggyBank->errors());
        }

    }

    /**
     * @return $this|\Illuminate\Http\RedirectResponse
     */
    public function storeRepeated()
    {

        $data = Input::all();
        unset($data['_token']);

        // extend the data array with the settings needed to create a repeated:
        $data['repeats'] = 1;
        $data['order'] = 0;

        $piggyBank = $this->_repository->store($data);
        if ($piggyBank->id) {
            Session::flash('success', 'New piggy bank "' . $piggyBank->name . '" created!');
            Event::fire('piggybanks.change');

            return Redirect::route('piggybanks.index');

        } else {
            Session::flash('error', 'Could not save piggy bank: ' . $piggyBank->errors()->first());

            return Redirect::route('piggybanks.create.repeated')->withInput()->withErrors($piggyBank->errors());
        }

    }

    /**
     * @return $this|\Illuminate\Http\RedirectResponse
     */
    public function update(Piggybank $piggyBank)
    {
        $piggyBank = $this->_repository->update($piggyBank, Input::all());
        if ($piggyBank->validate()) {
            Session::flash('success', 'Piggy bank "' . $piggyBank->name . '" updated.');
            Event::fire('piggybanks.change');

            return Redirect::route('piggybanks.index');
        } else {
            Session::flash('error', 'Could not update piggy bank: ' . $piggyBank->errors()->first());

            return Redirect::route('piggybanks.edit', $piggyBank->id)->withErrors($piggyBank->errors())->withInput();
        }


    }
}