Added support in DB::prohibitDestructiveCommands to preventing destru…

…ctive Rollback… (#54238) * Added support in DB::prohibitDestructiveCommands to prohibit RollbackCommand Signed-off-by: Rainer Bendig <hexathos@gmail.com> * fixed use of Illuminate\Console\Command; Signed-off-by: Rainer Bendig <hexathos@gmail.com> --------- Signed-off-by: Rainer Bendig <hexathos@gmail.com>
laravel · Jan 18, 2025 · fffdd53 · fffdd53
1 parent ad20dd5
commit fffdd53
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/Illuminate/Database/Console/Migrations/RollbackCommand.php b/src/Illuminate/Database/Console/Migrations/RollbackCommand.php
@@ -2,15 +2,17 @@
 
 namespace Illuminate\Database\Console\Migrations;
 
+use Illuminate\Console\Command;
 use Illuminate\Console\ConfirmableTrait;
+use Illuminate\Console\Prohibitable;
 use Illuminate\Database\Migrations\Migrator;
 use Symfony\Component\Console\Attribute\AsCommand;
 use Symfony\Component\Console\Input\InputOption;
 
 #[AsCommand('migrate:rollback')]
 class RollbackCommand extends BaseCommand
 {
-    use ConfirmableTrait;
+    use ConfirmableTrait, Prohibitable;
 
     /**
      * The console command name.
@@ -53,8 +55,9 @@ public function __construct(Migrator $migrator)
      */
     public function handle()
     {
-        if (! $this->confirmToProceed()) {
-            return 1;
+        if ($this->isProhibited() ||
+            ! $this->confirmToProceed()) {
+            return Command::FAILURE;
         }
 
         $this->migrator->usingConnection($this->option('database'), function () {

diff --git a/src/Illuminate/Support/Facades/DB.php b/src/Illuminate/Support/Facades/DB.php
@@ -5,6 +5,7 @@
 use Illuminate\Database\Console\Migrations\FreshCommand;
 use Illuminate\Database\Console\Migrations\RefreshCommand;
 use Illuminate\Database\Console\Migrations\ResetCommand;
+use Illuminate\Database\Console\Migrations\RollbackCommand;
 use Illuminate\Database\Console\WipeCommand;
 
 /**
@@ -132,6 +133,7 @@ public static function prohibitDestructiveCommands(bool $prohibit = true)
         FreshCommand::prohibit($prohibit);
         RefreshCommand::prohibit($prohibit);
         ResetCommand::prohibit($prohibit);
+        RollbackCommand::prohibit($prohibit);
         WipeCommand::prohibit($prohibit);
     }