// SPDX-License-Identifier: Apache-2.0

// SPDX-FileCopyrightText: Copyright OpenBMC Authors

#pragma once

#ifdef HAVE_ZSTD

#include <zstd.h>

#endif

#include <boost/beast/core/flat_buffer.hpp>

#include <optional>

#include <span>

namespace bmcweb

{

class ZstdCompressor

{

    boost::beast::flat_buffer compressionBuf;

#ifdef HAVE_ZSTD

    ZSTD_CCtx* cctx = nullptr;

#endif

  public:

    ZstdCompressor(const ZstdCompressor&) = delete;

    ZstdCompressor(ZstdCompressor&&) = delete;

    ZstdCompressor& operator=(const ZstdCompressor&) = delete;

    ZstdCompressor& operator=(ZstdCompressor&&) = delete;

    ZstdCompressor() = default;

    // must be called before compress.

    bool init(size_t sourceSize);

    std::optional<std::span<const uint8_t>> compress(

        std::span<const uint8_t> buffIn, bool more);

    ~ZstdCompressor();

};

} // namespace bmcweb