mirror of
				https://github.com/firefly-iii/firefly-iii.git
				synced 2025-11-04 05:15:39 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			335 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			335 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * RecurringTransactionTrait.php
 | 
						|
 * Copyright (c) 2019 james@firefly-iii.org
 | 
						|
 *
 | 
						|
 * This file is part of Firefly III (https://github.com/firefly-iii).
 | 
						|
 *
 | 
						|
 * This program is free software: you can redistribute it and/or modify
 | 
						|
 * it under the terms of the GNU Affero General Public License as
 | 
						|
 * published by the Free Software Foundation, either version 3 of the
 | 
						|
 * License, or (at your option) any later version.
 | 
						|
 *
 | 
						|
 * This program is distributed in the hope that it will be useful,
 | 
						|
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
 * GNU Affero General Public License for more details.
 | 
						|
 *
 | 
						|
 * You should have received a copy of the GNU Affero General Public License
 | 
						|
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
declare(strict_types=1);
 | 
						|
 | 
						|
namespace FireflyIII\Services\Internal\Support;
 | 
						|
 | 
						|
use FireflyIII\Exceptions\FireflyException;
 | 
						|
use FireflyIII\Factory\AccountFactory;
 | 
						|
use FireflyIII\Factory\BillFactory;
 | 
						|
use FireflyIII\Factory\BudgetFactory;
 | 
						|
use FireflyIII\Factory\CategoryFactory;
 | 
						|
use FireflyIII\Factory\PiggyBankFactory;
 | 
						|
use FireflyIII\Factory\TransactionCurrencyFactory;
 | 
						|
use FireflyIII\Models\Account;
 | 
						|
use FireflyIII\Models\AccountType;
 | 
						|
use FireflyIII\Models\Note;
 | 
						|
use FireflyIII\Models\Recurrence;
 | 
						|
use FireflyIII\Models\RecurrenceMeta;
 | 
						|
use FireflyIII\Models\RecurrenceRepetition;
 | 
						|
use FireflyIII\Models\RecurrenceTransaction;
 | 
						|
use FireflyIII\Models\RecurrenceTransactionMeta;
 | 
						|
use FireflyIII\Repositories\Account\AccountRepositoryInterface;
 | 
						|
use FireflyIII\Validation\AccountValidator;
 | 
						|
use Illuminate\Support\Facades\Log;
 | 
						|
 | 
						|
/**
 | 
						|
 * Trait RecurringTransactionTrait
 | 
						|
 */
 | 
						|
trait RecurringTransactionTrait
 | 
						|
{
 | 
						|
    public function updateNote(Recurrence $recurrence, string $note): bool
 | 
						|
    {
 | 
						|
        if ('' === $note) {
 | 
						|
            $dbNote = $recurrence->notes()->first();
 | 
						|
            if (null !== $dbNote) {
 | 
						|
                $dbNote->delete();
 | 
						|
            }
 | 
						|
 | 
						|
            return true;
 | 
						|
        }
 | 
						|
        $dbNote       = $recurrence->notes()->first();
 | 
						|
        if (null === $dbNote) {
 | 
						|
            $dbNote = new Note();
 | 
						|
            $dbNote->noteable()->associate($recurrence);
 | 
						|
        }
 | 
						|
        $dbNote->text = trim($note);
 | 
						|
        $dbNote->save();
 | 
						|
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    protected function createRepetitions(Recurrence $recurrence, array $repetitions): void
 | 
						|
    {
 | 
						|
        /** @var array $array */
 | 
						|
        foreach ($repetitions as $array) {
 | 
						|
            RecurrenceRepetition::create(
 | 
						|
                [
 | 
						|
                    'recurrence_id'     => $recurrence->id,
 | 
						|
                    'repetition_type'   => $array['type'],
 | 
						|
                    'repetition_moment' => $array['moment'] ?? '',
 | 
						|
                    'repetition_skip'   => $array['skip'] ?? 0,
 | 
						|
                    'weekend'           => $array['weekend'] ?? 1,
 | 
						|
                ]
 | 
						|
            );
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Store transactions of a recurring transactions. It's complex but readable.
 | 
						|
     *
 | 
						|
     * @throws FireflyException
 | 
						|
     *
 | 
						|
     * @SuppressWarnings(PHPMD.NPathComplexity)
 | 
						|
     */
 | 
						|
    protected function createTransactions(Recurrence $recurrence, array $transactions): void
 | 
						|
    {
 | 
						|
        app('log')->debug('Now in createTransactions()');
 | 
						|
        foreach ($transactions as $index => $array) {
 | 
						|
            app('log')->debug(sprintf('Now at transaction #%d', $index));
 | 
						|
            $sourceTypes     = config(sprintf('firefly.expected_source_types.source.%s', $recurrence->transactionType->type));
 | 
						|
            $destTypes       = config(sprintf('firefly.expected_source_types.destination.%s', $recurrence->transactionType->type));
 | 
						|
            $source          = $this->findAccount($sourceTypes, $array['source_id'], null);
 | 
						|
            $destination     = $this->findAccount($destTypes, $array['destination_id'], null);
 | 
						|
 | 
						|
            /** @var TransactionCurrencyFactory $factory */
 | 
						|
            $factory         = app(TransactionCurrencyFactory::class);
 | 
						|
            $currency        = $factory->find($array['currency_id'] ?? null, $array['currency_code'] ?? null);
 | 
						|
            $foreignCurrency = $factory->find($array['foreign_currency_id'] ?? null, $array['foreign_currency_code'] ?? null);
 | 
						|
            if (null === $currency) {
 | 
						|
                $currency = app('amount')->getDefaultCurrencyByUserGroup($recurrence->user->userGroup);
 | 
						|
            }
 | 
						|
 | 
						|
            app('log')->debug(
 | 
						|
                sprintf('Will set the validator type to %s based on the type of the recurrence (#%d).', $recurrence->transactionType->type, $recurrence->id)
 | 
						|
            );
 | 
						|
 | 
						|
            // once the accounts have been determined, we still verify their validity:
 | 
						|
            /** @var AccountValidator $validator */
 | 
						|
            $validator       = app(AccountValidator::class);
 | 
						|
            $validator->setUser($recurrence->user);
 | 
						|
            $validator->setTransactionType($recurrence->transactionType->type);
 | 
						|
 | 
						|
            if (!$validator->validateSource(['id' => $source->id])) {
 | 
						|
                throw new FireflyException(sprintf('Source invalid: %s', $validator->sourceError));
 | 
						|
            }
 | 
						|
 | 
						|
            if (!$validator->validateDestination(['id' => $destination->id])) {
 | 
						|
                throw new FireflyException(sprintf('Destination invalid: %s', $validator->destError));
 | 
						|
            }
 | 
						|
            if (array_key_exists('foreign_amount', $array) && '' === (string)$array['foreign_amount']) {
 | 
						|
                unset($array['foreign_amount']);
 | 
						|
            }
 | 
						|
            // TODO typeOverrule. The account validator may have a different opinion on the type of the transaction.
 | 
						|
            $transaction     = new RecurrenceTransaction(
 | 
						|
                [
 | 
						|
                    'recurrence_id'           => $recurrence->id,
 | 
						|
                    'transaction_currency_id' => $currency->id,
 | 
						|
                    'foreign_currency_id'     => $foreignCurrency?->id,
 | 
						|
                    'source_id'               => $source->id,
 | 
						|
                    'destination_id'          => $destination->id,
 | 
						|
                    'amount'                  => $array['amount'],
 | 
						|
                    'foreign_amount'          => array_key_exists('foreign_amount', $array) ? (string)$array['foreign_amount'] : null,
 | 
						|
                    'description'             => $array['description'],
 | 
						|
                ]
 | 
						|
            );
 | 
						|
            $transaction->save();
 | 
						|
 | 
						|
            if (array_key_exists('budget_id', $array)) {
 | 
						|
                $this->setBudget($transaction, (int)$array['budget_id']);
 | 
						|
            }
 | 
						|
            if (array_key_exists('bill_id', $array)) {
 | 
						|
                $this->setBill($transaction, (int)$array['bill_id']);
 | 
						|
            }
 | 
						|
            if (array_key_exists('category_id', $array)) {
 | 
						|
                $this->setCategory($transaction, (int)$array['category_id']);
 | 
						|
            }
 | 
						|
 | 
						|
            // same for piggy bank
 | 
						|
            if (array_key_exists('piggy_bank_id', $array)) {
 | 
						|
                $this->updatePiggyBank($transaction, (int)$array['piggy_bank_id']);
 | 
						|
            }
 | 
						|
 | 
						|
            if (array_key_exists('tags', $array) && is_array($array['tags'])) {
 | 
						|
                $this->updateTags($transaction, $array['tags']);
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    protected function findAccount(array $expectedTypes, ?int $accountId, ?string $accountName): Account
 | 
						|
    {
 | 
						|
        $result       = null;
 | 
						|
        $accountId    = (int)$accountId;
 | 
						|
        $accountName  = (string)$accountName;
 | 
						|
 | 
						|
        /** @var AccountRepositoryInterface $repository */
 | 
						|
        $repository   = app(AccountRepositoryInterface::class);
 | 
						|
        $repository->setUser($this->user);
 | 
						|
 | 
						|
        // if user has submitted an account ID, search for it.
 | 
						|
        $result       = $repository->find($accountId);
 | 
						|
        if (null !== $result) {
 | 
						|
            return $result;
 | 
						|
        }
 | 
						|
 | 
						|
        // if user has submitted a name, search for it:
 | 
						|
        $result       = $repository->findByName($accountName, $expectedTypes);
 | 
						|
        if (null !== $result) {
 | 
						|
            return $result;
 | 
						|
        }
 | 
						|
 | 
						|
        // maybe we can create it? Try to avoid LOAN and other asset types.
 | 
						|
        $cannotCreate = [AccountType::ASSET, AccountType::DEBT, AccountType::LOAN, AccountType::MORTGAGE, AccountType::CREDITCARD];
 | 
						|
 | 
						|
        /** @var AccountFactory $factory */
 | 
						|
        $factory      = app(AccountFactory::class);
 | 
						|
        $factory->setUser($this->user);
 | 
						|
 | 
						|
        /** @var string $expectedType */
 | 
						|
        foreach ($expectedTypes as $expectedType) {
 | 
						|
            if (in_array($expectedType, $cannotCreate, true)) {
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
            if (!in_array($expectedType, $cannotCreate, true)) {
 | 
						|
                try {
 | 
						|
                    $result = $factory->findOrCreate($accountName, $expectedType);
 | 
						|
                } catch (FireflyException $e) {
 | 
						|
                    app('log')->error($e->getMessage());
 | 
						|
                }
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        return $result ?? $repository->getCashAccount();
 | 
						|
    }
 | 
						|
 | 
						|
    private function setBudget(RecurrenceTransaction $transaction, int $budgetId): void
 | 
						|
    {
 | 
						|
        Log::debug(sprintf('Now in %s', __METHOD__));
 | 
						|
        $budgetFactory = app(BudgetFactory::class);
 | 
						|
        $budgetFactory->setUser($transaction->recurrence->user);
 | 
						|
        $budget        = $budgetFactory->find($budgetId, null);
 | 
						|
        if (null === $budget) {
 | 
						|
            // remove budget from recurring transaction:
 | 
						|
            $transaction->recurrenceTransactionMeta()->where('name', 'budget_id')->delete();
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        $meta          = $transaction->recurrenceTransactionMeta()->where('name', 'budget_id')->first();
 | 
						|
        if (null === $meta) {
 | 
						|
            $meta        = new RecurrenceTransactionMeta();
 | 
						|
            $meta->rt_id = $transaction->id;
 | 
						|
            $meta->name  = 'budget_id';
 | 
						|
        }
 | 
						|
        $meta->value   = $budget->id;
 | 
						|
        $meta->save();
 | 
						|
    }
 | 
						|
 | 
						|
    private function setBill(RecurrenceTransaction $transaction, int $billId): void
 | 
						|
    {
 | 
						|
        $billFactory = app(BillFactory::class);
 | 
						|
        $billFactory->setUser($transaction->recurrence->user);
 | 
						|
        $bill        = $billFactory->find($billId, null);
 | 
						|
        if (null === $bill) {
 | 
						|
            // remove bill from recurring transaction:
 | 
						|
            $transaction->recurrenceTransactionMeta()->where('name', 'bill_id')->delete();
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        $meta        = $transaction->recurrenceTransactionMeta()->where('name', 'bill_id')->first();
 | 
						|
        if (null === $meta) {
 | 
						|
            $meta        = new RecurrenceTransactionMeta();
 | 
						|
            $meta->rt_id = $transaction->id;
 | 
						|
            $meta->name  = 'bill_id';
 | 
						|
        }
 | 
						|
        $meta->value = $bill->id;
 | 
						|
        $meta->save();
 | 
						|
    }
 | 
						|
 | 
						|
    private function setCategory(RecurrenceTransaction $transaction, int $categoryId): void
 | 
						|
    {
 | 
						|
        $categoryFactory = app(CategoryFactory::class);
 | 
						|
        $categoryFactory->setUser($transaction->recurrence->user);
 | 
						|
        $category        = $categoryFactory->findOrCreate($categoryId, null);
 | 
						|
        if (null === $category) {
 | 
						|
            // remove category:
 | 
						|
            $transaction->recurrenceTransactionMeta()->where('name', 'category_id')->delete();
 | 
						|
            $transaction->recurrenceTransactionMeta()->where('name', 'category_name')->delete();
 | 
						|
 | 
						|
            return;
 | 
						|
        }
 | 
						|
        $transaction->recurrenceTransactionMeta()->where('name', 'category_name')->delete();
 | 
						|
        $meta            = $transaction->recurrenceTransactionMeta()->where('name', 'category_id')->first();
 | 
						|
        if (null === $meta) {
 | 
						|
            $meta        = new RecurrenceTransactionMeta();
 | 
						|
            $meta->rt_id = $transaction->id;
 | 
						|
            $meta->name  = 'category_id';
 | 
						|
        }
 | 
						|
        $meta->value     = $category->id;
 | 
						|
        $meta->save();
 | 
						|
    }
 | 
						|
 | 
						|
    protected function updatePiggyBank(RecurrenceTransaction $transaction, int $piggyId): void
 | 
						|
    {
 | 
						|
        /** @var PiggyBankFactory $factory */
 | 
						|
        $factory   = app(PiggyBankFactory::class);
 | 
						|
        $factory->setUser($transaction->recurrence->user);
 | 
						|
        $piggyBank = $factory->find($piggyId, null);
 | 
						|
        if (null !== $piggyBank) {
 | 
						|
            /** @var null|RecurrenceMeta $entry */
 | 
						|
            $entry        = $transaction->recurrenceTransactionMeta()->where('name', 'piggy_bank_id')->first();
 | 
						|
            if (null === $entry) {
 | 
						|
                $entry = RecurrenceTransactionMeta::create(['rt_id' => $transaction->id, 'name' => 'piggy_bank_id', 'value' => $piggyBank->id]);
 | 
						|
            }
 | 
						|
            $entry->value = $piggyBank->id;
 | 
						|
            $entry->save();
 | 
						|
        }
 | 
						|
        if (null === $piggyBank) {
 | 
						|
            // delete if present
 | 
						|
            $transaction->recurrenceTransactionMeta()->where('name', 'piggy_bank_id')->delete();
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    protected function updateTags(RecurrenceTransaction $transaction, array $tags): void
 | 
						|
    {
 | 
						|
        if (0 !== count($tags)) {
 | 
						|
            /** @var null|RecurrenceMeta $entry */
 | 
						|
            $entry        = $transaction->recurrenceTransactionMeta()->where('name', 'tags')->first();
 | 
						|
            if (null === $entry) {
 | 
						|
                $entry = RecurrenceTransactionMeta::create(['rt_id' => $transaction->id, 'name' => 'tags', 'value' => json_encode($tags)]);
 | 
						|
            }
 | 
						|
            $entry->value = json_encode($tags);
 | 
						|
            $entry->save();
 | 
						|
        }
 | 
						|
        if (0 === count($tags)) {
 | 
						|
            // delete if present
 | 
						|
            $transaction->recurrenceTransactionMeta()->where('name', 'tags')->delete();
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    protected function deleteRepetitions(Recurrence $recurrence): void
 | 
						|
    {
 | 
						|
        $recurrence->recurrenceRepetitions()->delete();
 | 
						|
    }
 | 
						|
 | 
						|
    protected function deleteTransactions(Recurrence $recurrence): void
 | 
						|
    {
 | 
						|
        app('log')->debug('deleteTransactions()');
 | 
						|
 | 
						|
        /** @var RecurrenceTransaction $transaction */
 | 
						|
        foreach ($recurrence->recurrenceTransactions as $transaction) {
 | 
						|
            $transaction->recurrenceTransactionMeta()->delete();
 | 
						|
 | 
						|
            $transaction->delete();
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 |