#pragma once

#include <memory>

#include <QHash>
#include <QString>
#include <QStringList>

#include "core/plugin_system/IProtocolFramerPlugin.h"
#include "core/plugin_system/IProtocolMapperPlugin.h"
#include "core/plugin_system/IProtocolPlugin.h"

namespace softbus::core::plugin_system
{

class PluginManager
{
public:
    PluginManager() = default;

    void registerProtocolPlugin(std::shared_ptr<IProtocolPlugin> plugin);
    std::shared_ptr<IProtocolPlugin> selectProtocolPlugin(const QString& protocolHint) const;
    void unregisterProtocolPlugin(const QString& pluginId);
    void registerProtocolFramerPlugin(std::shared_ptr<IProtocolFramerPlugin> plugin);
    std::shared_ptr<IProtocolFramerPlugin> selectProtocolFramerPlugin(const QString& protocolHint) const;
    void unregisterProtocolFramerPlugin(const QString& pluginId);
    void registerProtocolMapperPlugin(std::shared_ptr<IProtocolMapperPlugin> plugin);
    std::shared_ptr<IProtocolMapperPlugin> selectProtocolMapperPlugin(const QString& protocolHint) const;
    void unregisterProtocolMapperPlugin(const QString& pluginId);
    QStringList registeredProtocolPluginIds() const;
    QStringList registeredProtocolFramerPluginIds() const;
    QStringList registeredProtocolMapperPluginIds() const;

private:
    QHash<QString, std::shared_ptr<IProtocolPlugin>> m_protocolPlugins;
    QHash<QString, std::shared_ptr<IProtocolFramerPlugin>> m_protocolFramerPlugins;
    QHash<QString, std::shared_ptr<IProtocolMapperPlugin>> m_protocolMapperPlugins;
};

} // namespace softbus::core::plugin_system