#pragma once

#include "common/entity_manager_interface.hpp"
#include "common/events.hpp"
#include "device/base_device.hpp"
#include "inventory/modbus_inventory.hpp"
#include "port/base_port.hpp"

#include <sdbusplus/async.hpp>

namespace phosphor::modbus::rtu
{

namespace InventoryIntf = phosphor::modbus::rtu::inventory;
namespace PortIntf = phosphor::modbus::rtu::port;
namespace ModbusIntf = phosphor::modbus::rtu;
namespace DeviceIntf = phosphor::modbus::rtu::device;
namespace EventIntf = phosphor::modbus::events;

class DeviceManager
{
  public:
    DeviceManager() = delete;
    DeviceManager(const DeviceManager&) = delete;
    DeviceManager& operator=(const DeviceManager&) = delete;
    DeviceManager(DeviceManager&&) = delete;
    DeviceManager& operator=(DeviceManager&&) = delete;

    explicit DeviceManager(sdbusplus::async::context& ctx);

  private:
    using inventory_device_map_t =
        std::unordered_map<std::string, std::unique_ptr<InventoryIntf::Device>>;

    using port_map_t =
        std::unordered_map<std::string, std::unique_ptr<PortIntf::BasePort>>;

    using device_map_t =
        std::unordered_map<std::string,
                           std::unique_ptr<DeviceIntf::BaseDevice>>;

    auto processConfigAdded(const sdbusplus::message::object_path& objectPath,
                            const std::string& interfaceName)
        -> sdbusplus::async::task<>;

    auto processPortAdded(const sdbusplus::message::object_path& objectPath,
                          const std::string& interfaceName)
        -> sdbusplus::async::task<>;

    auto processInventoryAdded(
        const sdbusplus::message::object_path& objectPath)
        -> sdbusplus::async::task<>;

    auto processDeviceAdded(const sdbusplus::message::object_path& objectPath,
                            const std::string& interfaceName)
        -> sdbusplus::async::task<>;

    auto processConfigRemoved(const sdbusplus::message::object_path& objectPath,
                              const std::string& interfaceName)
        -> sdbusplus::async::task<>;

    sdbusplus::async::context& ctx;
    entity_manager::EntityManagerInterface entityManager;
    EventIntf::Events events;
    inventory_device_map_t inventoryDevices;
    port_map_t ports;
    device_map_t devices; // Modbus devices
};

} // namespace phosphor::modbus::rtu