Add lfe compiler

src/rebar.app.src

@@ -31,6 +31,7 @@
                      rebar_prv_install_deps,
                      rebar_prv_packages,
                      rebar_prv_erlydtl_compiler,
+                     rebar_prv_lfe_compiler,
                      rebar_prv_compile,
                      rebar_prv_app_discovery,
                      rebar_prv_shell,

src/rebar_prv_lfe_compiler.erl

@@ -0,0 +1,129 @@
+%% -*- erlang-indent-level: 4;indent-tabs-mode: nil -*-
+%% ex: ts=4 sw=4 et
+%% -------------------------------------------------------------------
+%%
+%% rebar: Erlang Build Tools
+%%
+%% Copyright (c) 2009 Dave Smith ([email protected]),
+%%                    Tim Dysinger ([email protected])
+%%                    Duncan McGreggor ([email protected])
+%%
+%% Permission is hereby granted, free of charge, to any person obtaining a copy
+%% of this software and associated documentation files (the "Software"), to deal
+%% in the Software without restriction, including without limitation the rights
+%% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+%% copies of the Software, and to permit persons to whom the Software is
+%% furnished to do so, subject to the following conditions:
+%%
+%% The above copyright notice and this permission notice shall be included in
+%% all copies or substantial portions of the Software.
+%%
+%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+%% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+%% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+%% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+%% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+%% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+%% THE SOFTWARE.
+%% -------------------------------------------------------------------
+-module(rebar_prv_lfe_compiler).
+
+-behaviour(provider).
+
+-export([init/1,
+         do/1,
+         format_error/1]).
+
+-include("rebar.hrl").
+
+-define(PROVIDER, lfecompile).
+-define(DESC, "Compile LFE source files.").
+-define(DEPS, []).
+
+%% ===================================================================
+%% Public API
+%% ===================================================================
+
+-spec init(rebar_state:t()) -> {ok, rebar_state:t()}.
+init(State) ->
+    Provider = providers:create([
+        {name, ?PROVIDER},
+        {module, ?MODULE},
+        {bare, false},
+        {deps, ?DEPS},
+        {example, "rebar lfecompile"},
+        {short_desc, ?DESC},
+        {desc, info(?DESC)},
+        {opts, [{undefined, undefined, undefined, undefined, ""}]}
+    ]),
+    State1 = rebar_state:add_provider(State, Provider),
+    {ok, State1}.
+
+do(State) ->
+    Cwd = rebar_utils:get_cwd(),
+    FirstFiles = check_files(rebar_state:get(State, lfe_first_files, [])),
+    Result = rebar_base_compiler:run(State,
+                                     FirstFiles,
+                                     filename:join(Cwd, "src"),
+                                     ".lfe",
+                                     filename:join(Cwd, "ebin"),
+                                     ".beam",
+                                     fun compile_lfe/3),
+    {Result, State}.
+
+-spec format_error(any()) ->  iolist().
+format_error(Reason) ->
+    io_lib:format("~p", [Reason]).
+
+%% ===================================================================
+%% Internal functions
+%% ===================================================================
+info(Description) ->
+    io_lib:format(
+        "~n~s~n"
+        "~n"
+        "No additional configuration options are required to compile~n"
+        "LFE (*.lfe) files. The rebar 'erl_opts' setting is reused by~n"
+        "LFE. For more information, see the rebar documentation for~n"
+        "'erl_opts'.",
+        [Description]).
+
+compile_lfe(Source, _Target, State) ->
+    case code:which(lfe_comp) of
+        non_existing ->
+            ?ERROR("~n"
+                   "*** MISSING LFE COMPILER ***~n"
+                   "  You must do one of the following:~n"
+                   "    a) Install LFE globally in your erl libs~n"
+                   "    b) Add LFE as a dep for your project, eg:~n"
+                   "       {lfe, \"0.9.0\",~n"
+                   "        {git, \"git://github.com/rvirding/lfe\",~n"
+                   "         {tag, \"v0.9.0\"}}}~n"
+                   "~n", []),
+            ?FAIL;
+        _ ->
+            ErlOpts = rebar_utils:erl_opts(State),
+            Opts = [{i, "include"}, {outdir, "ebin"}, return] ++ ErlOpts,
+            case lfe_comp:file(Source, Opts) of
+                {ok, _Mod, Ws} ->
+                    rebar_base_compiler:ok_tuple(State, Source, Ws);
+                {error, Es, Ws} ->
+                    rebar_base_compiler:error_tuple(State, Source,
+                                                    Es, Ws, Opts);
+                _ ->
+                    ?FAIL
+            end
+    end.
+
+%%
+%% Ensure all files in a list are present and abort if one is missing
+%%
+-spec check_files([file:filename()]) -> [file:filename()].
+check_files(FileList) ->
+    [check_file(F) || F <- FileList].
+
+check_file(File) ->
+    case filelib:is_regular(File) of
+        false -> ?ABORT("File ~p is missing, aborting\n", [File]);
+        true -> File
+    end.