support __MODULE__ and attributes in replace remote function code action

Author: lukaszsamson

apps/language_server/lib/language_server/providers/code_action/replace_remote_function.ex

@@ -119,7 +119,8 @@ defmodule ElixirLS.LanguageServer.Providers.CodeAction.ReplaceRemoteFunction do
 
       ast
       |> Macro.postwalk(fn
-        {:., function_meta, [{:__aliases__, module_meta, module_alias}, ^function_atom]} ->
+        {:., function_meta, [{:__aliases__, module_meta, module_alias = [h | _]}, ^function_atom]}
+        when is_atom(h) ->
           case expand_alias(source_file, module_alias, line_number) do
             {:ok, ^module} ->
               {:., function_meta, [{:__aliases__, module_meta, module_alias}, suggestion]}
@@ -128,9 +129,8 @@ defmodule ElixirLS.LanguageServer.Providers.CodeAction.ReplaceRemoteFunction do
               {:., function_meta, [{:__aliases__, module_meta, module_alias}, function_atom]}
           end
 
-        # erlang call
-        {:., function_meta, [^module, ^function_atom]} ->
-          {:., function_meta, [module, suggestion]}
+        {:., function_meta, [remote, ^function_atom]} ->
+          {:., function_meta, [remote, suggestion]}
 
         other ->
           other

apps/language_server/test/providers/code_action/replace_remote_function_test.exs

@@ -257,6 +257,36 @@ defmodule ElixirLS.LanguageServer.Providers.CodeAction.ReplaceRemoteFunctionTest
 
         assert result == "alias ElixirLS.Test.RemoteFunction, as: Remote\nRemote.foo(42)"
       end
+
+      test "handles __MODULE__" do
+        message = """
+        ElixirLS.Test.RemoteFunction.fou/1 is undefined or private. Did you mean:
+              * foo/1
+        """
+
+        {:ok, [result]} =
+          ~q{
+          __MODULE__.fou(42)
+          }
+          |> modify(message: message, suggestion: "__MODULE__.foo", line: 0)
+
+        assert result == "__MODULE__.foo(42)"
+      end
+
+      test "handles __MODULE__.Submodule alias" do
+        message = """
+        ElixirLS.Test.RemoteFunction.fou/1 is undefined or private. Did you mean:
+              * foo/1
+        """
+
+        {:ok, [result]} =
+          ~q{
+          __MODULE__.RemoteFunction.fou(42)
+          }
+          |> modify(message: message, suggestion: "__MODULE__.RemoteFunction.foo", line: 0)
+
+        assert result == "__MODULE__.RemoteFunction.foo(42)"
+      end
     end
   end