tahoma2d/toonz/sources/common/tmsgcore.cpp

#include "tmsgcore.h"
#include <qlocalserver.h>
#include <qlocalsocket.h>
#include <QStringList>
#include <QCoreApplication>

#include <QTcpServer>
#include <QTcpSocket>

TMsgCore *TMsgCore::instance()
{
	static TMsgCore *theInstance = 0;
	if (!theInstance)
		theInstance = new TMsgCore();
	return theInstance;
}

//----------------------------------

TMsgCore::TMsgCore()
	: m_tcpServer(0), m_clientSocket(0)
{
}

//----------------------------------

void TMsgCore::OnNewConnection() //server side
{
	QTcpSocket *socket;
	if (m_tcpServer)
		socket = m_tcpServer->nextPendingConnection();
	assert(socket);

	bool ret = connect(socket, SIGNAL(readyRead()), SLOT(OnReadyRead()));
	ret = ret && connect(socket, SIGNAL(disconnected()), SLOT(OnDisconnected()));
	assert(ret);
	m_sockets.insert(socket);
}

//----------------------------------------------------------

#define TMSG_PORT 10545

bool TMsgCore::openConnection() //server side
{
	//QHostAddress host = (remoteConnection) ? QHostAddress::AnyQHostAddress::LocalHost : ;

	if (m_tcpServer != 0 && m_tcpServer->serverAddress() == QHostAddress::Any)
		return true;
	if (m_tcpServer != 0)
		delete m_tcpServer;

	m_tcpServer = new QTcpServer();
	bool ret = connect(m_tcpServer, SIGNAL(newConnection()), SLOT(OnNewConnection()));
	//bool listen = m_tcpServer->listen(QString("tmsg")+QString::number(QCoreApplication::applicationPid()));
	bool listen = m_tcpServer->listen(QHostAddress::Any, TMSG_PORT); //QString("tmsg")+QString::number(QCoreApplication::applicationPid()));
	if (!listen) {
		QString err = m_tcpServer->errorString();
	}

	assert(ret && listen);
	return true;
}

//---------------------------

QString TMsgCore::getConnectionName() //server side
{
	assert(m_tcpServer);
	QString serverName = "pippo"; //m_tcpServer->serverName();
	return serverName;
}

//-----------------------------------------------------------------------

void TMsgCore::OnDisconnected() //server side
{
	std::set<QTcpSocket *>::iterator it = m_sockets.begin();
	while (it != m_sockets.end()) {
		if ((*it)->state() != QTcpSocket::ConnectedState)
			m_sockets.erase(it++);
		else
			it++;
	}

	//if (m_socketIn)
	//delete m_socketIn;
	//m_socketIn = 0;
}

//------------------------------------------------------------------

void TMsgCore::OnReadyRead() //server side
{
	std::set<QTcpSocket *>::iterator it = m_sockets.begin();
	for (; it != m_sockets.end(); it++) //devo scorrerli tutti perche' non so da quale socket viene il messaggio...
	{
		if ((*it)->state() == QTcpSocket::ConnectedState && (*it)->bytesAvailable() > 0)
			break;
	}
	if (it != m_sockets.end()) {
		readFromSocket(*it);
		OnReadyRead();
	}
}

//-------------------------------------------------

void TMsgCore::readFromSocket(QTcpSocket *socket) //server side
{
	static char data[1024];
	static QString prevMessage;
	QString str;
	int byteread;

	while ((byteread = socket->read(data, 1023)) != 0) {
		data[byteread] = '\0';
		str += QString(data);
	}
	QString message = prevMessage + str;
	prevMessage = QString();
	if (message.isEmpty())
		return;

	int lastEnd = message.lastIndexOf("#END");
	int lastbegin = message.lastIndexOf("#TMSG");
	if (lastbegin == -1 && lastEnd == -1) {
		prevMessage = message;
		return;
	} else if (lastbegin != -1 && lastEnd != -1 && lastEnd < lastbegin) {
		prevMessage = message.right(message.size() - lastbegin);
		message.chop(lastbegin);
	}

	QStringList messages = message.split("#TMSG", QString::SkipEmptyParts);

	for (int i = 0; i < messages.size(); i++) {
		QString str = messages.at(i).simplified();
		str.chop(4); //rimuovo i "#END" alla fine
		if (str.startsWith("ERROR"))
			DVGui::error(str.right(str.size() - 5));
		else if (str.startsWith("WARNING"))
			DVGui::warning(str.right(str.size() - 7));
		else if (str.startsWith("INFO"))
			DVGui::info(str.right(str.size() - 4));
		else
			assert(false);
	}
}

//------------------------------------------------------

TMsgCore::~TMsgCore()
{
	if (m_tcpServer == 0 && m_clientSocket != 0) //client side
	{
		//m_socketIn->waitForBytesWritten (-1);
		//   m_clientSocket->disconnectFromServer();
		// m_clientSocket->waitForDisconnected();
		delete m_clientSocket;
		m_clientSocket = 0;
	}
}

//----------------------------------

bool TMsgCore::send(DVGui::MsgType type, const QString &message) //client side
{
	if (receivers(SIGNAL(sendMessage(int, const QString &))) == 0) {
		if (m_clientSocket == 0 || m_clientSocket->state() != QTcpSocket::ConnectedState)
			return false;

		QString socketMessage = (type == DVGui::CRITICAL ? "#TMSG ERROR " : (type == DVGui::WARNING ? "#TMSG WARNING " : "#TMSG INFO ")) + message + " #END\n";

#if QT_VERSION >= 0x050000
		m_clientSocket->write(socketMessage.toLatin1());
#else
		m_clientSocket->write(socketMessage.toAscii());
#endif
		m_clientSocket->flush();
		//m_clientSocket->waitForBytesWritten (1000);
	} else
		Q_EMIT sendMessage(type, message);

	return true;
}

//-------------------------------------------------------------------

void TMsgCore::connectTo(const QString &address) //client side
{
	m_clientSocket = new QTcpSocket();
	m_clientSocket->connectToHost(address == "" ? QHostAddress::LocalHost : QHostAddress(address), TMSG_PORT);
	//m_clientSocket->connectToServer (connectionName, QIODevice::ReadWrite | QIODevice::Text);
	bool ret = m_clientSocket->waitForConnected(1000);
	if (!ret) {
		// std::cout << "cannot connect to "<< address.toStdString() << std::endl;
		//std::cout << "error "<< m_clientSocket->errorString().toStdString() << std::endl;
		int err = m_clientSocket->error();
	}
}

//-----------------------------------------------------------------------

void DVGui::MsgBox(MsgType type, const QString &text)
{
	TMsgCore::instance()->send(type, text);
}

//-----------------------------------------------------------------------

void DVGui::error(const QString &msg)
{
	DVGui::MsgBox(DVGui::CRITICAL, msg);
}

//-----------------------------------------------------------------------------

void DVGui::warning(const QString &msg)
{
	DVGui::MsgBox(DVGui::WARNING, msg);
}

//-----------------------------------------------------------------------------

void DVGui::info(const QString &msg)
{
	DVGui::MsgBox(DVGui::INFORMATION, msg);
}