#ifndef SOCKET_H
#define SOCKET_H

#include <iostream>
#include <string>
#include <sstream>
#include <conio.h>           // Needed for kbhit and getch
#include <SDL.h>
#include <SDL_net.h>         // Include SDL_net, which includes SDL.h for us
using namespace std;

class ClientSocket
{
	private:
		bool debug;                 // Flag to control whether the ServerSocket should display debug info

		unsigned int serverPort;    // The port of the server to connect to
		unsigned int bufferSize;    // The size of our message buffer

		string    serverHostname;   // The host name of the server (i.e. "localhost", "www.foo.com" etc.
		IPaddress serverIP;         // The IP address of the server to connect to (NOT in dot-quad format)
		string    dotQuadString;    // The IP address of the server to connect to as a dot-quad (i.e. 127.0.0.1)
		TCPsocket serverSocket;     // The server socket
		TCPsocket clientSocket;     // Our own client socket

		char   *pBuffer;            // A pointer to (what will be) an array of characters used to store the messages we send and receive
		string  userInput;          // User input string
		int     inputLength;        // The length of our userInput in characters

		SDLNet_SocketSet socketSet; // Our entire set of sockets (i.e. just the server socket and our client socket)

		bool shutdownClient;        // Flag to control when to shut down the client

	public:
		static const string       SERVER_NOT_FULL;
		static const string       SERVER_FULL;
		static const string       SHUTDOWN_SIGNAL;
		static const string       QUIT_SIGNAL;
		static const unsigned int CONNECTION_TIMEOUT_PERIOD;
		static const unsigned int SOCKET_SET_POLL_PERIOD;

		ClientSocket(string theServerAddress, unsigned int theServerPort, unsigned int theBufferSize);

		~ClientSocket();

		// Function to poll for clients connecting
		void connectToServer();

		// Function to check the server for incoming messages
		string checkForIncomingMessages();

		// Function to display a received message
		void displayMessage(string &receivedMessage);

		// Function to send a message to the server
		//void sendOutgoingMessage();

		// Function display the prompt + any input that hasn't been sent yet
		void displayPrompt();

		// Function to get keypresses in a non-blocking manner
		void getUserInput();

		// Function to get the current contents of our outgoing message
		string getCurrentUserInputContents();

		// Function to return the shutdown status, used to control when to terminate
		bool getShutdownStatus();
};

// Template function to convert most anything to a string
template<class T>
std::string toString(const T& t)
{
	ostringstream stream;
	stream << t;
	return stream.str();
}

#endif