Expand import routine.

2025-10-12 15:35:15 +00:00 · 2017-06-20 21:04:25 +02:00
parent b9f110ac2b
commit a905cce2c9
16 changed files with 762 additions and 117 deletions
--- a/app/Import/Object/ImportBudget.php
+++ b/app/Import/Object/ImportBudget.php
@@ -12,13 +12,52 @@ declare(strict_types=1);
 namespace FireflyIII\Import\Object;


+use FireflyIII\Models\Budget;
+use FireflyIII\Repositories\Budget\BudgetRepositoryInterface;
+use FireflyIII\User;
+use Illuminate\Support\Collection;
+use Log;
+
+/**
+ * Class ImportBudget
+ *
+ * @package FireflyIII\Import\Object
+ */
 class ImportBudget
 {

+    /** @var Budget */
+    private $budget;
    /** @var array */
    private $id = [];
    /** @var array */
    private $name = [];
+    /** @var BudgetRepositoryInterface */
+    private $repository;
+    /** @var  User */
+    private $user;
+
+    /**
+     * ImportBudget constructor.
+     */
+    public function __construct()
+    {
+        $this->budget     = new Budget;
+        $this->repository = app(BudgetRepositoryInterface::class);
+        Log::debug('Created ImportBudget.');
+    }
+
+    /**
+     * @return Budget
+     */
+    public function getBudget(): Budget
+    {
+        if (is_null($this->budget->id)) {
+            $this->store();
+        }
+
+        return $this->budget;
+    }

    /**
     * @param array $id
@@ -36,5 +75,155 @@ class ImportBudget
        $this->name = $name;
    }

+    /**
+     * @param User $user
+     */
+    public function setUser(User $user)
+    {
+        $this->user = $user;
+        $this->repository->setUser($user);
+    }
+
+    /**
+     * @return Budget
+     */
+    private function findExistingObject(): Budget
+    {
+        Log::debug('In findExistingObject() for Budget');
+        // 1: find by ID, or name
+
+        if (count($this->id) === 3) {
+            Log::debug(sprintf('Finding budget with ID #%d', $this->id['value']));
+            /** @var Budget $budget */
+            $budget = $this->repository->find(intval($this->id['value']));
+            if (!is_null($budget->id)) {
+                Log::debug(sprintf('Found unmapped budget by ID (#%d): %s', $budget->id, $budget->name));
+
+                return $budget;
+            }
+            Log::debug('Found nothing.');
+        }
+        // 2: find by name
+        if (count($this->name) === 3) {
+            /** @var Collection $budgets */
+            $budgets = $this->repository->getBudgets();
+            $name    = $this->name['value'];
+            Log::debug(sprintf('Finding budget with name %s', $name));
+            $filtered = $budgets->filter(
+                function (Budget $budget) use ($name) {
+                    if ($budget->name === $name) {
+                        Log::debug(sprintf('Found unmapped budget by name (#%d): %s', $budget->id, $budget->name));
+
+                        return $budget;
+                    }
+
+                    return null;
+                }
+            );
+
+            if ($filtered->count() === 1) {
+                return $filtered->first();
+            }
+            Log::debug('Found nothing.');
+        }
+
+        // 4: do not search by account number.
+        Log::debug('Found NO existing budgets.');
+
+        return new Budget;
+
+    }
+
+    /**
+     * @return Budget
+     */
+    private function findMappedObject(): Budget
+    {
+        Log::debug('In findMappedObject() for Budget');
+        $fields = ['id', 'name'];
+        foreach ($fields as $field) {
+            $array = $this->$field;
+            Log::debug(sprintf('Find mapped budget based on field "%s" with value', $field), $array);
+            // check if a pre-mapped object exists.
+            $mapped = $this->getMappedObject($array);
+            if (!is_null($mapped->id)) {
+                Log::debug(sprintf('Found budget #%d!', $mapped->id));
+
+                return $mapped;
+            }
+
+        }
+        Log::debug('Found no budget on mapped data or no map present.');
+
+        return new Budget;
+    }
+
+    /**
+     * @param array $array
+     *
+     * @return Budget
+     */
+    private function getMappedObject(array $array): Budget
+    {
+        Log::debug('In getMappedObject() for Budget');
+        if (count($array) === 0) {
+            Log::debug('Array is empty, nothing will come of this.');
+
+            return new Budget;
+        }
+
+        if (array_key_exists('mapped', $array) && is_null($array['mapped'])) {
+            Log::debug(sprintf('No map present for value "%s". Return NULL.', $array['value']));
+
+            return new Budget;
+        }
+
+        Log::debug('Finding a mapped budget based on', $array);
+
+        $search  = intval($array['mapped']);
+        $account = $this->repository->find($search);
+
+        Log::debug(sprintf('Found budget! #%d ("%s"). Return it', $account->id, $account->name));
+
+        return $account;
+    }
+
+    /**
+     * @return bool
+     */
+    private function store(): bool
+    {
+        // 1: find mapped object:
+        $mapped = $this->findMappedObject();
+        if (!is_null($mapped->id)) {
+            $this->budget = $mapped;
+
+            return true;
+        }
+        // 2: find existing by given values:
+        $found = $this->findExistingObject();
+        if (!is_null($found->id)) {
+            $this->budget = $found;
+
+            return true;
+        }
+        $name = $this->name['value'] ?? '';
+
+        if (strlen($name) === 0) {
+            return true;
+        }
+
+        Log::debug('Found no budget so must create one ourselves.');
+
+        $data = [
+            'name' => $name,
+        ];
+
+        $this->budget = $this->repository->store($data);
+        Log::debug(sprintf('Successfully stored new budget #%d: %s', $this->budget->id, $this->budget->name));
+
+        return true;
+    }
+

 }