pixelfed/app/Console/Commands/MediaCloudUrlRewrite.php

<?php

namespace App\Console\Commands;

use App\Avatar;
use App\Media;
use Illuminate\Console\Command;
use Illuminate\Contracts\Console\PromptsForMissingInput;
use Storage;

use function Laravel\Prompts\select;

class MediaCloudUrlRewrite extends Command implements PromptsForMissingInput
{
    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'media:cloud-url-rewrite {oldDomain} {newDomain}';

    /**
     * Prompt for missing input arguments using the returned questions.
     *
     * @return array
     */
    protected function promptForMissingArgumentsUsing()
    {
        return [
            'oldDomain' => 'The old S3 domain',
            'newDomain' => 'The new S3 domain',
        ];
    }

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Rewrite S3 media urls from local users';

    /**
     * Execute the console command.
     */
    public function handle()
    {
        $this->preflightCheck();
        $this->bootMessage();
        $this->confirmCloudUrl();
        $this->handleTask();
    }

    protected function preflightCheck()
    {
        if (! (bool) config_cache('pixelfed.cloud_storage')) {
            $this->info('Error: Cloud storage is not enabled! Please enable before proceeding.');
            $this->error('Aborting...');
            exit;
        }
    }

    protected function bootMessage()
    {
        $this->info('       ____  _           ______         __  ');
        $this->info('      / __ \(_)  _____  / / __/__  ____/ /  ');
        $this->info('     / /_/ / / |/_/ _ \/ / /_/ _ \/ __  /   ');
        $this->info('    / ____/ />  </  __/ / __/  __/ /_/ /    ');
        $this->info('   /_/   /_/_/|_|\___/_/_/  \___/\__,_/     ');
        $this->info(' ');
        $this->info('    Media Cloud Url Rewrite Tool');
        $this->info('    ===');
        $this->info('    Old S3: '.trim($this->argument('oldDomain')));
        $this->info('    New S3: '.trim($this->argument('newDomain')));
        $this->info(' ');
    }

    protected function confirmCloudUrl()
    {
        $disk = Storage::disk(config('filesystems.cloud'))->url('test');
        $domain = parse_url($disk, PHP_URL_HOST);
        if (trim($this->argument('newDomain')) !== $domain) {
            $this->error('Error: The new S3 domain you entered is not currently configured');
            exit;
        }

        if (! $this->confirm('Confirm this is correct')) {
            $this->error('Aborting...');
            exit;
        }
    }

    protected function handleTask()
    {
        $task = select(
            label: 'What action would you like to perform?',
            options: ['Migrate All', 'Migrate Media', 'Migrate Avatars']
        );

        switch ($task) {
            case 'Migrate All':
                $this->updateMediaUrls();
                $this->updateAvatarUrls();
                break;
            case 'Migrate Media':
                $this->updateMediaUrls();
                break;
            case 'Migrate Avatars':
                $this->updateAvatarUrls();
                break;
            default:
                $this->error('Invalid selection');

                return;
                break;
        }
    }

    protected function updateMediaUrls()
    {
        $this->info('Updating media urls...');
        $oldDomain = trim($this->argument('oldDomain'));
        $newDomain = trim($this->argument('newDomain'));
        $disk = Storage::disk(config('filesystems.cloud'));
        $count = Media::whereNotNull('cdn_url')->count();
        $bar = $this->output->createProgressBar($count);
        $counter = 0;
        $bar->start();
        foreach (Media::whereNotNull('cdn_url')->lazyById(1000, 'id') as $media) {
            if (strncmp($media->media_path, 'http', 4) === 0) {
                $bar->advance();

                continue;
            }
            $cdnHost = parse_url($media->cdn_url, PHP_URL_HOST);
            if ($oldDomain != $cdnHost || $newDomain == $cdnHost) {
                $bar->advance();

                continue;
            }

            $media->cdn_url = str_replace($oldDomain, $newDomain, $media->cdn_url);

            if ($media->thumbnail_url != null) {
                $thumbHost = parse_url($media->thumbnail_url, PHP_URL_HOST);
                if ($thumbHost == $oldDomain) {
                    $thumbUrl = $disk->url($media->thumbnail_path);
                    $media->thumbnail_url = $thumbUrl;
                }
            }

            if ($media->optimized_url != null) {
                $optiHost = parse_url($media->optimized_url, PHP_URL_HOST);
                if ($optiHost == $oldDomain) {
                    $optiUrl = str_replace($oldDomain, $newDomain, $media->optimized_url);
                    $media->optimized_url = $optiUrl;
                }
            }

            $media->save();
            $counter++;
            $bar->advance();
        }

        $bar->finish();

        $this->line(' ');
        $this->info('Finished! Updated '.$counter.' total records!');
        $this->line(' ');
        $this->info('Tip: Run `php artisan cache:clear` to purge cached urls');
    }

    protected function updateAvatarUrls()
    {
        $this->info('Updating avatar urls...');
        $oldDomain = trim($this->argument('oldDomain'));
        $newDomain = trim($this->argument('newDomain'));
        $disk = Storage::disk(config('filesystems.cloud'));
        $count = Avatar::count();
        $bar = $this->output->createProgressBar($count);
        $counter = 0;
        $bar->start();
        foreach (Avatar::lazyById(1000, 'id') as $avatar) {
            if (! $avatar->cdn_url) {
                $bar->advance();

                continue;
            }

            $cdnHost = parse_url($avatar->cdn_url, PHP_URL_HOST);
            if (strcasecmp($oldDomain, $cdnHost) !== 0 || strcasecmp($newDomain, $cdnHost) === 0) {
                $bar->advance();

                continue;
            }

            $avatar->cdn_url = str_replace($oldDomain, $newDomain, $avatar->cdn_url);

            $avatar->save();
            $counter++;
            $bar->advance();
        }

        $bar->finish();

        $this->line(' ');
        $this->info('Finished! Updated '.$counter.' total records!');
        $this->line(' ');
        $this->info('Tip: Run `php artisan cache:clear` to purge cached urls');
    }
}