add protobuf rules

3 files changed, 184 insertions, 3 deletions

diff --git a/xmake/core/project/target.lua b/xmake/core/project/target.lua
index 1745e08a4..a4c194a15 100644
--- a/xmake/core/project/target.lua
+++ b/xmake/core/project/target.lua
@@ -90,8 +90,8 @@ function _instance:_load_rules(suffix)
     return true
 end
 
--- do load target and rules
-function _instance:_load()
+-- load all
+function _instance:_load_all()
 
     -- do before_load with target rules
     local ok, errors = self:_load_rules("before")
@@ -119,8 +119,29 @@ function _instance:_load()
     if not ok then
         return false, errors
     end
+    return true
+end
 
-    -- ok
+-- do load target and rules
+function _instance:_load()
+
+    -- enter the environments of the target packages
+    local oldenvs = {}
+    for name, values in pairs(self:pkgenvs()) do
+        oldenvs[name] = os.getenv(name)
+        os.addenv(name, unpack(values))
+    end
+
+    -- load all
+    local ok, errors = self:_load_all()
+    if not ok then
+        return false, errors
+    end
+
+    -- leave the environments of the target packages
+    for name, values in pairs(oldenvs) do
+        os.setenv(name, values)
+    end
     return true
 end
 
diff --git a/xmake/rules/protobuf/proto.lua b/xmake/rules/protobuf/proto.lua
new file mode 100644
index 000000000..cc00d3b35
--- /dev/null
+++ b/xmake/rules/protobuf/proto.lua
@@ -0,0 +1,94 @@
+--!A cross-platform build utility based on Lua
+--
+-- Licensed under the Apache License, Version 2.0 (the "License");
+-- you may not use this file except in compliance with the License.
+-- You may obtain a copy of the License at
+--
+--     http://www.apache.org/licenses/LICENSE-2.0
+--
+-- Unless required by applicable law or agreed to in writing, software
+-- distributed under the License is distributed on an "AS IS" BASIS,
+-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+-- See the License for the specific language governing permissions and
+-- limitations under the License.
+-- 
+-- Copyright (C) 2015 - 2019, TBOOX Open Source Group.
+--
+-- @author      ruki
+-- @file        proto.lua
+--
+
+-- imports
+import("core.base.option")
+import("core.theme.theme")
+import("core.project.config")
+import("core.project.depend")
+import("core.tool.compiler")
+
+-- build protobuf file
+function main(target, sourcekind, sourcefile_proto, opt)
+
+    -- get protoc
+    local protoc = assert(target:data("protobuf.protoc"), "protoc not found!")
+
+    -- get c/c++ source file for protobuf
+    local sourcefile_cx = path.join(target:autogendir(), "rules", "protobuf", "src", path.basename(sourcefile_proto) .. ".pb" .. (sourcekind == "cxx" and ".cc" or ".c"))
+    local sourcefile_dir = path.directory(sourcefile_cx)
+
+    -- get object file
+    local objectfile = target:objectfile(sourcefile_cx)
+
+    -- load compiler 
+    local compinst = compiler.load(sourcekind, {target = target})
+
+    -- get compile flags
+    local configs = {includedirs = sourcefile_dir}
+    if sourcekind == "cxx" then
+        configs.languages = "c++11"
+    end
+    local compflags = compinst:compflags({target = target, sourcefile = sourcefile_cx, configs = configs})
+
+    -- add objectfile
+    table.insert(target:objectfiles(), objectfile)
+
+    -- load dependent info 
+    local dependfile = target:dependfile(objectfile)
+    local dependinfo = option.get("rebuild") and {} or (depend.load(dependfile) or {})
+
+    -- need build this object?
+    local depvalues = {compinst:program(), compflags}
+    if not depend.is_changed(dependinfo, {lastmtime = os.mtime(objectfile), values = depvalues}) then
+        return 
+    end
+
+    -- trace progress info
+    cprintf("${color.build.progress}" .. theme.get("text.build.progress_format") .. ":${clear} ", opt.progress)
+    if option.get("verbose") then
+        cprint("${dim color.build.object}compiling.proto %s", sourcefile_proto)
+    else
+        cprint("${color.build.object}compiling.proto %s", sourcefile_proto)
+    end
+
+    -- ensure the source file directory
+    if not os.isdir(sourcefile_dir) then
+        os.mkdir(sourcefile_dir)
+    end
+
+    -- compile protobuf 
+    os.vrunv(protoc, {sourcefile_proto, (sourcekind == "cxx" and "--cpp_out=" or "--c_out=") .. sourcefile_dir})
+
+    -- trace
+    if option.get("verbose") then
+        print(compinst:compcmd(sourcefile_cx, objectfile, {compflags = compflags}))
+    end
+
+    -- compile c/c++ source file for protobuf
+    dependinfo.files = {}
+    assert(compinst:compile(sourcefile_cx, objectfile, {dependinfo = dependinfo, compflags = compflags}))
+
+    -- update files and values to the dependent file
+    dependinfo.values = depvalues
+    table.insert(dependinfo.files, sourcefile_proto)
+    depend.save(dependinfo, dependfile)
+end
+
diff --git a/xmake/rules/protobuf/xmake.lua b/xmake/rules/protobuf/xmake.lua
new file mode 100644
index 000000000..3907a4c2d
--- /dev/null
+++ b/xmake/rules/protobuf/xmake.lua
@@ -0,0 +1,66 @@
+--!A cross-platform build utility based on Lua
+--
+-- Licensed under the Apache License, Version 2.0 (the "License");
+-- you may not use this file except in compliance with the License.
+-- You may obtain a copy of the License at
+--
+--     http://www.apache.org/licenses/LICENSE-2.0
+--
+-- Unless required by applicable law or agreed to in writing, software
+-- distributed under the License is distributed on an "AS IS" BASIS,
+-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+-- See the License for the specific language governing permissions and
+-- limitations under the License.
+-- 
+-- Copyright (C) 2015 - 2019, TBOOX Open Source Group.
+--
+-- @author      ruki
+-- @file        xmake.lua
+--
+
+
+-- define rule: protobuf-cpp 
+rule("protobuf.env")
+
+    -- load protoc
+    before_load(function (target)
+        import("lib.detect.find_tool")
+        local protoc = target:data("protobuf.protoc")
+        if not protoc then
+            protoc = find_tool("protoc") 
+            if protoc and protoc.program then
+                target:data_set("protobuf.protoc", protoc.program)
+            else
+                raise("protoc not found!")
+            end
+        end
+    end)
+
+-- define rule: protobuf.cpp 
+rule("protobuf.cpp")
+
+    -- add deps
+    add_deps("protobuf.env")
+
+    -- set extension
+    set_extensions(".proto")
+
+    -- build protobuf file
+    on_build_file(function (target, sourcefile_proto, opt)
+        import("proto")(target, "cxx", sourcefile_proto, opt)
+    end)
+
+
+-- define rule: protobuf.c 
+rule("protobuf.c")
+
+    -- add deps
+    add_deps("protobuf.env")
+
+    -- set extension
+    set_extensions(".proto")
+
+    -- build protobuf file
+    on_build_file(function (target, sourcefile_proto, opt)
+        import("proto")(target, "cc", sourcefile_proto, opt)
+    end)