mirror of
https://github.com/correl/rebar.git
synced 2024-12-18 03:00:17 +00:00
2e64e91c0c
Allows customizing the arguments handed over to the Erlang VM, e.g. to set a name or a cookie, e.g. by setting escript_emu_args to: %%! -name rebar -setcookie rebar
121 lines
4.9 KiB
Erlang
121 lines
4.9 KiB
Erlang
%% -*- 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_escripter).
|
|
|
|
-export([escriptize/2,
|
|
clean/2]).
|
|
|
|
-include("rebar.hrl").
|
|
-include_lib("kernel/include/file.hrl").
|
|
|
|
%% ===================================================================
|
|
%% Public API
|
|
%% ===================================================================
|
|
|
|
escriptize(Config, AppFile) ->
|
|
%% Extract the application name from the archive -- this is the default
|
|
%% name of the generated script
|
|
AppName = rebar_app_utils:app_name(AppFile),
|
|
|
|
%% Get the output filename for the escript -- this may include dirs
|
|
Filename = rebar_config:get_local(Config, escript_name, AppName),
|
|
ok = filelib:ensure_dir(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.
|
|
InclBeams = get_app_beams(
|
|
rebar_config:get_local(Config, escript_incl_apps, []), []),
|
|
|
|
%% Construct the archive of everything in ebin/ dir -- put it on the
|
|
%% top-level of the zip file so that code loading works properly.
|
|
Files = load_files("*", "ebin") ++ InclBeams,
|
|
case zip:create("mem", Files, [memory]) of
|
|
{ok, {"mem", ZipBin}} ->
|
|
%% Archive was successfully created. Prefix that binary with our
|
|
%% header and write to our escript file
|
|
Shebang = rebar_config:get(Config, escript_shebang,
|
|
"#!/usr/bin/env escript\n"),
|
|
Comment = rebar_config:get(Config, escript_comment, "%%\n"),
|
|
EmuArgs = rebar_config:get(Config, escript_emu_args, "%%!\n"),
|
|
Script = iolist_to_binary([Shebang, Comment, EmuArgs, ZipBin]),
|
|
case file:write_file(Filename, Script) of
|
|
ok ->
|
|
ok;
|
|
{error, WriteError} ->
|
|
?ERROR("Failed to write ~p script: ~p\n",
|
|
[AppName, WriteError]),
|
|
?FAIL
|
|
end;
|
|
{error, ZipError} ->
|
|
?ERROR("Failed to construct ~p escript: ~p\n",
|
|
[AppName, ZipError]),
|
|
?FAIL
|
|
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#00100),
|
|
ok.
|
|
|
|
clean(Config, AppFile) ->
|
|
%% Extract the application name from the archive -- this is the default
|
|
%% name of the generated script
|
|
AppName = rebar_app_utils:app_name(AppFile),
|
|
|
|
%% Get the output filename for the escript -- this may include dirs
|
|
Filename = rebar_config:get_local(Config, escript_name, AppName),
|
|
rebar_file_utils:delete_each([Filename]).
|
|
|
|
|
|
%% ===================================================================
|
|
%% Internal functions
|
|
%% ===================================================================
|
|
|
|
get_app_beams([], Acc) ->
|
|
Acc;
|
|
get_app_beams([App | Rest], Acc) ->
|
|
case code:lib_dir(App, ebin) of
|
|
{error, bad_name} ->
|
|
?ABORT("Failed to get ebin/ directory for "
|
|
"~p escript_incl_apps.", [App]);
|
|
Path ->
|
|
Acc2 = [{filename:join([App, ebin, F]),
|
|
file_contents(filename:join(Path, F))} ||
|
|
F <- filelib:wildcard("*", Path)],
|
|
get_app_beams(Rest, Acc2 ++ Acc)
|
|
end.
|
|
|
|
load_files(Wildcard, Dir) ->
|
|
[read_file(Filename, Dir) || Filename <- filelib:wildcard(Wildcard, Dir)].
|
|
|
|
read_file(Filename, Dir) ->
|
|
{Filename, file_contents(filename:join(Dir, Filename))}.
|
|
|
|
file_contents(Filename) ->
|
|
{ok, Bin} = file:read_file(Filename),
|
|
Bin.
|