Revert "Revert "rewrite then when value is first argument of named function""

This reverts commit 3a5cd94.

Author: novaugust

lib/style/pipes.ex

@@ -144,6 +144,10 @@ defmodule Styler.Style.Pipes do
   defp fix_pipe({:|>, m, [lhs, {fun, m2, nil}]}), do: {:|>, m, [lhs, {fun, m2, []}]}
   # a |> then(&fun/1) |> c => a |> fun() |> c()
   defp fix_pipe({:|>, m, [lhs, {:then, _, [{:&, _, [{:/, _, [{fun, m2, _}, _]}]}]}]}), do: {:|>, m, [lhs, {fun, m2, []}]}
+  # a |> then(&fun(&1, d)) |> c => a |> fun(d) |> c()
+  defp fix_pipe({:|>, m, [lhs, {:then, _, [{:&, _, [{fun, m2, [{:&, _, _} | args]}]}]}]}),
+    do: {:|>, m, [lhs, {fun, m2, args}]}
+
   # Credo.Check.Readability.PipeIntoAnonymousFunctions
   # rewrite anonymous function invocation to use `then/2`
   # `a |> (& &1).() |> c()` => `a |> then(& &1) |> c()`

test/style/pipes_test.exs

@@ -502,7 +502,9 @@ defmodule Styler.Style.PipesTest do
       assert_style("a |> then(&fun/1) |> c", "a |> fun() |> c()")
       assert_style("a |> then(&DateTime.from_is8601/1) |> c", "a |> DateTime.from_is8601() |> c()")
       assert_style("a |> then(&DateTime.from_is8601/1)", "DateTime.from_is8601(a)")
-      assert_style("a |> then(&fun(&1, d)) |> c", "a |> then(&fun(&1, d)) |> c()")
+      assert_style("a |> then(&fun(&1)) |> c", "a |> fun() |> c()")
+      assert_style("a |> then(&fun(&1, d)) |> c", "a |> fun(d) |> c()")
+      assert_style("a |> then(&fun(d, &1)) |> c", "a |> then(&fun(d, &1)) |> c()")
     end
 
     test "adds parens to 1-arity pipes" do