WIP: main

2020-12-26 15:03:51 +03:00 · 2020-12-26 15:03:51 +03:00 · 15730b470e
commit 15730b470e
parent f25f34565a
52 changed files with 2438 additions and 208 deletions
--- a/service/server/localserver.cpp
+++ b/service/server/localserver.cpp
@ -0,0 +1,168 @@
+#include <QCoreApplication>
+#include <QFileInfo>
+#include <QLocalServer>
+#include <QLocalSocket>
+
+#include "localserver.h"
+#include "utils.h"
+
+LocalServer::LocalServer(const QString& name, QObject *parent) : QObject(parent),
+    m_clientConnected(false),
+    m_clientConnection(nullptr)
+{
+    m_server = new QLocalServer(this);
+    m_server->setSocketOptions(QLocalServer::WorldAccessOption);
+
+    if (!m_server->listen(name)) {
+        qDebug() << QString("Unable to start the server: %1.").arg(m_server->errorString());
+        return;
+    }
+
+    connect(m_server, &QLocalServer::newConnection, this, &LocalServer::onNewConnection);
+
+    qDebug() << "Local server started";
+}
+
+LocalServer::~LocalServer()
+{
+    m_clientConnected = false;
+    m_server->disconnect();
+
+    qDebug() << "Local server stopped";
+}
+
+bool LocalServer::isRunning() const
+{
+    return m_server->isListening();
+}
+
+void LocalServer::onNewConnection()
+{
+    if (m_clientConnection) {
+        m_clientConnection->deleteLater();
+    }
+
+    m_clientConnection = m_server->nextPendingConnection();
+    connect(m_clientConnection, &QLocalSocket::disconnected, this, &LocalServer::onDisconnected);
+    m_clientConnected = true;
+
+    qDebug() << "On new connection";
+
+    for(;;) {
+        qApp->processEvents(QEventLoop::ExcludeUserInputEvents);
+        if (!m_clientConnected || !m_clientConnection) {
+            break;
+        }
+
+        if (m_clientConnection->waitForReadyRead() && m_clientConnection->canReadLine()) {
+            char buf[1024];
+            qint64 lineLength = m_clientConnection->readLine(buf, sizeof(buf));
+            if (lineLength != -1) {
+                QString line = buf;
+                line = line.simplified();
+                qDebug().noquote() << QString("Readed line: '%1'").arg(line);
+                Message icomingMessage(line);
+                if (!icomingMessage.isValid()) {
+                    qWarning().noquote() << "Message is not valid!";
+                    continue;
+                }
+
+                switch (icomingMessage.state()) {
+                case Message::State::Initialize:
+                    sendMessage(Message(Message::State::Initialize, QStringList({"Pong"})));
+                    break;
+                case Message::State::StartRequest:
+                    startProcess(icomingMessage.args());
+                    break;
+                case Message::State::FinishRequest:
+                    finishProcess(icomingMessage.args());
+                    break;
+                default:
+                    ;
+                }
+            }
+        }
+    }
+
+    qDebug() << "Released";
+}
+
+void LocalServer::finishProcess(const QStringList& args)
+{
+    Q_UNUSED(args)
+}
+
+void LocalServer::startProcess(const QStringList& messageArgs)
+{
+    if (messageArgs.size() < 1) {
+        return;
+    }
+
+    QProcess* process = new QProcess();
+    connect(process, SIGNAL(started()), this, SLOT(onStarted()));
+    connect(process, SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(onFinished(int, QProcess::ExitStatus)));
+
+    const QString program = messageArgs.at(0);
+    QStringList args;
+    for (int i = 1; i < messageArgs.size(); i++) {
+        args.append(messageArgs.at(i));
+    }
+
+    QFileInfo fi(program);
+    const QString baseName = fi.baseName();
+    if (!fi.exists()) {
+        qWarning() << "This program does not exist";
+        sendMessage(Message(Message::State::Started, QStringList({baseName})));
+        sendMessage(Message(Message::State::Finished, QStringList({baseName, QString::number(-1)})));
+        return;
+    }
+
+    process->setObjectName(baseName);
+
+    qDebug().noquote() << QString("Start process '%1' - '%2' with args '%3'")
+                .arg(baseName).arg(program).arg(args.join(","));
+
+    process->start(program, args);
+    m_processList.append(process);
+}
+
+void LocalServer::onFinished(int exitCode, QProcess::ExitStatus exitStatus)
+{
+    Q_UNUSED(exitStatus)
+
+    QProcess* process = (QProcess*)sender();
+    sendMessage(Message(Message::State::Finished, QStringList({process->objectName(), QString::number(exitCode)})));
+}
+
+void LocalServer::onStarted()
+{
+    QProcess* process = (QProcess*)sender();
+    sendMessage(Message(Message::State::Started, QStringList({process->objectName()})));
+}
+
+void LocalServer::onDisconnected()
+{
+    if (!m_clientConnected) {
+        return;
+    }
+
+    m_clientConnected = false;
+    QLocalSocket* clientConnection = (QLocalSocket*)sender();
+    clientConnection->deleteLater();
+
+    qDebug() << "Diconnected";
+}
+
+void LocalServer::sendMessage(const Message& message)
+{
+    if (!m_clientConnection || !m_clientConnected) {
+        qDebug()<< "Cannot send data, remote peer is not connected";
+        return;
+    }
+
+    const QString data = message.toString();
+    bool status = m_clientConnection->write(QString(data + "\n").toLocal8Bit());
+
+    qDebug().noquote() << QString("Send message '%1', status '%2'").arg(data).arg(Utils::toString(status));
+}
+