diff options
author | Dave Smith <dizzyd@dizzyd.com> | 2009-11-30 09:57:28 -0700 |
---|---|---|
committer | Dave Smith <dizzyd@dizzyd.com> | 2009-11-30 09:57:28 -0700 |
commit | 961c95b419ca0aa1e1c8ff6520450fe8e3368de5 (patch) | |
tree | 9b5d90f81eeddf715f087c78ed8a684d8d551c84 /src | |
parent | 60a8fc18adc4f9e880dc9b17a1f64bee54e63cdf (diff) |
Add protobuffs compiler module
Diffstat (limited to 'src')
-rw-r--r-- | src/rebar_protobuffs_compiler.erl | 132 |
1 files changed, 132 insertions, 0 deletions
diff --git a/src/rebar_protobuffs_compiler.erl b/src/rebar_protobuffs_compiler.erl new file mode 100644 index 0000000..540df9f --- /dev/null +++ b/src/rebar_protobuffs_compiler.erl @@ -0,0 +1,132 @@ +%% ------------------------------------------------------------------- +%% +%% 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() -> + case code:which(protobuffs_compile) of + non_existing -> + false; + _ -> + true + end. + + +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])), + [] = 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). |