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

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

#pragma once

#include "mutation/mutation.hh"
#include "db/system_keyspace.hh"
#include "service/session.hh"
#include "locator/tablets.hh"

namespace replica {

class tablet_mutation_builder {
    api::timestamp_type _ts;
    schema_ptr _s;
    mutation _m;
private:
    clustering_key get_ck(dht::token last_token) {
        return clustering_key::from_single_value(*_s, data_value(dht::token::to_int64(last_token)).serialize_nonnull());
    }
public:
    tablet_mutation_builder(api::timestamp_type ts, table_id table)
            : _ts(ts)
            , _s(db::system_keyspace::tablets())
            , _m(_s, partition_key::from_single_value(*_s,
                    data_value(table.uuid()).serialize_nonnull()
            ))
    { }

    tablet_mutation_builder& set_new_replicas(dht::token last_token, locator::tablet_replica_set replicas);
    tablet_mutation_builder& set_replicas(dht::token last_token, locator::tablet_replica_set replicas);
    tablet_mutation_builder& set_stage(dht::token last_token, locator::tablet_transition_stage stage);
    tablet_mutation_builder& set_transition(dht::token last_token, locator::tablet_transition_kind);
    tablet_mutation_builder& set_session(dht::token last_token, service::session_id);
    tablet_mutation_builder& del_session(dht::token last_token);
    tablet_mutation_builder& del_transition(dht::token last_token);
    tablet_mutation_builder& set_resize_decision(locator::resize_decision, const gms::feature_service&);
    tablet_mutation_builder& set_repair_scheduler_config(locator::repair_scheduler_config);
    tablet_mutation_builder& set_repair_time(dht::token last_token, db_clock::time_point repair_time);
    tablet_mutation_builder& set_sstables_repair_at(dht::token last_token, int64_t sstables_repaired_at);
    tablet_mutation_builder& set_repair_task_info(dht::token last_token, locator::tablet_task_info info, const gms::feature_service& features);
    tablet_mutation_builder& del_repair_task_info(dht::token last_token, const gms::feature_service& features);
    tablet_mutation_builder& set_migration_task_info(dht::token last_token, locator::tablet_task_info info, const gms::feature_service& features);
    tablet_mutation_builder& del_migration_task_info(dht::token last_token, const gms::feature_service& features);
    tablet_mutation_builder& set_resize_task_info(locator::tablet_task_info info, const gms::feature_service& features);
    tablet_mutation_builder& del_resize_task_info(const gms::feature_service& features);
    tablet_mutation_builder& set_base_table(table_id base_table);

    mutation build() {
        return std::move(_m);
    }
};

} // namespace replica