From b14eaa0dc67d28b296eb190ef9a971ddbf4d6a70 Mon Sep 17 00:00:00 2001
From: Dave Smith <dizzyd@dizzyd.com>
Date: Tue, 22 Dec 2009 06:45:01 -0700
Subject: [PATCH] Adding rudimentary escript generator

---
 ebin/rebar.app          |  4 ++-
 src/rebar_escripter.erl | 63 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 66 insertions(+), 1 deletion(-)
 create mode 100644 src/rebar_escripter.erl

diff --git a/ebin/rebar.app b/ebin/rebar.app
index b447cc8..642f35f 100644
--- a/ebin/rebar.app
+++ b/ebin/rebar.app
@@ -7,6 +7,7 @@
               rebar_core,
               rebar_ct,
               rebar_erlc_compiler,
+              rebar_escripter,
               rebar_eunit,
               rebar_file_utils,
               rebar_lfe_compiler,
@@ -34,7 +35,8 @@
                                 rebar_port_compiler,
                                 rebar_otp_app,
                                 rebar_ct,
-                                rebar_eunit]},
+                                rebar_eunit,
+                                rebar_escripter]},
                     {rel_dir, [ rebar_reltool ]}
                    ]}
         ]}
diff --git a/src/rebar_escripter.erl b/src/rebar_escripter.erl
new file mode 100644
index 0000000..b18b436
--- /dev/null
+++ b/src/rebar_escripter.erl
@@ -0,0 +1,63 @@
+%% -------------------------------------------------------------------
+%%
+%% 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]).
+
+-include("rebar.hrl").
+
+%% ===================================================================
+%% Public API
+%% ===================================================================
+
+escriptize(Config, AppFile) ->
+    %% Extract the application name from the archive -- this will be be what
+    %% we call the output script
+    {ok, AppName, _AppData} = rebar_app_utils:load_app_file(AppFile),
+    
+    %% 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 = filelib:wildcard("*", "ebin"),
+    case zip:create("mem", Files, [{cwd, "ebin"}, memory]) of
+        {ok, {"mem", ZipBin}} ->
+            %% Archive was successfully created. Prefix that binary with our
+            %% header and write to our escript file
+            Script = <<"#!/usr/bin/env escript\n", ZipBin/binary>>,
+            case file:write_file(AppName, 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
+    [] = os:cmd(?FMT("chmod u+x ~p", [AppName])),
+    ok.
+