Add ability to run migrate:single with class name instead file name

Author: MCMatters

README.md

@@ -21,4 +21,4 @@ Include the service provider within your `config/app.php` file.
 Available commands:
 
 * `php artisan db:drop-tables` — drops all tables.
-* `php artisan migrate:single {file}` — migrate single migration by filename.
+* `php artisan migrate:single {"file" or "class"}` — migrate single migration by file or class name.

src/Console/Commands/DropTables.php

@@ -5,6 +5,7 @@
 namespace McMatters\DbCommands\Console\Commands;
 
 use Illuminate\Console\Command;
+use Illuminate\Support\Facades\Config;
 use Illuminate\Support\Facades\DB;
 use Illuminate\Support\Facades\Schema;
 
@@ -30,7 +31,7 @@ class DropTables extends Command
      */
     public function handle()
     {
-        if (!$this->isForced() && !$this->isLocalEnvironment()) {
+        if (!$this->isForced() && $this->isProduction()) {
             $this->error(
                 'This action can\'t be performed in non-local environment.
                     Please use --force option if you really want to proceed.'
@@ -59,8 +60,8 @@ protected function isForced(): bool
     /**
      * @return bool
      */
-    protected function isLocalEnvironment(): bool
+    protected function isProduction(): bool
     {
-        return in_array(config('app.env'), ['production', 'live'], true);
+        return in_array(Config::get('app.env'), ['production', 'live'], true);
     }
 }

src/Console/Commands/MigrateSingle.php

@@ -4,8 +4,12 @@
 
 namespace McMatters\DbCommands\Console\Commands;
 
+use Illuminate\Contracts\Filesystem\FileNotFoundException;
 use Illuminate\Database\Console\Migrations\MigrateCommand;
+use Illuminate\Support\Arr;
+use Illuminate\Support\Str;
 use McMatters\DbCommands\Extenders\Database\Migrator;
+use RuntimeException;
 
 /**
  * Class MigrateSingle
@@ -17,7 +21,9 @@ class MigrateSingle extends MigrateCommand
     /**
      * @var string
      */
-    protected $signature = 'migrate:single {file : The file of migration to be executed.}
+    protected $signature = 'migrate:single
+        {--file : The file of migration to be executed.}
+        {--class : The class name of migration.}
         {--database= : The database connection to use.}
         {--force : Force the operation to run when in production.}
         {--pretend : Dump the SQL queries that would be run.}';
@@ -39,21 +45,72 @@ public function __construct(Migrator $migrator)
 
     /**
      * @return void
+     * @throws FileNotFoundException
+     * @throws RuntimeException
      */
     public function fire()
     {
         if (!$this->confirmToProceed()) {
             return;
         }
 
+        $this->checkRequirements();
         $this->prepareDatabase();
 
-        $file = $this->getMigrationPath().'/'.$this->argument('file');
+        $file = $this->getMigrationFile();
 
         $this->migrator->run($file, ['pretend' => $this->option('pretend')]);
 
         foreach ($this->migrator->getNotes() as $note) {
             $this->output->writeln($note);
         }
     }
+
+    /**
+     * @return string
+     * @throws FileNotFoundException
+     */
+    protected function getMigrationFile(): string
+    {
+        if ($this->hasOption('class')) {
+            $file = $this->getFileByClass();
+        } else {
+            $file = $this->getMigrationPath().'/'.$this->argument('file');
+        }
+
+        if (!$file || !file_exists($file)) {
+            throw new FileNotFoundException('File with migration not found.');
+        }
+
+        return $file;
+    }
+
+    /**
+     * @return string|null
+     */
+    protected function getFileByClass()
+    {
+        $class = $this->option('class');
+
+        if (!$class) {
+            return null;
+        }
+
+        $class = Str::snake($class);
+
+        $files = glob("{$this->getMigrationPath()}/[0-9_]*{$class}.php");
+
+        return Arr::first($files);
+    }
+
+    /**
+     * @return void
+     * @throws RuntimeException
+     */
+    protected function checkRequirements()
+    {
+        if (!$this->hasOption('file') && !$this->hasOption('class')) {
+            throw new RuntimeException('You must pass at least one argument "file" or "class"');
+        }
+    }
 }