#pragma once

/**
 * \file MessageDispatcher.h
 * \brief File containing the sp::MessageDispatcher class
 */

#include <map>
#include <memory>

namespace sp {

/**
 * \class MessageDispatcher
 * \brief Class used to dispatch messages
 */
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
class MessageDispatcher {
  private:
	std::map<MessageIdType, std::vector<MessageHandler*>> m_Handlers;

  public:
	using MessageBaseType = MessageBase;

	/**
	 * \brief Constructor
	 */
	MessageDispatcher() {}

	/**
	 * \brief Dispatch a packet
	 * \param packet The packet to dispatch
	 */
	void Dispatch(const MessageBase& a_Message);

	/**
	 * \brief Register a packet handler
	 * \param type The packet type
	 * \param handler The packet handler
	 */
	void RegisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler);
	
	/**
	 * \brief Unregister a packet handler
	 * \param type The packet type
	 * \param handler The packet handler
	 */
	void UnregisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler);
	
	/**
	 * \brief Unregister a packet handler
	 * \param handler The packet handler
	 */
	void UnregisterHandler(MessageHandler* a_Handler);
};

#include <sp/protocol/message/MessageDispatcherImpl.inl>

}  // namespace sp