/*
    This file is part of the Arduino_RPClite library.

    Copyright (C) Arduino s.r.l. and/or its affiliated companies

    This Source Code Form is subject to the terms of the Mozilla Public
    License, v. 2.0. If a copy of the MPL was not distributed with this
    file, You can obtain one at http://mozilla.org/MPL/2.0/.
    
*/

#ifndef RPCLITE_SERVER_H
#define RPCLITE_SERVER_H

#include <utility>

#include "request.h"
#include "error.h"
#include "dispatcher.h"
#include "decoder.h"
#include "decoder_manager.h"

#define MAX_CALLBACKS   100

class RPCServer {

public:
    explicit RPCServer(ITransport& t) : decoder(RpcDecoderManager::getInstance().getDecoder(t)) {}

    operator bool() const {return decoder != nullptr;}

    template<typename F>
    bool bind(const MsgPack::str_t& name, F&& func, MsgPack::str_t tag=""){
        return dispatcher.bind(name, func, tag);
    }

    bool hasTag(const MsgPack::str_t& name, MsgPack::str_t tag) const {
        return dispatcher.hasTag(name, tag);
    }

    bool run() {

        RPCRequest<> req;

        if (!get_rpc(req)) return false; // Populate local request

        process_request(req);      // Process local data

        return send_response(req);           // Send local data

    }

    template<size_t RpcSize = DEFAULT_RPC_BUFFER_SIZE>
    bool get_rpc(RPCRequest<RpcSize>& req, MsgPack::str_t tag="") {
        decoder->decode();

        const MsgPack::str_t method = decoder->fetch_rpc_method();

        if (method == "") return false;
        if (tag != "" && !dispatcher.hasTag(method, tag)) return false;

        req.size = decoder->get_request(req.buffer, RpcSize);
        return req.size > 0;
    }

    template<size_t RpcSize = DEFAULT_RPC_BUFFER_SIZE>
    void process_request(RPCRequest<RpcSize>& req) {

        if (!req.unpack_request_headers()) {
            req.reset();
            return;
        }

        req.pack_response_headers();

        dispatcher.call(req.method, req.unpacker, req.packer);

    }

    template<size_t RpcSize = DEFAULT_RPC_BUFFER_SIZE>
    bool send_response(const RPCRequest<RpcSize>& req) const {

        if (req.type == NO_MSG || req.packer.size() == 0) {
            return true; // No response to send
        }

        if (req.type == NOTIFY_MSG) return true;

        return decoder->send_response(req.packer);

    }

    uint32_t get_discarded_packets() const {return decoder->get_discarded_packets();}

private:
    RpcDecoder<>* decoder = nullptr;
    RpcFunctionDispatcher<MAX_CALLBACKS> dispatcher{};
    
};

#endif //RPCLITE_SERVER_H