-module(rebar_prv_compile).
|
|
|
|
-behaviour(provider).
|
|
|
|
-export([init/1,
|
|
do/1,
|
|
format_error/2,
|
|
build/2]).
|
|
|
|
-include("rebar.hrl").
|
|
|
|
-define(PROVIDER, compile).
|
|
-define(DEPS, [lock]).
|
|
|
|
-define(DEFAULT_JOBS, 3).
|
|
|
|
%% ===================================================================
|
|
%% Public API
|
|
%% ===================================================================
|
|
|
|
-spec init(rebar_state:t()) -> {ok, rebar_state:t()}.
|
|
init(State) ->
|
|
JobsHelp = io_lib:format(
|
|
"Number of concurrent workers the compiler may use. Default: ~B",
|
|
[?DEFAULT_JOBS]),
|
|
State1 = rebar_state:add_provider(State, providers:create([{name, ?PROVIDER},
|
|
{module, ?MODULE},
|
|
{bare, false},
|
|
{deps, ?DEPS},
|
|
{example, "rebar compile"},
|
|
{short_desc, "Compile apps .app.src and .erl files."},
|
|
{desc, ""},
|
|
{opts, [
|
|
{jobs, $j, "jobs", integer, JobsHelp}
|
|
]}])),
|
|
{ok, State1}.
|
|
|
|
-spec do(rebar_state:t()) -> {ok, rebar_state:t()} | {error, string()}.
|
|
do(State) ->
|
|
{ok, State1} = handle_args(State),
|
|
Jobs = rebar_state:get(State1, jobs),
|
|
|
|
ProjectApps = rebar_state:project_apps(State1),
|
|
Deps = rebar_state:get(State1, deps_to_build, []),
|
|
|
|
%% Need to allow global config vars used on deps
|
|
%% Right now no way to differeniate and just give deps a new state
|
|
lists:foreach(fun(AppInfo) ->
|
|
AppDir = rebar_app_info:dir(AppInfo),
|
|
C = rebar_config:consult(AppDir),
|
|
S = rebar_state:new(rebar_state:new(), C, AppDir),
|
|
S1 = rebar_state:set(S, jobs, Jobs),
|
|
|
|
%% Legacy hook support
|
|
run_compile_hooks(AppDir, pre_hooks, S1),
|
|
build(S1, AppInfo),
|
|
run_compile_hooks(AppDir, post_hooks, S1)
|
|
end, Deps),
|
|
|
|
%% Use the project State for building project apps
|
|
Cwd = rebar_utils:get_cwd(),
|
|
run_compile_hooks(Cwd, pre_hooks, State1),
|
|
%% Set hooks to empty so top-level hooks aren't run for each project app
|
|
State2 = rebar_state:set(rebar_state:set(State1, post_hooks, []), pre_hooks, []),
|
|
lists:foreach(fun(AppInfo) ->
|
|
AppDir = rebar_app_info:dir(AppInfo),
|
|
C = rebar_config:consult(AppDir),
|
|
S = rebar_state:new(State2, C, AppDir),
|
|
|
|
%% Legacy hook support
|
|
run_compile_hooks(AppDir, pre_hooks, S),
|
|
build(S, AppInfo),
|
|
run_compile_hooks(AppDir, post_hooks, S)
|
|
end, ProjectApps),
|
|
run_compile_hooks(Cwd, post_hooks, State1),
|
|
|
|
{ok, State1}.
|
|
|
|
-spec format_error(any(), rebar_state:t()) -> {iolist(), rebar_state:t()}.
|
|
format_error(Reason, State) ->
|
|
{io_lib:format("~p", [Reason]), State}.
|
|
|
|
build(State, AppInfo) ->
|
|
?INFO("Compiling ~s~n", [rebar_app_info:name(AppInfo)]),
|
|
rebar_erlc_compiler:compile(State, ec_cnv:to_list(rebar_app_info:dir(AppInfo))),
|
|
{ok, AppInfo1} = rebar_otp_app:compile(State, AppInfo),
|
|
AppInfo1.
|
|
|
|
%% ===================================================================
|
|
%% Internal functions
|
|
%% ===================================================================
|
|
|
|
handle_args(State) ->
|
|
{Args, _} = rebar_state:command_parsed_args(State),
|
|
Jobs = proplists:get_value(jobs, Args, ?DEFAULT_JOBS),
|
|
{ok, rebar_state:set(State, jobs, Jobs)}.
|
|
|
|
run_compile_hooks(Dir, Type, State) ->
|
|
Hooks = rebar_state:get(State, Type, []),
|
|
lists:foreach(fun({_, compile, _}=Hook) ->
|
|
apply_hook(Dir, [], Hook);
|
|
({compile, _}=Hook)->
|
|
apply_hook(Dir, [], Hook);
|
|
(_) ->
|
|
continue
|
|
end, Hooks).
|
|
|
|
apply_hook(Dir, Env, {Arch, Command, Hook}) ->
|
|
case rebar_utils:is_arch(Arch) of
|
|
true ->
|
|
apply_hook(Dir, Env, {Command, Hook});
|
|
false ->
|
|
ok
|
|
end;
|
|
apply_hook(Dir, Env, {Command, Hook}) ->
|
|
Msg = lists:flatten(io_lib:format("Hook for ~p failed!~n", [Command])),
|
|
rebar_utils:sh(Hook, [{cd, Dir}, {env, Env}, {abort_on_error, Msg}]).
|