BookStack Code Mirror - bookstack/blob - app/Console/Commands/UpdateUrlCommand.php

   1 <?php
   2
   3 namespace BookStack\Console\Commands;
   4
   5 use Illuminate\Console\Command;
   6 use Illuminate\Database\Connection;
   7
   8 class UpdateUrlCommand extends Command
   9 {
  10     /**
  11      * The name and signature of the console command.
  12      *
  13      * @var string
  14      */
  15     protected $signature = 'bookstack:update-url
  16                             {oldUrl : URL to replace}
  17                             {newUrl : URL to use as the replacement}
  18                             {--force : Force the operation to run, ignoring confirmations}';
  19
  20     /**
  21      * The console command description.
  22      *
  23      * @var string
  24      */
  25     protected $description = 'Find and replace the given URLs in your BookStack database';
  26
  27     /**
  28      * Execute the console command.
  29      */
  30     public function handle(Connection $db): int
  31     {
  32         $oldUrl = str_replace("'", '', $this->argument('oldUrl'));
  33         $newUrl = str_replace("'", '', $this->argument('newUrl'));
  34
  35         $urlPattern = '/https?:\/\/(.+)/';
  36         if (!preg_match($urlPattern, $oldUrl) || !preg_match($urlPattern, $newUrl)) {
  37             $this->error('The given urls are expected to be full urls starting with http:// or https://');
  38
  39             return 1;
  40         }
  41
  42         if (!$this->checkUserOkayToProceed($oldUrl, $newUrl)) {
  43             return 1;
  44         }
  45
  46         $columnsToUpdateByTable = [
  47             'attachments' => ['path'],
  48             'pages'       => ['html', 'text', 'markdown'],
  49             'chapters'    => ['description_html'],
  50             'books'       => ['description_html'],
  51             'bookshelves' => ['description_html'],
  52             'images'      => ['url'],
  53             'settings'    => ['value'],
  54             'comments'    => ['html', 'text'],
  55         ];
  56
  57         foreach ($columnsToUpdateByTable as $table => $columns) {
  58             foreach ($columns as $column) {
  59                 $changeCount = $this->replaceValueInTable($db, $table, $column, $oldUrl, $newUrl);
  60                 $this->info("Updated {$changeCount} rows in {$table}->{$column}");
  61             }
  62         }
  63
  64         $jsonColumnsToUpdateByTable = [
  65             'settings' => ['value'],
  66         ];
  67
  68         foreach ($jsonColumnsToUpdateByTable as $table => $columns) {
  69             foreach ($columns as $column) {
  70                 $oldJson = trim(json_encode($oldUrl), '"');
  71                 $newJson = trim(json_encode($newUrl), '"');
  72                 $changeCount = $this->replaceValueInTable($db, $table, $column, $oldJson, $newJson);
  73                 $this->info("Updated {$changeCount} JSON encoded rows in {$table}->{$column}");
  74             }
  75         }
  76
  77         $this->info('URL update procedure complete.');
  78         $this->info('============================================================================');
  79         $this->info('Be sure to run "php artisan cache:clear" to clear any old URLs in the cache.');
  80         $this->info('============================================================================');
  81
  82         return 0;
  83     }
  84
  85     /**
  86      * Perform a find+replace operations in the provided table and column.
  87      * Returns the count of rows changed.
  88      */
  89     protected function replaceValueInTable(
  90         Connection $db,
  91         string $table,
  92         string $column,
  93         string $oldUrl,
  94         string $newUrl
  95     ): int {
  96         $oldQuoted = $db->getPdo()->quote($oldUrl);
  97         $newQuoted = $db->getPdo()->quote($newUrl);
  98
  99         return $db->table($table)->update([
 100             $column => $db->raw("REPLACE({$column}, {$oldQuoted}, {$newQuoted})"),
 101         ]);
 102     }
 103
 104     /**
 105      * Warn the user of the dangers of this operation.
 106      * Returns a boolean indicating if they've accepted the warnings.
 107      */
 108     protected function checkUserOkayToProceed(string $oldUrl, string $newUrl): bool
 109     {
 110         if ($this->option('force')) {
 111             return true;
 112         }
 113
 114         $dangerWarning = "This will search for \"{$oldUrl}\" in your database and replace it with  \"{$newUrl}\".\n";
 115         $dangerWarning .= 'Are you sure you want to proceed?';
 116         $backupConfirmation = 'This operation could cause issues if used incorrectly. Have you made a backup of your existing database?';
 117
 118         return $this->confirm($dangerWarning) && $this->confirm($backupConfirmation);
 119     }
 120 }