Expand alias __MODULE__ manually. Fixes gyson#27

turion · Dec 9, 2020 · 5fbb3f2 · 5fbb3f2
1 parent 57daac5
commit 5fbb3f2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/ex_type/custom_env.ex b/lib/ex_type/custom_env.ex
@@ -102,6 +102,13 @@ defmodule ExType.CustomEnv do
   defmacro defmodule(alias, do: block) do
     {:__aliases__, meta, tokens} = alias
 
+    # Manually expand `alias __MODULE__`, see https://github.com/gyson/ex_type/issues/27
+    block = block
+      |> Macro.prewalk(fn
+          {:alias, meta_alias, [{:__MODULE__, meta_module, nil}]} -> {:alias, meta_alias, [{:__aliases__, meta_module, tokens}]}
+          token -> token
+        end)
+
     # Don't move nested modules into custom env (https://github.com/gyson/ex_type/issues/23)
     env = __CALLER__
     new_alias_prefix = case env.module do