175 lines
6.0 KiB
Erlang
175 lines
6.0 KiB
Erlang
%% Elixir compiler front-end to the Erlang backend.
|
|
-module(elixir_compiler).
|
|
-export([string/3, quoted/3, bootstrap/0,
|
|
file/2, file_to_path/3, eval_forms/3]).
|
|
-include("elixir.hrl").
|
|
|
|
string(Contents, File, Callback) ->
|
|
Forms = elixir:'string_to_quoted!'(Contents, 1, File, []),
|
|
quoted(Forms, File, Callback).
|
|
|
|
quoted(Forms, File, Callback) ->
|
|
Previous = get(elixir_module_binaries),
|
|
|
|
try
|
|
put(elixir_module_binaries, []),
|
|
elixir_lexical:run(File, fun(Pid) ->
|
|
Env = elixir:env_for_eval([{line, 1}, {file, File}]),
|
|
eval_forms(Forms, [], Env#{lexical_tracker := Pid}),
|
|
Callback(File, Pid)
|
|
end),
|
|
lists:reverse(get(elixir_module_binaries))
|
|
after
|
|
put(elixir_module_binaries, Previous)
|
|
end.
|
|
|
|
file(File, Callback) ->
|
|
{ok, Bin} = file:read_file(File),
|
|
string(elixir_utils:characters_to_list(Bin), File, Callback).
|
|
|
|
file_to_path(File, Dest, Callback) when is_binary(File), is_binary(Dest) ->
|
|
file(File, fun(CallbackFile, CallbackLexical) ->
|
|
_ = [binary_to_path(Mod, Dest) || Mod <- get(elixir_module_binaries)],
|
|
Callback(CallbackFile, CallbackLexical)
|
|
end).
|
|
|
|
%% Evaluates the given code through the Erlang compiler.
|
|
%% It may end-up evaluating the code if it is deemed a
|
|
%% more efficient strategy depending on the code snippet.
|
|
|
|
eval_forms(Forms, Vars, E) ->
|
|
case (?key(E, module) == nil) andalso allows_fast_compilation(Forms) of
|
|
true ->
|
|
Binding = [{Key, Value} || {_Name, _Kind, Key, Value} <- Vars],
|
|
{Result, _Binding, EE, _S} = elixir:eval_forms(Forms, Binding, E),
|
|
{Result, EE};
|
|
false ->
|
|
compile(Forms, Vars, E)
|
|
end.
|
|
|
|
compile(Quoted, Vars, E) ->
|
|
Args = list_to_tuple([V || {_, _, _, V} <- Vars]),
|
|
{Expanded, EE} = elixir_expand:expand(Quoted, E),
|
|
elixir_env:check_unused_vars(EE),
|
|
|
|
{Module, Fun, Purgeable} =
|
|
elixir_erl_compiler:spawn(fun spawned_compile/3, [Expanded, Vars, E]),
|
|
|
|
{dispatch(Module, Fun, Args, Purgeable), EE}.
|
|
|
|
spawned_compile(ExExprs, Vars, #{line := Line, file := File} = E) ->
|
|
Dict = [{{Name, Kind}, {0, Value}} || {Name, Kind, Value, _} <- Vars],
|
|
S = elixir_env:env_to_scope_with_vars(E, Dict),
|
|
{ErlExprs, _} = elixir_erl_pass:translate(ExExprs, S),
|
|
|
|
Module = retrieve_compiler_module(),
|
|
Fun = code_fun(?key(E, module)),
|
|
Forms = code_mod(Fun, ErlExprs, Line, File, Module, Vars),
|
|
|
|
{Module, Binary} = elixir_erl_compiler:noenv_forms(Forms, File, [nowarn_nomatch]),
|
|
code:load_binary(Module, "", Binary),
|
|
{Module, Fun, is_purgeable(Module, Binary)}.
|
|
|
|
dispatch(Module, Fun, Args, Purgeable) ->
|
|
Res = Module:Fun(Args),
|
|
code:delete(Module),
|
|
Purgeable andalso code:purge(Module),
|
|
return_compiler_module(Module, Purgeable),
|
|
Res.
|
|
|
|
code_fun(nil) -> '__FILE__';
|
|
code_fun(_) -> '__MODULE__'.
|
|
|
|
code_mod(Fun, Expr, Line, File, Module, Vars) when is_binary(File), is_integer(Line) ->
|
|
Tuple = {tuple, Line, [{var, Line, K} || {_, _, K, _} <- Vars]},
|
|
Relative = elixir_utils:relative_to_cwd(File),
|
|
|
|
[{attribute, Line, file, {elixir_utils:characters_to_list(Relative), 1}},
|
|
{attribute, Line, module, Module},
|
|
{attribute, Line, compile, no_auto_import},
|
|
{attribute, Line, export, [{Fun, 1}, {'__RELATIVE__', 0}]},
|
|
{function, Line, Fun, 1, [
|
|
{clause, Line, [Tuple], [], [Expr]}
|
|
]},
|
|
{function, Line, '__RELATIVE__', 0, [
|
|
{clause, Line, [], [], [elixir_erl:elixir_to_erl(Relative)]}
|
|
]}].
|
|
|
|
retrieve_compiler_module() ->
|
|
elixir_code_server:call(retrieve_compiler_module).
|
|
|
|
return_compiler_module(Module, Purgeable) ->
|
|
elixir_code_server:cast({return_compiler_module, Module, Purgeable}).
|
|
|
|
is_purgeable(Module, Binary) ->
|
|
beam_lib:chunks(Binary, [labeled_locals]) == {ok, {Module, [{labeled_locals, []}]}}.
|
|
|
|
allows_fast_compilation({'__block__', _, Exprs}) ->
|
|
lists:all(fun allows_fast_compilation/1, Exprs);
|
|
allows_fast_compilation({defmodule, _, _}) ->
|
|
true;
|
|
allows_fast_compilation(_) ->
|
|
false.
|
|
|
|
%% Bootstrapper
|
|
|
|
bootstrap() ->
|
|
{ok, _} = application:ensure_all_started(elixir),
|
|
elixir_config:put(bootstrap, true),
|
|
elixir_config:put(docs, false),
|
|
elixir_config:put(relative_paths, false),
|
|
elixir_config:put(ignore_module_conflict, true),
|
|
[bootstrap_file(File) || File <- bootstrap_main()].
|
|
|
|
bootstrap_file(File) ->
|
|
try
|
|
Lists = file(filename:absname(File), fun(_, _) -> ok end),
|
|
_ = [binary_to_path(X, "lib/elixir/ebin") || X <- Lists],
|
|
io:format("Compiled ~ts~n", [File])
|
|
catch
|
|
?WITH_STACKTRACE(Kind, Reason, Stacktrace)
|
|
io:format("~p: ~p~nstacktrace: ~p~n", [Kind, Reason, Stacktrace]),
|
|
erlang:halt(1)
|
|
end.
|
|
|
|
bootstrap_main() ->
|
|
[<<"lib/elixir/lib/kernel.ex">>,
|
|
<<"lib/elixir/lib/macro/env.ex">>,
|
|
<<"lib/elixir/lib/keyword.ex">>,
|
|
<<"lib/elixir/lib/module.ex">>,
|
|
<<"lib/elixir/lib/list.ex">>,
|
|
<<"lib/elixir/lib/macro.ex">>,
|
|
<<"lib/elixir/lib/code.ex">>,
|
|
<<"lib/elixir/lib/code/identifier.ex">>,
|
|
<<"lib/elixir/lib/module/checker.ex">>,
|
|
<<"lib/elixir/lib/module/locals_tracker.ex">>,
|
|
<<"lib/elixir/lib/module/parallel_checker.ex">>,
|
|
<<"lib/elixir/lib/kernel/typespec.ex">>,
|
|
<<"lib/elixir/lib/kernel/utils.ex">>,
|
|
<<"lib/elixir/lib/exception.ex">>,
|
|
<<"lib/elixir/lib/protocol.ex">>,
|
|
<<"lib/elixir/lib/stream/reducers.ex">>,
|
|
<<"lib/elixir/lib/enum.ex">>,
|
|
<<"lib/elixir/lib/inspect/algebra.ex">>,
|
|
<<"lib/elixir/lib/inspect.ex">>,
|
|
<<"lib/elixir/lib/access.ex">>,
|
|
<<"lib/elixir/lib/range.ex">>,
|
|
<<"lib/elixir/lib/regex.ex">>,
|
|
<<"lib/elixir/lib/string.ex">>,
|
|
<<"lib/elixir/lib/string/chars.ex">>,
|
|
<<"lib/elixir/lib/io.ex">>,
|
|
<<"lib/elixir/lib/path.ex">>,
|
|
<<"lib/elixir/lib/file.ex">>,
|
|
<<"lib/elixir/lib/system.ex">>,
|
|
<<"lib/elixir/lib/kernel/cli.ex">>,
|
|
<<"lib/elixir/lib/kernel/error_handler.ex">>,
|
|
<<"lib/elixir/lib/kernel/parallel_compiler.ex">>,
|
|
<<"lib/elixir/lib/kernel/lexical_tracker.ex">>].
|
|
|
|
binary_to_path({ModuleName, Binary}, CompilePath) ->
|
|
Path = filename:join(CompilePath, atom_to_list(ModuleName) ++ ".beam"),
|
|
case file:write_file(Path, Binary) of
|
|
ok -> Path;
|
|
{error, Reason} -> error('Elixir.File.Error':exception([{action, "write to"}, {path, Path}, {reason, Reason}]))
|
|
end.
|