chore: fixes for dev migrations feature

Author: zachdaniel

lib/migration_generator/migration_generator.ex

@@ -381,15 +381,64 @@ defmodule AshSqlite.MigrationGenerator do
     end
   end
 
+  if Mix.env() == :test do
+    defp with_repo_not_in_test(repo, fun) do
+      fun.(repo)
+    end
+  else
+    defp with_repo_not_in_test(repo, fun) do
+      Ecto.Migrator.with_repo(repo, fun)
+    end
+  end
+
   defp remove_dev_migrations_and_snapshots(dev_migrations, repo, opts, snapshots) do
-    # Remove dev migration files
-    Enum.each(dev_migrations, fn migration_name ->
-      opts
-      |> migration_path(repo)
-      |> Path.join(migration_name)
-      |> File.rm!()
+    dev_migrations =
+      Enum.map(dev_migrations, fn migration ->
+        opts
+        |> migration_path(repo)
+        |> Path.join(migration)
+      end)
+
+    with_repo_not_in_test(repo, fn repo ->
+      {repo, query, opts} = Ecto.Migration.SchemaMigration.versions(repo, [], nil)
+
+      repo.transaction(fn ->
+        Ecto.Migration.SchemaMigration.ensure_schema_migrations_table!(
+          repo,
+          repo.config(),
+          []
+        )
+
+        versions = repo.all(query, opts)
+
+        dev_migrations
+        |> Enum.map(&extract_migration_info/1)
+        |> Enum.filter(& &1)
+        |> Enum.map(&load_migration!/1)
+        |> Enum.sort()
+        |> Enum.filter(fn {version, _} ->
+          version in versions
+        end)
+        |> Enum.each(fn {version, mod} ->
+          Ecto.Migration.Runner.run(
+            repo,
+            [],
+            version,
+            mod,
+            :forward,
+            :down,
+            :down,
+            all: true
+          )
+
+          Ecto.Migration.SchemaMigration.down(repo, repo.config(), version, [])
+        end)
+      end)
     end)
 
+    # Remove dev migration files
+    Enum.each(dev_migrations, &File.rm!(&1))
+
     # Remove dev snapshots
     Enum.each(snapshots, fn snapshot ->
       snapshot_folder =
@@ -411,6 +460,34 @@ defmodule AshSqlite.MigrationGenerator do
     end)
   end
 
+  defp load_migration!({version, _, file}) when is_binary(file) do
+    loaded_modules = file |> compile_file() |> Enum.map(&elem(&1, 0))
+
+    if mod = Enum.find(loaded_modules, &migration?/1) do
+      {version, mod}
+    else
+      raise Ecto.MigrationError,
+            "file #{Path.relative_to_cwd(file)} does not define an Ecto.Migration"
+    end
+  end
+
+  defp compile_file(file) do
+    Code.compile_file(file)
+  end
+
+  defp migration?(mod) do
+    function_exported?(mod, :__migration__, 0)
+  end
+
+  defp extract_migration_info(file) do
+    base = Path.basename(file)
+
+    case Integer.parse(Path.rootname(base)) do
+      {integer, "_" <> name} -> {integer, name, file}
+      _ -> nil
+    end
+  end
+
   defp add_order_to_operations({snapshot, operations}) do
     operations_with_order = Enum.map(operations, &add_order_to_operation(&1, snapshot.attributes))