mirror of https://github.com/pixelfed/pixelfed
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
211 lines
6.3 KiB
PHP
211 lines
6.3 KiB
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');
|
|
}
|
|
}
|