From adbf009ffa8c5347d01b0dbdc418670192c8dec1 Mon Sep 17 00:00:00 2001
From: Mitchell Hanberg <mitch@mitchellhanberg.com>
Date: Mon, 3 Jul 2023 07:58:24 -0400
Subject: [PATCH] feat(definition): remote function definition (#80)

---
 test/next_ls/symbol_table_test.exs | 11 +++++-
 test/next_ls_test.exs              | 61 +++++++++++++++++++++++++++---
 2 files changed, 66 insertions(+), 6 deletions(-)

diff --git a/test/next_ls/symbol_table_test.exs b/test/next_ls/symbol_table_test.exs
index ed1afbc4..65c5b71d 100644
--- a/test/next_ls/symbol_table_test.exs
+++ b/test/next_ls/symbol_table_test.exs
@@ -6,7 +6,16 @@ defmodule NextLS.SymbolTableTest do
 
   setup %{tmp_dir: dir} do
     File.mkdir_p!(dir)
-    pid = start_supervised!({SymbolTable, [path: dir]})
+
+    # this fails with `{:error, incompatible_arguments}` on CI a lot, and I have no idea why
+    pid =
+      try do
+        start_supervised!({SymbolTable, [path: dir]})
+      rescue
+        _ ->
+          Process.sleep(250)
+          start_supervised!({SymbolTable, [path: dir]})
+      end
 
     Process.link(pid)
     [pid: pid, dir: dir]
diff --git a/test/next_ls_test.exs b/test/next_ls_test.exs
index be0e89b3..ecae3fa4 100644
--- a/test/next_ls_test.exs
+++ b/test/next_ls_test.exs
@@ -468,6 +468,16 @@ defmodule NextLSTest do
 
   describe "two" do
     setup %{cwd: cwd} do
+      remote = Path.join(cwd, "lib/remote.ex")
+
+      File.write!(remote, """
+      defmodule Remote do
+        def bang!() do
+          "‼️"
+        end
+      end
+      """)
+
       imported = Path.join(cwd, "lib/imported.ex")
 
       File.write!(imported, """
@@ -484,6 +494,7 @@ defmodule NextLSTest do
       defmodule Foo do
         import Imported
         def run() do
+          Remote.bang!()
           process()
         end
 
@@ -494,7 +505,7 @@ defmodule NextLSTest do
       end
       """)
 
-      [bar: bar, imported: imported]
+      [bar: bar, imported: imported, remote: remote]
     end
 
     setup :with_lsp
@@ -517,7 +528,7 @@ defmodule NextLSTest do
         id: 4,
         jsonrpc: "2.0",
         params: %{
-          position: %{line: 3, character: 6},
+          position: %{line: 4, character: 6},
           textDocument: %{uri: uri}
         }
       })
@@ -525,11 +536,11 @@ defmodule NextLSTest do
       assert_result 4, %{
         "range" => %{
           "start" => %{
-            "line" => 6,
+            "line" => 7,
             "character" => 0
           },
           "end" => %{
-            "line" => 6,
+            "line" => 7,
             "character" => 0
           }
         },
@@ -555,7 +566,7 @@ defmodule NextLSTest do
         id: 4,
         jsonrpc: "2.0",
         params: %{
-          position: %{line: 7, character: 5},
+          position: %{line: 8, character: 5},
           textDocument: %{uri: uri}
         }
       })
@@ -576,6 +587,46 @@ defmodule NextLSTest do
         "uri" => ^uri
       }
     end
+
+    test "go to remote function definition", %{client: client, bar: bar, remote: remote} do
+      assert :ok ==
+               notify(client, %{
+                 method: "initialized",
+                 jsonrpc: "2.0",
+                 params: %{}
+               })
+
+      assert_notification "window/logMessage", %{"message" => "[NextLS] Runtime ready..."}
+      assert_notification "window/logMessage", %{"message" => "[NextLS] Compiled!"}
+
+      uri = uri(bar)
+
+      request(client, %{
+        method: "textDocument/definition",
+        id: 4,
+        jsonrpc: "2.0",
+        params: %{
+          position: %{line: 3, character: 12},
+          textDocument: %{uri: uri}
+        }
+      })
+
+      uri = uri(remote)
+
+      assert_result 4, %{
+        "range" => %{
+          "start" => %{
+            "line" => 1,
+            "character" => 0
+          },
+          "end" => %{
+            "line" => 1,
+            "character" => 0
+          }
+        },
+        "uri" => ^uri
+      }
+    end
   end
 
   defp with_lsp(%{tmp_dir: tmp_dir}) do