elixirscript
diff --git a/‎lib/elixir_script/passes/find_used_modules.ex‎
Lines changed: 4 additions & 0 deletions b/‎lib/elixir_script/passes/find_used_modules.ex‎
Lines changed: 4 additions & 0 deletions
diff --git a/‎lib/elixir_script/state.ex‎
Lines changed: 15 additions & 0 deletions b/‎lib/elixir_script/state.ex‎
Lines changed: 15 additions & 0 deletions
@@ -97,6 +97,7 @@ defmodule ElixirScript.FindUsedModules do
     ModuleState.put_module(pid, module, %{protocol: true, impls: impls, functions: functions})
 
     Enum.each(implementations, fn {impl, info} ->
+      ModuleState.add_used_module(pid, module, impl)
       walk_module(impl, info, pid)
     end)
   end
@@ -136,6 +137,7 @@ defmodule ElixirScript.FindUsedModules do
   defp walk(form, state) when is_atom(form) and form not in [BitString, Function, PID, Port, Reference, Any, Elixir] do
     if ElixirScript.Translate.Module.is_elixir_module(form) and !ElixirScript.Translate.Module.is_js_module(form, state) do
       if ModuleState.get_module(state.pid, form) == nil do
+        ModuleState.add_used_module(state.pid, state.module, form)
         do_execute(form, state.pid)
       end
     end
@@ -165,6 +167,7 @@ defmodule ElixirScript.FindUsedModules do
   defp walk({:%, _, [module, params]}, state) do
     if ElixirScript.Translate.Module.is_elixir_module(module) and !ElixirScript.Translate.Module.is_js_module(module, state) do
       if ModuleState.get_module(state.pid, module) == nil do
+        ModuleState.add_used_module(state.pid, state.module, module)
         do_execute(module, state.pid)
       end
     end
@@ -300,6 +303,7 @@ defmodule ElixirScript.FindUsedModules do
     cond do
       ElixirScript.Translate.Module.is_elixir_module(module) ->
         if ModuleState.get_module(state.pid, module) == nil do
+          ModuleState.add_used_module(state.pid, state.module, module)
           do_execute(module, state.pid)
         end
       true ->
 
@@ -31,6 +31,21 @@ defmodule ElixirScript.State do
     end)
   end
 
+  def add_used_module(pid, module, used_module) do
+    Agent.update(pid, fn(state) ->
+      module_info = Keyword.get(state.modules, module)
+
+      used_module = Map.get(module_info, :used_module, [])
+      used_module = Enum.uniq([used_module | used_module])
+
+      module_info = Map.put(module_info, :used_module, used_module)
+      modules = Keyword.put(state.modules, module, module_info)
+
+      %{ state | modules: modules }
+    end)
+  end
+
+
   def has_used?(pid, module, func) do
     Agent.get(pid, fn(state) ->
       module_info = Keyword.get(state.modules, module)