firefly-iii/app/Http/Controllers/Transaction/SplitController.php

<?php
/**
 * SplitController.php
 * Copyright (C) 2016 thegrumpydictator@gmail.com
 *
 * This software may be modified and distributed under the terms
 * of the MIT license.  See the LICENSE file for details.
 */

namespace FireflyIII\Http\Controllers\Transaction;


use ExpandedForm;
use FireflyIII\Crud\Split\JournalInterface;
use FireflyIII\Exceptions\FireflyException;
use FireflyIII\Http\Controllers\Controller;
use FireflyIII\Http\Requests\SplitJournalFormRequest;
use FireflyIII\Models\TransactionJournal;
use FireflyIII\Repositories\Account\AccountRepositoryInterface;
use FireflyIII\Repositories\Budget\BudgetRepositoryInterface;
use FireflyIII\Repositories\Currency\CurrencyRepositoryInterface;
use FireflyIII\Repositories\PiggyBank\PiggyBankRepositoryInterface;
use Illuminate\Http\Request;
use Log;
use Session;
use View;

/**
 * Class SplitController
 *
 * @package FireflyIII\Http\Controllers\Transaction
 */
class SplitController extends Controller
{
    /**
     *
     */
    public function __construct()
    {
        parent::__construct();
        View::share('mainTitleIcon', 'fa-share-alt');
        View::share('title', trans('firefly.split-transactions'));
    }

    public function edit(TransactionJournal $journal)
    {
        $count = $journal->transactions()->count();
        if ($count === 2) {
            return redirect(route('transactions.edit', [$journal->id]));
        }

        /** @var CurrencyRepositoryInterface $currencyRepository */
        $currencyRepository = app(CurrencyRepositoryInterface::class);
        /** @var AccountRepositoryInterface $accountRepository */
        $accountRepository = app(AccountRepositoryInterface::class);

        /** @var BudgetRepositoryInterface $budgetRepository */
        $budgetRepository = app(BudgetRepositoryInterface::class);

        /** @var PiggyBankRepositoryInterface $piggyBankRepository */
        $piggyBankRepository = app(PiggyBankRepositoryInterface::class);

        $what          = strtolower(TransactionJournal::transactionTypeStr($journal));
        $currencies    = ExpandedForm::makeSelectList($currencyRepository->get());
        $assetAccounts = ExpandedForm::makeSelectList($accountRepository->getAccounts(['Default account', 'Asset account']));
        $budgets       = ExpandedForm::makeSelectListWithEmpty($budgetRepository->getActiveBudgets());
        $piggyBanks    = ExpandedForm::makeSelectListWithEmpty($piggyBankRepository->getPiggyBanks());
        $amount        = TransactionJournal::amountPositive($journal);

        // get source account:
        $sourceAccounts      = TransactionJournal::sourceAccountList($journal);
        $destinationAccounts = TransactionJournal::destinationAccountList($journal);

        // get the transactions:

        return view(
            'split.journals.edit',
            compact('currencies', 'amount', 'piggyBanks', 'sourceAccounts', 'destinationAccounts', 'assetAccounts', 'budgets', 'what', 'journal')
        );
    }

    /**
     * @param Request $request
     *
     * @return mixed
     * @throws FireflyException
     */
    public function journalFromStore(Request $request)
    {
        if ($request->old('journal_currency_id')) {
            $preFilled = $this->arrayFromOldData($request->old());
        } else {
            $preFilled = $this->arrayFromSession();
        }

        Session::flash('preFilled', $preFilled);
        View::share('subTitle', trans('firefly.split-new-transaction'));

        /** @var CurrencyRepositoryInterface $currencyRepository */
        $currencyRepository = app(CurrencyRepositoryInterface::class);
        /** @var AccountRepositoryInterface $accountRepository */
        $accountRepository = app(AccountRepositoryInterface::class);

        /** @var BudgetRepositoryInterface $budgetRepository */
        $budgetRepository = app(BudgetRepositoryInterface::class);

        /** @var PiggyBankRepositoryInterface $piggyBankRepository */
        $piggyBankRepository = app(PiggyBankRepositoryInterface::class);


        $currencies    = ExpandedForm::makeSelectList($currencyRepository->get());
        $assetAccounts = ExpandedForm::makeSelectList($accountRepository->getAccounts(['Default account', 'Asset account']));
        $budgets       = ExpandedForm::makeSelectListWithEmpty($budgetRepository->getActiveBudgets());
        $piggyBanks    = ExpandedForm::makeSelectListWithEmpty($piggyBankRepository->getPiggyBanks());

        //Session::flash('warning', 'This feature is very experimental. Beware.');


        return view('split.journals.from-store', compact('currencies', 'piggyBanks', 'assetAccounts', 'budgets'))->with('data', $preFilled);


    }

    /**
     * @param SplitJournalFormRequest $request
     * @param JournalInterface        $repository
     *
     * @return \Illuminate\Http\RedirectResponse|\Illuminate\Routing\Redirector
     * @throws FireflyException
     */
    public function postJournalFromStore(SplitJournalFormRequest $request, JournalInterface $repository)
    {
        $data = $request->getSplitData();

        // store an empty journal first. This will be the place holder.
        $journal = $repository->storeJournal($data);
        // Then, store each transaction individually.

        if (is_null($journal->id)) {
            throw new FireflyException('Could not store transaction.');
        }
        foreach ($data['transactions'] as $transaction) {
            $transactions = $repository->storeTransaction($journal, $transaction);
        }

        // TODO move to repository.
        $journal->completed = true;
        $journal->save();

        // forget temp journal data
        Session::forget('temporary_split_data');

        // this is where we originally came from.
        return redirect(session('transactions.create.url'));
    }

    /**
     * @param array $old
     *
     * @return array
     */
    private function arrayFromOldData(array $old): array
    {
        // this array is pretty much equal to what we expect it to be.
        Log::debug('Prefilled', $old);

        return $old;
    }

    /**
     * @return array
     * @throws FireflyException
     */
    private function arrayFromSession(): array
    {
        // expect data to be in session or in post?
        $data = session('temporary_split_data');

        if (!is_array($data)) {
            Log::error('Could not find transaction data in your session. Please go back and try again.', ['data' => $data]); // translate me.
            throw new FireflyException('Could not find transaction data in your session. Please go back and try again.'); // translate me.
        }

        Log::debug('Journal data', $data);

        $preFilled = [
            'what'                             => $data['what'],
            'journal_description'              => $data['description'],
            'journal_source_account_id'        => $data['source_account_id'],
            'journal_source_account_name'      => $data['source_account_name'],
            'journal_destination_account_id'   => $data['destination_account_id'],
            'journal_destination_account_name' => $data['destination_account_name'],
            'journal_amount'                   => $data['amount'],
            'journal_currency_id'              => $data['amount_currency_id_amount'],
            'date'                             => $data['date'],
            'interest_date'                    => $data['interest_date'],
            'book_date'                        => $data['book_date'],
            'process_date'                     => $data['process_date'],

            'description'              => [],
            'destination_account_id'   => [],
            'destination_account_name' => [],
            'amount'                   => [],
            'budget_id'                => [],
            'category'                 => [],
            'piggy_bank_id'            => [],
        ];

        // create the first transaction:
        $preFilled['description'][]              = $data['description'];
        $preFilled['destination_account_id'][]   = $data['destination_account_id'];
        $preFilled['destination_account_name'][] = $data['destination_account_name'];
        $preFilled['amount'][]                   = $data['amount'];
        $preFilled['budget_id'][]                = $data['budget_id'];
        $preFilled['category'][]                 = $data['category'];
        $preFilled['piggy_bank_id'][]            = $data['piggy_bank_id'];

        //        echo '<pre>';
        //        var_dump($data);
        //        var_dump($preFilled);
        //        exit;
        Log::debug('Prefilled', $preFilled);

        return $preFilled;
    }

}