From 2171b6d711840b69c240bda23b25fe36101e5bf5 Mon Sep 17 00:00:00 2001
From: Mane Darbinyan <mane.darbinyann@gmail.com>
Date: Wed, 18 Dec 2024 12:39:03 +0400
Subject: [PATCH] Fix phantom migration rollback failures caused by
 cross-branch acronyms

---
 .../patches/migration_context.rb              |  7 +++++
 test/rake_task_test.rb                        | 27 +++++++++++++++++++
 test/support/test_utils.rb                    |  6 +++++
 3 files changed, 40 insertions(+)

diff --git a/lib/actual_db_schema/patches/migration_context.rb b/lib/actual_db_schema/patches/migration_context.rb
index 6db97c1..df35bad 100644
--- a/lib/actual_db_schema/patches/migration_context.rb
+++ b/lib/actual_db_schema/patches/migration_context.rb
@@ -79,6 +79,8 @@ def show_info_for(migration)
       end
 
       def migrate(migration)
+        migration.name = extract_class_name(migration.filename)
+
         message = "[ActualDbSchema] Rolling back phantom migration #{migration.version} #{migration.name} " \
             "(from branch: #{branch_for(migration.version.to_s)})"
         puts colorize(message, :gray)
@@ -89,6 +91,11 @@ def migrate(migration)
         File.delete(migration.filename)
       end
 
+      def extract_class_name(filename)
+        content = File.read(filename)
+        content.match(/^class\s+([A-Za-z0-9_]+)\s+</)[1]
+      end
+
       def branch_for(version)
         metadata.fetch(version, {})[:branch] || "unknown"
       end
diff --git a/test/rake_task_test.rb b/test/rake_task_test.rb
index 7e9f650..12455d6 100644
--- a/test/rake_task_test.rb
+++ b/test/rake_task_test.rb
@@ -95,6 +95,33 @@ def down
         assert_equal %w[20130906111510_irreversible.rb], utils.migrated_files
       end
     end
+
+    describe "with acronyms defined" do
+      before do
+        utils.define_migration_file("20241218064344_ts360.rb", <<~RUBY)
+          class Ts360 < ActiveRecord::Migration[6.0]
+            def up
+              TestingState.up << :ts360
+            end
+
+            def down
+              TestingState.down << :ts360
+            end
+          end
+        RUBY
+      end
+
+      it "rolls back the phantom migrations without failing" do
+        utils.prepare_phantom_migrations
+        assert_equal %i[first second ts360], TestingState.up
+        assert_empty ActualDbSchema.failed
+        utils.define_acronym("TS360")
+        utils.run_migrations
+        assert_equal %i[ts360 second first], TestingState.down
+        assert_empty ActualDbSchema.failed
+        assert_empty utils.migrated_files
+      end
+    end
   end
 
   describe "db:rollback_branches:manual" do
diff --git a/test/support/test_utils.rb b/test/support/test_utils.rb
index bbd3ea8..30203cc 100644
--- a/test/support/test_utils.rb
+++ b/test/support/test_utils.rb
@@ -144,6 +144,12 @@ def branch_for(version)
     metadata.fetch(version.to_s, {})[:branch]
   end
 
+  def define_acronym(acronym)
+    ActiveSupport::Inflector.inflections(:en) do |inflect|
+      inflect.acronym acronym
+    end
+  end
+
   private
 
   def cleanup_call(prefix_name = nil)