Add escriptize provider and a minimal test

2 files changed, 233 insertions, 11 deletions

diff --git a/src/rebar.app.src b/src/rebar.app.src
index 8c544aa..bd98277 100644
--- a/src/rebar.app.src
+++ b/src/rebar.app.src
@@ -23,30 +23,31 @@
         %% Default log level
         {log_level, warn},
 
-        {providers, [rebar_prv_as,
+        {providers, [rebar_prv_app_discovery,
+                     rebar_prv_as,
                      rebar_prv_clean,
+                     rebar_prv_common_test,
+                     rebar_prv_compile,
                      rebar_prv_cover,
                      rebar_prv_deps,
                      rebar_prv_dialyzer,
                      rebar_prv_do,
-                     rebar_prv_eunit,
                      rebar_prv_edoc,
-                     rebar_prv_lock,
+                     rebar_prv_erlydtl_compiler,
+                     rebar_prv_escriptize,
+                     rebar_prv_eunit,
+                     rebar_prv_help,
                      rebar_prv_install_deps,
+                     rebar_prv_lock,
+                     rebar_prv_new,
                      rebar_prv_packages,
-                     rebar_prv_erlydtl_compiler,
-                     rebar_prv_compile,
-                     rebar_prv_app_discovery,
+                     rebar_prv_release,
                      rebar_prv_shell,
                      rebar_prv_tar,
-                     rebar_prv_new,
                      rebar_prv_update,
                      rebar_prv_upgrade,
-                     rebar_prv_release,
                      rebar_prv_version,
-                     rebar_prv_common_test,
                      rebar_prv_wtf,
-                     rebar_prv_xref,
-                     rebar_prv_help]}
+                     rebar_prv_xref]}
         ]}
  ]}.
diff --git a/src/rebar_prv_escriptize.erl b/src/rebar_prv_escriptize.erl
new file mode 100644
index 0000000..0254d24
--- /dev/null
+++ b/src/rebar_prv_escriptize.erl
@@ -0,0 +1,221 @@
+%% -*- erlang-indent-level: 4;indent-tabs-mode: nil -*-
+%% ex: ts=4 sw=4 et
+%% -------------------------------------------------------------------
+%%
+%% rebar: Erlang Build Tools
+%%
+%% Copyright (c) 2009 Dave Smith (dizzyd@dizzyd.com)
+%%
+%% 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_escriptize).
+
+-behaviour(provider).
+
+-export([init/1, do/1, format_error/1]).
+
+-define(PROVIDER, escriptize).
+-define(DEPS, [compile]).
+
+-include("rebar.hrl").
+
+-include_lib("providers/include/providers.hrl").
+-include_lib("kernel/include/file.hrl").
+
+%%%=============================================================================
+%%% 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 escriptize"},
+                                {opts, []},
+                                {short_desc, "Generate escript archive"},
+                                {desc, desc()}
+                                ]),
+    {ok, rebar_state:add_provider(State, Provider)}.
+
+desc() ->
+    "Generate an escript executable containing "
+        "the project's and its dependencies' BEAM files.".
+
+do(State) ->
+    escriptize(State).
+
+escriptize(State0) ->
+    App1 = case rebar_state:project_apps(State0) of
+               [App] ->
+                   App;
+               Apps ->
+                   case rebar_state:get(escript_main_app, State0, undefined) of
+                       undefined ->
+                           ?PRV_ERROR(no_main_app);
+                       Name ->
+                           rebar_app_utils:find(Name, Apps)
+                   end
+           end,
+
+    AppName = rebar_app_info:name(App1),
+    AppNameStr = ec_cnv:to_list(AppName),
+
+    %% Get the output filename for the escript -- this may include dirs
+    Filename = filename:join([rebar_dir:base_dir(State0), "bin",
+                              rebar_state:get(State0, escript_name, AppName)]),
+    ok = filelib:ensure_dir(Filename),
+    State = rebar_state:escript_path(State0, Filename),
+
+    %% Look for a list of other applications (dependencies) to include
+    %% in the output file. We then use the .app files for each of these
+    %% to pull in all the .beam files.
+    InclApps = lists:usort(rebar_state:get(State, escript_incl_apps, [])
+                           ++ all_deps(State)),
+    InclBeams = get_app_beams(InclApps),
+
+    %% Look for a list of extra files to include in the output file.
+    %% For internal rebar-private use only. Do not use outside rebar.
+    InclExtra = get_extra(State),
+
+    %% Construct the archive of everything in ebin/ dir -- put it on the
+    %% top-level of the zip file so that code loading works properly.
+    EbinPrefix = filename:join(AppNameStr, "ebin"),
+    EbinFiles = usort(load_files(EbinPrefix, "*", "ebin")),
+
+    ExtraFiles = usort(InclBeams ++ InclExtra),
+    Files = get_nonempty(EbinFiles ++ ExtraFiles),
+
+    DefaultEmuArgs = ?FMT("%%! -escript main ~s -pa ~s/~s/ebin\n",
+                          [AppNameStr, AppNameStr, AppNameStr]),
+    EscriptSections =
+        [ {shebang,
+           def("#!", State, escript_shebang, "#!/usr/bin/env escript\n")}
+        , {comment, def("%%", State, escript_comment, "%%\n")}
+        , {emu_args, def("%%!", State, escript_emu_args, DefaultEmuArgs)}
+        , {archive, Files, []} ],
+    case escript:create(Filename, EscriptSections) of
+        ok -> ok;
+        {error, EscriptError} ->
+            throw(?PRV_ERROR({escript_creation_failed, AppName, EscriptError}))
+    end,
+
+    %% Finally, update executable perms for our script
+    {ok, #file_info{mode = Mode}} = file:read_file_info(Filename),
+    ok = file:change_mode(Filename, Mode bor 8#00111),
+    {ok, State}.
+
+-spec format_error(any()) -> iolist().
+format_error({write_failed, AppName, WriteError}) ->
+    io_lib:format("Failed to write ~p script: ~p", [AppName, WriteError]);
+format_error({zip_error, AppName, ZipError}) ->
+    io_lib:format("Failed to construct ~p escript: ~p", [AppName, ZipError]);
+format_error({bad_name, App}) ->
+    io_lib:format("Failed to get ebin/ directory for "
+                   "escript_incl_app: ~p", [App]);
+format_error(no_main_app) ->
+    io_lib:format("Multiple project apps and {rebar_escript_plugin, [{main_app, atom()}]}."
+                 " not set in rebar.config", []).
+
+%% ===================================================================
+%% Internal functions
+%% ===================================================================
+
+get_app_beams(Apps) ->
+    get_app_beams(Apps, []).
+
+get_app_beams([], Acc) ->
+    Acc;
+get_app_beams([App | Rest], Acc) ->
+    case code:lib_dir(App, ebin) of
+        {error, bad_name} ->
+            throw(?PRV_ERROR({bad_name, App}));
+        Path ->
+            Prefix = filename:join(atom_to_list(App), "ebin"),
+            Acc2 = load_files(Prefix, "*.beam", Path),
+            get_app_beams(Rest, Acc2 ++ Acc)
+    end.
+
+get_extra(State) ->
+    Extra = rebar_state:get(State, escript_incl_extra, []),
+    lists:foldl(fun({Wildcard, Dir}, Files) ->
+                        load_files(Wildcard, Dir) ++ Files
+                end, [], Extra).
+
+load_files(Wildcard, Dir) ->
+    load_files("", Wildcard, Dir).
+
+load_files(Prefix, Wildcard, Dir) ->
+    [read_file(Prefix, Filename, Dir)
+     || Filename <- filelib:wildcard(Wildcard, Dir)].
+
+read_file(Prefix, Filename, Dir) ->
+    Filename1 = case Prefix of
+                    "" ->
+                        Filename;
+                    _ ->
+                        filename:join([Prefix, Filename])
+                end,
+    [dir_entries(filename:dirname(Filename1)),
+     {Filename1, file_contents(filename:join(Dir, Filename))}].
+
+file_contents(Filename) ->
+    {ok, Bin} = file:read_file(Filename),
+    Bin.
+
+%% Given a filename, return zip archive dir entries for each sub-dir.
+%% Required to work around issues fixed in OTP-10071.
+dir_entries(File) ->
+    Dirs = dirs(File),
+    [{Dir ++ "/", <<>>} || Dir <- Dirs].
+
+%% Given "foo/bar/baz", return ["foo", "foo/bar", "foo/bar/baz"].
+dirs(Dir) ->
+    dirs1(filename:split(Dir), "", []).
+
+dirs1([], _, Acc) ->
+    lists:reverse(Acc);
+dirs1([H|T], "", []) ->
+    dirs1(T, H, [H]);
+dirs1([H|T], Last, Acc) ->
+    Dir = filename:join(Last, H),
+    dirs1(T, Dir, [Dir|Acc]).
+
+usort(List) ->
+    lists:ukeysort(1, lists:flatten(List)).
+
+get_nonempty(Files) ->
+    [{FName,FBin} || {FName,FBin} <- Files, FBin =/= <<>>].
+
+all_deps(State) ->
+    [list_to_existing_atom(binary_to_list(rebar_app_info:name(App)))
+     || App <- rebar_state:all_deps(State)].
+
+def(Rm, State, Key, Default) ->
+    Value0 = rebar_state:get(State, Key, Default),
+    case Rm of
+        "#!"  -> "#!"  ++ Value = Value0, rm_newline(Value);
+        "%%"  -> "%%"  ++ Value = Value0, rm_newline(Value);
+        "%%!" -> "%%!" ++ Value = Value0, rm_newline(Value)
+    end.
+
+rm_newline(String) ->
+    [C || C <- String, C =/= $\n].