#pragma once

#include <sp/common/DataBuffer.h>
#include <sp/io/IoInterface.h>
#include <sp/io/MessageEncapsulator.h>
#include <vector>

namespace sp {

template <typename TMessageFactory>
class MessageStream {
  private:
	std::vector<MessageEncapsulator> m_Encapsulators;
	std::shared_ptr<IoInterface> m_Stream;

	using MessageBaseType = typename TMessageFactory::MessageBaseType;
	using MessageIdType = typename MessageBaseType::MessageIdType;

  public:
	MessageStream(std::shared_ptr<IoInterface>&& a_Stream) : m_Stream(std::move(a_Stream)) {}

	std::unique_ptr<MessageBaseType> ReadMessage();
	std::unique_ptr<MessageBaseType> ReadMessage(MessageIdType a_Id);

	void WriteMessage(const MessageBaseType& a_Message, bool a_WriteId = true);

  private:
	DataBuffer ReadAndDecapsulate();
	std::unique_ptr<MessageBaseType> MakeMessage(DataBuffer& buffer, MessageIdType a_Id);
};

}  // namespace sp

#include <sp/io/MessageStream.inl>