1 files changed, 40 insertions, 6 deletions
diff --git a/src/rlx_prv_archive.erl b/src/rlx_prv_archive.erl
index 3ad06b4..8370659 100644
--- a/src/rlx_prv_archive.erl
+++ b/src/rlx_prv_archive.erl
@@ -22,7 +22,7 @@
 %%% into a release directory.
 -module(rlx_prv_archive).
 
--behaviour(rlx_provider).
+-behaviour(provider).
 
 -export([init/1,
          do/1,
@@ -30,19 +30,32 @@
 
 -include("relx.hrl").
 
+-define(PROVIDER, tar).
+-define(DEPS, [release]).
+
 %%============================================================================
 %% API
 %%============================================================================
+
 -spec init(rlx_state:t()) -> {ok, rlx_state:t()}.
 init(State) ->
-    {ok, State}.
+    State1 = rlx_state:add_provider(State, providers:create([{name, ?PROVIDER},
+                                                             {module, ?MODULE},
+                                                             {bare, false},
+                                                             {deps, ?DEPS},
+                                                             {example, "tar"},
+                                                             {short_desc, ""},
+                                                             {desc, ""},
+                                                             {opts, []}])),
+
+    {ok, State1}.
 
 -spec do(rlx_state:t()) -> {ok, rlx_state:t()} | relx:error().
-do(State) ->    
+do(State) ->
     {RelName, RelVsn} = rlx_state:default_configured_release(State),
     Release = rlx_state:get_realized_release(State, RelName, RelVsn),
-    OutputDir = rlx_state:output_dir(State),        
-    make_tar(State, Release, OutputDir).        
+    OutputDir = rlx_state:output_dir(State),
+    make_tar(State, Release, OutputDir).
 
 format_error({tar_unknown_generation_error, Module, Vsn}) ->
     io_lib:format("Tarball generation error of ~s ~s",
@@ -94,6 +107,7 @@ update_tar(State, TempDir, OutputDir, Name, Vsn, ErtsVersion) ->
     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),
     ok =
         erl_tar:create(TarFile,
                        [{"lib", filename:join(TempDir, "lib")},
@@ -108,9 +122,29 @@ update_tar(State, TempDir, OutputDir, Name, Vsn, ErtsVersion) ->
                                 [];
                             _ ->
                                 [{"erts-"++ErtsVersion, filename:join(OutputDir, "erts-"++ErtsVersion)}]
-                        end], [compressed]),
+                        end]++OverlayFiles, [compressed]),
     ec_cmd_log:info(rlx_state:log(State),
                  "tarball ~s successfully created!~n", [TarFile]),
     ec_file:remove(TempDir, [recursive]),
     {ok, State}.
 
+overlay_files(undefined, _) ->
+    [];
+overlay_files(Overlay, OutputDir) ->
+    [{to(O), filename:join(OutputDir, to(O))} || O <- Overlay, filter(O)].
+
+to({copy, _, To}) ->
+    To;
+to({mkdir, To}) ->
+    To;
+to({template, _, To}) ->
+    To.
+
+filter({copy, _, _}) ->
+    true;
+filter({mkdir, _}) ->
+    true;
+filter({template, _, _}) ->
+    true;
+filter(_) ->
+    false.