firefly-iii/app/controllers/TransactionController.php

<?php


use FireflyIII\Exception\FireflyException;
use Illuminate\Support\Collection;
use Illuminate\Support\MessageBag;

/**
 * Class TransactionController
 *
 */
class TransactionController extends BaseController
{


    /**
     * Construct a new transaction controller with two of the most often used helpers.
     *
     */
    public function __construct()
    {
        View::share('title', 'Transactions');
        View::share('mainTitleIcon', 'fa-repeat');
    }

    /**
     *
     * TODO this needs cleaning up and thinking over.
     *
     * @param TransactionJournal $journal
     *
     * @return array|\Illuminate\Http\JsonResponse
     */
    public function alreadyRelated(TransactionJournal $journal)
    {

        $ids = [];
        /** @var TransactionGroup $group */
        foreach ($journal->transactiongroups()->get() as $group) {
            /** @var TransactionJournal $jrnl */
            foreach ($group->transactionjournals()->get() as $jrnl) {
                if ($jrnl->id != $journal->id) {
                    $ids[] = $jrnl->id;
                }
            }
        }
        $unique = array_unique($ids);
        if (count($unique) > 0) {

            /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
            $repository = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');
            $set        = $repository->getByIds($unique);
            $set->each(
                function (TransactionJournal $journal) {
                    $journal->amount = mf($journal->getAmount());
                }
            );

            return Response::json($set->toArray());
        } else {
            return (new Collection)->toArray();
        }
    }

    /**
     * Shows the view helping the user to create a new transaction journal.
     *
     * @param string $what
     *
     * @return \Illuminate\View\View
     */
    public function create($what = 'deposit')
    {
        /** @var \FireflyIII\Database\Account\Account $accountRepository */
        $accountRepository = App::make('FireflyIII\Database\Account\Account');

        /** @var \FireflyIII\Database\Budget\Budget $budgetRepository */
        $budgetRepository = App::make('FireflyIII\Database\Budget\Budget');

        /** @var \FireflyIII\Database\PiggyBank\PiggyBank $piggyRepository */
        $piggyRepository = App::make('FireflyIII\Database\PiggyBank\PiggyBank');

        /** @var \FireflyIII\Database\PiggyBank\RepeatedExpense $repRepository */
        $repRepository = App::make('FireflyIII\Database\PiggyBank\RepeatedExpense');

        // get asset accounts with names and id's .
        $assetAccounts = FFForm::makeSelectList($accountRepository->getAssetAccounts());

        // get budgets as a select list.
        $budgets    = FFForm::makeSelectList($budgetRepository->get());
        $budgets[0] = '(no budget)';

        // get the piggy banks.
        $list       = $piggyRepository->get()->merge($repRepository->get());
        $piggies    = FFForm::makeSelectList($list);
        $piggies[0] = '(no piggy bank)';
        asort($piggies);

        $preFilled = Session::has('preFilled') ? Session::get('preFilled') : [];
        $respondTo = ['account_id', 'account_from_id'];
        foreach ($respondTo as $r) {
            if (!is_null(Input::get($r))) {
                $preFilled[$r] = Input::get($r);
            }
        }
        Session::put('preFilled', $preFilled);

        return View::make('transactions.create')->with('accounts', $assetAccounts)->with('budgets', $budgets)->with('what', $what)->with('piggies', $piggies)
                   ->with('subTitle', 'Add a new ' . $what);
    }

    /**
     * Shows the form that allows a user to delete a transaction journal.
     *
     * @param TransactionJournal $transactionJournal
     *
     * @return $this
     */
    public function delete(TransactionJournal $transactionJournal)
    {
        $type = strtolower($transactionJournal->transactionType->type);

        return View::make('transactions.delete')->with('journal', $transactionJournal)->with(
            'subTitle', 'Delete ' . $type . ' "' . $transactionJournal->description . '"'
        );


    }

    /**
     * @param TransactionJournal $transactionJournal
     *
     * @return \Illuminate\Http\RedirectResponse
     */
    public function destroy(TransactionJournal $transactionJournal)
    {
        $type = $transactionJournal->transactionType->type;

        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
        $repository = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');
        $repository->destroy($transactionJournal);

        $return = 'withdrawal';


        switch ($type) {
            case 'Deposit':
                $return = 'deposit';
                break;
            case 'Transfer':
                $return = 'transfers';
                break;
        }

        return Redirect::route('transactions.index', $return);
    }

    /**
     * TODO this needs cleaning up and thinking over.
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function doRelate()
    {
        $id     = intval(Input::get('id'));
        $sister = intval(Input::get('relateTo'));

        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
        $repository = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');

        $journal = $repository->find($id);
        $sis     = $repository->find($sister);

        if ($journal && $sis) {
            $group           = new TransactionGroup;
            $group->relation = 'balance';
            $group->user_id  = $repository->getUser()->id;
            $group->save();
            $group->transactionjournals()->save($journal);
            $group->transactionjournals()->save($sis);

            return Response::json(true);
        }

        return Response::json(false);


    }

    /**
     * Shows the view to edit a transaction.
     *
     * @param TransactionJournal $journal
     *
     * @return $this
     */
    public function edit(TransactionJournal $journal)
    {
        /*
         * All the repositories we need:
         */

        /** @var \FireflyIII\Database\Account\Account $accountRepository */
        $accountRepository = App::make('FireflyIII\Database\Account\Account');

        /** @var \FireflyIII\Database\Budget\Budget $budgetRepository */
        $budgetRepository = App::make('FireflyIII\Database\Budget\Budget');

        /** @var \FireflyIII\Database\PiggyBank\PiggyBank $piggyRepository */
        $piggyRepository = App::make('FireflyIII\Database\PiggyBank\PiggyBank');


        // type is useful for display:
        $what = strtolower($journal->transactiontype->type);

        // get asset accounts with names and id's.

        $budgets  = FFForm::makeSelectList($budgetRepository->get(), true);
        $accounts = FFForm::makeSelectList($accountRepository->getAssetAccounts());
        $piggies  = FFForm::makeSelectList($piggyRepository->get(), true);


        /*
         * Data to properly display the edit form.
         */
        $preFilled = [
            'date'         => $journal->date->format('Y-m-d'),
            'category'     => '',
            'budget_id'    => 0,
            'piggybank_id' => 0
        ];

        /*
         * Fill in the category.
         */
        $category = $journal->categories()->first();
        if (!is_null($category)) {
            $preFilled['category'] = $category->name;
        }

        /*
         * Switch on the type of transaction edited by the user and fill in other
         * relevant fields:
         */
        switch ($what) {
            case 'withdrawal':
                if (floatval($journal->transactions[0]->amount) < 0) {
                    // transactions[0] is the asset account that paid for the withdrawal.
                    $preFilled['account_id']      = $journal->transactions[0]->account->id;
                    $preFilled['expense_account'] = $journal->transactions[1]->account->name;
                    $preFilled['amount']          = floatval($journal->transactions[1]->amount);
                } else {
                    // transactions[1] is the asset account that paid for the withdrawal.
                    $preFilled['account_id']      = $journal->transactions[1]->account->id;
                    $preFilled['expense_account'] = $journal->transactions[0]->account->name;
                    $preFilled['amount']          = floatval($journal->transactions[0]->amount);
                }


                $budget = $journal->budgets()->first();
                if (!is_null($budget)) {
                    $preFilled['budget_id'] = $budget->id;
                }
                break;
            case 'deposit':
                if (floatval($journal->transactions[0]->amount) < 0) {
                    // transactions[0] contains the account the money came from.
                    $preFilled['account_id']      = $journal->transactions[1]->account->id;
                    $preFilled['revenue_account'] = $journal->transactions[0]->account->name;
                    $preFilled['amount']          = floatval($journal->transactions[1]->amount);
                } else {
                    // transactions[1] contains the account the money came from.
                    $preFilled['account_id']      = $journal->transactions[0]->account->id;
                    $preFilled['revenue_account'] = $journal->transactions[1]->account->name;
                    $preFilled['amount']          = floatval($journal->transactions[0]->amount);

                }

                break;
            case 'transfer':
                if (floatval($journal->transactions[0]->amount) < 0) {
                    // zero = from account.
                    $preFilled['account_from_id'] = $journal->transactions[0]->account->id;
                    $preFilled['account_to_id']   = $journal->transactions[1]->account->id;
                    $preFilled['amount']          = floatval($journal->transactions[1]->amount);
                } else {
                    // one = from account
                    $preFilled['account_from_id'] = $journal->transactions[1]->account->id;
                    $preFilled['account_to_id']   = $journal->transactions[0]->account->id;
                    $preFilled['amount']          = floatval($journal->transactions[0]->amount);
                }
                if ($journal->piggybankevents()->count() > 0) {
                    $preFilled['piggybank_id'] = $journal->piggybankevents()->first()->piggybank_id;
                }
                break;
        }

        /*
         * Show the view.
         */

        return View::make('transactions.edit')->with('journal', $journal)->with('accounts', $accounts)->with(
            'what', $what
        )->with('budgets', $budgets)->with('data', $preFilled)->with('piggies', $piggies)->with(
            'subTitle', 'Edit ' . $what . ' "' . $journal->description . '"'
        );
    }

    /**
     * @param $what
     *
     * @return $this
     */
    public function index($what)
    {

        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
        $repository = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');

        switch ($what) {
            case 'expenses':
            case 'withdrawal':
                $subTitleIcon = 'fa-long-arrow-left';
                $subTitle     = 'Expenses';
                $journals     = $repository->getWithdrawalsPaginated(50);
                break;
            case 'revenue':
            case 'deposit':
                $subTitleIcon = 'fa-long-arrow-right';
                $subTitle     = 'Revenue, income and deposits';
                $journals     = $repository->getDepositsPaginated(50);
                break;
            case 'transfer':
            case 'transfers':
                $subTitleIcon = 'fa-arrows-h';
                $subTitle     = 'Transfers';
                $journals     = $repository->getTransfersPaginated(50);
                break;
        }

        return View::make('transactions.index', compact('subTitle', 'what', 'subTitleIcon', 'journals'));

    }

    /**
     * @param TransactionJournal $journal
     *
     * @return \Illuminate\View\View
     */
    public function relate(TransactionJournal $journal)
    {
        $groups  = $journal->transactiongroups()->get();
        $members = new Collection;
        /** @var TransactionGroup $group */
        foreach ($groups as $group) {
            /** @var TransactionJournal $jrnl */
            foreach ($group->transactionjournals()->get() as $jrnl) {
                if ($jrnl->id != $journal->id) {
                    $members->push($jrnl);
                }
            }
        }

        return View::make('transactions.relate', compact('journal', 'members'));
    }

    /**
     * TODO this needs cleaning up and thinking over.
     *
     * @param TransactionJournal $journal
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function relatedSearch(TransactionJournal $journal)
    {
        $search = e(trim(Input::get('searchValue')));

        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
        $repository = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');

        $result = $repository->searchRelated($search, $journal);
        $result->each(
            function (TransactionJournal $j) {
                $j->amount = mf($j->getAmount());
            }
        );

        return Response::json($result->toArray());
    }

    /**
     * @param TransactionJournal $journal
     *
     * @return $this
     */
    public function show(TransactionJournal $journal)
    {
        $journal->transactions->each(
            function (\Transaction $t) use ($journal) {
                $t->before = floatval(
                    $t->account->transactions()->leftJoin(
                        'transaction_journals', 'transaction_journals.id', '=', 'transactions.transaction_journal_id'
                    )->where('transaction_journals.date', '<=', $journal->date->format('Y-m-d'))->where(
                        'transaction_journals.created_at', '<=', $journal->created_at->format('Y-m-d H:i:s')
                    )->where('transaction_journals.id', '!=', $journal->id)->sum('transactions.amount')
                );
                $t->after  = $t->before + $t->amount;
            }
        );
        $members = new Collection;
        /** @var TransactionGroup $group */
        foreach ($journal->transactiongroups()->get() as $group) {
            /** @var TransactionJournal $jrnl */
            foreach ($group->transactionjournals()->get() as $jrnl) {
                if ($jrnl->id != $journal->id) {
                    $members->push($jrnl);
                }
            }
        }

        return View::make('transactions.show', compact('journal', 'members'))->with(
            'subTitle', $journal->transactionType->type . ' "' . $journal->description . '"'
        );
    }

    /**
     * @param $what
     *
     * @return $this|\Illuminate\Http\RedirectResponse
     * @throws FireflyException
     */
    public function store($what)
    {
        $data             = Input::except('_token');
        $data['what']     = $what;
        $data['currency'] = 'EUR';

        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
        $repository = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');

        switch ($data['post_submit_action']) {
            default:
                throw new FireflyException('Cannot handle post_submit_action "' . e($data['post_submit_action']) . '"');
                break;
            case 'create_another':
            case 'store':
                $messages = $repository->validate($data);
                /** @var MessageBag $messages ['errors'] */
                if ($messages['errors']->count() > 0) {
                    Session::flash('warnings', $messages['warnings']);
                    Session::flash('successes', $messages['successes']);
                    Session::flash('error', 'Could not save transaction: ' . $messages['errors']->first());

                    return Redirect::route('transactions.create', $what)->withInput()->withErrors($messages['errors']);
                }
                // store!
                $journal = $repository->store($data);
                Session::flash('success', 'New transaction stored!');

                /*
                 * Trigger a search for the related (if selected)
                 * piggy bank and store an event.
                 */
                Event::fire('transactionJournal.store', [$journal, Input::get('piggybank_id')]); // new and used.
                /*
                 * Also trigger on both transactions.
                 */
                /** @var Transaction $transaction */
                foreach ($journal->transactions as $transaction) {
                    Event::fire('transaction.store', [$transaction]);
                }

                if ($data['post_submit_action'] == 'create_another') {
                    return Redirect::route('transactions.create', $what)->withInput();
                } else {
                    return Redirect::route('transactions.index', $what);
                }
                break;
            case 'validate_only':
                $messageBags = $repository->validate($data);
                Session::flash('warnings', $messageBags['warnings']);
                Session::flash('successes', $messageBags['successes']);
                Session::flash('errors', $messageBags['errors']);

                return Redirect::route('transactions.create', $what)->withInput();
                break;
        }
    }

    /**
     * TODO this needs cleaning up and thinking over.
     *
     * @param TransactionJournal $journal
     *
     * @return \Illuminate\Http\JsonResponse
     * @throws Exception
     */
    public function unrelate(TransactionJournal $journal)
    {
        $groups    = $journal->transactiongroups()->get();
        $relatedTo = intval(Input::get('relation'));
        /** @var TransactionGroup $group */
        foreach ($groups as $group) {
            foreach ($group->transactionjournals()->get() as $jrnl) {
                if ($jrnl->id == $relatedTo) {
                    // remove from group:
                    $group->transactionjournals()->detach($relatedTo);
                }
            }
            if ($group->transactionjournals()->count() == 1) {
                $group->delete();
            }
        }

        return Response::json(true);

    }

    /**
     * @param TransactionJournal $journal
     *
     * @return $this
     * @throws FireflyException
     */
    public function update(TransactionJournal $journal)
    {
        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repos */
        $repos = App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');

        $data             = Input::except('_token');
        $data['currency'] = 'EUR';
        $data['what']     = strtolower($journal->transactionType->type);


        switch (Input::get('post_submit_action')) {
            case 'update':
            case 'return_to_edit':
                $messageBag = $repos->update($journal, $data);
                if ($messageBag->count() == 0) {
                    // has been saved, return to index:
                    Session::flash('success', 'Transaction updated!');
                    Event::fire('transactionJournal.update', [$journal]); // new and used.

                    /*
                     * Also trigger on both transactions.
                     */
                    /** @var Transaction $transaction */
                    foreach ($journal->transactions as $transaction) {
                        Event::fire('transaction.update', [$transaction]);
                    }

                    if (Input::get('post_submit_action') == 'return_to_edit') {
                        return Redirect::route('transactions.edit', $journal->id)->withInput();
                    } else {
                        return Redirect::route('transactions.index', $data['what']);
                    }
                } else {
                    Session::flash('error', 'Could not update transaction: ' . $journal->getErrors()->first());

                    return Redirect::route('transactions.edit', $journal->id)->withInput()->withErrors(
                        $journal->getErrors()
                    );
                }

                break;
            case 'validate_only':
                $messageBags = $repos->validate($data);

                Session::flash('warnings', $messageBags['warnings']);
                Session::flash('successes', $messageBags['successes']);
                Session::flash('errors', $messageBags['errors']);

                return Redirect::route('transactions.edit', $journal->id)->withInput();
                break;
            default:
                throw new FireflyException('Method ' . Input::get('post_submit_action') . ' not implemented yet.');
                break;
        }


    }

}