chore: fix rollback behavior

Author: zachdaniel

lib/data_layer.ex

@@ -311,7 +311,15 @@ defmodule AshSqlite.DataLayer do
   end
 
   def rollback(args) do
-    repos = AshSqlite.Mix.Helpers.repos!([], args)
+    {opts, _, _} =
+      OptionParser.parse(args,
+        switches: [
+          repo: :string
+        ],
+        aliases: [r: :repo]
+      )
+
+    repos = AshSqlite.Mix.Helpers.repos!(opts, args)
 
     show_for_repo? = Enum.count_until(repos, 2) == 2
 

lib/mix/helpers.ex

@@ -94,7 +94,7 @@ defmodule AshSqlite.Mix.Helpers do
   def delete_flag(args, arg) do
     case Enum.split_while(args, &(&1 != arg)) do
       {left, [_ | rest]} ->
-        left ++ rest
+        delete_flag(left ++ rest, arg)
 
       _ ->
         args
@@ -104,7 +104,7 @@ defmodule AshSqlite.Mix.Helpers do
   def delete_arg(args, arg) do
     case Enum.split_while(args, &(&1 != arg)) do
       {left, [_, _ | rest]} ->
-        left ++ rest
+        delete_arg(left ++ rest, arg)
 
       _ ->
         args