big push

include/td/protocol/ByteBuffer.h

@@ -0,0 +1,94 @@
+#pragma once
+
+#include <godot_cpp/variant/packed_byte_array.hpp>
+#include <godot_cpp/variant/string.hpp>
+#include <stdexcept>
+#include <vector>
+
+namespace td {
+namespace protocol {
+
+class PlayerInfo;
+
+class ByteBuffer {
+  private:
+	godot::PackedByteArray m_Buffer;
+	std::size_t m_ReadOffset;
+
+  public:
+	class ReadError : public std::runtime_error {
+	  public:
+		ReadError(const std::string& msg) : std::runtime_error(msg) {}
+	};
+
+	ByteBuffer(godot::PackedByteArray&& a_Buffer) : m_Buffer(std::move(a_Buffer)), m_ReadOffset(0) {}
+	ByteBuffer() : m_ReadOffset(0) {
+		m_Buffer.resize(0);
+	}
+
+	const godot::PackedByteArray& GetByteArray() const {
+		return m_Buffer;
+	}
+
+	godot::PackedByteArray& GetByteArray() {
+		return m_Buffer;
+	}
+
+	// Integers
+	ByteBuffer& operator<<(int8_t a_Data);
+	ByteBuffer& operator>>(int8_t& a_Data);
+	ByteBuffer& operator<<(uint8_t a_Data);
+	ByteBuffer& operator>>(uint8_t& a_Data);
+
+	ByteBuffer& operator<<(int16_t a_Data);
+	ByteBuffer& operator>>(int16_t& a_Data);
+	ByteBuffer& operator<<(uint16_t a_Data);
+	ByteBuffer& operator>>(uint16_t& a_Data);
+
+	ByteBuffer& operator<<(int32_t a_Data);
+	ByteBuffer& operator>>(int32_t& a_Data);
+	ByteBuffer& operator<<(uint32_t a_Data);
+	ByteBuffer& operator>>(uint32_t& a_Data);
+
+	ByteBuffer& operator<<(int64_t a_Data);
+	ByteBuffer& operator>>(int64_t& a_Data);
+	ByteBuffer& operator<<(uint64_t a_Data);
+	ByteBuffer& operator>>(uint64_t& a_Data);
+
+	ByteBuffer& operator<<(float a_Data);
+	ByteBuffer& operator>>(float& a_Data);
+	ByteBuffer& operator<<(double a_Data);
+	ByteBuffer& operator>>(double& a_Data);
+
+	ByteBuffer& operator<<(const godot::String& a_Data);
+	ByteBuffer& operator>>(godot::String& a_Data);
+
+	ByteBuffer& operator<<(const godot::Vector3& a_Data);
+	ByteBuffer& operator>>(godot::Vector3& a_Data);
+
+	template <typename T>
+	ByteBuffer& operator<<(const std::vector<T>& a_Data) {
+		*this << static_cast<std::uint32_t>(a_Data.size());
+		for (const T& data : a_Data) {
+			*this << data;
+		}
+		return *this;
+	}
+
+	template <typename T>
+	ByteBuffer& operator>>(std::vector<T>& a_Data) {
+		std::uint32_t arraySize;
+		*this >> arraySize;
+		a_Data.resize(arraySize);
+		for (std::uint32_t i = 0; i < arraySize; i++) {
+			*this >> a_Data[i];
+		}
+		return *this;
+	}
+
+	ByteBuffer& operator<<(const PlayerInfo& a_Data);
+	ByteBuffer& operator>>(PlayerInfo& a_Data);
+};
+
+}  // namespace protocol
+}  // namespace td