/*
 *     .============.
 *    //  M A K E  / \
 *   //  C++ DEV  /   \
 *  //  E A S Y  /  \/ \
 * ++ ----------.  \/\  .
 *  \\     \     \ /\  /
 *   \\     \     \   /
 *    \\     \     \ /
 *     -============'
 *
 * Copyright (c) 2023 Hevake and contributors, all rights reserved.
 *
 * This file is part of cpp-tbox (https://github.com/cpp-main/cpp-tbox)
 * Use of this source code is governed by MIT license that can be found
 * in the LICENSE file in the root of the source tree. All contributing
 * project authors may be found in the CONTRIBUTORS.md file in the root
 * of the source tree.
 */
#include "proto.h"
#include <tbox/base/log.h>
#include <tbox/base/json.hpp>
#include <tbox/base/assert.h>
#include <tbox/util/json.h>

namespace tbox {
namespace jsonrpc {

void Proto::sendRequest(int id, const std::string &method, const Json &js_params)
{
    Json js = {
        {"jsonrpc", "2.0"},
        {"method", method}
    };

    if (id != 0)
        js["id"] = id;

    if (!js_params.is_null())
        js["params"] = js_params;

    sendJson(js);
}

void Proto::sendRequest(const std::string &id, const std::string &method, const Json &js_params)
{
    Json js = {
        {"jsonrpc", "2.0"},
        {"method", method}
    };

    if (!id.empty())
        js["id"] = id;

    if (!js_params.is_null())
        js["params"] = js_params;

    sendJson(js);
}

void Proto::sendRequest(int id, const std::string &method)
{
    sendRequest(id, method, Json());
}

void Proto::sendRequest(const std::string &id, const std::string &method)
{
    sendRequest(id, method, Json());
}

void Proto::sendResult(int id, const Json &js_result)
{
    Json js = {
        {"jsonrpc", "2.0"},
        {"result", js_result}
    };

    if (id != 0)
        js["id"] = id;

    sendJson(js);
}

void Proto::sendResult(const std::string &id, const Json &js_result)
{
    Json js = {
        {"jsonrpc", "2.0"},
        {"result", js_result}
    };

    if (!id.empty())
        js["id"] = id;

    sendJson(js);
}

void Proto::sendError(int id, int errcode, const std::string &message)
{
    Json js = {
        {"jsonrpc", "2.0"},
        {"error", { {"code", errcode } } }
    };

    if (id != 0)
        js["id"] = id;

    if (!message.empty())
        js["error"]["message"] = message;

    sendJson(js);
}

void Proto::sendError(const std::string &id, int errcode, const std::string &message)
{
    Json js = {
        {"jsonrpc", "2.0"},
        {"error", { {"code", errcode } } }
    };

    if (!id.empty())
        js["id"] = id;

    if (!message.empty())
        js["error"]["message"] = message;

    sendJson(js);
}

void Proto::setRecvCallback(RecvRequestIntCallback &&req_cb, RecvRespondIntCallback &&rsp_cb)
{
    if (req_cb && rsp_cb) {
        id_type_ = IdType::kInt;
        recv_request_int_cb_ = std::move(req_cb);
        recv_respond_int_cb_ = std::move(rsp_cb);

    } else {
        LogWarn("int req_cb or rsp_cb is nullptr");
    }
}

void Proto::setRecvCallback(RecvRequestStrCallback &&req_cb, RecvRespondStrCallback &&rsp_cb)
{
    if (req_cb && rsp_cb) {
        id_type_ = IdType::kString;
        recv_request_str_cb_ = std::move(req_cb);
        recv_respond_str_cb_ = std::move(rsp_cb);

    } else {
        LogWarn("str req_cb or rsp_cb is nullptr");
    }
}

void Proto::setSendCallback(SendDataCallback &&cb)
{
    send_data_cb_ = std::move(cb);
}

void Proto::cleanup()
{
    recv_request_int_cb_ = nullptr;
    recv_respond_int_cb_ = nullptr;
    recv_request_str_cb_ = nullptr;
    recv_respond_str_cb_ = nullptr;
    send_data_cb_ = nullptr;
}

void Proto::onRecvJson(const Json &js) const
{
    if (js.is_object()) {

        std::string version;
        if (!util::json::GetField(js, "jsonrpc", version) || version != "2.0") {
            LogNotice("no jsonrpc field, or version not match");
            return;
        }

        if (js.contains("method")) {
            handleAsMethod(js);     //! 按请求进行处理
        } else if (js.contains("result")) {
            handleAsResult(js);     //! 按结果回复进行处理，必须要有id字段
        } else if (js.contains("error")) {
            handleAsError(js);      //! 按错误回复进行处理
        } else {
            LogNotice("not jsonrpc format");
        }

    } else if (js.is_array()) {
        for (auto &js_item : js) {
            onRecvJson(js_item);
        }
    }
}

void Proto::handleAsMethod(const Json &js) const
{
    std::string method;
    if (!util::json::GetField(js, "method", method)) {
        LogNotice("method type not string");
        return;
    }

    auto &js_params = js.contains("params") ? js["params"] : Json();

    if (id_type_ == IdType::kInt) {
        int id = 0;
        if (js.contains("id") && !util::json::GetField(js, "id", id)) {
            LogNotice("id type not int");
            return;
        }
        recv_request_int_cb_(id, method, js_params);

    } else if (id_type_ == IdType::kString) {
        std::string id;
        if (js.contains("id") && !util::json::GetField(js, "id", id)) {
            LogNotice("id type not string");
            return;
        }
        recv_request_str_cb_(id, method, js_params);

    } else {
        LogWarn("please invoke setRecvCallback() first.");
    }
}

void Proto::handleAsResult(const Json &js) const
{
    if (!js.contains("id")) {
        LogNotice("no id in respond");
        return;
    }

    Response response;
    response.js_result = std::move(js["result"]);

    if (id_type_ == IdType::kInt) {
        int id = 0;
        if (!util::json::GetField(js, "id", id)) {
            LogNotice("id type not int in respond");
            return;
        }
        recv_respond_int_cb_(id, response);

    } else if (id_type_ == IdType::kString) {
        std::string id;
        if (!util::json::GetField(js, "id", id)) {
            LogNotice("id type not string in respond");
            return;
        }
        recv_respond_str_cb_(id, response);

    } else {
        LogWarn("please invoke setRecvCallback() first.");
    }
}

void Proto::handleAsError(const Json &js) const
{
    Response response;

    auto &js_error = js["error"];
    if (!util::json::GetField(js_error, "code", response.error.code)) {
        LogNotice("no code field in error");
        return;
    }
    util::json::GetField(js_error, "message", response.error.message);

    if (id_type_ == IdType::kInt) {
        int id = 0;
        if (js.contains("id") && !util::json::GetField(js, "id", id)) {
            LogNotice("id type not int in error");
            return;
        }
        recv_respond_int_cb_(id, response);

    } else if (id_type_ == IdType::kString) {
        std::string id;
        if (js.contains("id") && !util::json::GetField(js, "id", id)) {
            LogNotice("id type not string in error");
            return;
        }
        recv_respond_str_cb_(id, response);

    } else {
        LogWarn("please invoke setRecvCallback() first.");
    }
}

}
}