fix: Map/Keyword.merge([one: :key]) with explicit list

Author: novaugust

lib/style/single_node.ex

@@ -176,6 +176,11 @@ defmodule Styler.Style.SingleNode do
     defp style({:|>, pm, [lhs, {{:., dm, [{_, _, [unquote(m)]} = module, :merge]}, m, [[{key, value}]]}]}),
       do: {:|>, pm, [lhs, {{:., dm, [module, :put]}, m, [key, value]}]}
 
+    # lhs |> Map.merge([key: value]) => lhs |> Map.put(:key, value)
+    defp style(
+           {:|>, pm, [lhs, {{:., dm, [{_, _, [unquote(m)]} = module, :merge]}, m, [{:__block__, _, [[{key, value}]]}]}]}
+         ), do: {:|>, pm, [lhs, {{:., dm, [module, :put]}, m, [key, value]}]}
+
     # Map.merge(foo, %{one_key: :bar}) => Map.put(foo, :one_key, :bar)
     defp style({{:., dm, [{_, _, [unquote(m)]} = module, :merge]}, m, [lhs, {:%{}, _, [{key, value}]}]}),
       do: {{:., dm, [module, :put]}, m, [lhs, key, value]}
@@ -184,6 +189,10 @@ defmodule Styler.Style.SingleNode do
     defp style({{:., dm, [{_, _, [unquote(m)]} = module, :merge]}, m, [lhs, [{key, value}]]}),
       do: {{:., dm, [module, :put]}, m, [lhs, key, value]}
 
+    # Map.merge(foo, [one_key: :bar]) => Map.put(foo, :one_key, :bar)
+    defp style({{:., dm, [{_, _, [unquote(m)]} = module, :merge]}, m, [lhs, {:__block__, _, [[{key, value}]]}]}),
+      do: {{:., dm, [module, :put]}, m, [lhs, key, value]}
+
     # (lhs |>) Map.drop([key]) => Map.delete(key)
     defp style({{:., dm, [{_, _, [unquote(m)]} = module, :drop]}, m, [{:__block__, _, [[{op, _, _} = key]]}]})
          when op != :|, do: {{:., dm, [module, :delete]}, m, [key]}

test/style/single_node_test.exs

@@ -95,6 +95,12 @@ defmodule Styler.Style.SingleNodeTest do
     test "in a pipe" do
       for module <- ~w(Map Keyword) do
         assert_style("foo |> #{module}.merge(%{one_key: :bar}) |> bop()", "foo |> #{module}.put(:one_key, :bar) |> bop()")
+        assert_style("foo |> #{module}.merge([one_key: :bar]) |> bop()", "foo |> #{module}.put(:one_key, :bar) |> bop()")
+        assert_style("foo |> #{module}.merge(one_key: :bar) |> bop()", "foo |> #{module}.put(:one_key, :bar) |> bop()")
+
+        assert_style("#{module}.merge(foo, %{one_key: :bar})", "#{module}.put(foo, :one_key, :bar)")
+        assert_style("#{module}.merge(foo, [one_key: :bar])", "#{module}.put(foo, :one_key, :bar)")
+        assert_style("#{module}.merge(foo, one_key: :bar)", "#{module}.put(foo, :one_key, :bar)")
       end
     end