From 5f98b6b65af7374d4cb6097b756b7a600ecc06ac Mon Sep 17 00:00:00 2001
From: Tuncer Ayaz <tuncer.ayaz@gmail.com>
Date: Mon, 22 Mar 2010 18:31:01 +0100
Subject: [PATCH] Add ASN.1 support (via Ruslan Babayev)

Add support for compiling ASN.1 files with asn1ct.
Patch from Ruslan Babayev.
---
 THANKS                      |  1 +
 ebin/rebar.app              |  2 ++
 src/rebar_asn1_compiler.erl | 69 +++++++++++++++++++++++++++++++++++++
 3 files changed, 72 insertions(+)
 create mode 100644 src/rebar_asn1_compiler.erl

diff --git a/THANKS b/THANKS
index cb4ce96..3b4fe8b 100644
--- a/THANKS
+++ b/THANKS
@@ -8,3 +8,4 @@ Tuncer Ayaz
 Ian Wilkinson
 Juan Jose Comellas for getopt.erl
 OJ Reeves
+Ruslan Babayev
diff --git a/ebin/rebar.app b/ebin/rebar.app
index 712d0af..a43de73 100644
--- a/ebin/rebar.app
+++ b/ebin/rebar.app
@@ -10,6 +10,7 @@
               rebar_ct,
               rebar_deps,
               rebar_dialyzer,
+              rebar_asn1_compiler,
               rebar_erlc_compiler,
               rebar_escripter,
               rebar_eunit,
@@ -51,6 +52,7 @@
          {modules, [
                     {app_dir, [
                                rebar_protobuffs_compiler,
+                               rebar_asn1_compiler,
                                rebar_erlc_compiler,
                                rebar_lfe_compiler,
                                rebar_erlydtl_compiler,
diff --git a/src/rebar_asn1_compiler.erl b/src/rebar_asn1_compiler.erl
new file mode 100644
index 0000000..39614f0
--- /dev/null
+++ b/src/rebar_asn1_compiler.erl
@@ -0,0 +1,69 @@
+%% -*- tab-width: 4;erlang-indent-level: 4;indent-tabs-mode: nil -*-
+%% ex: ts=4 sw=4 et
+%% -------------------------------------------------------------------
+%%
+%% rebar: Erlang Build Tools
+%%
+%% Copyright (c) 2009, 2010 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_asn1_compiler).
+-author('ruslan@babayev.com').
+
+-export([compile/2,
+         clean/2]).
+
+-include("rebar.hrl").
+
+%% ===================================================================
+%% Public API
+%% ===================================================================
+
+-spec compile(Config::#config{}, AppFile::string()) -> 'ok'.
+compile(Config, _AppFile) ->
+    rebar_base_compiler:run(Config, filelib:wildcard("asn1/*.asn1"),
+                            "asn1", ".asn1", "src", ".erl",
+                            fun compile_asn1/3).
+
+-spec clean(Config::#config{}, AppFile::string()) -> 'ok'.
+clean(_Config, _AppFile) ->
+    rebar_file_utils:delete_each(asn_generated_files("asn1", "src")),
+    ok.
+
+-spec compile_asn1(string(), string(), #config{}) -> ok.
+compile_asn1(Source, Target, Config) ->
+    ok = rebar_utils:ensure_dir(Target),
+    Opts = [{outdir, "src"}, noobj] ++ rebar_config:get(Config, asn1_opts, []),
+    case asn1ct:compile(Source, Opts) of
+        ok ->
+            ok;
+        {error, _Reason} ->
+            ?FAIL
+    end.
+
+asn_generated_files(AsnDir, SrcDir) ->
+    lists:foldl(
+        fun(AsnFile, Acc) ->
+                Base = filename:rootname(filename:basename(AsnFile)),
+                filelib:wildcard(filename:join([SrcDir, Base ++ ".*"])) ++ Acc
+        end,
+        [],
+        filelib:wildcard(filename:join([AsnDir, "*.asn1"]))
+       ).