local cjson = require "cjson"

local rpc = require "oui.rpc"

local ubus = require "oui.ubus"

cjson.encode_empty_table_as_object(false)

if ngx.req.get_method() ~= "POST" then -- 检查 HTTP 请求方法是否为 POST

    ngx.exit(ngx.HTTP_FORBIDDEN) -- 如果不是，返回 403 Forbidden。

    end

ngx.req.read_body() -- 获取请求体

local data = ngx.req.get_body_data() -- 如果数据直接存在于 body 中，则获取

if not data then -- 否则从临时文件中读取，这可能处理较大的请求体。

    local name = ngx.req.get_body_file()

    local f = io.open(name, "r")

    data = f:read("*a")

    f:close()

end

local function rpc_method_challenge(id, params)

    local res = ubus.call("gl-session", "challenge", params)

    local code, data = res.code, res.data

    if code ~= 0 then

        local resp = rpc.error_response(id, code, data)

        ngx.say(cjson.encode(resp))

        return

    end

    local resp = rpc.result_response(id, data)

    ngx.say(cjson.encode(resp))

end -- 调用 gl-session 服务的 challenge 方法，生成认证挑战码（用于后续登录验证）。

local function rpc_method_login(id, params)

    local res = ubus.call("gl-session", "login", params)

    local code, data = res.code, res.data

    if code ~= 0 then

        local resp = rpc.error_response(id, code, data)

        ngx.say(cjson.encode(resp))

        return

    end

    ngx.header["Set-Cookie"] = "Admin-Token=" .. data.sid

    local resp = rpc.result_response(id, data)

    ngx.say(cjson.encode(resp))

end -- 调用 login 方法验证用户凭证，成功时设置 Cookie Admin-Token 为会话 ID（sid）。

local function rpc_method_logout(id, params)

    ubus.call("gl-session", "logout", params)

    local resp = rpc.result_response(id)

    ngx.say(cjson.encode(resp))

end -- 终止当前会话。

local function rpc_method_alive(id, params)

    local res = ubus.call("gl-session", "touch", params)

    if res.code ~= 0 then

        local resp = rpc.error_response(id, res.code, data)

        ngx.say(cjson.encode(resp))

        return

    end

    local resp = rpc.result_response(id)

    ngx.say(cjson.encode(resp))

end -- 通过 touch 方法刷新会话有效期。

local function rpc_method_call(id, params)

    if #params < 3 then

        local resp = rpc.error_response(id, rpc.ERROR_CODE_INVALID_PARAMS)

        ngx.say(cjson.encode(resp))

        return

    end

    local sid, object, method, args = params[1], params[2], params[3], params[4]

    if type(sid) ~= "string" or type(object) ~= "string" or type(method) ~= "string" then

        local resp = rpc.error_response(id, rpc.ERROR_CODE_INVALID_PARAMS)

        ngx.say(cjson.encode(resp))

        return

    end

    if args and type(args) ~= "table" then

        local resp = rpc.error_response(id, rpc.ERROR_CODE_INVALID_PARAMS)

        ngx.say(cjson.encode(resp))

        return

    end -- 参数校验：检查 sid、object、method 类型，args 是否为表。

    ngx.ctx.sid = sid

    if not rpc.is_no_auth(object, method) then

        -- rpc.is_no_auth (object, method)：跳过认证的方法（如公共 API）。

        if not rpc.access("rpc", object .. "." .. method) then

            -- rpc.access ("rpc", object .. "." .. method)：验证调用权限。

            local resp = rpc.error_response(id, rpc.ERROR_CODE_ACCESS)

            ngx.say(cjson.encode(resp))

            return

        end

    end

    local res = rpc.call(object, method, args)

    -- 调用目标方法：通过 rpc.call 执行实际逻辑，返回结果或错误。

    if type(res) == "number" then

        local resp = rpc.error_response(id, res)

        ngx.say(cjson.encode(resp))

        return

    end

    if type(res) ~= "table" then res = {} end

    local resp = rpc.result_response(id, res)

    ngx.say(cjson.encode(resp))

end

local methods= {

    ["challenge"] = rpc_method_challenge,

    ["login"] = rpc_method_login,

    ["logout"] = rpc_method_logout,

    ["alive"] = rpc_method_alive,

    ["call"] = rpc_method_call

} --RPC 方法：challenge、login、logout、alive、call

local ok, json_data = pcall(cjson.decode, data)

if not ok then

    local resp = rpc.error_response(nil, rpc.ERROR_CODE_PARSE_ERROR)

    ngx.say(cjson.encode(resp))

    return

end

if type(json_data) ~= "table" then

    local resp = rpc.error_response(nil, rpc.ERROR_CODE_PARSE_ERROR)

    ngx.say(cjson.encode(resp))

    return

end

if type(json_data.method) ~= "string" then

    local resp = rpc.error_response(json_data.id, rpc.ERROR_CODE_INVALID_REQUEST)

    ngx.say(cjson.encode(resp))

    return

end

if json_data.params and type(json_data.params) ~= "table" then

    local resp = rpc.error_response(json_data.id, rpc.ERROR_CODE_INVALID_REQUEST)

    ngx.say(cjson.encode(resp))

    return

end

if not methods[json_data.method] then

    local resp = rpc.error_response(json_data.id, rpc.ERROR_CODE_METHOD_NOT_FOUND)

    ngx.say(cjson.encode(resp))

    return

end

methods[json_data.method](json_data.id, json_data.params or {})

M.session = function()

    local session = ubus.call("gl-session", "session", { sid = ngx.ctx.sid })

    local __oui_session = {

        is_local = ngx.var.remote_addr == "127.0.0.1" or ngx.var.remote_addr == "::1",

        -- is_local 为本地

        remote_addr = ngx.var.remote_addr,

        remote_port = ngx.var.remote_port

    }

    if not session then return __oui_session end

    utils.update_ngx_session("/tmp/gl_token_" .. ngx.ctx.sid)

    session.remote_addr = ngx.var.remote_addr

    session.remote_port = ngx.var.remote_port

    return session

end

M.access = function(scope, entry, need)

    local headers = ngx.req.get_headers()

    local s = M.session()

    local aclgroup = s.aclgroup

    if s.is_local and headers["glinet"] then

        -- 判断是否本地请求并且使用 glinet 服务，则通过

        return true

    end

    -- The admin acl group is always allowed

    if aclgroup == "root" then return true end

    if not aclgroup or aclgroup == "" then return false end

    local perm = db.get_perm(aclgroup, scope, entry)

    if not need then return false end

    if need == "r" then

        return perm:find("[r,w]") ~= nil

    else

        return perm:find(need) ~= nil

    end

end

M.call = function(object, method, args)

    ngx.log(ngx.DEBUG, "call: '", object, ".", method, "'")

    if not objects[object] then -- 如果没有属性加载就从 /usr/lib/oui-httpd/rpc/ 加载脚本

        local script = "/usr/lib/oui-httpd/rpc/" .. object

        -- 如果脚本文件存在且加载成功，将对象的方法注册到 objects 表中。

        if not fs.access(script) then

            -- 如果无法从 /usr/lib/oui-httpd/rpc/ 目录下加载脚本文件或者找不到对象或方法。

            return glc_call(object, method, args)

            -- 则调用 glc_call 执行。

        end

        local ok, tb = pcall(dofile, script)

        if not ok then

            ngx.log(ngx.ERR, tb)

            return glc_call(object, method, args)

        end

        if type(tb) == "table" then

            local funs = {}

            for k, v in pairs(tb) do

                if type(v) == "function" then

                    funs[k] = v

                end

            end

            objects[object] = funs

        end

    end

    local fn = objects[object] and objects[object][method]

    if not fn  then

        return glc_call(object, method, args)

    end

    return fn(args)

end

local function glc_call(object, method, args)

    ngx.log(ngx.DEBUG, "call C: '", object, ".", method, "'")

    -- 查看 /usr/lib/oui-httpd/rpc/ 目录下是二进制 s2s.so 文件，

    -- 无法直接加载，则通过 glc_call 调用 /cgi-bin/glc 执行 C 程序实现的 RPC 方法。

    local res = ngx.location.capture("/cgi-bin/glc", {

        method = ngx.HTTP_POST,

        body = cjson.encode({

            object = object,

            method = method,

            args = args or {}

        })

    })

    if res.status ~= ngx.HTTP_OK then return M.ERROR_CODE_INTERNAL_ERROR end

    local body = res.body

    local code = tonumber(body:match("(-?%d+)"))

    if code ~= M.ERROR_CODE_NONE then

        local err_msg = body:match("%d+ (.+)")

        if err_msg then

            ngx.log(ngx.ERR, err_msg)

        end

        return code

    end

    local msg = body:match("%d+ (.*)")

    return cjson.decode(msg)

end