amnezia-client/client/core/utils/managementServer.cpp

#include <QDebug>
#include <QTcpServer>
#include <QTcpSocket>

#include "core/utils/managementServer.h"

ManagementServer::ManagementServer(QObject *parent) : QObject(parent),
    m_tcpServer(nullptr)
{

}

ManagementServer::~ManagementServer()
{

}

bool ManagementServer::isOpen() const
{
    return (m_socket && m_socket->isOpen());
}

void ManagementServer::stop()
{
    if (m_tcpServer) m_tcpServer->close();
}

void ManagementServer::onAcceptError(QAbstractSocket::SocketError socketError)
{
    qDebug().noquote() << QString("Accept error: %1").arg(socketError);
}

qint64 ManagementServer::writeCommand(const QString& message)
{
    if (!isOpen()) {
        return 0;
    }

    const QString command = message + "\n";
    qint64 bytesWritten = m_socket->write(command.toStdString().c_str());
    m_socket->flush();
    return bytesWritten;
}

void ManagementServer::onNewConnection()
{
    qDebug() << "New incoming connection";

    m_socket = QPointer<QTcpSocket>(m_tcpServer->nextPendingConnection());
    if (m_tcpServer) m_tcpServer->close();

    QObject::connect(m_socket.data(), &QTcpSocket::disconnected, this, &ManagementServer::onSocketDisconnected);
    QObject::connect(m_socket.data(), &QTcpSocket::errorOccurred, this, &ManagementServer::onSocketError);
    QObject::connect(m_socket.data(), &QTcpSocket::readyRead, this, &ManagementServer::onReadyRead);
}

void ManagementServer::onSocketError(QAbstractSocket::SocketError socketError)
{
    Q_UNUSED(socketError)
    qDebug().noquote() << QString("Management server error: %1").arg(m_socket->errorString());
}

void ManagementServer::onSocketDisconnected()
{
    if (m_socket) m_socket->deleteLater();
}

QPointer<QTcpSocket> ManagementServer::socket() const
{
    if (!isOpen()) {
        return nullptr;
    }
    return m_socket;
}

void ManagementServer::onReadyRead()
{
    emit readyRead();
}

bool ManagementServer::start(const QString& host, unsigned int port)
{
    if (m_tcpServer) m_tcpServer->close();

    m_tcpServer =  QSharedPointer<QTcpServer>(new QTcpServer(this), [](QTcpServer *s){
        if (s) s->deleteLater();
    });
    m_tcpServer->setMaxPendingConnections(1);

    connect(m_tcpServer.data(), SIGNAL(acceptError(QAbstractSocket::SocketError)), this, SLOT(onAcceptError(QAbstractSocket::SocketError)));
    connect(m_tcpServer.data(), SIGNAL(newConnection()), this, SLOT(onNewConnection()));

    if (m_tcpServer->listen(QHostAddress(host), port)) {
        emit serverStarted();
        return true;
    }

    qDebug().noquote() << QString("Can't start TCP server, %1,%2")
                      .arg(m_tcpServer->serverError())
                      .arg(m_tcpServer->errorString());
    return false;
}

QString ManagementServer::readLine()
{
    if (!isOpen()) {
        qDebug() << "Socket is not opened";
        return QString();
    }

    return m_socket->readLine();
}