From 5fe24a8c977425385deafaa0d7f84552cc2260ef Mon Sep 17 00:00:00 2001
From: Tim Dysinger <tim@dysinger.net>
Date: Fri, 18 Dec 2009 19:00:15 -1000
Subject: [PATCH] Added optional LFE compilation abilities to rebar

---
 ebin/rebar.app             |  2 ++
 src/rebar_lfe_compiler.erl | 64 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 66 insertions(+)
 create mode 100644 src/rebar_lfe_compiler.erl

diff --git a/ebin/rebar.app b/ebin/rebar.app
index f21b3b7..b447cc8 100644
--- a/ebin/rebar.app
+++ b/ebin/rebar.app
@@ -9,6 +9,7 @@
               rebar_erlc_compiler,
               rebar_eunit,
               rebar_file_utils,
+              rebar_lfe_compiler,
               rebar_log,
               rebar_otp_app,
               rebar_port_compiler,
@@ -29,6 +30,7 @@
          {modules, [
                     {app_dir, [ rebar_protobuffs_compiler,
                                 rebar_erlc_compiler,
+                                rebar_lfe_compiler,
                                 rebar_port_compiler,
                                 rebar_otp_app,
                                 rebar_ct,
diff --git a/src/rebar_lfe_compiler.erl b/src/rebar_lfe_compiler.erl
new file mode 100644
index 0000000..cc2cde5
--- /dev/null
+++ b/src/rebar_lfe_compiler.erl
@@ -0,0 +1,64 @@
+%% -------------------------------------------------------------------
+%%
+%% rebar: Erlang Build Tools
+%%
+%% Copyright (c) 2009 Dave Smith (dizzyd@dizzyd.com),
+%%                    Tim Dysinger (tim@dysinger.net)
+%%
+%% 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_lfe_compiler).
+
+-export([compile/2]).
+
+-include("rebar.hrl").
+
+%% ===================================================================
+%% Public API
+%% ===================================================================
+
+compile(Config, _AppFile) ->
+    FirstFiles = rebar_config:get_list(Config, lfe_first_files, []),
+    rebar_erlc_compiler:do_compile(Config, "src/*.lfe", "ebin", ".lfe", ".beam",
+                                   fun compile_lfe/2, FirstFiles).
+
+
+%% ===================================================================
+%% Internal functions
+%% ===================================================================
+
+compile_lfe(Source, Config) ->
+    case code:which(lfe_comp) of
+        non_existing ->
+            ?CONSOLE("~n===============================================~n" ++
+                     " You need to install LFE to compile LFE source~n" ++
+                     "Download the latest tarball release from github~n" ++
+                     "   http://github.com/rvirding/lfe/downloads~n" ++
+                     "  and install it into your erlang library dir~n" ++
+                     "===============================================~n~n", []),
+            ?FAIL;
+        _ ->
+            Opts = [{i, "include"}, {outdir, "ebin"}, report, return] ++
+                rebar_config:get_list(Config, lfe_opts, []),
+            case lfe_comp:file(Source,Opts) of
+                {ok, _, []} -> ok;
+                _ -> ?FAIL
+            end
+    end.