1 files changed, 127 insertions, 0 deletions
diff --git a/xmake/modules/private/service/distributed_build/server.lua b/xmake/modules/private/service/distributed_build/server.lua
new file mode 100644
index 000000000..88b3d689c
--- /dev/null
+++ b/xmake/modules/private/service/distributed_build/server.lua
@@ -0,0 +1,127 @@
+--!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-present, TBOOX Open Source Group.
+--
+-- @author      ruki
+-- @file        server.lua
+--
+
+-- imports
+import("private.service.config")
+import("private.service.message")
+import("private.service.server")
+import("private.service.stream", {alias = "socket_stream"})
+import("private.service.distributed_build.session", {alias = "server_session"})
+import("lib.detect.find_tool")
+
+-- define module
+local distributed_build_server = distributed_build_server or server()
+local super = distributed_build_server:class()
+
+-- init server
+function distributed_build_server:init(daemon)
+    super.init(self, daemon)
+    if self:daemon() then
+        config.load()
+    end
+
+    -- init address
+    local address = assert(config.get("distributed_build.server.listen"), "config(distributed_build.server.listen): not found!")
+    super.address_set(self, address)
+
+    -- init handler
+    super.handler_set(self, self._on_handle)
+
+    -- init sessions
+    self._SESSIONS = {}
+end
+
+-- get class
+function distributed_build_server:class()
+    return distributed_build_server
+end
+
+-- on handle message
+function distributed_build_server:_on_handle(stream, msg)
+    local session_id = msg:session_id()
+    local session = self:_session(session_id)
+    vprint("%s: %s: <session %s>: on handle message(%d)", self, stream:sock(), session_id, msg:code())
+    vprint(msg:body())
+    session:stream_set(stream)
+    local respmsg = msg:clone()
+    local session_errs
+    local session_ok = try
+    {
+        function()
+            if self:need_verfiy() then
+                local ok, errors = self:verify_user(msg:token(), stream:sock():peeraddr())
+                if not ok then
+                    session_errs = errors
+                    return false
+                end
+            end
+            if msg:is_connect() then
+                session:open()
+            elseif msg:is_disconnect() then
+                session:close()
+                self._SESSIONS[session_id] = nil
+            else
+                assert(session:is_connected(), "session has not been connected!")
+                -- TODO
+            end
+            return true
+        end,
+        catch
+        {
+            function (errors)
+                if errors then
+                    session_errs = tostring(errors)
+                    vprint(session_errs)
+                end
+            end
+        }
+    }
+    respmsg:status_set(session_ok)
+    if not session_ok and session_errs then
+        respmsg:errors_set(session_errs)
+    end
+    local ok = stream:send_msg(respmsg) and stream:flush()
+    vprint("%s: %s: <session %s>: send %s", self, stream:sock(), session_id, ok and "ok" or "failed")
+end
+
+-- get session
+function distributed_build_server:_session(session_id)
+    local session = self._SESSIONS[session_id]
+    if not session then
+        session = server_session(session_id)
+        self._SESSIONS[session_id] = session
+    end
+    return session
+end
+
+-- close session
+function distributed_build_server:_session_close(session_id)
+    self._SESSIONS[session_id] = nil
+end
+
+function distributed_build_server:__tostring()
+    return "<distributed_build_server>"
+end
+
+function main(daemon)
+    local instance = distributed_build_server()
+    instance:init(daemon ~= nil)
+    return instance
+end