#include "mxsocket.h"


namespace mx
{


	mxSocket::mxSocket()
	{
		current_socket = 0;
		connected = false;
	}

	mxSocket::~mxSocket()
	{

	}

	mxSocket::mxSocket(mx_socket nsock)
	{
		current_socket = nsock;
		connected = true;
	}
	
	mxSocket::mxSocket(const mxSocket &nsock)
	{
		current_socket = nsock.current_socket;
		connected = nsock.connected;
	}

	mxSocket& mxSocket::operator=(const mxSocket &sock)
	{

		current_socket = sock.current_socket;
		connected = sock.connected;
		return *this;
	}

	bool mxSocket::createSocket()
	{

		current_socket = socket(AF_INET,SOCK_STREAM,0);
		return true;
	}

	bool mxSocket::listenAt(int port)
	{

		static struct sockaddr_in addy;
		memset((char*)&addy,0, sizeof(addy));
		addy.sin_port = htons(port);
		addy.sin_family = AF_INET;
		addy.sin_addr.s_addr = INADDR_ANY;
		int yup = 1;

		setsockopt(current_socket, SOL_SOCKET,SO_REUSEADDR,(const char*)&yup,sizeof(yup));

		bind(current_socket, (struct sockaddr *)&addy, sizeof(addy));
		listen(current_socket,5);

		return true;
	}


	mx_socket mxSocket::acceptNewSocket(std::string &ip)
	{
#ifdef _WIN32
		int caddy_len;
#else
		socklen_t caddy_len;
#endif
		static struct sockaddr_in caddy;
		int s;
		s = accept(current_socket, (struct sockaddr *)&caddy, &caddy_len);
		ip = std::string(inet_ntoa(caddy.sin_addr));
		connected = true;
		return s;
	}

	mx_socket mxSocket::acceptsocket()
	{
#ifdef _WIN32
		int caddy_len;
#else
		socklen_t caddy_len;
#endif

		mx_socket SOCK;
		struct sockaddr_in  caddy;
		SOCK = accept(current_socket, (struct sockaddr *)&caddy, &caddy_len);
		connected = true;
		return SOCK;
	}


	bool mxSocket::connectTo(std::string where, unsigned int port)
	{
		struct sockaddr_in saddy;
		saddy.sin_port = htons(port);
		saddy.sin_addr.s_addr = inet_addr(where.c_str());
		saddy.sin_family = AF_INET;

		if(connect(current_socket, (const sockaddr*)&saddy, sizeof(sockaddr_in)) == -1) {
	
			std::cout << "error: could not connect: \n";
			return false;
		}

		connected = true;
		return true;
	}


	ssize_t mxSocket::Read(void *data, size_t size)
	{

		size_t len = size;
		size_t total = 0;
		int cur_;
		char *cdata = (char*)data;

		while ( (len != 0) && ( (cur_ = recv(current_socket, cdata, len, 0) ) != 0) )
		{

			if(cur_ == -1) {


				std::cerr << "mxsocket: error on read\n";
				return -1;
	
			} 
			len -= cur_;
			cdata += cur_;
			total += cur_;
		}

		return total;

	}
	
	ssize_t mxSocket::Write(void *data, size_t size)
	{
		
		size_t len = size;
		size_t total = 0;
		int cur_;
		char *cdata = (char*)data;

		while ( (len != 0) && (( cur_ = send(current_socket, cdata, len, 0) ) != 0) )
		{

			if(cur_ == -1) {

	
				std::cerr << "mxsocket: error on write\n";
				return -1;
			
			}
			len -= cur_;
			cdata += cur_;
			total += cur_;
		}

		return total;
	}


	int mxSocket::closeSocket()
	{

#ifdef _WIN32
		return closesocket(current_socket);
#else
		return close(current_socket);
#endif
		return 1;

	}


}