#include "TcpServer.h"

#include <QTcpSocket>

/*==============================================================*/
TcpServer::TcpServer(QObject *parent): QTcpServer(parent)
{
 connect(this, SIGNAL(newConnection()), this, SLOT(acceptNewClient()));
}

/*==============================================================*/
//accept a new client and set the SIGNAL and SLOT
void TcpServer::acceptNewClient()
{
 int socketDescriptor;
 QTcpSocket *tcpClientSocket = nextPendingConnection();
 socketDescriptor = tcpClientSocket->socketDescriptor();

 connect(tcpClientSocket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
 connect(tcpClientSocket, SIGNAL(disconnected()), tcpClientSocket, SLOT(deleteLater()));
 connect(tcpClientSocket, SIGNAL(readyRead()), this, SLOT(clientDataReceived()));

 tcpClientSocketList.append(tcpClientSocket); //add client

 //send the client address and descriptor to NetAssistWidget
 QString peerPortStr = QString::number(tcpClientSocket->peerPort());
 QString rdClientAddress_Port = tcpClientSocket->peerAddress().toString() + ":" + peerPortStr;
 emit addClientLink(rdClientAddress_Port, socketDescriptor);
}

/*==============================================================*/
//send data to a client or all client
void TcpServer::sendDataToClient(char *msg, int length, int socketDescriptor, int socketDescriptorEx)
{
 for(int i = 0; i < tcpClientSocketList.count(); i++)
 {
 QTcpSocket *item = tcpClientSocketList.at(i);

 if(socketDescriptor == 0) {
 if(item->socketDescriptor() != socketDescriptorEx) {
 if(item->write(msg, length) != length)
 {
 continue;
 }
 }
 } else {
 if(item->socketDescriptor() == socketDescriptor) {
 if(item->write(msg, length) != length)
 {
 break;
 }
 }
 }
 }

}

/*==============================================================*/
//send disconnect a valid client tcpsocket
void TcpServer::clientDisconnected()
{
 for(int i = 0; i < tcpClientSocketList.count(); i++)
 {
 QTcpSocket *item = tcpClientSocketList.at(i);
 if(item->state() == 0)
 {
 QString peerPortStr = QString::number(item->peerPort());
 QString rdClientAddress_Port = item->peerAddress().toString() + ":" + peerPortStr;
 //send the client address and descriptor to NetAssistWidget
 emit removeClientLink(rdClientAddress_Port, item->socketDescriptor());

 tcpClientSocketList.removeAt(i); //remove Client

 break;
 }
 }
}

/*==============================================================*/
//processe a client data
void TcpServer::clientDataReceived()
{
 for(int i = 0; i < tcpClientSocketList.count(); i++)
 {
 QTcpSocket *item = tcpClientSocketList.at(i);
 while(item->bytesAvailable() > 0)
 {
 QByteArray datagram;
 datagram.resize(item->bytesAvailable());
 item->read(datagram.data(), datagram.size());
 emit updateTcpServer((char *)datagram.data(), datagram.size(), item->socketDescriptor());
 }
 }
}