#pragma GCC diagnostic ignored "-Wconversion-null"

#define _WINSOCK_DEPRECATED_NO_WARNINGS

#include <WinSock2.h> //For win sockets
#include <iostream> //For std::cout, std::endl, std::cin.getline

SOCKET Connection; //This client's connection to the server
int ID = 0;

enum Packet
{
	P_ChatMessage,
	P_Test
};

bool ProcessPacket(Packet packettype)
{
	switch (packettype)
	{
        case P_ChatMessage:
        {
            int remitentLength;
            recv(Connection, (char*)&remitentLength, sizeof(int), NULL);
            char * remitentMSG = new char[remitentLength + 1];
            recv(Connection, remitentMSG, remitentLength, NULL);
            remitentMSG[remitentLength] = '\0';
            std::cout << remitentMSG;

            int bufferlength; //length of the chat message
            recv(Connection, (char*)&bufferlength, sizeof(int), NULL); //receive buffer
            char * buffer = new char[bufferlength + 1]; //Allocate buffer
            recv(Connection, buffer, bufferlength, NULL);
            buffer[bufferlength] = '\0'; //Set last character of buffer to be a null terminator so we aren't printing memory that we shouldn't be looking at
            std::cout << buffer << std::endl; //print out buffer
            delete[] buffer; //Deallocate buffer
            break;
        }

        case P_Test:
            std::cout << "You have received the test packet!" << std::endl;
            break;

        default:
            std::cout << "Unknown packet: " << packettype << std::endl;
            break;
	}
	return true;
}

void ClientThread()
{
	Packet packettype;
	while (true)
	{
		//First get the packet type
		recv(Connection, (char*)&packettype, sizeof(Packet), NULL); //Receive packet type from server

		if (!ProcessPacket(packettype)) //If the packet is not properly processed
			break; //break out of our client handler loop

        Sleep(1000);
	}
}

int main()
{
	//Winsock Startup
	WSAData wsaData;
	WORD DllVersion = MAKEWORD(2, 1);
	if (WSAStartup(DllVersion, &wsaData) != 0)
	{
		MessageBoxA(NULL, "Winsock startup failed", "Error", MB_OK | MB_ICONERROR);
		return 0;
	}

	SOCKADDR_IN addr; //Address to be binded to our Connection socket
	int sizeofaddr = sizeof(addr); //Need sizeofaddr for the connect function
	addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //Address = localhost (this pc)
	addr.sin_port = htons(1111); //Port = 1111
	addr.sin_family = AF_INET; //IPv4 Socket

	Connection = socket(AF_INET, SOCK_STREAM, NULL); //Set Connection socket
	if (connect(Connection, (SOCKADDR*)&addr, sizeofaddr) != 0) //If we are unable to connect...
	{
		MessageBoxA(NULL, "Failed to Connect", "Error", MB_OK | MB_ICONERROR);
		return 0; //Failed to Connect
	}

    recv(Connection, (char*)&ID, sizeof(int), NULL);

	std::cout << "Connected under the ID: " << ID << std::endl;

	CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)ClientThread, NULL, NULL, NULL); //Create the client thread that will receive any data that the server sends.

	std::string userinput; //holds the user's chat message
	while (true)
	{
		std::getline(std::cin,userinput); //Get line if user presses enter and fill the buffer
		int bufferlength = userinput.size(); //Find buffer length
		Packet chatmessagepacket = P_ChatMessage; //Create packet type: Chat Message to be sent to the server
		send(Connection, (char*)&chatmessagepacket, sizeof(Packet), NULL); //Send packet type: Chat Message
		send(Connection, (char*)&bufferlength, sizeof(int), NULL); //Send length of buffer
		send(Connection, userinput.c_str(), bufferlength, NULL); //Send buffer
		Sleep(20);
	}

	return 0;
}