#include <iostream>

#include <examples/PacketExample.h>
#include <sp/io/File.h>

class CustomPacketHandler : public sp::PacketHandler {
	void Handle(const KeepAlivePacket& packet) {
		std::cout << "KeepAlive handled ! " << packet.GetKeepAliveId() << "\n";
	}

	void Handle(const DisconnectPacket& packet) {
		std::cout << "Disconnect handled ! " << packet.GetReason() << "\n";
	}

	void Handle(const UpgradeTowerPacket& packet) {
		std::cout << "UpgradeTower handled !\n";
	}
};

using FileStream = sp::io::Stream<sp::io::FileTag, sp::PacketDispatcher, sp::PacketFactory, sp::option::ZlibCompress>;

int main() {
	auto handler = std::make_shared<CustomPacketHandler>();

	FileStream stream(sp::io::File{"test.txt", sp::io::FileTag::In | sp::io::FileTag::Out});
	stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler);
	stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler);

	auto zlibOptions = sp::option::ZlibCompress{};

	stream.SendMessage(KeepAlivePacket{96}, zlibOptions);
	stream.SendMessage(KeepAlivePacket{69}, zlibOptions);
	stream.SendMessage(
		DisconnectPacket{
			"This is in the "
			"fiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiile !"},
		zlibOptions);

	stream.RecieveMessages(zlibOptions);

	return 0;
}