FileSink.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 <fstream>
 
// Project includes
#include <dfx-core/Node.hpp>
#include <dfx-utilities/EnumString.hpp>
#include <dfx-utilities/FileSystem.hpp>
 
namespace dfx::Node
{
class FileSink : public Core::Node
{
    DFX_NODE("FileSink")
 
public:
    enum class Mode
    {
        Append,
        Truncate,
    };
 
public:
    FileSink(Id id, std::string name);
 
protected:
    void startImpl() override;
 
    void initializeImpl(nlohmann::json config) override;
    void handleMessage(Core::InputPort const & port, Core::MessagePtr message) override;
 
private:
    fs::path _path;
    Mode     _mode = Mode::Append;
    bool     _flushAfterWrite = false;
    bool     _appendNewline = false;
 
private:
    std::ofstream _ofs;
};
} // !namespace dfx::Node
 
DECLARE_ENUM_STRING_FUNCTIONS(dfx::Node::FileSink::Mode);