firefly-iii/app/Http/Controllers/ExportController.php

<?php
declare(strict_types = 1);
/**
 * ExportController.php
 * Copyright (C) 2016 thegrumpydictator@gmail.com
 *
 * This software may be modified and distributed under the terms
 * of the MIT license.  See the LICENSE file for details.
 */

namespace FireflyIII\Http\Controllers;

use Carbon\Carbon;
use ExpandedForm;
use FireflyIII\Exceptions\FireflyException;
use FireflyIII\Export\Processor;
use FireflyIII\Http\Requests;
use FireflyIII\Http\Requests\ExportFormRequest;
use FireflyIII\Models\ExportJob;
use FireflyIII\Repositories\Account\AccountRepositoryInterface as ARI;
use FireflyIII\Repositories\ExportJob\ExportJobRepositoryInterface as EJRI;
use Log;
use Preferences;
use Response;
use Storage;
use View;

/**
 * Class ExportController
 *
 * @package FireflyIII\Http\Controllers
 */
class ExportController extends Controller
{
    /**
     *
     */
    public function __construct()
    {
        parent::__construct();
        View::share('mainTitleIcon', 'fa-file-archive-o');
        View::share('title', trans('firefly.export_data'));
    }

    /**
     * @param ExportJob $job
     *
     * @return mixed
     * @throws FireflyException
     */
    public function download(ExportJob $job)
    {
        $disk   = Storage::disk('export');
        $file   = $job->key . '.zip';
        $date   = date('Y-m-d \a\t H-i-s');
        $name   = 'Export job on ' . $date . '.zip';
        $quoted = sprintf('"%s"', addcslashes($name, '"\\'));

        if (!$disk->exists($file)) {
            throw new FireflyException('Against all expectations, zip file "' . $file . '" does not exist.');
        }


        $job->change('export_downloaded');
        Log::debug('Will send user file "' . $file . '".');

        return response($disk->get($file), 200)
            ->header('Content-Description', 'File Transfer')
            ->header('Content-Type', 'application/octet-stream')
            ->header('Content-Disposition', 'attachment; filename=' . $quoted)
            ->header('Content-Transfer-Encoding', 'binary')
            ->header('Connection', 'Keep-Alive')
            ->header('Expires', '0')
            ->header('Cache-Control', 'must-revalidate, post-check=0, pre-check=0')
            ->header('Pragma', 'public')
            ->header('Content-Length', $disk->size($file));

    }

    /**
     * @param ExportJob $job
     *
     * @return \Illuminate\Http\JsonResponse
     */
    public function getStatus(ExportJob $job)
    {
        return Response::json(['status' => trans('firefly.' . $job->status)]);
    }

    /**
     * @param ARI  $repository
     *
     * @param EJRI $jobs
     *
     * @return \Illuminate\Contracts\View\Factory|\Illuminate\View\View
     */
    public function index(ARI $repository, EJRI $jobs)
    {
        // create new export job.
        $job = $jobs->create();
        // delete old ones.
        $jobs->cleanup();

        // does the user have shared accounts?
        $accounts      = $repository->getAccounts(['Default account', 'Asset account']);
        $accountList   = ExpandedForm::makeSelectList($accounts);
        $checked       = array_keys($accountList);
        $formats       = array_keys(config('firefly.export_formats'));
        $defaultFormat = Preferences::get('export_format', config('firefly.default_export_format'))->data;
        $first         = session('first')->format('Y-m-d');
        $today         = Carbon::create()->format('Y-m-d');

        return view('export.index', compact('job', 'checked', 'accountList', 'formats', 'defaultFormat', 'first', 'today'));

    }

    /**
     * @param ExportFormRequest $request
     * @param ARI               $repository
     *
     * @param EJRI              $jobs
     *
     * @return string
     * @throws \FireflyIII\Exceptions\FireflyException
     */
    public function postIndex(ExportFormRequest $request, ARI $repository, EJRI $jobs)
    {
        set_time_limit(0);
        $job      = $jobs->findByKey($request->get('job'));
        $settings = [
            'accounts'           => $repository->get($request->get('accounts')),
            'startDate'          => new Carbon($request->get('export_start_range')),
            'endDate'            => new Carbon($request->get('export_end_range')),
            'exportFormat'       => $request->get('exportFormat'),
            'includeAttachments' => intval($request->get('include_attachments')) === 1,
            'includeConfig'      => intval($request->get('include_config')) === 1,
            'includeOldUploads'  => intval($request->get('include_old_uploads')) === 1,
            'job'                => $job,
        ];

        $job->change('export_status_make_exporter');
        $processor = new Processor($settings);

        /*
         * Collect journals:
         */
        $job->change('export_status_collecting_journals');
        $processor->collectJournals();
        $job->change('export_status_collected_journals');
        /*
         * Transform to exportable entries:
         */
        $job->change('export_status_converting_to_export_format');
        $processor->convertJournals();
        $job->change('export_status_converted_to_export_format');
        /*
         * Transform to (temporary) file:
         */
        $job->change('export_status_creating_journal_file');
        $processor->exportJournals();
        $job->change('export_status_created_journal_file');
        /*
         *  Collect attachments, if applicable.
         */
        if ($settings['includeAttachments']) {
            $job->change('export_status_collecting_attachments');
            $processor->collectAttachments();
            $job->change('export_status_collected_attachments');
        }

        /*
         * Collect old uploads
         */
        if ($settings['includeOldUploads']) {
            $job->change('export_status_collecting_old_uploads');
            $processor->collectOldUploads();
            $job->change('export_status_collected_old_uploads');
        }

        /*
         * Generate / collect config file.
         */
        if ($settings['includeConfig']) {
            $job->change('export_status_creating_config_file');
            $processor->createConfigFile();
            $job->change('export_status_created_config_file');
        }

        /*
         * Create ZIP file:
         */
        $job->change('export_status_creating_zip_file');
        $processor->createZipFile();
        $job->change('export_status_created_zip_file');

        $job->change('export_status_finished');

        return Response::json('ok');
    }
}