trade_handler.hpp

/* -*- mode: c++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 *
 * Copyright (C) 2026 Marco Craveiro <marco.craveiro@gmail.com>
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 3 of the License, or (at your option) any later
 * version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */
#ifndef ORES_TRADING_MESSAGING_TRADE_HANDLER_HPP
#define ORES_TRADING_MESSAGING_TRADE_HANDLER_HPP
 
#include <optional>
#include <boost/uuid/string_generator.hpp>
#include "ores.logging/make_logger.hpp"
#include "ores.nats/domain/message.hpp"
#include "ores.nats/service/client.hpp"
#include "ores.database/domain/context.hpp"
#include "ores.security/jwt/jwt_authenticator.hpp"
#include "ores.service/messaging/handler_helpers.hpp"
#include "ores.service/service/request_context.hpp"
#include "ores.trading.api/messaging/trade_protocol.hpp"
#include "ores.trading.core/service/activity_type_service.hpp"
#include "ores.trading.core/service/trade_service.hpp"
#include "ores.utility/uuid/tenant_id.hpp"
 
namespace ores::trading::messaging {
 
namespace {
inline auto& trade_handler_lg() {
    static auto instance = ores::logging::make_logger(
        "ores.trading.messaging.trade_handler");
    return instance;
}
} // namespace
 
using ores::service::messaging::reply;
using ores::service::messaging::decode;
using ores::service::messaging::error_reply;
using ores::service::messaging::has_permission;
using namespace ores::logging;
 
class trade_handler {
public:
    trade_handler(ores::nats::service::client& nats,
        ores::database::context ctx,
        std::optional<ores::security::jwt::jwt_authenticator> verifier)
        : nats_(nats), ctx_(std::move(ctx)), verifier_(std::move(verifier)) {}
 
    void list_activity_types(ores::nats::message msg) {
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Handling " << msg.subject;
        auto req_ctx_expected = ores::service::service::make_request_context(
            ctx_, msg, verifier_);
        if (!req_ctx_expected) {
            error_reply(nats_, msg, req_ctx_expected.error());
            return;
        }
        const auto& req_ctx = *req_ctx_expected;
        // Activity types are system-level configuration; look them up under
        // the system tenant so all tenants see the same standard set.
        const auto sys_ctx = req_ctx.with_tenant(
            ores::utility::uuid::tenant_id::system(), req_ctx.actor());
        service::activity_type_service svc(sys_ctx);
        get_activity_types_response resp;
        try {
            resp.activity_types = svc.list_types();
        } catch (...) {}
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Completed " << msg.subject;
        reply(nats_, msg, resp);
    }
 
    void list(ores::nats::message msg) {
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Handling " << msg.subject;
        auto ctx_expected = ores::service::service::make_request_context(
            ctx_, msg, verifier_);
        if (!ctx_expected) {
            error_reply(nats_, msg, ctx_expected.error());
            return;
        }
        const auto& ctx = *ctx_expected;
        service::trade_service svc(ctx);
        get_trades_response resp;
        try {
            if (auto req = decode<get_trades_request>(msg)) {
                const auto offset =
                    static_cast<std::uint32_t>(req->offset);
                const auto limit =
                    static_cast<std::uint32_t>(req->limit);
                if (!req->book_id.empty()) {
                    boost::uuids::string_generator gen;
                    const auto book_uuid = gen(req->book_id);
                    resp.trades = svc.list_trades_filtered(
                        offset, limit,
                        std::optional<boost::uuids::uuid>(book_uuid),
                        std::nullopt);
                    resp.total_available_count =
                        static_cast<int>(svc.count_trades_filtered(
                            std::optional<boost::uuids::uuid>(book_uuid),
                            std::nullopt));
                } else {
                    resp.trades = svc.list_trades(offset, limit);
                    resp.total_available_count =
                        static_cast<int>(svc.count_trades());
                }
            }
        } catch (...) {}
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Completed " << msg.subject;
        reply(nats_, msg, resp);
    }
 
    void save(ores::nats::message msg) {
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Handling " << msg.subject;
        auto ctx_expected = ores::service::service::make_request_context(
            ctx_, msg, verifier_);
        if (!ctx_expected) {
            error_reply(nats_, msg, ctx_expected.error());
            return;
        }
        const auto& ctx = *ctx_expected;
        if (!has_permission(ctx, "trading::trades:write")) {
            error_reply(nats_, msg, ores::service::error_code::forbidden);
            return;
        }
        service::trade_service svc(ctx);
        if (auto req = decode<save_trade_request>(msg)) {
            try {
                svc.save_trades(req->trades);
                BOOST_LOG_SEV(trade_handler_lg(), debug)
                    << "Completed " << msg.subject;
                reply(nats_, msg, save_trade_response{.success = true});
            } catch (const std::exception& e) {
                BOOST_LOG_SEV(trade_handler_lg(), error)
                    << msg.subject << " failed: " << e.what();
                reply(nats_, msg, save_trade_response{
                    .success = false, .message = e.what()});
            }
        } else {
            BOOST_LOG_SEV(trade_handler_lg(), warn)
                << "Failed to decode: " << msg.subject;
        }
    }
 
    void remove(ores::nats::message msg) {
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Handling " << msg.subject;
        auto ctx_expected = ores::service::service::make_request_context(
            ctx_, msg, verifier_);
        if (!ctx_expected) {
            error_reply(nats_, msg, ctx_expected.error());
            return;
        }
        const auto& ctx = *ctx_expected;
        if (!has_permission(ctx, "trading::trades:delete")) {
            error_reply(nats_, msg, ores::service::error_code::forbidden);
            return;
        }
        service::trade_service svc(ctx);
        if (auto req = decode<delete_trade_request>(msg)) {
            try {
                for (const auto& id : req->ids)
                    svc.remove_trade(id);
                BOOST_LOG_SEV(trade_handler_lg(), debug)
                    << "Completed " << msg.subject;
                reply(nats_, msg, delete_trade_response{.success = true});
            } catch (const std::exception& e) {
                BOOST_LOG_SEV(trade_handler_lg(), error)
                    << msg.subject << " failed: " << e.what();
                reply(nats_, msg, delete_trade_response{
                    .success = false, .message = e.what()});
            }
        } else {
            BOOST_LOG_SEV(trade_handler_lg(), warn)
                << "Failed to decode: " << msg.subject;
        }
    }
 
    void history(ores::nats::message msg) {
        BOOST_LOG_SEV(trade_handler_lg(), debug)
            << "Handling " << msg.subject;
        auto ctx_expected = ores::service::service::make_request_context(
            ctx_, msg, verifier_);
        if (!ctx_expected) {
            error_reply(nats_, msg, ctx_expected.error());
            return;
        }
        const auto& ctx = *ctx_expected;
        service::trade_service svc(ctx);
        if (auto req = decode<get_trade_history_request>(msg)) {
            try {
                auto versions = svc.get_trade_history(req->id);
                BOOST_LOG_SEV(trade_handler_lg(), debug)
                    << "Completed " << msg.subject;
                reply(nats_, msg, get_trade_history_response{
                    .success = true,
                    .versions = std::move(versions)});
            } catch (const std::exception& e) {
                BOOST_LOG_SEV(trade_handler_lg(), error)
                    << msg.subject << " failed: " << e.what();
                reply(nats_, msg, get_trade_history_response{
                    .success = false, .message = e.what()});
            }
        } else {
            BOOST_LOG_SEV(trade_handler_lg(), warn)
                << "Failed to decode: " << msg.subject;
        }
    }
 
private:
    ores::nats::service::client& nats_;
    ores::database::context ctx_;
    std::optional<ores::security::jwt::jwt_authenticator> verifier_;
};
 
} // namespace ores::trading::messaging
 
#endif