mirror of
https://github.com/firefly-iii/firefly-iii.git
synced 2025-02-25 18:45:27 -06:00
304 lines
11 KiB
PHP
304 lines
11 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
/**
|
|
* ExportData.php
|
|
* Copyright (c) 2020 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/>.
|
|
*/
|
|
|
|
namespace FireflyIII\Console\Commands\Export;
|
|
|
|
use Carbon\Carbon;
|
|
use Exception;
|
|
use FireflyIII\Console\Commands\VerifiesAccessToken;
|
|
use FireflyIII\Exceptions\FireflyException;
|
|
use FireflyIII\Models\AccountType;
|
|
use FireflyIII\Repositories\Account\AccountRepositoryInterface;
|
|
use FireflyIII\Repositories\Journal\JournalRepositoryInterface;
|
|
use FireflyIII\Support\Export\ExportDataGenerator;
|
|
use FireflyIII\User;
|
|
use Illuminate\Console\Command;
|
|
use Illuminate\Support\Collection;
|
|
use InvalidArgumentException;
|
|
use League\Csv\CannotInsertRecord;
|
|
use Log;
|
|
|
|
/**
|
|
* Class ExportData
|
|
*/
|
|
class ExportData extends Command
|
|
{
|
|
use VerifiesAccessToken;
|
|
|
|
/**
|
|
* The console command description.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $description = 'Command to export data from Firefly III.';
|
|
/**
|
|
* The name and signature of the console command.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $signature = 'firefly-iii:export-data
|
|
{--user=1 : The user ID that the export should run for.}
|
|
{--token= : The user\'s access token.}
|
|
{--start= : First transaction to export. Defaults to your very first transaction. Only applies to transaction export.}
|
|
{--end= : Last transaction to export. Defaults to today. Only applies to transaction export.}
|
|
{--accounts= : From which accounts or liabilities to export. Only applies to transaction export. Defaults to all of your asset accounts.}
|
|
{--export_directory=./ : Where to store the export files.}
|
|
{--export-transactions : Create a file with all your transactions and their meta data. This flag and the other flags can be combined.}
|
|
{--export-accounts : Create a file with all your accounts and some meta data.}
|
|
{--export-budgets : Create a file with all your budgets and some meta data.}
|
|
{--export-categories : Create a file with all your categories and some meta data.}
|
|
{--export-tags : Create a file with all your tags and some meta data.}
|
|
{--export-recurring : Create a file with all your recurring transactions and some meta data.}
|
|
{--export-rules : Create a file with all your rules and some meta data.}
|
|
{--export-bills : Create a file with all your bills and some meta data.}
|
|
{--export-piggies : Create a file with all your piggy banks and some meta data.}
|
|
{--force : Force overwriting of previous exports if found.}';
|
|
/** @var AccountRepositoryInterface */
|
|
private $accountRepository;
|
|
/** @var JournalRepositoryInterface */
|
|
private $journalRepository;
|
|
/** @var User */
|
|
private $user;
|
|
|
|
/**
|
|
* Execute the console command.
|
|
*
|
|
* @throws FireflyException
|
|
* @throws CannotInsertRecord
|
|
* @return int
|
|
*/
|
|
public function handle(): int
|
|
{
|
|
// verify access token
|
|
if (!$this->verifyAccessToken()) {
|
|
$this->error('Invalid access token. Check /profile.');
|
|
|
|
return 1;
|
|
}
|
|
// set up repositories.
|
|
$this->stupidLaravel();
|
|
$this->user = $this->getUser();
|
|
$this->journalRepository->setUser($this->user);
|
|
$this->accountRepository->setUser($this->user);
|
|
// get the options.
|
|
try {
|
|
$options = $this->parseOptions();
|
|
} catch (FireflyException $e) {
|
|
$this->error(sprintf('Could not work with your options: %s', $e));
|
|
|
|
return 1;
|
|
}
|
|
// make export object and configure it.
|
|
|
|
/** @var ExportDataGenerator $exporter */
|
|
$exporter = app(ExportDataGenerator::class);
|
|
$exporter->setUser($this->user);
|
|
$exporter->setStart($options['start']);
|
|
$exporter->setEnd($options['end']);
|
|
$exporter->setExportTransactions($options['export']['transactions']);
|
|
$exporter->setExportAccounts($options['export']['accounts']);
|
|
$exporter->setExportBudgets($options['export']['budgets']);
|
|
$exporter->setExportCategories($options['export']['categories']);
|
|
$exporter->setExportTags($options['export']['tags']);
|
|
$exporter->setExportRecurring($options['export']['recurring']);
|
|
$exporter->setExportRules($options['export']['rules']);
|
|
$exporter->setExportBills($options['export']['bills']);
|
|
$exporter->setExportPiggies($options['export']['piggies']);
|
|
|
|
$data = $exporter->export();
|
|
|
|
if (0 === count($data)) {
|
|
$this->error('You must export *something*. Use --export-transactions or another option. See docs.firefly-iii.org');
|
|
|
|
return 1;
|
|
}
|
|
|
|
try {
|
|
$this->exportData($options, $data);
|
|
} catch (FireflyException $e) {
|
|
$this->error(sprintf('Could not store data: %s', $e->getMessage()));
|
|
|
|
// app('telemetry')->feature('executed-command-with-error', $this->signature);
|
|
return 1;
|
|
}
|
|
|
|
// app('telemetry')->feature('executed-command', $this->signature);
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @param array $options
|
|
* @param array $data
|
|
*
|
|
* @throws FireflyException
|
|
*/
|
|
private function exportData(array $options, array $data): void
|
|
{
|
|
$date = date('Y_m_d');
|
|
foreach ($data as $key => $content) {
|
|
$file = sprintf('%s%s_%s.csv', $options['directory'], $date, $key);
|
|
if (false === $options['force'] && file_exists($file)) {
|
|
throw new FireflyException(sprintf('File "%s" exists already. Use --force to overwrite.', $file));
|
|
}
|
|
if (true === $options['force'] && file_exists($file)) {
|
|
$this->warn(sprintf('File "%s" exists already but will be replaced.', $file));
|
|
}
|
|
// continue to write to file.
|
|
file_put_contents($file, $content);
|
|
$this->info(sprintf('Wrote %s-export to file "%s".', $key, $file));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @throws FireflyException
|
|
* @return Collection
|
|
*/
|
|
private function getAccountsParameter(): Collection
|
|
{
|
|
$final = new Collection;
|
|
$accounts = new Collection;
|
|
$accountList = $this->option('accounts');
|
|
$types = [AccountType::ASSET, AccountType::LOAN, AccountType::DEBT, AccountType::MORTGAGE];
|
|
if (null !== $accountList && '' !== (string) $accountList) {
|
|
$accountIds = explode(',', $accountList);
|
|
$accounts = $this->accountRepository->getAccountsById($accountIds);
|
|
}
|
|
if (null === $accountList) {
|
|
$accounts = $this->accountRepository->getAccountsByType($types);
|
|
}
|
|
// filter accounts,
|
|
/** @var AccountType $account */
|
|
foreach ($accounts as $account) {
|
|
if (in_array($account->accountType->type, $types, true)) {
|
|
$final->push($account);
|
|
}
|
|
}
|
|
if (0 === $final->count()) {
|
|
throw new FireflyException('Ended up with zero valid accounts to export from.');
|
|
}
|
|
|
|
return $final;
|
|
}
|
|
|
|
/**
|
|
* @param string $field
|
|
*
|
|
* @throws FireflyException
|
|
* @throws Exception
|
|
* @return Carbon
|
|
*/
|
|
private function getDateParameter(string $field): Carbon
|
|
{
|
|
$date = Carbon::now()->subYear();
|
|
if (null !== $this->option($field)) {
|
|
try {
|
|
$date = Carbon::createFromFormat('Y-m-d', $this->option($field));
|
|
} catch (InvalidArgumentException $e) {
|
|
Log::error($e->getMessage());
|
|
$this->error(sprintf('%s date "%s" must be formatted YYYY-MM-DD. Field will be ignored.', $field, $this->option('start')));
|
|
}
|
|
|
|
return $date;
|
|
}
|
|
if ('start' === $field) {
|
|
$journal = $this->journalRepository->firstNull();
|
|
$date = null === $journal ? Carbon::now()->subYear() : $journal->date;
|
|
$date->startOfDay();
|
|
|
|
return $date;
|
|
}
|
|
if ('end' === $field) {
|
|
$date = new Carbon;
|
|
$date->endOfDay();
|
|
|
|
return $date;
|
|
}
|
|
|
|
// fallback
|
|
return $date;
|
|
}
|
|
|
|
/**
|
|
* @throws FireflyException
|
|
*
|
|
* @return string
|
|
*/
|
|
private function getExportDirectory(): string
|
|
{
|
|
$directory = (string) $this->option('export_directory');
|
|
if (null === $directory) {
|
|
$directory = './';
|
|
}
|
|
if (!is_writable($directory)) {
|
|
throw new FireflyException(sprintf('Directory "%s" isn\'t writeable.', $directory));
|
|
}
|
|
|
|
return $directory;
|
|
}
|
|
|
|
/**
|
|
* @throws FireflyException
|
|
* @return array
|
|
*/
|
|
private function parseOptions(): array
|
|
{
|
|
$start = $this->getDateParameter('start');
|
|
$end = $this->getDateParameter('end');
|
|
$accounts = $this->getAccountsParameter();
|
|
$export = $this->getExportDirectory();
|
|
|
|
return [
|
|
'export' => [
|
|
'transactions' => $this->option('export-transactions'),
|
|
'accounts' => $this->option('export-accounts'),
|
|
'budgets' => $this->option('export-budgets'),
|
|
'categories' => $this->option('export-categories'),
|
|
'tags' => $this->option('export-tags'),
|
|
'recurring' => $this->option('export-recurring'),
|
|
'rules' => $this->option('export-rules'),
|
|
'bills' => $this->option('export-bills'),
|
|
'piggies' => $this->option('export-piggies'),
|
|
],
|
|
'start' => $start,
|
|
'end' => $end,
|
|
'accounts' => $accounts,
|
|
'directory' => $export,
|
|
'force' => $this->option('force'),
|
|
];
|
|
}
|
|
|
|
/**
|
|
* Laravel will execute ALL __construct() methods for ALL commands whenever a SINGLE command is
|
|
* executed. This leads to noticeable slow-downs and class calls. To prevent this, this method should
|
|
* be called from the handle method instead of using the constructor to initialize the command.
|
|
*
|
|
* @codeCoverageIgnore
|
|
*/
|
|
private function stupidLaravel(): void
|
|
{
|
|
$this->journalRepository = app(JournalRepositoryInterface::class);
|
|
$this->accountRepository = app(AccountRepositoryInterface::class);
|
|
}
|
|
|
|
|
|
}
|