render overlay filenames for tar archive

2 files changed, 41 insertions, 11 deletions

diff --git a/src/rlx_prv_archive.erl b/src/rlx_prv_archive.erl
index 8ec7973..0532dc5 100644
--- a/src/rlx_prv_archive.erl
+++ b/src/rlx_prv_archive.erl
@@ -104,10 +104,13 @@ make_tar(State, Release, OutputDir) ->
     end.
 
 update_tar(State, TempDir, OutputDir, Name, Vsn, ErtsVersion) ->
+    {RelName, RelVsn} = rlx_state:default_configured_release(State),
+    Release = rlx_state:get_realized_release(State, RelName, RelVsn),
     TarFile = filename:join(OutputDir, Name++"-"++Vsn++".tar.gz"),
     file:rename(filename:join(OutputDir, Name++".tar.gz"), TarFile),
     erl_tar:extract(TarFile, [{cwd, TempDir}, compressed]),
-    OverlayFiles = overlay_files(rlx_state:get(State, overlay, undefined), OutputDir),
+    OverlayVars = rlx_prv_overlay:generate_overlay_vars(State, Release),
+    OverlayFiles = overlay_files(OverlayVars, rlx_state:get(State, overlay, undefined), OutputDir),
     ok =
         erl_tar:create(TarFile,
                        [{"lib", filename:join(TempDir, "lib")},
@@ -130,10 +133,15 @@ update_tar(State, TempDir, OutputDir, Name, Vsn, ErtsVersion) ->
     ec_file:remove(TempDir, [recursive]),
     {ok, State}.
 
-overlay_files(undefined, _) ->
+overlay_files(_, undefined, _) ->
     [];
-overlay_files(Overlay, OutputDir) ->
-    [{to(O), filename:join(OutputDir, to(O))} || O <- Overlay, filter(O)].
+overlay_files(OverlayVars, Overlay, OutputDir) ->
+    [begin
+         To = to(O),
+         ToTemplateName = rlx_prv_overlay:make_template_name("rlx_template_to_template", To),
+         File = rlx_prv_overlay:render_string(OverlayVars, To, ToTemplateName),
+         {ec_cnv:to_list(File), ec_cnv:to_list(filename:join(OutputDir, File))}
+     end || O <- Overlay, filter(O)].
 
 to({copy, _, To}) ->
     To;
diff --git a/src/rlx_prv_overlay.erl b/src/rlx_prv_overlay.erl
index f7e2be8..f14963e 100644
--- a/src/rlx_prv_overlay.erl
+++ b/src/rlx_prv_overlay.erl
@@ -28,6 +28,10 @@
          do/1,
          format_error/1]).
 
+-export([generate_overlay_vars/2,
+         make_template_name/2,
+         render_string/3]).
+
 -define(DIRECTORY_RE, ".*(\/|\\\\)$").
 
 -define(ERLYDTL_COMPILE_OPTS, [report_warnings, return_errors, {auto_escape, false}, {out_dir, false}]).
@@ -61,7 +65,12 @@ do(State) ->
     Release = rlx_state:get_realized_release(State, RelName, RelVsn),
     case rlx_release:realized(Release) of
         true ->
-            generate_overlay_vars(State, Release);
+            case generate_overlay_vars(State, Release) of
+                {error, Reason} ->
+                    {error, Reason};
+                OverlayVars ->
+                    do_overlay(State, OverlayVars)
+            end;
         false ->
             ?RLX_ERROR({unresolved_release, RelName, RelVsn})
     end.
@@ -126,20 +135,20 @@ format_errors(_, []) -> [].
 
 
 -spec generate_overlay_vars(rlx_state:t(), rlx_release:t()) ->
-                                   {ok, rlx_state:t()} | relx:error().
+                                   proplists:proplist() | relx:error().
 generate_overlay_vars(State, Release) ->
     StateVars = generate_state_vars(State),
     ReleaseVars = generate_release_vars(Release),
     get_overlay_vars_from_file(State, StateVars ++ ReleaseVars).
 
 -spec get_overlay_vars_from_file(rlx_state:t(), proplists:proplist()) ->
-                                        {ok, rlx_state:t()} | relx:error().
+                                        proplists:proplist() | relx:error().
 get_overlay_vars_from_file(State, OverlayVars) ->
     case rlx_state:get(State, overlay_vars, undefined) of
         undefined ->
-            do_overlay(State, OverlayVars);
+            OverlayVars;
         [] ->
-            do_overlay(State, OverlayVars);
+            OverlayVars;
         [H | _]=FileNames when is_list(H) ->
             read_overlay_vars(State, OverlayVars, FileNames);
         FileName when is_list(FileName) ->
@@ -147,12 +156,12 @@ get_overlay_vars_from_file(State, OverlayVars) ->
     end.
 
 -spec read_overlay_vars(rlx_state:t(), proplists:proplist(), [file:name()]) ->
-                                   {ok, rlx_state:t()} | relx:error().
+                               proplists:proplist() | relx:error().
 read_overlay_vars(State, OverlayVars, FileNames) ->
     Terms = merge_overlay_vars(State, FileNames),
     case render_overlay_vars(OverlayVars, Terms, []) of
         {ok, NewTerms} ->
-            do_overlay(State, OverlayVars ++ NewTerms);
+            OverlayVars ++ NewTerms;
         Error ->
             Error
     end.
@@ -431,6 +440,19 @@ write_template(OverlayVars, FromFile, ToFile) ->
             Error
     end.
 
+render_string(OverlayVars, Data, TemplateName) ->
+    case erlydtl:compile(erlang:iolist_to_binary(Data), TemplateName, ?ERLYDTL_COMPILE_OPTS) of
+        {ok, TemplateName} ->
+            case render(TemplateName, OverlayVars) of
+                {ok, IoList} ->
+                    erlang:iolist_to_binary(IoList);
+                {error, Error} ->
+                    ?RLX_ERROR({render_failed, Data, Error})
+            end;
+        {error, Reason, _Warnings} ->
+            ?RLX_ERROR({unable_to_compile_template, Data, Reason})
+    end.
+
 -spec file_render_do(proplists:proplist(), iolist(), module(),
                      fun((term()) -> {ok, rlx_state:t()} | relx:error())) ->
                             {ok, rlx_state:t()} | relx:error().