firefly-iii/app/Support/Import/JobConfiguration/FinTS/ChooseAccountHandler.php

<?php
/**
 * ChooseAccountHandler.php
 * Copyright (c) 2017 thegrumpydictator@gmail.com
 *
 * This file is part of Firefly III.
 *
 * Firefly III is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Firefly III 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Firefly III. If not, see <http://www.gnu.org/licenses/>.
 */
declare(strict_types=1);

namespace FireflyIII\Support\Import\JobConfiguration\FinTS;


use FireflyIII\Exceptions\FireflyException;
use FireflyIII\Import\JobConfiguration\FinTSConfigurationSteps;
use FireflyIII\Models\AccountType;
use FireflyIII\Models\ImportJob;
use FireflyIII\Repositories\ImportJob\ImportJobRepositoryInterface;
use FireflyIII\Support\FinTS\FinTS;
use Illuminate\Support\MessageBag;
use FireflyIII\Repositories\Account\AccountRepositoryInterface;

class ChooseAccountHandler implements FinTSConfigurationInterface
{
    /** @var ImportJob */
    private $importJob;
    /** @var ImportJobRepositoryInterface */
    private $repository;
    /** @var AccountRepositoryInterface */
    private $accountRepository;

    /**
     * Store data associated with current stage.
     *
     * @param array $data
     *
     * @return MessageBag
     */
    public function configureJob(array $data): MessageBag
    {
        $config                  = $this->importJob->configuration;
        $config['fints_account'] = (string)($data['fints_account'] ?? '');
        $config['local_account'] = (string)($data['local_account'] ?? '');
        $config['from_date']     = (string)($data['from_date'] ?? '');
        $config['to_date']       = (string)($data['to_date'] ?? '');
        $this->repository->setConfiguration($this->importJob, $config);

        try {
            $finTS = new FinTS($this->importJob->configuration);
            $finTS->getAccount($config['fints_account']);
        } catch (FireflyException $e) {
            return new MessageBag([$e->getMessage()]);
        }

        $this->repository->setStage($this->importJob, FinTSConfigurationSteps::GO_FOR_IMPORT);

        return new MessageBag();
    }

    /**
     * Get the data necessary to show the configuration screen.
     *
     * @return array
     * @throws \FireflyIII\Exceptions\FireflyException
     */
    public function getNextData(): array
    {
        $finTS             = new FinTS($this->importJob->configuration);
        $finTSAccounts     = $finTS->getAccounts();
        $finTSAccountsData = [];
        foreach ($finTSAccounts as $account) {
            $finTSAccountsData[$account->getAccountNumber()] = $account->getIban();
        }

        $localAccounts = [];
        foreach ($this->accountRepository->getAccountsByType([AccountType::ASSET]) as $localAccount) {
            $display_name = $localAccount->name;
            if ($localAccount->iban) {
                $display_name .= " - $localAccount->iban";
            }
            $localAccounts[$localAccount->id] = $display_name;
        }

        $data = [
            'fints_accounts' => $finTSAccountsData,
            'fints_account' => $this->importJob->configuration['fints_account'] ?? null,
            'local_accounts' => $localAccounts,
            'local_account' => $this->importJob->configuration['local_account'] ?? null,
            'from_date' => $this->importJob->configuration['from_date'] ?? (new \DateTime('now - 1 month'))->format('Y-m-d'),
            'to_date' => $this->importJob->configuration['to_date'] ?? (new \DateTime('now'))->format('Y-m-d')
        ];
        return $data;
    }

    /**
     * @param ImportJob $importJob
     */
    public function setImportJob(ImportJob $importJob): void
    {
        $this->importJob         = $importJob;
        $this->repository        = app(ImportJobRepositoryInterface::class);
        $this->accountRepository = app(AccountRepositoryInterface::class);
        $this->repository->setUser($importJob->user);
    }


}