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

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

#pragma once

#include "sstable_writer.hh"
#include "sstables_manager.hh"
#include "schema/schema_fwd.hh"
#include "mutation/mutation_fragment.hh"
#include "metadata_collector.hh"
#include "mutation/mutation_fragment_stream_validator.hh"

namespace sstables {

struct sstable_writer::writer_impl {
    sstable& _sst;
    const schema& _schema;
    const sstable_writer_config _cfg;
    // NOTE: _collector and _c_stats are used to generation of statistics file
    // when writing a new sstable.
    metadata_collector _collector;
    column_stats _c_stats;
    mutation_fragment_stream_validating_filter _validator;
    sstable_enabled_features _features = sstable_enabled_features::all();

    writer_impl(sstable& sst, const schema& schema, const sstable_writer_config& cfg)
        : _sst(sst)
        , _schema(schema)
        , _cfg(cfg)
        , _collector(_schema, sst.get_filename(), sst.manager().get_local_host_id())
        , _validator(format("sstable writer {}", _sst.get_filename()), _schema, _cfg.validation_level)
    {
        if (!cfg.correct_pi_block_width) {
            _features.disable(CorrectLastPiBlockWidth);
        }
        sst.set_features(_features);
    }

    virtual void consume_new_partition(const dht::decorated_key& dk) = 0;
    virtual void consume(tombstone t) = 0;
    virtual stop_iteration consume(static_row&& sr) = 0;
    virtual stop_iteration consume(clustering_row&& cr) = 0;
    virtual stop_iteration consume(range_tombstone_change&& rtc) = 0;
    virtual stop_iteration consume_end_of_partition() = 0;
    virtual void consume_end_of_stream() = 0;
    virtual ~writer_impl() {}
    void set_repaired_at(uint64_t repaired_at) {
        _collector.set_repaired_at(repaired_at);
    }
    virtual uint64_t data_file_position_for_tests() const = 0;
};

}