Capture.hpp

// SPDX-FileCopyrightText: 2025-2026 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 <memory>
 
// Project includes
#include "sinks/FileSink.hpp"
#include "sinks/PipeSink.hpp"
#include "sinks/TcpSink.hpp"
#include <dfx-hooks/Graph.hpp>
#include <dfx-hooks/Port.hpp>
#include <dfx-utilities/Thread.hpp>
 
namespace dfx::Core
{
using NodeWPtr = std::weak_ptr<Node>;
} // !namespace dfx::Core
 
namespace dfx::Pcapng
{
class Worker;

class Capture : public Hooks::Port, public Hooks::Graph
{
public:
    explicit Capture(bool getHwAndOsInfo = true);

    ~Capture();

    template<typename ... Args>
    void setFileSink(Args && ... args)
    { _setFileSinkImpl(std::make_unique<FileSink>(std::forward<Args>(args)...)); }

    void unsetFileSink();

    template<typename ... Args>
    void setPipeSink(Args && ... args)
    { _setPipeSinkImpl(std::make_unique<PipeSink>(std::forward<Args>(args)...)); }

    void unsetPipeSink();

    template<typename ... Args>
    void setTcpSink(Args && ... args)
    { _setTcpSinkImpl(std::make_unique<TcpSink>(std::forward<Args>(args)...)); }

    void unsetTcpSink();
 
public:
    void setUserAppInfo(std::string info);
 
public:
    void onMessageSent(Core::OutputPort const & port, Core::Message const & message) override;

    void onMessageReceived(Core::InputPort const & port, Core::Message const & message) override;

    void onNodeCreated(Core::NodePtr node) override;

    void onNodeRemoved(Core::NodePtr node) override;

    void onChannelCreated(Core::ChannelPtr) override {}
    void onChannelRemoved(Core::ChannelPtr) override {}
 
private:
    void _setFileSinkImpl(SinkPtr sink);
    void _setPipeSinkImpl(SinkPtr sink);
    void _setTcpSinkImpl(SinkPtr sink);
 
    Worker & _worker();
    Worker const & _worker() const;
 
private:
    std::vector<Core::NodeWPtr> _nodes;
    Utils::Thread _thread;
};
} // !namespace dfx::Pcapng