Improve variable handing in key generators (#221)

cabol · Feb 24, 2024 · 18b4d26 · 18b4d26
1 parent 4071a9b
commit 18b4d26
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/lib/nebulex/caching/decorators.ex b/lib/nebulex/caching/decorators.ex
@@ -886,10 +886,16 @@ if Code.ensure_loaded?(Decorator.Define) do
       walk(ast, acc)
     end
 
-    defp walk({var, [{:line, _} | _], nil} = ast, acc) do
-      case "#{var}" do
-        "_" <> _ -> acc
-        _ -> [ast | acc]
+    defp walk({var, _meta, context} = ast, acc) when is_atom(context) and is_atom(var) do
+      cond do
+        match?("_" <> _, "#{var}") ->
+          acc
+
+        Macro.special_form?(var, 0) ->
+          acc
+
+        true ->
+          [ast | acc]
       end
     end