Sink.hpp

// SPDX-FileCopyrightText: 2025 Vincent Leroy
// SPDX-License-Identifier: MIT
//
// This file is part of dfx.
//
// Licensed under the MIT License. See the LICENSE file in the project root
// for full license information.
 
#pragma once
 
// Standard includes
#include <cstdint>
#include <memory>
#include <vector>
 
// Project includes
#include <dfx-fdwatch/Poller.hpp>
#include <dfx-utilities/CopyMoveControl.hpp>
 
namespace dfx::Pcapng
{
class Sink
{
public:
    Sink() noexcept = default;

    DISABLE_COPY_AND_MOVE(Sink);

    virtual ~Sink() = default;

    virtual void init(std::vector<uint8_t> initialHeader) = 0;

    void setPoller(FdWatch::Poller * poller) { _poller = poller; }

    virtual void write(std::vector<uint8_t> data) = 0;
 
protected:
    FdWatch::Poller * _poller = nullptr;
};

using SinkPtr = std::unique_ptr<Sink>;
} // !namespace dfx::Pcapng