fix: connect variable to usage inside when guard

Author: gp-pereira

lib/refactorex/dataflow.ex

@@ -71,9 +71,18 @@ defmodule Refactorex.Dataflow do
       {:test, _, scope} ->
         analyze_sealed_scope(dataflow, scope)
 
+      {:->, _, [[{:when, _, [left, guard]}], right]} ->
+        analyze_scope(dataflow, left, [guard, right])
+
       {:->, _, [left, right]} ->
         analyze_scope(dataflow, left, right)
 
+      {:<-, _, [{:when, _, [left, guard]}, right]} ->
+        dataflow
+        |> recursive_analyze(right)
+        |> add_commands(gen_commands(left))
+        |> recursive_analyze(guard)
+
       {id, _, [left, right]} when id in ~w(= <-)a ->
         dataflow
         |> recursive_analyze(right)

test/dataflow_test.exs

@@ -119,6 +119,33 @@ defmodule Refactorex.DataflowTest do
              )
   end
 
+  test "lists all variables reused by guards" do
+    assert %{
+             variable_at(:arg, 2, 3) => [
+               variable_at(:arg, 10, 17),
+               variable_at(:arg, 7, 16),
+               variable_at(:arg, 6, 17)
+             ]
+           } =
+             Dataflow.analyze(
+               """
+               defp bar() do
+                 arg = 10
+
+                 1..30
+                 |> Enum.map(fn
+                   i when i <= arg -> i
+                   i when i > arg -> i + 10
+                 end)
+
+                with true when arg > 10 <- true do
+                end
+               end
+               """
+               |> Sourceror.parse_string!()
+             )
+  end
+
   test "lists all variables after a reassignment" do
     assert %{
              variable_at(:arg, 1, 9) => [