#include "../pchheader.hpp"
#include "../util/util.hpp"
#include "../msg/json/usrmsg_json.hpp"
#include "user_comm_session.hpp"
#include "usr.hpp"

namespace jusrmsg = msg::usrmsg::json;

namespace usr
{
    /**
     * This gets hit every time a client connects to HP via the public port (configured in config).
     * @return returns 0 if connection is successful and user challenge is sent, otherwise -1.
     */
    int user_comm_session::handle_connect()
    {
        // Allow connection only if the maximum capacity is not reached. 0 means allowing unlimited connections.
        if ((conf::cfg.user.max_connections == 0) || (ctx.users.size() < conf::cfg.user.max_connections))
        {
            LOG_DEBUG << "User client connected " << display_name();

            // As soon as a user connects, we issue them a challenge message. We remember the
            // challenge we issued and later verify the user's response with it.
            std::vector<uint8_t> msg;
            jusrmsg::create_user_challenge(msg, issued_challenge);
            send(msg);

            // Set the challenge-issued value to true.
            challenge_status = comm::CHALLENGE_STATUS::CHALLENGE_ISSUED;
            return 0;
        }
        else
        {
            LOG_DEBUG << "Dropping the user connection. Maximum user capacity reached. [" << display_name() << "] (limit: " << conf::cfg.user.max_connections << ").";
            return -1;
        }
    }

    /**
     * This gets hit every time we receive some data from a client connected to the HP public port.
     */
    int user_comm_session::handle_message(std::string_view msg)
    {
        // Adding message size to user message characters(bytes) per minute counter.
        increment_metric(comm::SESSION_THRESHOLDS::MAX_RAWBYTES_PER_MINUTE, msg.size());

        // First check whether this session is pending challenge.
        // Meaning we have previously issued a challenge to the client.
        if (challenge_status == comm::CHALLENGE_STATUS::CHALLENGE_ISSUED)
        {
            if (verify_challenge(msg, *this) == 0)
                return 0;

            LOG_DEBUG << "User challenge verification failed. " << display_name();
        }
        // Check whether this session belongs to an authenticated (challenge-verified) user.
        else if (challenge_status == comm::CHALLENGE_STATUS::CHALLENGE_VERIFIED)
        {
            // Check whether this user is among authenticated users
            // and perform authenticated msg processing.

            const auto itr = ctx.users.find(pubkey);
            if (itr != ctx.users.end())
            {
                // This is an authed user.
                connected_user &user = itr->second;
                if (handle_authed_user_message(user, msg) != 0)
                {
                    increment_metric(comm::SESSION_THRESHOLDS::MAX_BADMSGS_PER_MINUTE, 1);
                    LOG_DEBUG << "Bad message from user " << display_name();
                }
            }
            else
            {
                increment_metric(comm::SESSION_THRESHOLDS::MAX_BADMSGS_PER_MINUTE, 1);
                LOG_DEBUG << "User session id not found: " << display_name();
            }

            return 0;
        }

        // If for any reason we reach this point, we should drop the connection because none of the
        // valid cases match.
        LOG_DEBUG << "Dropping the user connection " << display_name();
        return -1;
    }

    /**
     * This gets hit every time a client disconnects from the HP public port.
     */
    void user_comm_session::handle_close()
    {
        // Session belongs to an authed user.
        if (challenge_status == comm::CHALLENGE_STATUS::CHALLENGE_VERIFIED)
            remove_user(pubkey);
    }

} // namespace usr