#include <arpa/inet.h>
#include <netdb.h>
#include <pthread.h>
#include <stdio.h>
#include <string.h>
#include <sys/time.h>
#include <unistd.h>

#include "disp.h"
#include "log.h"
#include "public.h"

struct waitClientArgs {
	sockaddr_in newSockAddr;
	socklen_t newSockAddrSize;
	int auxInt; // used in client mode because I didn't want to make another
		    // struct to keep track of
};

void
closeClient()
{
	long e = end1.tv_sec - start1.tv_sec;

	gettimeofday(&end1, NULL);
	close(clientSocketDescriptor);
	
	puts("********Session********");
	printf("Bytes written: %i\n", bytesWritten);
	printf("Bytes read: %i\n", bytesRead);
	printf("Elapsed time: %ld secs\n", e);
	puts("Connection closed.");
}

void *
pollForSever(void *args)
{
	waitClientArgs *aaa = static_cast<waitClientArgs *>(args);
	int socketDescriptor = (int)aaa->auxInt;
	char msg[1024];
	while (1) {
	server_message_loop:
		memset(&msg, 0, sizeof(msg)); // clear the buffer
		bytesRead += recv(socketDescriptor, (char *)&msg, sizeof(msg), 0);

		// not needed for proofs of concept testing
		/*if(!strcmp(msg, "exit"))
		{
		    writeToFile(logFileName, msg);
		    displayFile(logFileName, linePos, LOG_LENGTH);
		    break;
		}*/
		// cout << "Server: " << msg << endl;
		// printf("Server: %s\n");

		if (msg[0] == '\0')
			goto server_message_loop;

		writeToFile(logFileName, msg);

		if (linesInFile(logFileName) > LOG_LENGTH)
			linePos++;

		displayFile(logFileName, linePos, LOG_LENGTH);
	}

	return nullptr;
}

void
setupClient()
{
	// we need 2 things: ip address and port number, in that order
	// if(argc != 3)
	//{
	//     cerr << "Usage: ip_address port" << endl; exit(0);
	// } //grab the IP address and port number
	// create a message buffer
	char msg[1024];
	// setup a socket and connection tools
	struct hostent *host = gethostbyname(IP_ADDRESS.c_str());
	sockaddr_in sendSockAddr;
	bzero((char *)&sendSockAddr, sizeof(sendSockAddr));
	sendSockAddr.sin_family = AF_INET;
	sendSockAddr.sin_addr.s_addr = inet_addr(inet_ntoa(*(struct in_addr *)*host->h_addr_list));
	sendSockAddr.sin_port = htons(PORT_NUM);
	clientSocketDescriptor = socket(AF_INET, SOCK_STREAM, 0);
	// try to connect...
	int status =
	    connect(clientSocketDescriptor, (sockaddr *)&sendSockAddr, sizeof(sendSockAddr));

	if (status < 0)
		writeToFile(logFileName, "Error connecting to socket!");

	writeToFile(logFileName, "Connected to the server!");
	int bytesRead, bytesWritten = 0;
	struct timeval start1, end1;
	gettimeofday(&start1, NULL);
	auto *aaa = new waitClientArgs{}; // it looks stupid but it works
	aaa->auxInt = clientSocketDescriptor;
	int rc = pthread_create(&client_wait_thread, nullptr, pollForSever, aaa);
	pthread_detach(client_wait_thread);
}