#pragma once

#include <boost/container/flat_map.hpp>

#include <sdbusplus/asio/connection.hpp>

#include <cstdlib>

#include <functional>

#include <memory>

namespace estoraged

{

using BasicVariantType =

    std::variant<std::vector<std::string>, std::string, int64_t, uint64_t,

                 double, int32_t, uint32_t, int16_t, uint16_t, uint8_t, bool>;

/* Map of properties for the config interface. */

using StorageData = boost::container::flat_map<std::string, BasicVariantType>;

using ManagedStorageType =

    boost::container::flat_map<sdbusplus::object_path, StorageData>;

const constexpr char* emmcConfigInterface =

    "xyz.openbmc_project.Configuration.EmmcDevice";

/** @class GetStorageConfiguration

 *  @brief Object used to find Entity Manager config objects.

 *  @details eStoraged will create a new D-Bus object for each config object.

 */

class GetStorageConfiguration :

    public std::enable_shared_from_this<GetStorageConfiguration>

{

  public:

    /** @brief Constructor for GetStorageConfiguration

     *

     *  @param[in] connection - shared_ptr to D-Bus connection object.

     *  @param[in] callbackFunc - callback to run after finding the config

     *    objects.

     */

    GetStorageConfiguration(

        std::shared_ptr<sdbusplus::asio::connection> connection,

        std::function<void(ManagedStorageType& resp)>&& callbackFunc) :

        dbusConnection(std::move(connection)), callback(std::move(callbackFunc))

    {}

    GetStorageConfiguration& operator=(const GetStorageConfiguration&) = delete;

    GetStorageConfiguration(const GetStorageConfiguration&) = delete;

    GetStorageConfiguration(GetStorageConfiguration&&) = default;

    GetStorageConfiguration& operator=(GetStorageConfiguration&&) = default;

    /** @brief Destructor for GetStorageConfiguration.

     *  @details This will execute the callback provided to the constructor.

     */

    ~GetStorageConfiguration();

    /** @brief Find the Entity Manager config objects for eStoraged. */

    void getConfiguration();

    /** @brief Get the D-Bus properties from the config object.

     *

     *  @param[in] path - D-Bus object path of the config object.

     *  @param[in] owner - D-Bus service that owns the config object.

     *  @param[in] retries - (optional) Number of times to retry, if needed.

     */

    void getStorageInfo(const std::string& path, const std::string& owner,

                        size_t retries = 5);

    /** @brief Map containing config objects with corresponding properties. */

    ManagedStorageType respData;

    /** @brief Connection to D-Bus. */

    std::shared_ptr<sdbusplus::asio::connection> dbusConnection;

  private:

    /** @brief callback to process the config object data in respData. */

    std::function<void(ManagedStorageType& resp)> callback;

};

} // namespace estoraged