// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

#pragma once

// Includes
#include "DDSDataTypes.h"
#include "Listener.h"

#include "datatypes/VehicleDataSourceTypes.h"
// DDS lib related headers
#include "SensorDataListener.h"
#include <fastdds/dds/domain/DomainParticipant.hpp>
#include <fastdds/dds/domain/DomainParticipantFactory.hpp>
#include <fastdds/dds/subscriber/DataReader.hpp>
#include <fastdds/dds/subscriber/DataReaderListener.hpp>
#include <fastdds/dds/subscriber/SampleInfo.hpp>
#include <fastdds/dds/subscriber/Subscriber.hpp>
#include <fastdds/dds/subscriber/qos/DataReaderQos.hpp>
#include <fastdds/dds/topic/TypeSupport.hpp>
#include <memory>

namespace Aws
{
namespace IoTFleetWise
{
namespace VehicleNetwork
{
using namespace eprosima::fastdds::dds;
using namespace Aws::IoTFleetWise::Platform::Linux;

/**
 * @brief Abstract DDS Subscriber Interface. Every DDS Node that wants to subscribe to data on a
 * DDS topic should implement from this interface.
 * This interface raises a notification from the SensorDataListener upon data arrival.
 */
class IDDSSubscriber : public ThreadListeners<SensorDataListener>, public DataReaderListener
{
public:
    ~IDDSSubscriber() override = default;

    /**
     * @brief Inits a DDS Node subscriber using the source properties.
     * @param dataSourceConfig DDS Source configuration.
     * @return True if the subscriber has been setup correctly. This includes that the domain
     * has been found and the participant has been attached the domain, and the topic has been
     * created.
     */
    virtual bool init( const DDSDataSourceConfig &dataSourceConfig ) = 0;

    /**
     * @brief Connect to the DDS Topic and start the worker thread of the subscriber.
     * @return True if the thread is active and connection has been setup.
     */
    virtual bool connect() = 0;

    /**
     * @brief Stops the thread and disconnect from the DDS Topic.
     * @return True if the thread is stopped and connection has been closed.
     */
    virtual bool disconnect() = 0;

    /**
     * @brief Checks if domain participant is alive and still connected to the DDS domain.
     * @return True if the connection is healthy.
     */
    virtual bool isAlive() = 0;

    /**
     * @return the DDS Domain ID
     */
    inline DDSDomainID
    getDDSDomainID() const
    {
        return mDDSDomainID;
    }

    /**
     * @return the DDS Topic
     */
    inline DDSTopicName
    getDDSTopic() const
    {
        return mDDSTopic;
    }

    /**
     * @return the Subscriber name
     */
    inline DDSWriterName
    getDDSSubscriberName() const
    {
        return mDDSReaderName;
    }

    /**
     * @return the Source Type
     */
    inline SensorSourceType
    getSensorSourceType() const
    {
        return mType;
    }

    /**
     * @return the DDS Topic QoS
     */
    inline DDSTopicQoS
    getDDSTopicQoS() const
    {
        return mTopicQoS;
    }

protected:
    /**
     * @brief Thread safe Channel ID generator
     * @return returns a unique identifier of a channel.
     */
    static uint32_t
    generateChannelID()
    {
        static std::atomic<uint32_t> channelID{ 0 };
        return ++channelID;
    }

    SensorSourceType mType;
    uint32_t mID;
    DDSDomainID mDDSDomainID;
    DDSTopicName mDDSTopic;
    DDSWriterName mDDSReaderName;
    std::string mTemporaryCacheLocation;
    DDSTopicQoS mTopicQoS;
};
using DDSSubscriberPtr = std::unique_ptr<IDDSSubscriber>;
} // namespace VehicleNetwork
} // namespace IoTFleetWise
} // namespace Aws