avoid storing anonymous function AST in CallInfo

Author: lukaszsamson

lib/elixir_sense/core/compiler/state.ex

@@ -348,6 +348,16 @@ defmodule ElixirSense.Core.Compiler.State do
         column + Keyword.get(meta, :column_correction, 0)
       end
 
+    # For anonymous function calls, func can be an AST tuple (e.g., {:fn, [], []})
+    # Replace it with nil to avoid issues in consumers
+    func =
+      case func do
+        f when is_atom(f) -> f
+        {:attribute, _} = attr -> attr
+        {:variable, _, _} = var -> var
+        _ -> nil
+      end
+
     call = %CallInfo{mod: mod, func: func, arity: arity, position: {line, column}, kind: kind}
 
     calls =

lib/elixir_sense/core/state/call_info.ex

@@ -5,8 +5,8 @@ defmodule ElixirSense.Core.State.CallInfo do
   @type t :: %ElixirSense.Core.State.CallInfo{
           arity: non_neg_integer | nil,
           position: {non_neg_integer, pos_integer | nil},
-          func: atom,
-          mod: module | {:attribute, atom},
+          func: atom | {:attribute, atom} | {:variable, atom, any} | nil,
+          mod: module | {:attribute, atom} | {:variable, atom, any} | nil,
           kind: atom
         }
   defstruct arity: 0,

test/elixir_sense/core/metadata_builder_test.exs

@@ -8954,6 +8954,24 @@ defmodule ElixirSense.Core.MetadataBuilderTest do
              } = state.calls
     end
 
+    test "registers calls capture expression anonymous" do
+      state =
+        """
+        defmodule NyModule do
+          def func do
+            (& 1 + &1).(1)
+          end
+        end
+        """
+        |> string_to_state
+
+      # The anonymous function call should have func: nil instead of a tuple
+      assert Enum.any?(
+               state.calls[3],
+               &match?(%CallInfo{arity: 1, func: nil, kind: :anonymous_function}, &1)
+             )
+    end
+
     test "registers calls on ex_unit DSL" do
       state =
         """