mirror of
https://github.com/correl/rebar.git
synced 2024-11-15 03:00:18 +00:00
107 lines
4.2 KiB
Erlang
107 lines
4.2 KiB
Erlang
%% -*- 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_dia_compiler).
|
|
|
|
-export([compile/2,
|
|
clean/2]).
|
|
|
|
%% for internal use only
|
|
-export([info/2]).
|
|
|
|
-include("rebar.hrl").
|
|
|
|
%% ===================================================================
|
|
%% Public API
|
|
%% ===================================================================
|
|
|
|
-spec compile(rebar_config:config(), file:filename()) -> 'ok'.
|
|
compile(Config, _AppFile) ->
|
|
rebar_base_compiler:run(Config, filelib:wildcard("dia/*.dia"),
|
|
"dia", ".dia", "src", ".erl",
|
|
fun compile_dia/3).
|
|
|
|
-spec clean(rebar_config:config(), file:filename()) -> 'ok'.
|
|
clean(_Config, _AppFile) ->
|
|
GeneratedFiles = dia_generated_files("dia", "src", "include"),
|
|
ok = rebar_file_utils:delete_each(GeneratedFiles),
|
|
ok.
|
|
|
|
%% ===================================================================
|
|
%% Internal functions
|
|
%% ===================================================================
|
|
|
|
info(help, compile) ->
|
|
info_help("Build Diameter (*.dia) sources");
|
|
info(help, clean) ->
|
|
info_help("Delete generated Diameter files").
|
|
|
|
info_help(Description) ->
|
|
?CONSOLE(
|
|
"~s.~n"
|
|
"~n"
|
|
"Valid rebar.config options:~n"
|
|
" {dia_opts, []} (see diameter_codegen:from_dict/4 documentation)~n",
|
|
[Description]).
|
|
|
|
-spec compile_dia(file:filename(), file:filename(),
|
|
rebar_config:config()) -> ok.
|
|
compile_dia(Source, Target, Config) ->
|
|
ok = filelib:ensure_dir(Target),
|
|
ok = filelib:ensure_dir(filename:join("include", "dummy.hrl")),
|
|
Opts = [{outdir, "src"}] ++ rebar_config:get(Config, dia_opts, []),
|
|
case diameter_dict_util:parse({path, Source}, []) of
|
|
{ok, Spec} ->
|
|
FileName = dia_filename(Source, Spec),
|
|
_ = diameter_codegen:from_dict(FileName, Spec, Opts, erl),
|
|
_ = diameter_codegen:from_dict(FileName, Spec, Opts, hrl),
|
|
HrlFile = filename:join("src", FileName ++ ".hrl"),
|
|
case filelib:is_regular(HrlFile) of
|
|
true ->
|
|
ok = rebar_file_utils:mv(HrlFile, "include");
|
|
false ->
|
|
ok
|
|
end;
|
|
{error, Reason} ->
|
|
?ERROR("~s~n", [diameter_dict_util:format_error(Reason)])
|
|
end.
|
|
|
|
dia_generated_files(DiaDir, SrcDir, IncDir) ->
|
|
F = fun(File, Acc) ->
|
|
{ok, Spec} = diameter_dict_util:parse({path, File}, []),
|
|
FileName = dia_filename(File, Spec),
|
|
[filename:join([IncDir, FileName ++ ".hrl"]) |
|
|
filelib:wildcard(filename:join([SrcDir, FileName ++ ".*"]))] ++ Acc
|
|
end,
|
|
lists:foldl(F, [], filelib:wildcard(filename:join([DiaDir, "*.dia"]))).
|
|
|
|
dia_filename(File, Spec) ->
|
|
case proplists:get_value(name, Spec) of
|
|
undefined ->
|
|
filename:rootname(filename:basename(File));
|
|
Name ->
|
|
Name
|
|
end.
|