#ifndef BYTEME_RAW_BUFFER_WRITER_HPP
#define BYTEME_RAW_BUFFER_WRITER_HPP

#include <cstddef>
#include <vector>

#include "Writer.hpp"
#include "utils.hpp"

/**
 * @file RawBufferWriter.hpp
 *
 * @brief Write bytes to a raw buffer without any extra transformations.
 */

namespace byteme {

/**
 * @brief Options for the `RawBufferWriter` constructor.
 */
struct RawBufferWriterOptions {
    /**
     * Initial size of the output buffer to reserve.
     * Setting this to an estimate of the total number of written bytes may avoid unnecessary memory allocations.
     */
    std::size_t reserve = 0;
};

/**
 * @brief Write bytes to a raw buffer.
 *
 * This class will append bytes to an internal instance of a `std::vector` without any further transformations.
 * Not much else to say here.
 */
class RawBufferWriter final : public Writer {
public:
    /**
     * @param options Further options.
     */
    RawBufferWriter(const RawBufferWriterOptions& options) {
        output.reserve(options.reserve);
    }

public:
    using Writer::write;

    void write(const unsigned char* buffer, std::size_t n) {
        output.insert(output.end(), buffer, buffer + n);
    }

    void finish() {}

public:
    /**
     * @cond
     */
    // Exposed for back-compatibility only.
    std::vector<unsigned char> output;
    /**
     * @endcond
     */

    /**
     * @return Contents of the output buffer.
     * This should only be used after `finish()` is called.
     */
    std::vector<unsigned char>& get_output() {
        return output;
    }
};

}

#endif
