looqs/gui/ipcpreviewclient.cpp

#include <QLocalSocket>
#include <QApplication>
#include "ipc.h"
#include "ipcpreviewclient.h"
#include "previewresultpdf.h"
#include "previewresultplaintext.h"

bool IPCPreviewClient::connect()
{
	if(socket->state() == QLocalSocket::ConnectedState)
	{
		socket->disconnectFromServer();
		if(socket->state() == QLocalSocket::ConnectedState)
		{
			socket->waitForDisconnected(100);
		}
	}
	socket->connectToServer(socketPath);
	socket->waitForConnected(100);
	return socket->state() == QLocalSocket::ConnectedState;
}

QSharedPointer<PreviewResult> IPCPreviewClient::deserialize(QByteArray &array)
{
	QDataStream stream{&array, QIODevice::ReadOnly};

	PreviewResultType type;
	stream >> type;
	if(type == PreviewResultType::PDF)
	{
		return PreviewResultPdf::deserialize(array);
	}
	if(type == PreviewResultType::PlainText)
	{
		return PreviewResultPlainText::deserialize(array);
	}
	return QSharedPointer<PreviewResult>(nullptr);
}

IPCPreviewClient::IPCPreviewClient()
{
	this->socket = new QLocalSocket(this);
}

void IPCPreviewClient::setSocketPath(QString socketPath)
{
	this->socketPath = socketPath;
}

void IPCPreviewClient::startGeneration(RenderConfig config, const QVector<RenderTarget> &targets)
{
	this->start(config, targets);
}

void IPCPreviewClient::start(RenderConfig config, const QVector<RenderTarget> &targets)
{
	if(targets.count() == 0)
	{
		return;
	}

	if(!connect() || !socket->isOpen())
	{
		emit error("Could not connect to IPC worker");
		return;
	}

	if(socket->isOpen() && socket->isWritable())
	{
		QDataStream stream(socket);
		stream.startTransaction();
		stream << GeneratePreviews;
		stream << config;
		stream << targets;
		stream.commitTransaction();
		socket->flush();

		qsizetype numTarget = 0;
		if(socket->isOpen() && socket->isReadable() && socket->state() == QLocalSocket::ConnectedState)
		{
			do
			{
				socket->waitForReadyRead(5000);
				stream.startTransaction();
				stream >> numTarget;
			} while(!stream.commitTransaction() && socket->state() == QLocalSocket::ConnectedState);
			if(numTarget != targets.count())
			{
				qDebug() << "Target count mismatch";
				emit error("IPC Error: Server reports less targets than it should");
				return;
			}
		}
		else
		{
			emit error("Error while trying to process previews: " + socket->errorString());
			return;
		}

		int processed = 0;
		++this->currentPreviewGeneration;
		while(socket->isOpen() && socket->isReadable() && processed < targets.count())
		{
			QByteArray array;
			do
			{
				socket->waitForReadyRead(100);
				stream.startTransaction();
				stream >> array;
			} while(!stream.commitTransaction() && socket->state() == QLocalSocket::ConnectedState);
			emit previewReceived(deserialize(array), this->currentPreviewGeneration);
			++processed;
		}
		if(processed != targets.count())
		{
			emit error("IPC worker didn't send enough previews. This is a bug, please report");
		}
	}
	socket->disconnectFromServer();
	emit finished();
}

void IPCPreviewClient::stopGeneration()
{
	if(!connect() || !socket->isOpen())
	{
		emit error("Could not connect to IPC worker");
		return;
	}
	QDataStream stream(socket);
	stream << StopGeneratePreviews;
	socket->flush();
}