Spatie\MediaLibrary\Commands\RegenerateCommand

Complexity

Total Complexity

10

Size/Duplication

Total Lines	95
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes	2
Bugs	0	Features	1

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	7	1
B	handle()	0	22	4
B	getMediaToBeRegenerated()	0	20	5

<?php

namespace Spatie\MediaLibrary\Commands;

use Exception;
use Illuminate\Console\Command;
use Illuminate\Console\ConfirmableTrait;
use Illuminate\Support\Collection;
use Spatie\{
    MediaLibrary\FileManipulator,
    MediaLibrary\Media,
    MediaLibrary\MediaRepository
};

class RegenerateCommand extends Command
{
    use ConfirmableTrait;

    /**
     * The console command name.
     *
     * @var string
     */
    protected $signature = 'medialibrary:regenerate {modelType?} {--ids=*}
    {-- force : Force the operation to run when in production}';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Regenerate the derived images of media';

    /**
     * @var \Spatie\MediaLibrary\MediaRepository
     */
    protected $mediaRepository;

    /**
     * @var \Spatie\MediaLibrary\FileManipulator
     */
    protected $fileManipulator;

    /**
     * @var array
     */
    protected $erroredMediaIds = [];

    /**
     * RegenerateCommand constructor.
     *
     * @param MediaRepository $mediaRepository
     * @param FileManipulator $fileManipulator
     */
    public function __construct(MediaRepository $mediaRepository, FileManipulator $fileManipulator)
    {
        parent::__construct();

        $this->mediaRepository = $mediaRepository;
        $this->fileManipulator = $fileManipulator;
    }

    /**
     * Handle regeneration.
     */
    public function handle()
    {
        if (!$this->confirmToProceed()) {
            return;
        }

        $this->getMediaToBeRegenerated()->each(function (Media $media) {
            try {
                $this->fileManipulator->createDerivedFiles($media);
                $this->info("Media {$media->id} regenerated");
            } catch (Exception $exception) {
                $this->error("Media {$media->id} could not be regenerated because `{$exception->getMessage()}`");
                $this->erroredMediaIds[] = $media->id;
            }
        });

        if (count($this->erroredMediaIds)) {
            $this->warn('The derived files of these media ids could not be regenerated: ' . implode(',', $this->erroredMediaIds));
        }

        $this->info('All done!');
    }

    public function getMediaToBeRegenerated(): Collection
    {
        $modelType = $this->argument('modelType') ?? '';
        $mediaIds = $this->option('ids');

        if ($modelType === '' && !$mediaIds) {
            return $this->mediaRepository->all();
        }

        if ($mediaIds) {

            if (!is_array($mediaIds)) {
                $mediaIds = explode(',', $mediaIds);
            }

            return $this->mediaRepository->getByIds($mediaIds);
        }

        return $this->mediaRepository->getByModelType($modelType);

It seems like $modelType defined by $this->argument('modelType') ?? '' on line 91 can also be of type array; however, Spatie\MediaLibrary\MediaRepository::getByModelType() does only seem to accept string, maybe add an additional type check?

    }
}