From cf820007fcdce4b6da142321a687c523fc3749c9 Mon Sep 17 00:00:00 2001
From: Bryan Fink <bryan@basho.com>
Date: Wed, 31 Mar 2010 11:35:06 -0400
Subject: [PATCH] add handling for yecc's .yrl files to erlc compiler

---
 src/rebar_erlc_compiler.erl | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/src/rebar_erlc_compiler.erl b/src/rebar_erlc_compiler.erl
index 0320380..2684561 100644
--- a/src/rebar_erlc_compiler.erl
+++ b/src/rebar_erlc_compiler.erl
@@ -40,6 +40,9 @@
 
 -spec compile(Config::#config{}, AppFile::string()) -> 'ok'.
 compile(Config, _AppFile) ->
+    rebar_base_compiler:run(Config, rebar_config:get_list(Config, yrl_first_files, []),
+                            "src", ".yrl", "src", ".erl",
+                            fun compile_yrl/3),
     doterl_compile(Config, "ebin"),
     rebar_base_compiler:run(Config, rebar_config:get_list(Config, mib_first_files, []),
                             "mibs", ".mib", "priv/mibs", ".bin",
@@ -52,6 +55,11 @@ clean(_Config, _AppFile) ->
     %%       much slower.
     ok = rebar_file_utils:rm_rf("ebin/*.beam priv/mibs/*.bin"),
 
+    YrlFiles = rebar_utils:find_files("src", "^.*\\.yrl\$"),
+    rebar_file_utils:delete_each(
+      [ binary_to_list(iolist_to_binary(re:replace(F, "\\.yrl$", ".erl")))
+        || F <- YrlFiles  ]),
+
     %% Erlang compilation is recursive, so it's possible that we have a nested
     %% directory structure in ebin with .beam files within. As such, we want
     %% to scan whatever is left in the ebin/ directory for sub-dirs which
@@ -198,6 +206,33 @@ compile_mib(Source, Target, Config) ->
             ?FAIL
     end.
 
+-spec compile_yrl(Source::string(), Target::string(), Config::#config{}) -> 'ok'.
+compile_yrl(Source, Target, Config) ->
+    case yrl_needs_compile(Source, Target) of
+        true ->
+            Opts = [{parserfile, Target}, {return, true}
+                    |rebar_config:get(Config, yrl_opts, [])],
+            case yecc:file(Source, Opts) of
+                {ok, _, []} ->
+                    ok;
+                {ok, _, _Warnings} ->
+                    case lists:member(fail_on_warnings, Config) of
+                        true ->
+                            ?FAIL;
+                        false ->
+                            ok
+                    end;
+                _X ->
+                    ?FAIL
+            end;
+        false ->
+            skipped
+    end.
+
+-spec yrl_needs_compile(Source::string(), Target::string()) -> boolean().
+yrl_needs_compile(Source, Target) ->
+    filelib:last_modified(Target) < filelib:last_modified(Source).
+
 gather_src([], Srcs) ->
     Srcs;
 gather_src([Dir|Rest], Srcs) ->