TransportFactory.hpp

// SPDX-FileCopyrightText: 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 <functional>
 
// Project includes
#include <dfx-core/transports/destinations/DestinationTransport.hpp>
#include <dfx-core/transports/sources/SourceTransport.hpp>
#include <dfx-utilities/StringMap.hpp>
 
namespace dfx::Graph
{
class Controller;

class TransportFactory
{
public:
    using SourceBuilder = std::move_only_function<Core::SourceTransportPtr (Core::Endpoint, nlohmann::json)>;

    using DestinationBuilder = std::move_only_function<Core::DestinationTransportPtr (Core::Endpoint, nlohmann::json)>;
 
public:
    TransportFactory();
 
public:
    void registerBuilder(std::string scheme, SourceBuilder sourceBuilder, DestinationBuilder destinationBuilder);

    Core::SourceTransportPtr createSource(std::string_view uri, nlohmann::json config,
                                          Controller & controller);

    Core::SourceTransportPtr createSource(Core::Endpoint::Descriptor descriptor, nlohmann::json config,
                                          Controller & controller);

    Core::DestinationTransportPtr createDestination(std::string_view uri, nlohmann::json config,
                                                    Controller & controller);

    Core::DestinationTransportPtr createDestination(Core::Endpoint::Descriptor descriptor, nlohmann::json config,
                                                    Controller & controller);
 
public:
    static Core::Endpoint::Descriptor descriptorFromString(std::string_view str);

    static Core::Endpoint endpointFromDescriptor(Core::Endpoint::Descriptor desc, Core::Port * port = nullptr);
 
private:
    Utils::UnorderedStringMap<SourceBuilder> _sourceBuilders;
    Utils::UnorderedStringMap<DestinationBuilder> _destinationBuilders;
};
} // !namespace dfx::Graph