#include "wiServer.h"

#ifndef WINSTORE_SUPPORT

using namespace std;


wiServer::wiServer(const string& newName, const string& ipaddress, int port)
{
	name=newName;
	if(ListenOnPort(port,ipaddress.length()<=1?"0.0.0.0":ipaddress.c_str())){
		stringstream ss("");
		ss<<"Listening as "<<name<<" ,IP: "<<ipaddress<<" [port: "<<port<<"]";
		wiBackLog::post(ss.str().c_str());
		success=true;
	}
	else{
		stringstream ss("");
		ss<<"Creating server on address: "<<ipaddress<< " [port "<<port<<"] FAILED with: "<<WSAGetLastError();
		wiBackLog::post(ss.str().c_str());
		success=false;
	}
}


wiServer::~wiServer(void)
{
	for (map<SOCKET,string>::iterator it = clients.begin(); it != clients.end(); ++it) {
		closesocket(it->first);
	}
	wiNetwork::~wiNetwork();
}

bool wiServer::ListenOnPort(int portno, const char* ipaddress)
{
	int error = WSAStartup(SCK_VERSION2,&w);

	if(error)
		return false;

	if(w.wVersion != SCK_VERSION2){
		WSACleanup();
		return false;
	}

	SOCKADDR_IN addr;
	addr.sin_family = AF_INET;
	addr.sin_port= htons(portno);
	//addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY); //or inet_addr("0.0.0.0");
	addr.sin_addr.S_un.S_addr = inet_addr(ipaddress);

	s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	

	//set master socket to allow multiple connections , this is just a good habit, it will work without this
	int opt = TRUE;
	if( setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt)) < 0 )
		return false;


	if( setsockopt(s, SOL_SOCKET, TCP_NODELAY, (char *)&opt, sizeof(opt)) < 0 )
		return false;

	if(s == INVALID_SOCKET)
		return false;

	if(::bind(s, (LPSOCKADDR)&addr, sizeof(addr)) == SOCKET_ERROR)
		return false;  //couldn't bind (for example if you try to bind more than once to the same socket

	//if( listen(s,SOMAXCONN) == SOCKET_ERROR )
	if( listen(s,SOMAXCONN) == SOCKET_ERROR )
		return false;



	return true;
}
SOCKET wiServer::CreateAccepter(){
	struct sockaddr_in caller;
	int addrlen = sizeof(caller);
	SOCKET newsock = accept(s,(struct sockaddr *) &caller, &addrlen);

	//inform user of socket number - used in send and receive commands
	if(newsock != SOCKET_ERROR){
		//printf("\nNew connection , socket fd is %d , ip is : %s , port : %d \n" , newsock , inet_ntoa(caller.sin_addr) , ntohs(caller.sin_port));
	}
	else{
		//printf("\nConnection FAILED on socket: fd is %d , ip is : %s , port : %d \n" , newsock , inet_ntoa(caller.sin_addr) , ntohs(caller.sin_port));
	}

	return newsock;
}



vector<string> wiServer::listClients()
{
	vector<string> ret(0);
	for (map<SOCKET,string>::iterator it = clients.begin(); it != clients.end(); ++it) {
		stringstream ss("");
		ss<<it->second<<":"<<it->first;
		ret.push_back(ss.str());
	}
	return ret;
}

bool wiServer::sendText(const string& text, int packettype, const string& clientName, int clientID){
	int sentTo=0;

	if(clientName.length()<=0){ //send to everyone
		for (map<SOCKET,string>::iterator it = clients.begin(); it != clients.end(); ++it) {
			sentTo += wiNetwork::sendData(packettype,it->first) && wiNetwork::sendText(text,it->first);
		}
	}
	else if(clientID<0){ //send to all of same name
		for (map<SOCKET,string>::iterator it = clients.begin(); it != clients.end(); ++it) {
			if(!clientName.compare(it->second)){
				sentTo += wiNetwork::sendData(packettype,it->first) && wiNetwork::sendText(text,it->first);
			}
		}
	}
	else{ //send to specific client
		if(clients.find(clientID) != clients.end()){
			sentTo += wiNetwork::sendData(packettype,clientID) && wiNetwork::sendText(text,clientID);
		}
	}

	if(sentTo<=0)
		wiBackLog::post("No client was found with the specified parameters");

	return sentTo>0;
}

bool wiServer::changeName(const string& newName){
	wiNetwork::changeName(newName);
	return sendText(newName, wiNetwork::PACKET_TYPE_CHANGENAME);
}
bool wiServer::sendMessage(const string& text, const string& clientName, int clientID){
	return sendText(text, wiNetwork::PACKET_TYPE_TEXTMESSAGE, clientName, clientID);
}

#endif