Don't move nested modules into custom env

turion · Dec 26, 2020 · abf7a7c · abf7a7c
1 parent 98cb354
commit abf7a7c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lib/ex_type/custom_env.ex b/lib/ex_type/custom_env.ex
@@ -101,7 +101,15 @@ defmodule ExType.CustomEnv do
   # create new module with name: ExType.Module.XXX
   defmacro defmodule(alias, do: block) do
     {:__aliases__, meta, tokens} = alias
-    new_alias = {:__aliases__, meta, [:ExType, :Module | tokens]}
+
+    # 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
+      nil -> [:ExType, :Module]
+      _ -> []
+    end
+
+    new_alias = {:__aliases__, meta, new_alias_prefix ++ tokens}
 
     quote do
       Kernel.defmodule unquote(new_alias) do