#pragma once

#include <boost/pfr.hpp>
#include <sp/common/ByteSwapping.h>
#include <sp/common/DataBuffer.h>
#include <sp/protocol/BitField.h>

namespace sp {
namespace details {

template <typename T>
void WriteBitField(DataBuffer& a_Buffer, std::uint64_t& a_DataRaw, std::size_t& a_Offset) {
	T filled = static_cast<T>(a_DataRaw);
	ToNetwork(filled);
	a_Buffer << filled;
	a_Offset = 0;
	a_DataRaw = 0;
}

template <typename T, std::size_t BitSize>
void WriteField(DataBuffer& a_Buffer, const BitField<T, BitSize>& a_Data, std::uint64_t& a_DataRaw, std::size_t& a_Offset) {
	T cut = *a_Data & ((1 << a_Data.GetBitSize()) - 1);
	std::size_t pushCount = sizeof(T) * 8 - a_Offset - a_Data.GetBitSize();
	a_DataRaw |= cut << pushCount;
	a_Offset += a_Data.GetBitSize();
	if (a_Offset == sizeof(T) * 8) {
		WriteBitField<T>(a_Buffer, a_DataRaw, a_Offset);
	}
}

template <typename T>
void WriteField(DataBuffer& a_Buffer, const T& a_Data, std::uint64_t& a_DataRaw, std::size_t& a_Offset) {
	T swapped = a_Data;
	ToNetwork(swapped);
	a_Buffer << swapped;
}

template <typename T, std::size_t BitSize>
void ReadField(DataBuffer& a_Buffer, BitField<T, BitSize>& a_Data, std::size_t& a_Offset) {
	a_Buffer >> *a_Data;
	FromNetwork(*a_Data);

	*a_Data >>= sizeof(T) * 8 - a_Offset - a_Data.GetBitSize();
	*a_Data &= (1 << a_Data.GetBitSize()) - 1;

	if (a_Offset != sizeof(T) * 8) {
		a_Buffer.SetReadOffset(a_Buffer.GetReadOffset() - sizeof(T));
		a_Offset += a_Data.GetBitSize();
	} else {
		a_Offset = 0;
	}
}

template <typename T>
void ReadField(DataBuffer& a_Buffer, T& a_Data, std::size_t& a_Offset) {
	a_Buffer >> a_Data;
	FromNetwork(a_Data);
}

template <typename TData>
DataBuffer WriteMessage(const TData& a_MessageData) {
	DataBuffer buffer;
	std::size_t currentOffset = 0;
	std::uint64_t dataRaw = 0;
	boost::pfr::for_each_field(a_MessageData,
		[&buffer, &dataRaw, &currentOffset](const auto& a_Field) { WriteField(buffer, a_Field, dataRaw, currentOffset); });
	return buffer;
}

template <typename TData>
void ReadMessage(DataBuffer& a_Buffer, TData& a_MessageData) {
	std::size_t currentOffset = 0;
	boost::pfr::for_each_field(
		a_MessageData, [&a_Buffer, &currentOffset](auto& a_Field) { ReadField(a_Buffer, a_Field, currentOffset); });
}

}  // namespace details
}  // namespace sp