Allow more models to accept transactions #2828

app/Api/V1/Requests/AttachmentStoreRequest.php

@@ -23,9 +23,6 @@ declare(strict_types=1);
 
 namespace FireflyIII\Api\V1\Requests;
 
-use FireflyIII\Models\Bill;
-use FireflyIII\Models\ImportJob;
-use FireflyIII\Models\TransactionJournal;
 use FireflyIII\Rules\IsValidAttachmentModel;
 
 /**
@@ -69,14 +66,14 @@ class AttachmentStoreRequest extends Request
      */
     public function rules(): array
     {
-        $models = implode(
-            ',',
-            [
-                str_replace('FireflyIII\\Models\\', '', Bill::class),
-                str_replace('FireflyIII\\Models\\', '', ImportJob::class),
-                str_replace('FireflyIII\\Models\\', '', TransactionJournal::class),
-            ]
+        $models = config('firefly.valid_attachment_models');
+        $models = array_map(
+
+            static function (string $className) {
+                return str_replace('FireflyIII\\Models\\', '', $className);
+            }, $models
         );
+        $models = implode(',', $models);
         $model  = $this->string('attachable_type');
 
         return [

app/Rules/IsValidAttachmentModel.php

@@ -24,15 +24,24 @@ declare(strict_types=1);
 namespace FireflyIII\Rules;
 
 
+use FireflyIII\Models\Account;
 use FireflyIII\Models\Bill;
+use FireflyIII\Models\Budget;
+use FireflyIII\Models\Category;
 use FireflyIII\Models\ImportJob;
+use FireflyIII\Models\PiggyBank;
+use FireflyIII\Models\Tag;
 use FireflyIII\Models\Transaction;
 use FireflyIII\Models\TransactionJournal;
+use FireflyIII\Repositories\Account\AccountRepositoryInterface;
 use FireflyIII\Repositories\Bill\BillRepositoryInterface;
+use FireflyIII\Repositories\Budget\BudgetRepositoryInterface;
+use FireflyIII\Repositories\Category\CategoryRepositoryInterface;
 use FireflyIII\Repositories\ImportJob\ImportJobRepositoryInterface;
 use FireflyIII\Repositories\Journal\JournalAPIRepositoryInterface;
 use FireflyIII\Repositories\Journal\JournalRepositoryInterface;
-use FireflyIII\User;
+use FireflyIII\Repositories\PiggyBank\PiggyBankRepositoryInterface;
+use FireflyIII\Repositories\Tag\TagRepositoryInterface;
 use Illuminate\Contracts\Validation\Rule;
 use Log;
 
@@ -80,54 +89,151 @@ class IsValidAttachmentModel implements Rule
         if (!auth()->check()) {
             return false;
         }
-
-
-        if (Bill::class === $this->model) {
-            /** @var BillRepositoryInterface $repository */
-            $repository = app(BillRepositoryInterface::class);
-            /** @var User $user */
-            $user = auth()->user();
-            $repository->setUser($user);
-            $bill = $repository->find((int)$value);
-
-            return null !== $bill;
-        }
-
-        if (ImportJob::class === $this->model) {
-            /** @var ImportJobRepositoryInterface $repository */
-            $repository = app(ImportJobRepositoryInterface::class);
-            /** @var User $user */
-            $user = auth()->user();
-            $repository->setUser($user);
-            $importJob = $repository->find((int)$value);
-
-            return null !== $importJob;
-        }
-
-        if (Transaction::class === $this->model) {
-            /** @var JournalAPIRepositoryInterface $repository */
-            $repository = app(JournalAPIRepositoryInterface::class);
-
-            /** @var User $user */
-            $user = auth()->user();
-            $repository->setUser($user);
-            $transaction = $repository->findTransaction((int)$value);
-
-            return null !== $transaction;
-        }
-
-        if (TransactionJournal::class === $this->model) {
-            $repository = app(JournalRepositoryInterface::class);
-            $user       = auth()->user();
-            $repository->setUser($user);
-            $result = $repository->findNull((int)$value);
-
-            return null !== $result;
-        }
-        Log::error(sprintf('No model was recognized from string "%s"', $this->model));
+        $methods = [
+            Account::class            => 'validateAccount',
+            Bill::class               => 'validateBill',
+            Budget::class             => 'validateBudget',
+            Category::class           => 'validateCategory',
+            ImportJob::class          => 'validateImportJob',
+            PiggyBank::class          => 'validatePiggyBank',
+            Tag::class                => 'validateTag',
+            Transaction::class        => 'validateTransaction',
+            TransactionJournal::class => 'validateJournal',
+        ];
+        if (!isset($methods[$this->model])) {
+            Log::error(sprintf('Cannot validate model "%s" in %s.', $this->model, __METHOD__));
 
             return false;
         }
+        $method = $methods[$this->model];
+
+        return $this->$method((int) $value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateTag(int $value): bool
+    {
+        /** @var TagRepositoryInterface $repository */
+        $repository = app(TagRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findNull($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validatePiggyBank(int $value): bool
+    {
+        /** @var PiggyBankRepositoryInterface $repository */
+        $repository = app(PiggyBankRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findNull($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateBudget(int $value): bool
+    {
+        /** @var BudgetRepositoryInterface $repository */
+        $repository = app(BudgetRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findNull($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateCategory(int $value): bool
+    {
+        /** @var CategoryRepositoryInterface $repository */
+        $repository = app(CategoryRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findNull($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateAccount(int $value): bool
+    {
+        /** @var AccountRepositoryInterface $repository */
+        $repository = app(AccountRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findNull($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateJournal(int $value): bool
+    {
+        $repository = app(JournalRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findNull($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateTransaction(int $value): bool
+    {
+        /** @var JournalAPIRepositoryInterface $repository */
+        $repository = app(JournalAPIRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->findTransaction((int) $value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateImportJob(int $value): bool
+    {
+        /** @var ImportJobRepositoryInterface $repository */
+        $repository = app(ImportJobRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->find($value);
+    }
+
+    /**
+     * @param int $value
+     *
+     * @return bool
+     */
+    private function validateBill(int $value): bool
+    {
+        /** @var BillRepositoryInterface $repository */
+        $repository = app(BillRepositoryInterface::class);
+        $repository->setUser(auth()->user());
+
+        return null !== $repository->find($value);
+    }
 
     /**
      * @param string $model
@@ -141,6 +247,7 @@ class IsValidAttachmentModel implements Rule
         $model   = str_replace($search, $replace, $model);
 
         $model = sprintf('FireflyIII\Models\%s', $model);
+
         return $model;
     }
 }

config/firefly.php

@@ -38,6 +38,7 @@ use FireflyIII\Models\Recurrence;
 use FireflyIII\Models\Rule;
 use FireflyIII\Models\RuleGroup;
 use FireflyIII\Models\Tag;
+use FireflyIII\Models\Transaction;
 use FireflyIII\Models\TransactionCurrency;
 use FireflyIII\Models\TransactionGroup;
 use FireflyIII\Models\TransactionJournal;
@@ -167,6 +168,17 @@ return [
         'latitude'   => env('MAP_DEFAULT_LAT', '51.983333'),
         'zoom_level' => env('MAP_DEFAULT_ZOOM', '6'),
     ],
+    'valid_attachment_models'      => [
+        Account::class,
+        Bill::class,
+        Budget::class,
+        Category::class,
+        ImportJob::class,
+        PiggyBank::class,
+        Tag::class,
+        Transaction::class,
+        TransactionJournal::class,
+    ],
     'allowedMimes'                 => [
         /* plain files */
         'text/plain',