db:truncate with tables argument and foreigen key checks option

Author: Sairahcaz

src/Commands/DBTruncateCommand.php

@@ -11,27 +11,46 @@ class DBTruncateCommand extends Command
 {
     use ConfirmableTrait;
 
-    public $signature = 'db:truncate';
+    protected $signature = 'db:truncate {tables?* : Optionally specify only specific tables to truncate}
+               {--force : Force the operation to run when in production}
+               {--checks=true : Enable or disable foreign key checks during truncation}';
 
-    public $description = 'Truncate all tables';
+    protected $description = 'Truncate all tables';
 
     public function handle(): int
     {
         if (! $this->confirmToProceed()) {
             return 1;
         }
 
-        $this->components->info('Preparing database.');
+        $foreignKeyChecks = $this->hasOption('checks') && filter_var($this->option('checks'), FILTER_VALIDATE_BOOLEAN);
 
-        collect(Schema::getAllTables())->each(function ($tableDefinition) {
-            $table = current($tableDefinition);
+        if (!$foreignKeyChecks) {
+            $this->components->warn('Disabling foreign key checks!');
+            Schema::disableForeignKeyConstraints();
+        }
+
+        $this->components->info('Start truncating tables.');
+
+        $tables = $this->argument('tables') ?: collect(Schema::getAllTables())->map(function ($tableDefinition) {
+            return current($tableDefinition);
+        });
+
+        collect($tables)->each(function ($table) {
             $this->components->task("Truncating table: $table", function () use ($table) {
                 DB::table($table)->truncate();
             });
         });
 
         $this->newLine();
 
+        $this->components->info('Finished truncating tables.');
+
+        if (!$foreignKeyChecks) {
+            $this->components->warn('Re-enabling foreign key checks!');
+            Schema::enableForeignKeyConstraints();
+        }
+
         return self::SUCCESS;
     }
 }

src/LaravelUsefulAdditionsServiceProvider.php

@@ -18,6 +18,6 @@ public function configurePackage(Package $package): void
         $package
             ->name('laravel-useful-additions')
             ->hasConfigFile()
-            ->hasCommand(DBTruncateCommand::class);;
+            ->hasCommand(DBTruncateCommand::class);
     }
 }