Added withOperation method to migration

Author: andrey-helldar

src/Listeners/Listener.php

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DragonCode\LaravelDeployOperations\Listeners;
+
+use DragonCode\LaravelDeployOperations\Console\OperationsCommand;
+use DragonCode\LaravelDeployOperations\Console\RollbackCommand;
+use DragonCode\LaravelDeployOperations\Constants\Options;
+use Illuminate\Database\Migrations\Migration;
+use Illuminate\Support\Facades\Artisan;
+
+use function array_merge;
+
+abstract class Listener
+{
+    protected function withOperation(Migration $migration): ?string
+    {
+        if (method_exists($migration, 'withOperation')) {
+            return $migration->withOperation();
+        }
+
+        return null;
+    }
+
+    protected function run(string $method, string $operation): void
+    {
+        match ($method) {
+            'up'   => $this->call(OperationsCommand::class, $operation),
+            'down' => $this->call(RollbackCommand::class, $operation, ['--force' => true]),
+        };
+    }
+
+    protected function call(string $command, string $filename, array $parameters = []): void
+    {
+        Artisan::call($command, array_merge([
+            '--' . Options::Path => $filename,
+        ], $parameters));
+    }
+}

src/Listeners/MigrationEndedListener.php

@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DragonCode\LaravelDeployOperations\Listeners;
+
+use Illuminate\Database\Events\MigrationEnded;
+
+class MigrationEndedListener extends Listener
+{
+    public function handle(MigrationEnded $event): void
+    {
+        if ($operation = $this->withOperation($event->migration)) {
+            $this->run($event->method, $operation);
+        }
+    }
+}

src/ServiceProvider.php

@@ -5,6 +5,9 @@
 namespace DragonCode\LaravelDeployOperations;
 
 use DragonCode\LaravelDeployOperations\Concerns\HasAbout;
+use DragonCode\LaravelDeployOperations\Listeners\MigrationEndedListener;
+use Illuminate\Database\Events\MigrationEnded;
+use Illuminate\Support\Facades\Event;
 use Illuminate\Support\ServiceProvider as BaseServiceProvider;
 
 class ServiceProvider extends BaseServiceProvider
@@ -21,6 +24,8 @@ public function boot(): void
             $this->registerCommands();
             $this->registerMigrations();
         }
+
+        $this->registerEvents();
     }
 
     public function register(): void
@@ -42,6 +47,11 @@ protected function registerCommands(): void
         ]);
     }
 
+    protected function registerEvents(): void
+    {
+        Event::listen(MigrationEnded::class, MigrationEndedListener::class);
+    }
+
     protected function registerMigrations(): void
     {
         $this->loadMigrationsFrom(__DIR__ . '/../database/migrations');

tests/Commands/OperationsTest.php

@@ -7,6 +7,7 @@
 use DragonCode\LaravelDeployOperations\Constants\Names;
 use DragonCode\LaravelDeployOperations\Jobs\OperationJob;
 use Exception;
+use Illuminate\Database\Console\Migrations\MigrateCommand;
 use Illuminate\Support\Facades\DB;
 use Illuminate\Support\Facades\Queue;
 use Illuminate\Support\Str;
@@ -746,4 +747,35 @@ public function testSync()
         $this->assertDatabaseOperationHas($this->table, 'foo_bar');
         $this->assertDatabaseOperationDoesntLike($this->table, 'every_time');
     }
+
+    public function testViaMigrationMethod()
+    {
+        $this->loadMigrationsFrom(__DIR__ . '/../fixtures/migrations_with_operations');
+
+        $this->copyViaMigrations();
+
+        $table = 'test';
+
+        $this->artisan(Names::Install)->assertExitCode(0);
+
+        $this->assertDatabaseCount($table, 0);
+        $this->assertDatabaseCount($this->table, 0);
+        $this->assertDatabaseOperationDoesntLike($this->table, 'custom');
+        $this->assertDatabaseOperationDoesntLike($this->table, 'invoke');
+        $this->assertDatabaseOperationDoesntLike($this->table, 'up_down');
+        $this->assertDatabaseOperationDoesntLike($table, 'custom', column: 'value');
+        $this->assertDatabaseOperationDoesntLike($table, 'invoke', column: 'value');
+        $this->assertDatabaseOperationDoesntLike($table, 'up_down', column: 'value');
+
+        $this->artisan(MigrateCommand::class)->assertExitCode(0);
+
+        $this->assertDatabaseCount($table, 2);
+        $this->assertDatabaseCount($this->table, 2);
+        $this->assertDatabaseOperationDoesntLike($this->table, 'custom');
+        $this->assertDatabaseOperationHas($this->table, 'invoke');
+        $this->assertDatabaseOperationHas($this->table, 'up_down');
+        $this->assertDatabaseOperationDoesntLike($table, 'custom', column: 'value');
+        $this->assertDatabaseOperationHas($table, 'invoke', column: 'value');
+        $this->assertDatabaseOperationHas($table, 'up_down', column: 'value');
+    }
 }

tests/Concerns/Files.php

@@ -50,6 +50,14 @@ protected function copyDI(): void
         );
     }
 
+    protected function copyViaMigrations(): void
+    {
+        File::copyDirectory(
+            __DIR__ . '/../fixtures/app/via_migrations',
+            $this->targetDirectory()
+        );
+    }
+
     protected function copySuccessFailureMethod(): void
     {
         File::copy(

tests/fixtures/app/via_migrations/2025_03_31_213407_custom.php

@@ -0,0 +1,22 @@
+<?php
+
+declare(strict_types=1);
+
+use DragonCode\LaravelDeployOperations\Operation;
+use Illuminate\Database\Query\Builder;
+use Illuminate\Support\Facades\DB;
+use Tests\Concerns\Some;
+
+return new class extends Operation {
+    public function __invoke(Some $some): void
+    {
+        $this->table()->insert([
+            'value' => $some->get('custom'),
+        ]);
+    }
+
+    protected function table(): Builder
+    {
+        return DB::table('test');
+    }
+};

tests/fixtures/app/via_migrations/2025_03_31_234251_invoke.php

@@ -0,0 +1,22 @@
+<?php
+
+declare(strict_types=1);
+
+use DragonCode\LaravelDeployOperations\Operation;
+use Illuminate\Database\Query\Builder;
+use Illuminate\Support\Facades\DB;
+use Tests\Concerns\Some;
+
+return new class extends Operation {
+    public function __invoke(Some $some): void
+    {
+        $this->table()->insert([
+            'value' => $some->get('invoke'),
+        ]);
+    }
+
+    protected function table(): Builder
+    {
+        return DB::table('test');
+    }
+};

tests/fixtures/app/via_migrations/2025_03_31_234312_up_down.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+use DragonCode\LaravelDeployOperations\Operation;
+use Illuminate\Database\Query\Builder;
+use Illuminate\Support\Facades\DB;
+use Tests\Concerns\Some;
+
+return new class extends Operation {
+    public function up(Some $some): void
+    {
+        $this->table()->insert([
+            'value' => $some->get('up_down'),
+        ]);
+    }
+
+    public function down(Some $some): void
+    {
+        $this->table()
+            ->where('value', $some->get('up_down'))
+            ->delete();
+    }
+
+    protected function table(): Builder
+    {
+        return DB::table('test');
+    }
+};

tests/fixtures/migrations_with_operations/2025_03_31_213847_call_invokable.php

@@ -0,0 +1,19 @@
+<?php
+
+declare(strict_types=1);
+
+use Illuminate\Database\Migrations\Migration;
+
+return new class extends Migration {
+    public function up(): void
+    {
+        //
+    }
+
+    public function down(): void {}
+
+    public function withOperation(): string
+    {
+        return '2025_03_31_234251_invoke';
+    }
+};

tests/fixtures/migrations_with_operations/2025_03_31_213921_call_up_down.php

@@ -0,0 +1,19 @@
+<?php
+
+declare(strict_types=1);
+
+use Illuminate\Database\Migrations\Migration;
+
+return new class extends Migration {
+    public function up(): void
+    {
+        //
+    }
+
+    public function down(): void {}
+
+    public function withOperation(): string
+    {
+        return '2025_03_31_234312_up_down';
+    }
+};