finish serialize

src/td/protocol/command/CommandFactory.cpp

@@ -8,15 +8,13 @@ namespace td {
 namespace protocol {
 namespace CommandFactory {
 
-using CommandCreator = std::function<std::unique_ptr<Command>()>;
+using CommandCreator = std::function<std::shared_ptr<Command>()>;
 
-#define DeclareCommand(CommandName, ...) std::make_unique<commands::CommandName>(),
+#define DeclareCommand(CommandName, ...) std::make_shared<commands::CommandName>(),
 
-static std::array<std::unique_ptr<Command>, static_cast<std::size_t>(CommandType::COMMAND_COUNT)> Commands = {
-	DeclareAllCommand()
-};
+static std::array<std::shared_ptr<Command>, static_cast<std::size_t>(CommandType::COMMAND_COUNT)> Commands = {DeclareAllCommand()};
 
-const std::unique_ptr<Command>& CreateReadOnlyCommand(CommandType a_Type) {
+const std::shared_ptr<Command>& CreateReadOnlyCommand(CommandType a_Type) {
 	assert(a_Type < CommandType::COMMAND_COUNT);
 	return Commands[static_cast<std::size_t>(a_Type)];
 }

src/td/protocol/command/CommandSerializer.cpp

@@ -0,0 +1,228 @@
+#include <td/protocol/command/CommandSerializer.h>
+
+#include <td/protocol/command/CommandFactory.h>
+#include <td/protocol/command/CommandVisitor.h>
+
+#include <td/misc/Format.h>
+#include <td/misc/Log.h>
+
+namespace td {
+namespace protocol {
+namespace CommandSerializer {
+
+#define DeclareCommand(CommandName, ...)                                                                                              \
+	void Visit(const commands::CommandName& a_Command) override {                                                                     \
+		const auto& commandData = a_Command.m_Data;                                                                                   \
+		SerializeCommandData(commandData);                                                                                            \
+	}                                                                                                                                 \
+                                                                                                                                      \
+	void SerializeCommandData(const commands::CommandName::CommandDataType& a_Command);
+
+
+
+
+class Serializer : public CommandVisitor {
+  private:
+	DataBuffer& m_Buffer;
+
+  public:
+	Serializer(DataBuffer& a_Buffer) : m_Buffer(a_Buffer) {}
+
+	void Serialize(const Command& a_Command) {
+		m_Buffer << static_cast<CommandID>(a_Command.GetType());
+		Check(a_Command);
+	}
+
+	DeclareAllCommand()
+};
+
+#undef DeclareCommand
+
+
+
+
+
+#define DeclareCommand(CommandName, ...)                                                                                              \
+	void Visit(const commands::CommandName& a_Command) override {                                                                     \
+		auto commandPtr = CommandFactory::CreateCommand<commands::CommandName>();                                                     \
+		auto& commandData = commandPtr->m_Data;                                                                                       \
+                                                                                                                                      \
+		DeserializeCommandData(commandData);                                                                                          \
+                                                                                                                                      \
+		m_Command = std::move(commandPtr);                                                                                            \
+	}                                                                                                                                 \
+                                                                                                                                      \
+	void DeserializeCommandData(commands::CommandName::CommandDataType& a_Command);
+
+
+
+class Deserializer : public CommandVisitor {
+  private:
+	DataBuffer& m_Buffer;
+	CommandPtr m_Command;
+
+  public:
+	Deserializer(DataBuffer& a_Buffer) : m_Buffer(a_Buffer) {}
+
+	bool Deserialize(const CommandPtr& a_Command) {
+		try {
+			Check(*a_Command.get());
+		} catch (std::exception& e) {
+			utils::LOGE(utils::Format("[CommandSerializer::Deserializer] %s", e.what()));
+			return false;
+		}
+		return true;
+	}
+
+	CommandPtr& GetCommand() {
+		return m_Command;
+	}
+
+	DeclareAllCommand()
+};
+
+
+
+
+
+DataBuffer Serialize(const Command& a_Command) {
+	DataBuffer buffer;
+
+	Serializer serializer(buffer);
+	serializer.Serialize(a_Command);
+
+	return buffer;
+}
+
+std::shared_ptr<Command> Deserialize(DataBuffer& a_Data) {
+	CommandID commandId;
+	a_Data >> commandId;
+
+	if (commandId >= static_cast<CommandID>(CommandType::COMMAND_COUNT))
+		return nullptr;
+
+	CommandType commandType = CommandType(commandId);
+
+	// for double-dispatch
+	const CommandPtr& emptyCommand = CommandFactory::CreateReadOnlyCommand(commandType);
+
+	Deserializer deserializer(a_Data);
+
+	if (deserializer.Deserialize(emptyCommand)) {
+		CommandPtr command = std::move(deserializer.GetCommand());
+		return command;
+	}
+
+	return nullptr;
+}
+
+
+
+
+
+//---------------------------------------------
+// Command serializer implementation
+//----------------------------------------------
+
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::End& a_Command) {}
+
+void Deserializer::DeserializeCommandData(cdata::End& a_Command) {}
+
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::PlaceTower& a_Command) {
+	m_Buffer << (static_cast<std::uint8_t>(a_Command.m_Type) << 4 | a_Command.m_Placer & 0xF) << a_Command.m_Position;
+}
+
+void Deserializer::DeserializeCommandData(cdata::PlaceTower& a_Command) {
+	std::uint8_t union1;
+	m_Buffer >> union1 >> a_Command.m_Position;
+	a_Command.m_Type = td::TowerType(union1 >> 4);
+	a_Command.m_Placer = union1 & 0xF;
+}
+
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::PlayerJoin& a_Command) {
+	m_Buffer << a_Command.m_ID << a_Command.m_Name;
+}
+
+void Deserializer::DeserializeCommandData(cdata::PlayerJoin& a_Command) {
+	m_Buffer >> a_Command.m_ID >> a_Command.m_Name;
+}
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::SpawnTroop& a_Command) {
+	m_Buffer << static_cast<std::uint8_t>(static_cast<std::uint8_t>(a_Command.m_Type) << 3 | a_Command.m_Level & 0x7)
+			 << a_Command.m_Position << a_Command.m_Sender;
+}
+
+void Deserializer::DeserializeCommandData(cdata::SpawnTroop& a_Command) {
+	std::uint8_t union1;
+	m_Buffer >> union1 >> a_Command.m_Position >> a_Command.m_Sender;
+	a_Command.m_Type = td::EntityType(union1 >> 3);
+	a_Command.m_Level = union1 & 0x7;
+}
+
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::TeamChange& a_Command) {
+	m_Buffer << static_cast<std::uint8_t>(a_Command.m_Player << 1 | static_cast<std::uint8_t>(a_Command.m_NewTeam));
+}
+
+void Deserializer::DeserializeCommandData(cdata::TeamChange& a_Command) {
+	std::uint8_t union1;
+	m_Buffer >> union1;
+	a_Command.m_Player = union1 >> 1;
+	a_Command.m_NewTeam = td::Team(union1 & 1);
+}
+
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::UpgradeTower& a_Command) {
+	m_Buffer << static_cast<std::uint16_t>(a_Command.m_Tower << 4 | a_Command.m_Upgrade & 0xF);
+}
+
+void Deserializer::DeserializeCommandData(cdata::UpgradeTower& a_Command) {
+	std::uint16_t union1;
+	m_Buffer >> union1;
+	a_Command.m_Tower = union1 >> 4;
+	a_Command.m_Upgrade = union1 & 0xF;
+}
+
+
+
+
+
+void Serializer::SerializeCommandData(const cdata::UseItem& a_Command) {
+	m_Buffer << static_cast<std::uint8_t>(static_cast<std::uint8_t>(a_Command.m_Item) << 4 | a_Command.m_User & 0xF)
+			 << a_Command.m_Position;
+}
+
+void Deserializer::DeserializeCommandData(cdata::UseItem& a_Command) {
+	std::uint8_t union1;
+	m_Buffer >> union1 >> a_Command.m_Position;
+	a_Command.m_Item = td::ShopItem(union1 >> 4);
+	a_Command.m_User = union1 & 0xF;
+}
+
+
+
+
+}  // namespace CommandSerializer
+}  // namespace protocol
+}  // namespace td

src/td/protocol/packet/PacketSerializer.cpp

@@ -1,5 +1,6 @@
 #include <td/protocol/packet/PacketSerializer.h>
 
+#include <td/protocol/command/CommandSerializer.h>
 #include <td/protocol/packet/PacketFactory.h>
 #include <td/protocol/packet/PacketVisitor.h>
 
@@ -212,11 +213,33 @@ void Deserializer::DeserializePacketData(pdata::ChatMessage& a_Packet) {
 
 
 void Serializer::SerializePacketData(const pdata::LockSteps& a_Packet) {
-	m_Buffer << a_Packet.m_FirstFrameNumber << a_Packet.m_LockSteps;
+	m_Buffer << a_Packet.m_FirstFrameNumber;
+	for (int i = 0; i < LOCKSTEP_BUFFER_SIZE; i++) {
+		auto& lockStep = a_Packet.m_LockSteps[i];
+		m_Buffer << VarInt{lockStep.size()};
+
+		for (int j = 0; j < lockStep.size(); i++) {
+			auto command = lockStep[j];
+			m_Buffer << CommandSerializer::Serialize(*command);
+		}
+	}
 }
 
 void Deserializer::DeserializePacketData(pdata::LockSteps& a_Packet) {
-	m_Buffer >> a_Packet.m_FirstFrameNumber >> a_Packet.m_LockSteps;
+	m_Buffer >> a_Packet.m_FirstFrameNumber;
+	for (int i = 0; i < LOCKSTEP_BUFFER_SIZE; i++) {
+		VarInt lockStepSize;
+		m_Buffer >> lockStepSize;
+
+		for (std::size_t j = 0; j < lockStepSize.GetValue(); i++) {
+			auto command = CommandSerializer::Deserialize(m_Buffer);
+			if (command) {
+				a_Packet.m_LockSteps[i].push_back(command);
+			} else {
+				throw std::runtime_error("Failed to deserialise command");
+			}
+		}
+	}
 }