#define MS_CLASS "RTC::RTCP::CompoundPacket"
// #define MS_LOG_DEV_LEVEL 3

#include "RTC/RTCP/CompoundPacket.hpp"
#include "Logger.hpp"
#include "RTC/Consts.hpp"

namespace RTC
{
	namespace RTCP
	{
		/* Instance methods. */

		size_t CompoundPacket::GetSize()
		{
			size_t size{ 0 };

			if (this->senderReportPacket.GetCount() > 0u)
			{
				size += this->senderReportPacket.GetSize();
			}

			if (this->receiverReportPacket.GetCount() > 0u)
			{
				size += this->receiverReportPacket.GetSize();
			}

			if (this->sdesPacket.GetCount() > 0u)
			{
				size += this->sdesPacket.GetSize();
			}

			if (this->xrPacket.Begin() != this->xrPacket.End())
			{
				size += this->xrPacket.GetSize();
			}

			return size;
		}

		void CompoundPacket::Serialize(uint8_t* data)
		{
			MS_TRACE();

			this->header = data;

			// Fill it.
			size_t offset{ 0 };

			MS_ASSERT(
			  this->senderReportPacket.GetCount() > 0u || this->receiverReportPacket.GetCount() > 0u,
			  "no Sender or Receiver report present");

			if (this->senderReportPacket.GetCount() > 0u)
			{
				offset += this->senderReportPacket.Serialize(this->header);
			}

			if (this->receiverReportPacket.GetCount() > 0u)
			{
				offset += this->receiverReportPacket.Serialize(this->header + offset);
			}

			if (this->sdesPacket.GetCount() > 0u)
			{
				offset += this->sdesPacket.Serialize(this->header + offset);
			}

			if (this->xrPacket.Begin() != this->xrPacket.End())
			{
				this->xrPacket.Serialize(this->header + offset);
			}
		}

		bool CompoundPacket::Add(
		  SenderReport* senderReport,
		  SdesChunk* sdesChunk,
		  DelaySinceLastRr::SsrcInfo* delaySinceLastRrSsrcInfo)
		{
			// Add the items into the packet.

			if (senderReport)
			{
				this->senderReportPacket.AddReport(senderReport);
			}

			if (sdesChunk)
			{
				this->sdesPacket.AddChunk(sdesChunk);
			}

			if (delaySinceLastRrSsrcInfo)
			{
				// Add a DLRR block into the XR packet if no present.
				if (!this->delaySinceLastRr)
				{
					this->delaySinceLastRr = new RTC::RTCP::DelaySinceLastRr();
					this->xrPacket.AddReport(this->delaySinceLastRr);
				}

				this->delaySinceLastRr->AddSsrcInfo(delaySinceLastRrSsrcInfo);
			}

			// New items can hold in the packet, report it.
			if (GetSize() <= RTC::Consts::RtcpPacketMaxSize)
			{
				return true;
			}

			// New items can not hold in the packet, remove them,
			// delete and report it.

			if (senderReport)
			{
				this->senderReportPacket.RemoveReport(senderReport);
				delete senderReport;
			}

			if (sdesChunk)
			{
				this->sdesPacket.RemoveChunk(sdesChunk);
				delete sdesChunk;
			}

			if (delaySinceLastRrSsrcInfo)
			{
				// NOTE: This method deletes the removed instances in place.
				this->delaySinceLastRr->RemoveLastSsrcInfos(1);
			}

			return false;
		}

		bool CompoundPacket::Add(
		  std::vector<SenderReport*>& senderReports,
		  std::vector<SdesChunk*>& sdesChunks,
		  std::vector<DelaySinceLastRr::SsrcInfo*>& delaySinceLastRrSsrcInfos)
		{
			// Add the items into the packet.

			for (auto* report : senderReports)
			{
				this->senderReportPacket.AddReport(report);
			}

			for (auto* chunk : sdesChunks)
			{
				this->sdesPacket.AddChunk(chunk);
			}

			// Add a DLRR block into the XR packet if no present.
			if (!delaySinceLastRrSsrcInfos.empty() && !this->delaySinceLastRr)
			{
				this->delaySinceLastRr = new RTC::RTCP::DelaySinceLastRr();
				this->xrPacket.AddReport(this->delaySinceLastRr);
			}

			for (auto* ssrcInfo : delaySinceLastRrSsrcInfos)
			{
				this->delaySinceLastRr->AddSsrcInfo(ssrcInfo);
			}

			// New items can hold in the packet, report it.
			if (GetSize() <= RTC::Consts::RtcpPacketMaxSize)
			{
				return true;
			}

			// New items can not hold in the packet, remove them,
			// delete and report it.

			for (auto* report : senderReports)
			{
				this->senderReportPacket.RemoveReport(report);
				delete report;
			}

			for (auto* chunk : sdesChunks)
			{
				this->sdesPacket.RemoveChunk(chunk);
				delete chunk;
			}

			if (!delaySinceLastRrSsrcInfos.empty())
			{
				// NOTE: This method deletes the instances in place.
				this->delaySinceLastRr->RemoveLastSsrcInfos(delaySinceLastRrSsrcInfos.size());
			}

			return false;
		}

		bool CompoundPacket::Add(
		  std::vector<ReceiverReport*>& receiverReports,
		  ReceiverReferenceTime* receiverReferenceTimeReport)
		{
			// Add the items into the packet.

			for (auto* report : receiverReports)
			{
				this->receiverReportPacket.AddReport(report);
			}

			if (receiverReferenceTimeReport)
			{
				this->xrPacket.AddReport(receiverReferenceTimeReport);
			}

			// New items can hold in the packet, report it.
			if (GetSize() <= RTC::Consts::RtcpPacketMaxSize)
			{
				return true;
			}

			// New items can not hold in the packet, remove them,
			// delete and report it.

			for (auto* report : receiverReports)
			{
				this->receiverReportPacket.RemoveReport(report);
				delete report;
			}

			if (receiverReferenceTimeReport)
			{
				this->xrPacket.RemoveReport(receiverReferenceTimeReport);
				delete receiverReferenceTimeReport;
			}

			return false;
		}

		void CompoundPacket::Dump(int indentation)
		{
			MS_TRACE();

			MS_DUMP_CLEAN(indentation, "<CompoundPacket>");

			if (HasSenderReport())
			{
				this->senderReportPacket.Dump(indentation + 1);

				if (this->receiverReportPacket.GetCount() != 0u)
				{
					this->receiverReportPacket.Dump(indentation + 1);
				}
			}
			else
			{
				this->receiverReportPacket.Dump(indentation + 1);
			}

			if (this->sdesPacket.GetCount() != 0u)
			{
				this->sdesPacket.Dump(indentation + 1);
			}

			if (this->xrPacket.Begin() != this->xrPacket.End())
			{
				this->xrPacket.Dump(indentation + 1);
			}

			MS_DUMP_CLEAN(indentation, "</CompoundPacket>");
		}

		void CompoundPacket::AddSenderReport(SenderReport* report)
		{
			MS_TRACE();

			this->senderReportPacket.AddReport(report);
		}

		void CompoundPacket::AddReceiverReport(ReceiverReport* report)
		{
			MS_TRACE();

			this->receiverReportPacket.AddReport(report);
		}

		void CompoundPacket::AddSdesChunk(SdesChunk* chunk)
		{
			MS_TRACE();

			this->sdesPacket.AddChunk(chunk);
		}
	} // namespace RTCP
} // namespace RTC
