Update MediaCloudUrlRewrite command, add avatar support. Fixes #6069

app/Console/Commands/MediaCloudUrlRewrite.php

@@ -2,53 +2,58 @@
 
 namespace App\Console\Commands;
 
-use Illuminate\Console\Command;
+use App\Avatar;
 use App\Media;
-use Cache, Storage;
+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
-    */
+     * 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
-    */
+     * 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'
+            'newDomain' => 'The new S3 domain',
         ];
     }
+
     /**
-    * The console command description.
-    *
-    * @var string
-    */
+     * The console command description.
+     *
+     * @var string
+     */
     protected $description = 'Rewrite S3 media urls from local users';
 
     /**
-    * Execute the console command.
-    */
+     * 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!');
+        if (! (bool) config_cache('pixelfed.cloud_storage')) {
+            $this->info('Error: Cloud storage is not enabled! Please enable before proceeding.');
             $this->error('Aborting...');
             exit;
         }
@@ -64,8 +69,8 @@ class MediaCloudUrlRewrite extends Command implements PromptsForMissingInput
         $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('    Old S3: '.trim($this->argument('oldDomain')));
+        $this->info('    New S3: '.trim($this->argument('newDomain')));
         $this->info(' ');
     }
 
@@ -73,22 +78,46 @@ class MediaCloudUrlRewrite extends Command implements PromptsForMissingInput
     {
         $disk = Storage::disk(config('filesystems.cloud'))->url('test');
         $domain = parse_url($disk, PHP_URL_HOST);
-        if(trim($this->argument('newDomain')) !== $domain) {
+        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')) {
+        if (! $this->confirm('Confirm this is correct')) {
             $this->error('Aborting...');
             exit;
         }
+    }
 
-        $this->updateUrls();
+    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 updateUrls()
+    protected function updateMediaUrls()
     {
-        $this->info('Updating urls...');
+        $this->info('Updating media urls...');
         $oldDomain = trim($this->argument('oldDomain'));
         $newDomain = trim($this->argument('newDomain'));
         $disk = Storage::disk(config('filesystems.cloud'));
@@ -96,30 +125,32 @@ class MediaCloudUrlRewrite extends Command implements PromptsForMissingInput
         $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) {
+        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) {
+            if ($oldDomain != $cdnHost || $newDomain == $cdnHost) {
                 $bar->advance();
+
                 continue;
             }
 
             $media->cdn_url = str_replace($oldDomain, $newDomain, $media->cdn_url);
 
-            if($media->thumbnail_url != null) {
+            if ($media->thumbnail_url != null) {
                 $thumbHost = parse_url($media->thumbnail_url, PHP_URL_HOST);
-                if($thumbHost == $oldDomain) {
+                if ($thumbHost == $oldDomain) {
                     $thumbUrl = $disk->url($media->thumbnail_path);
                     $media->thumbnail_url = $thumbUrl;
                 }
             }
 
-            if($media->optimized_url != null) {
+            if ($media->optimized_url != null) {
                 $optiHost = parse_url($media->optimized_url, PHP_URL_HOST);
-                if($optiHost == $oldDomain) {
+                if ($optiHost == $oldDomain) {
                     $optiUrl = str_replace($oldDomain, $newDomain, $media->optimized_url);
                     $media->optimized_url = $optiUrl;
                 }
@@ -133,7 +164,46 @@ class MediaCloudUrlRewrite extends Command implements PromptsForMissingInput
         $bar->finish();
 
         $this->line(' ');
-        $this->info('Finished! Updated ' . $counter . ' total records!');
+        $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');
     }