Merge pull request elixir-editors#58 from mattbaker/find-refs-in-umbrella

Support Find References in umbrella apps

Author: JakeBecker

apps/language_server/lib/language_server/build.ex

@@ -8,21 +8,23 @@ defmodule ElixirLS.LanguageServer.Build do
       {nil, nil}
     else
       spawn_monitor(fn ->
-        {us, _} =
-          :timer.tc(fn ->
-            IO.puts("Compiling with Mix env #{Mix.env()}")
-
-            case reload_project() do
-              {:ok, mixfile_diagnostics} ->
-                {status, diagnostics} = compile()
-                Server.build_finished(parent, {status, mixfile_diagnostics ++ diagnostics})
-
-              {:error, mixfile_diagnostics} ->
-                Server.build_finished(parent, {:error, mixfile_diagnostics})
-            end
-          end)
-
-        JsonRpc.log_message(:info, "Compile took #{div(us, 1000)} milliseconds")
+        with_build_lock(fn ->
+          {us, _} =
+            :timer.tc(fn ->
+              IO.puts("Compiling with Mix env #{Mix.env()}")
+
+              case reload_project() do
+                {:ok, mixfile_diagnostics} ->
+                  {status, diagnostics} = compile()
+                  Server.build_finished(parent, {status, mixfile_diagnostics ++ diagnostics})
+
+                {:error, mixfile_diagnostics} ->
+                  Server.build_finished(parent, {:error, mixfile_diagnostics})
+              end
+            end)
+
+          JsonRpc.log_message(:info, "Compile took #{div(us, 1000)} milliseconds")
+        end)
       end)
     end
   end
@@ -93,6 +95,10 @@ defmodule ElixirLS.LanguageServer.Build do
     }
   end
 
+  def with_build_lock(func) do
+    :global.trans({__MODULE__, self()}, func)
+  end
+
   defp reload_project do
     mixfile = Path.absname(System.get_env("MIX_EXS") || "mix.exs")
 

apps/language_server/lib/language_server/providers/references.ex

@@ -5,13 +5,15 @@ defmodule ElixirLS.LanguageServer.Providers.References do
   any function or module identified at the provided location.
   """
 
-  alias ElixirLS.LanguageServer.SourceFile
+  alias ElixirLS.LanguageServer.{SourceFile, Build}
   alias ElixirSense.Core.{Metadata, Parser, Source, Introspection}
 
   def references(text, line, character, _include_declaration) do
-    xref_at_cursor(text, line, character)
-    |> Enum.filter(fn %{line: line} -> is_integer(line) end)
-    |> Enum.map(&build_location/1)
+    Build.with_build_lock(fn ->
+      xref_at_cursor(text, line, character)
+      |> Enum.filter(fn %{line: line} -> is_integer(line) end)
+      |> Enum.map(&build_location/1)
+    end)
   end
 
   def supported? do
@@ -48,19 +50,39 @@ defmodule ElixirLS.LanguageServer.Providers.References do
   defp add_arity({mod, fun}, %{scope: {fun, arity}, module: mod}), do: {mod, fun, arity}
   defp add_arity({mod, fun}, _env), do: {mod, fun, nil}
 
-  def callers(nil), do: []
-  def callers(mfa), do: Mix.Tasks.Xref.calls() |> Enum.filter(caller_filter(mfa))
+  defp callers(mfa) do
+    if Mix.Project.umbrella?() do
+      umbrella_calls()
+    else
+      Mix.Tasks.Xref.calls()
+    end
+    |> Enum.filter(caller_filter(mfa))
+  end
+
+  def umbrella_calls() do
+    build_dir = Path.expand(Mix.Project.config()[:build_path])
+
+    Mix.Project.apps_paths()
+    |> Enum.flat_map(fn {app, path} ->
+      Mix.Project.in_project(app, path, [build_path: build_dir], fn _ ->
+        Mix.Tasks.Xref.calls()
+        |> Enum.map(fn %{file: file} = call ->
+          Map.put(call, :file, Path.expand(file))
+        end)
+      end)
+    end)
+  end
 
   defp caller_filter({module, nil, nil}), do: &match?(%{callee: {^module, _, _}}, &1)
   defp caller_filter({module, func, nil}), do: &match?(%{callee: {^module, ^func, _}}, &1)
   defp caller_filter({module, func, arity}), do: &match?(%{callee: {^module, ^func, ^arity}}, &1)
 
-  defp build_location(call) do
+  defp build_location(%{file: file, line: line}) do
     %{
-      "uri" => SourceFile.path_to_uri(call.file),
+      "uri" => SourceFile.path_to_uri(file),
       "range" => %{
-        "start" => %{"line" => call.line - 1, "character" => 0},
-        "end" => %{"line" => call.line - 1, "character" => 0}
+        "start" => %{"line" => line - 1, "character" => 0},
+        "end" => %{"line" => line - 1, "character" => 0}
       }
     }
   end

apps/language_server/lib/language_server/server.ex

@@ -339,10 +339,13 @@ defmodule ElixirLS.LanguageServer.Server do
 
   defp handle_request(references_req(_id, uri, line, character, include_declaration), state) do
     fun = fn ->
-      {
-        :ok,
-        References.references(state.source_files[uri].text, line, character, include_declaration)
-      }
+      {:ok,
+       References.references(
+         state.source_files[uri].text,
+         line,
+         character,
+         include_declaration
+       )}
     end
 
     {:async, fun, state}

apps/language_server/test/fixtures/references/lib/a.ex

@@ -0,0 +1,5 @@
+defmodule A do
+  def fun do
+    B.fun()
+  end
+end

apps/language_server/test/fixtures/references/lib/b.ex

@@ -0,0 +1,5 @@
+defmodule B do
+  def fun do
+    :ok
+  end
+end

apps/language_server/test/fixtures/references/mix.exs

@@ -0,0 +1,11 @@
+defmodule References.MixProject do
+  use Mix.Project
+
+  def project do
+    [app: :references, version: "0.1.0"]
+  end
+
+  def application do
+    []
+  end
+end

apps/language_server/test/fixtures/umbrella/apps/app1/lib/app1.ex

@@ -0,0 +1,5 @@
+defmodule App1 do
+  def hello() do
+    App2.hello()
+  end
+end

apps/language_server/test/fixtures/umbrella/apps/app1/mix.exs

@@ -0,0 +1,15 @@
+defmodule App1.Mixfile do
+  use Mix.Project
+
+  def project do
+    [app: :app1, version: "0.1.0", deps: deps()]
+  end
+
+  def application do
+    []
+  end
+
+  defp deps do
+    [{:app2, in_umbrella: true}]
+  end
+end

apps/language_server/test/fixtures/umbrella/apps/app2/lib/app2.ex

@@ -0,0 +1,5 @@
+defmodule App2 do
+  def hello do
+    :app2
+  end
+end

apps/language_server/test/fixtures/umbrella/apps/app2/mix.exs

@@ -0,0 +1,11 @@
+defmodule App2.Mixfile do
+  use Mix.Project
+
+  def project do
+    [app: :app2, version: "0.1.0"]
+  end
+
+  def application do
+    []
+  end
+end