--!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, Xmake Open Source Community.
--
-- @author      ruki
-- @file        server.lua
--

-- imports
import("core.base.global")
import("private.service.server_config", {alias = "config"})
import("private.service.message")
import("private.service.server")
import("private.service.distcc_build.server_session")
import("lib.detect.find_tool")

-- define module
local distcc_build_server = distcc_build_server or server()
local super = distcc_build_server:class()

-- init server
function distcc_build_server:init(daemon)
    super.init(self, daemon)

    -- init address
    local address = assert(config.get("distcc_build.listen"), "config(distcc_build.listen): not found!")
    super.address_set(self, address)

    -- init handler
    super.handler_set(self, self._on_handle)

    -- init sessions
    self._SESSIONS = {}

    -- init timeout
    self._SEND_TIMEOUT = config.get("distcc_build.send_timeout") or config.get("send_timeout") or -1
    self._RECV_TIMEOUT = config.get("distcc_build.recv_timeout") or config.get("recv_timeout") or -1
end

-- get class
function distcc_build_server:class()
    return distcc_build_server
end

-- get work directory
function distcc_build_server:workdir()
    local workdir = config.get("distcc_build.workdir")
    if not workdir then
        workdir = path.join(global.directory(), "service", "server", "distcc_build")
    end
    return workdir
end

-- on handle message
function distcc_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(respmsg)
            else
                assert(session:is_connected(), "session has not been connected!")
                if msg:is_compile() then
                    local ok, errors = session:compile(respmsg)
                    if not ok then
                        session_errs = errors
                        return false
                    end
                elseif msg:is_clean() then
                    session:clean()
                end
            end
            return true
        end,
        catch
        {
            function (errors)
                if errors then
                    session_errs = tostring(errors)
                end
            end
        }
    }
    respmsg:status_set(session_ok)
    if not session_ok and session_errs then
        respmsg:errors_set(session_errs)
        vprint(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 distcc_build_server:_session(session_id)
    local session = self._SESSIONS[session_id]
    if not session then
        session = server_session(self, session_id)
        self._SESSIONS[session_id] = session
    end
    return session
end

-- close session
function distcc_build_server:_session_close(session_id)
    self._SESSIONS[session_id] = nil
end

function distcc_build_server:__tostring()
    return "<distcc_build_server>"
end

function main(daemon)
    local instance = distcc_build_server()
    instance:init(daemon ~= nil)
    return instance
end