wip

Author: brendt

packages/upgrade/src/Tempest2/MigrationRector.php

@@ -3,6 +3,8 @@
 namespace Tempest\Upgrade\Tempest2;
 
 use PhpParser\Node;
+use PhpParser\Node\Name;
+use PhpParser\Node\Stmt\ClassMethod;
 use Rector\Rector\AbstractRector;
 
 final class MigrationRector extends AbstractRector
@@ -14,18 +16,74 @@ public function getNodeTypes(): array
         ];
     }
 
-    public function refactor(Node $node): ?int
+    public function refactor(Node $node): void
     {
         if (! $node instanceof Node\Stmt\Class_) {
-            return null;
+            return;
         }
 
+        // Check whether this class implements Tempest\Database\DatabaseMigration
         $implements = $node->implements;
-ld($implements);
-//        if (! in_array('', $implements)) {
 
-//        }
+        $implementsDatabaseMigration = array_find_key(
+            $implements,
+            static fn (Node\Name $name) => $name->toString() === 'Tempest\Database\DatabaseMigration',
+        );
 
-        ld($implements);
+        if ($implementsDatabaseMigration === null) {
+            return;
+        }
+
+        // Unset the old interface
+        unset($implements[$implementsDatabaseMigration]);
+
+        // Add the new MigrateUp interface
+        $implements[] = new Node\Name('\Tempest\Database\MigratesUp');
+        $node->getMethod('up')->returnType = new Name('QueryStatement');
+
+
+        // Check whether the migration has a down method implemented or not
+        $downStatements = $node->getMethod('down')->stmts;
+
+        $migratesDown = true;
+
+        foreach ($downStatements as $statement) {
+            if (! $statement instanceof Node\Stmt\Return_) {
+                continue;
+            }
+
+            if (! $statement->expr instanceof Node\Expr\ConstFetch) {
+                continue;
+            }
+
+            $migratesDown = $statement->expr->name->toString() !== 'null';
+
+            break;
+        }
+
+        if ($migratesDown) {
+            // If the migration has a down method implemented, we'll add the new MigrateDown interface
+            $implements[] = new Node\Name('\Tempest\Database\MigratesDown');
+            $node->getMethod('down')->returnType = new Name('QueryStatement');
+        } else {
+            // If the migration does not have a down method implemented, we'll remove it entirely
+            $statements = $node->stmts;
+
+            foreach ($node->stmts as $key => $statement) {
+                if (! $statement instanceof ClassMethod) {
+                    continue;
+                }
+
+                if ($statement->name->toString() !== 'down') {
+                    continue;
+                }
+
+                unset($statements[$key]);
+
+                $node->stmts = $statements;
+            }
+        }
+
+        $node->implements = $implements;
     }
 }

packages/upgrade/src/Tempest2/RemoveDatabaseMigrationImportRector.php

@@ -0,0 +1,30 @@
+<?php
+
+namespace Tempest\Upgrade\Tempest2;
+
+use PhpParser\Node;
+use PhpParser\NodeVisitor;
+use Rector\Rector\AbstractRector;
+
+final class RemoveDatabaseMigrationImportRector extends AbstractRector
+{
+    public function getNodeTypes(): array
+    {
+        return [
+            Node\UseItem::class
+        ];
+    }
+
+    public function refactor(Node $node): ?int
+    {
+        if (! $node instanceof Node\UseItem) {
+            return null;
+        }
+
+        if ($node->name->toString() === 'Tempest\Database\DatabaseMigration') {
+            return NodeVisitor::REMOVE_NODE;
+        }
+
+        return null;
+    }
+}

packages/upgrade/src/tempest2.php

@@ -5,6 +5,7 @@
 use Rector\Renaming\Rector\PropertyFetch\RenamePropertyRector;
 use Rector\Renaming\ValueObject\RenameProperty;
 use Tempest\Upgrade\Tempest2\MigrationRector;
+use Tempest\Upgrade\Tempest2\RemoveDatabaseMigrationImportRector;
 use Tempest\Upgrade\Tempest2\RemoveIdImportRector;
 
 return static function (RectorConfig $config) : void {
@@ -63,4 +64,5 @@
     ]);
 
     $config->rule(RemoveIdImportRector::class);
+    $config->rule(RemoveDatabaseMigrationImportRector::class);
 };