Add __info__ to each module

bryanjos · bryanjos · commit 2b38d06a639c · 2017-04-11T22:04:00.000-05:00
diff --git a/lib/elixir_script/translator.ex b/lib/elixir_script/translator.ex
@@ -523,15 +523,18 @@ defmodule ElixirScript.Translator do
   end
 
   defp do_translate({function, _, [{:when, _, [{name, _, _params} | _guards] }, _] } = ast, env) when function in @generator_types do
-    Def.process_function(name, [ast], %{ env | context: :generator})
+    {js_ast, _} = Def.process_function(name, [ast], %{ env | context: :generator})
+    {js_ast, env}
   end
 
   defp do_translate({function, _, [{name, _, params}, _]} = ast, env) when function in @generator_types and is_atom(params) do
-    Def.process_function(name, [ast], %{ env | context: :generator})
+    {js_ast, _} = Def.process_function(name, [ast], %{ env | context: :generator})
+    {js_ast, env}
   end
 
   defp do_translate({function, _, [{name, _, _params}, _]} = ast, env) when function in @generator_types do
-    Def.process_function(name, [ast], %{ env | context: :generator})
+    {js_ast, _} = Def.process_function(name, [ast], %{ env | context: :generator})
+    {js_ast, env}
   end
 
   defp do_translate({function, _, [{:when, _, [{name, _, _params} | _guards] }, _] } = ast, env) when function in @function_types do
diff --git a/lib/elixir_script/translator/kernel/defmodule.ex b/lib/elixir_script/translator/kernel/defmodule.ex
@@ -57,6 +57,8 @@ defmodule ElixirScript.Translator.Defmodule do
       []
     end
 
+    info_prop = [JS.property(Identifier.make_identifier("__info__"), Identifier.make_identifier("__info__"), :init, true)]
+
     body = Enum.map(body, fn(x) ->
       case x do
         %ESTree.CallExpression{} ->
@@ -69,7 +71,7 @@ defmodule ElixirScript.Translator.Defmodule do
     body = Group.inflate_groups(body)
 
     exported_object = JS.object_expression(
-      struct_prop ++
+      info_prop ++ struct_prop ++
       Enum.map(exported_functions, fn({key, _value}) ->
         JS.property(Identifier.make_identifier(key), Identifier.make_identifier(key), :init, true)
       end)
@@ -78,7 +80,7 @@ defmodule ElixirScript.Translator.Defmodule do
     exported_functions = Enum.map(exported_functions, fn({_key, value}) -> value end)
     private_functions = Enum.map(private_functions, fn({_key, value}) -> value end)
 
-    body = private_functions ++ exported_functions ++ body
+    body = private_functions ++ exported_functions ++ [make_info_function(env)] ++ body
     {body, exported_object}
   end
 
@@ -230,4 +232,36 @@ defmodule ElixirScript.Translator.Defmodule do
     JS.variable_declaration([declarator], :const)
   end
 
+  def make_info_function(env) do
+    functions = Keyword.get(env.functions, env.module, [])
+    macros = Keyword.get(env.macros, env.module, [])
+
+    info_case = quote do
+      case kind do
+        :functions ->
+          unquote(functions)
+        :macros ->
+          unquote(macros)
+        :module ->
+          unquote(env.module)
+      end
+    end
+
+    translated_case = ElixirScript.Translator.translate!(info_case, env)
+
+    declarator = JS.variable_declarator(
+      Identifier.make_identifier("__info__"),
+      JS.function_expression(
+        [JS.identifier("kind")],
+        [],
+        JS.block_statement([
+          JS.return_statement(translated_case)
+        ])
+      )
+    )
+
+    JS.variable_declaration([declarator], :const)    
+    
+  end
+
 end
diff --git a/test/translator/defmodule_test.exs b/test/translator/defmodule_test.exs
@@ -9,7 +9,9 @@ defmodule ElixirScript.Translator.Defmodule.Test do
     end
 
     js_code = """
-    const __exports = {};
+    const __exports = {
+        __info__
+    };
     """
 
     assert_translation(ex_ast, js_code)
@@ -31,19 +33,25 @@ defmodule ElixirScript.Translator.Defmodule.Test do
     end
 
     js_code = """
-         const something = Bootstrap.Core.Patterns.defmatchgen(Bootstrap.Core.Patterns.clause([], function*() {
+         const something = Bootstrap.Core.Patterns.defmatch(Bootstrap.Core.Patterns.clause([], function() {
              return ul;
          }));
 
+         const __info__ = function(kind) {
+                 return Bootstrap.Core.Patterns.defmatch(Bootstrap.Core.Patterns.clause([Symbol.for('functions')], function() {
+                     return Object.freeze([new Bootstrap.Core.Tuple(Symbol.for('something'), 0)]);
+                 }), Bootstrap.Core.Patterns.clause([Symbol.for('macros')], function() {
+                     return Object.freeze([]);
+                 }), Bootstrap.Core.Patterns.clause([Symbol.for('module')], function() {
+                     return Symbol.for('Elixir.Elephant');
+                 })).call(this, kind);
+             };
+
          const ul = '#todo-list';
 
          const something_else = Bootstrap.Core.Patterns.defmatchgen(Bootstrap.Core.Patterns.clause([], function*() {
              return null;
          }));
-
-         const __exports = {
-             something
-         };
     """
 
     assert_translation(ex_ast, js_code)
