fix: clean args and properly scope rollback task

closes #592

Author: zachdaniel

lib/data_layer.ex

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

lib/mix/helpers.ex

@@ -119,7 +119,7 @@ defmodule AshPostgres.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
@@ -129,7 +129,7 @@ defmodule AshPostgres.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