mirror of
https://github.com/correl/rebar.git
synced 2024-11-24 03:00:14 +00:00
130 lines
4.7 KiB
Erlang
130 lines
4.7 KiB
Erlang
%% -*- tab-width: 4;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_protobuffs_compiler).
|
|
|
|
-export([compile/2,
|
|
clean/2]).
|
|
|
|
-include("rebar.hrl").
|
|
|
|
%% ===================================================================
|
|
%% Public API
|
|
%% ===================================================================
|
|
|
|
compile(_Config, _AppFile) ->
|
|
case filelib:wildcard("src/*.proto") of
|
|
[] ->
|
|
ok;
|
|
FoundFiles ->
|
|
%% Check for protobuffs library -- if it's not present, fail since we have
|
|
%% .proto files that need building
|
|
case protobuffs_is_present() of
|
|
true ->
|
|
%% Build a list of output files - { Proto, Beam, Hrl }
|
|
Targets = [{Proto, beam_file(Proto), hrl_file(Proto)} ||
|
|
Proto <- FoundFiles],
|
|
|
|
%% Compile each proto file
|
|
compile_each(Targets);
|
|
false ->
|
|
?ERROR("Protobuffs library not present in code path!\n", []),
|
|
?FAIL
|
|
end
|
|
end.
|
|
|
|
|
|
clean(_Config, _AppFile) ->
|
|
%% Get a list of generated .beam and .hrl files and then delete them
|
|
Protos = filelib:wildcard("src/*.proto"),
|
|
Targets = [fq_beam_file(F) || F <- Protos] ++ [fq_hrl_file(F) || F <- Protos],
|
|
case Targets of
|
|
[] ->
|
|
ok;
|
|
_ ->
|
|
delete_each(Targets)
|
|
end.
|
|
|
|
|
|
%% ===================================================================
|
|
%% Internal functions
|
|
%% ===================================================================
|
|
|
|
protobuffs_is_present() ->
|
|
code:which(protobuffs_compile) =/= non_existing.
|
|
|
|
beam_file(Proto) ->
|
|
filename:basename(Proto, ".proto") ++ "_pb.beam".
|
|
|
|
hrl_file(Proto) ->
|
|
filename:basename(Proto, ".proto") ++ "_pb.hrl".
|
|
|
|
fq_beam_file(Proto) ->
|
|
filename:join(["ebin", filename:basename(Proto, ".proto") ++ "_pb.beam"]).
|
|
|
|
fq_hrl_file(Proto) ->
|
|
filename:join(["include", filename:basename(Proto, ".proto") ++ "_pb.hrl"]).
|
|
|
|
needs_compile(Proto, Beam) ->
|
|
ActualBeam = filename:join(["ebin", filename:basename(Beam)]),
|
|
filelib:last_modified(ActualBeam) < filelib:last_modified(Proto).
|
|
|
|
compile_each([]) ->
|
|
ok;
|
|
compile_each([{Proto, Beam, Hrl} | Rest]) ->
|
|
case needs_compile(Proto, Beam) of
|
|
true ->
|
|
?CONSOLE("Compiling ~s\n", [Proto]),
|
|
case protobuffs_compile:scan_file(Proto) of
|
|
ok ->
|
|
%% Compilation worked, but we need to move the .beam and .hrl file
|
|
%% into the ebin/ and include/ directories respectively
|
|
%% TODO: Protobuffs really needs to be better about this...sigh.
|
|
[] = os:cmd(?FMT("mv ~s ebin", [Beam])),
|
|
ok = filelib:ensure_dir(filename:join("include", Hrl)),
|
|
[] = os:cmd(?FMT("mv ~s include", [Hrl])),
|
|
ok;
|
|
Other ->
|
|
?ERROR("Protobuff compile of ~s failed: ~p\n", [Proto, Other]),
|
|
?FAIL
|
|
end;
|
|
false ->
|
|
ok
|
|
end,
|
|
compile_each(Rest).
|
|
|
|
delete_each([]) ->
|
|
ok;
|
|
delete_each([File | Rest]) ->
|
|
case file:delete(File) of
|
|
ok ->
|
|
ok;
|
|
{error, enoent} ->
|
|
ok;
|
|
{error, Reason} ->
|
|
?ERROR("Failed to delete ~s: ~p\n", [File, Reason])
|
|
end,
|
|
delete_each(Rest).
|