#include "Network.hpp"

Network::Network(std::string address, unsigned short port) : m_address(address), m_port(port) {
	LOG(INFO) << "Connecting to server " << m_address << ":" << m_port;
	sf::Socket::Status status = m_socket.connect(sf::IpAddress(m_address), m_port);
	m_socket.setBlocking(true);
	if (status != sf::Socket::Done) {
		if (status == sf::Socket::Error) {
			LOG(ERROR) << "Can't connect to remote server";
		} else {
			LOG(ERROR) << "Connection failed with unknown reason";
			throw std::runtime_error("Connection is failed");
			throw 13;
		}
	}
	LOG(INFO) << "Connected to server";
}

Network::~Network() {
	m_socket.disconnect();
	LOG(INFO) << "Disconnected";
}

void Network::SendHandshake(std::string username) {
	//Handshake packet
	Packet handshakePacket = PacketBuilder::CHandshaking0x00(316, m_address, m_port, 2);
	SendPacket(handshakePacket);

	//LoginStart packet
	Field fName;
	fName.SetString(username);
	Packet loginPacket(0);
	loginPacket.AddField(fName);
	SendPacket(loginPacket);
}

void DumpPacket(Packet &packet, std::string DumpName) {
	return;
	byte *buff = new byte[packet.GetLength()];
	packet.CopyToBuff(buff);
	std::ofstream fs(DumpName, std::ios::out | std::ios::binary);
	fs.write(reinterpret_cast<const char *>(buff), packet.GetLength());
	fs.close();
	delete buff;
}

static int pn = 0;

void Network::SendPacket(Packet &packet) {
	m_socket.setBlocking(true);
	byte *packetData = new byte[packet.GetLength()];
	packet.CopyToBuff(packetData);
	m_socket.send(packetData, packet.GetLength());
	std::ostringstream out;
	out << "s" << pn++ << "-";
	out << "0x" << (packet.GetId() < 15 ? "0" : "") << std::hex << packet.GetId() << std::dec;
	DumpPacket(packet, out.str());

	delete[] packetData;
}

Packet Network::ReceivePacket() {
	byte bufLen[5] = {0};
	size_t rec = 0;
	for (int i = 0; i < 5; i++) {
		byte buff = 0;
		size_t r = 0;
		m_socket.receive(&buff, 1, r);
		rec += r;
		bufLen[i] = buff;
		if ((buff & 0b10000000) == 0) {
			break;
		}
	}
	Field fLen = FieldParser::Parse(VarIntType, bufLen);
	size_t packetLen = fLen.GetVarInt() + fLen.GetLength();
	if (packetLen > 1024 * 1024 * 15)
		LOG(WARNING) << "OMG SIZEOF PACKAGE IS " << packetLen;
	if (packetLen < rec) {
		return Packet(bufLen);
	}
	byte *bufPack = new byte[packetLen];
	std::copy(bufLen, bufLen + rec, bufPack);
	size_t dataLen = rec;
	while (m_socket.receive(bufPack + dataLen, packetLen - dataLen, rec) == sf::Socket::Done && dataLen < packetLen) {
		dataLen += rec;
	}
	if (dataLen < packetLen) {
		LOG(ERROR) << "Received data is " << dataLen << " but " << packetLen << " is promoted";
		throw std::runtime_error("Data is losted");
	} else {
		Packet p(bufPack);
		delete[] bufPack;

		std::ostringstream out;
		out << "r" << pn++ << "-";
		out << "0x" << (p.GetId() < 15 ? "0" : "") << std::hex << p.GetId() << std::dec;
		DumpPacket(p, out.str());
		return p;
	}
}