Thinktomorrow\AssetLibrary\Commands\ImageToAssetMigrateCommand

Complexity

Total Complexity

17

Size/Duplication

Total Lines	152
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	getResultsFromDatabase()	0	13	2
A	setArguments()	0	8	1
B	handle()	0	50	8
A	mapResults()	0	19	3
A	handleResetFlag()	0	12	3

<?php

namespace Thinktomorrow\AssetLibrary\Commands;

use Illuminate\Console\Command;
use Illuminate\Support\Facades\DB;
use Thinktomorrow\AssetLibrary\Models\AssetUploader;
use Spatie\MediaLibrary\Exceptions\FileCannotBeAdded\UnreachableUrl;

class ImageToAssetMigrateCommand extends Command
{
    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'assetlibrary:migrate-image
                                    {table}
                                    {urlcolumn}
                                    {linkedmodel}
                                    {idcolumn=id}
                                    {ordercolumn?}
                                    {localecolumn?}
                                    {--force}
                                    {--reset}
                                    {--dry}';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Import images from imageurl on a model to assets managed by assetlibrary.';

    private $table;
    private $urlcolumn;
    private $linkedmodel;
    private $idcolumn;
    private $ordercolumn;
    private $localecolumn;

    private $nomodel     = 0;
    private $unreachable = 0;
    private $files       = 0;

    /**
     * Execute the console command.
     *
     * @return mixed
     */
    public function handle()
    {
        \ini_set('memory_limit', '256M');

        $this->setArguments();

        $results        = $this->getResultsFromDatabase();
        $bar            = $this->output->createProgressBar(count($results));
        $orderedResults = $this->mapResults($results);

        $this->handleResetFlag($orderedResults);

        $this->info("\n".'Migrating images.');

        foreach ($orderedResults as $result) {
            foreach ($result['images'] as $line) {
                $bar->advance();

                if (! $line || $this->option('dry')) {
                    $line ?? $this->unreachable++;
                    $this->option('dry') ? $this->files++ : '';
                    continue;
                }

                try {
                    $asset = AssetUploader::uploadFromUrl(public_path($line));
                } catch (UnreachableUrl $ex) {
                    // increment the amount of unreachable files counter
                    $this->unreachable++;

                    continue;
                }

                $asset->setOrder($result['order'] ?? null)->attachToModel($result['model']);

                if ($this->option('force')) {
                    unlink(public_path($line));
                }

                // increment the amount of files migrated counter
                $this->files++;
            }
        }

        $bar->finish();

        $this->info('Migrating done.');
        $this->info('Migrated '.$this->files.' files.');
        $this->info('Couldn\'t reach '.$this->unreachable.' files.');
        $this->info('Couldn\'t find '.$this->nomodel.' model(s).');
    }

    private function getResultsFromDatabase()
    {
        $columns = [$this->urlcolumn, $this->idcolumn, $this->ordercolumn, $this->localecolumn];

        $builder = DB::table($this->table)->select($columns);

        if (! $this->ordercolumn) {
            $builder = $builder->orderBy($this->idcolumn);
        }

        $results = $builder->get();

        return $results;
    }

    private function setArguments()
    {
        $this->table        = $this->argument('table');
        $this->urlcolumn    = $this->argument('urlcolumn');
        $this->linkedmodel  = $this->argument('linkedmodel');
        $this->idcolumn     = $this->argument('idcolumn');
        $this->ordercolumn  = $this->argument('ordercolumn');
        $this->localecolumn = $this->argument('localecolumn');
    }

    private function handleResetFlag($orderedResults)
    {
        if ($this->option('reset') && ! $this->option('dry')) {
            $this->info('Resetting the assets on the models');
            $resetbar = $this->output->createProgressBar(count($orderedResults));

            $orderedResults->each(function ($entry) use ($resetbar) {
                $entry['model']->deleteAllAssets();
                $resetbar->advance();
            });

            $resetbar->finish();
        }
    }

    private function mapResults($results)
    {
        return $results->map(function ($result) {
            $formattedResults = [];

            $formattedResults['images'][] = $result->{$this->urlcolumn};
            $formattedResults['model']    = $this->linkedmodel::find($result->{$this->idcolumn});

            if ($this->ordercolumn) {
                $formattedResults['order'] = $result->{$this->ordercolumn};
            }

            return $formattedResults;
        })->reject(function ($value) {
            if ($result = $value['model'] == null) {
                $this->nomodel++;
            }

            return $result;
        });
    }
}