/*
 * Copyright (C) 2015-present ScyllaDB
 */

/*
 * SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
 */

#include <seastar/core/coroutine.hh>

#include "gossiper.hh"
#include "api/api-doc/gossiper.json.hh"
#include "gms/endpoint_state.hh"
#include "gms/gossiper.hh"
#include "api/api.hh"

namespace api {
using namespace seastar::httpd;
using namespace json;

void set_gossiper(http_context& ctx, routes& r, gms::gossiper& g) {
    httpd::gossiper_json::get_down_endpoint.set(r, [&g] (std::unique_ptr<request> req) -> future<json::json_return_type> {
        auto res = co_await g.get_unreachable_members_synchronized();
        co_return json::json_return_type(res | std::views::transform([] (auto& ep) { return fmt::to_string(ep); }) | std::ranges::to<std::vector>());
    });


    httpd::gossiper_json::get_live_endpoint.set(r, [&g] (std::unique_ptr<request> req) -> future<json::json_return_type> {
        auto res = co_await g.get_live_members_synchronized();
        co_return json::json_return_type(res | std::views::transform([] (auto& ep) { return fmt::to_string(ep); }) | std::ranges::to<std::vector>());
    });

    httpd::gossiper_json::get_endpoint_downtime.set(r, [&g] (std::unique_ptr<request> req) -> future<json::json_return_type> {
        gms::inet_address ep(req->get_path_param("addr"));
        // synchronize unreachable_members on all shards
        co_await g.get_unreachable_members_synchronized();
        co_return g.get_endpoint_downtime(g.get_host_id(ep));
    });

    httpd::gossiper_json::get_current_generation_number.set(r, [&g] (std::unique_ptr<http::request> req) {
        gms::inet_address ep(req->get_path_param("addr"));
        return g.get_current_generation_number(g.get_host_id(ep)).then([] (gms::generation_type res) {
            return make_ready_future<json::json_return_type>(res.value());
        });
    });

    httpd::gossiper_json::get_current_heart_beat_version.set(r, [&g] (std::unique_ptr<http::request> req) {
        gms::inet_address ep(req->get_path_param("addr"));
        return g.get_current_heart_beat_version(g.get_host_id(ep)).then([] (gms::version_type res) {
            return make_ready_future<json::json_return_type>(res.value());
        });
    });

    httpd::gossiper_json::assassinate_endpoint.set(r, [&g](std::unique_ptr<http::request> req) {
        return g.assassinate_endpoint(req->get_path_param("addr")).then([] {
            return make_ready_future<json::json_return_type>(json_void());
        });
    });

    httpd::gossiper_json::force_remove_endpoint.set(r, [&g](std::unique_ptr<http::request> req) {
        gms::inet_address ep(req->get_path_param("addr"));
        return g.force_remove_endpoint(g.get_host_id(ep), gms::null_permit_id).then([] () {
            return make_ready_future<json::json_return_type>(json_void());
        });
    });
}

void unset_gossiper(http_context& ctx, routes& r) {
    httpd::gossiper_json::get_down_endpoint.unset(r);
    httpd::gossiper_json::get_live_endpoint.unset(r);
    httpd::gossiper_json::get_endpoint_downtime.unset(r);
    httpd::gossiper_json::get_current_generation_number.unset(r);
    httpd::gossiper_json::get_current_heart_beat_version.unset(r);
    httpd::gossiper_json::assassinate_endpoint.unset(r);
    httpd::gossiper_json::force_remove_endpoint.unset(r);
}

}