Move sources to src/ && lots of features

1 files changed, 308 insertions, 0 deletions

diff --git a/src/session.cpp b/src/session.cpp
new file mode 100644
index 0000000..d668e85
--- /dev/null
+++ b/src/session.cpp
@@ -0,0 +1,308 @@
+#pragma once
+
+#include "session.h"
+
+Session::Session(QObject *parent, QQmlApplicationEngine* engine, MainQMLAdaptor* main_gui) : QObject(parent) {
+    _initDBus();
+
+    _engine = engine;
+
+    _main_gui = main_gui;
+
+    statusTimer = new QTimer(this);
+    connect(statusTimer, &QTimer::timeout, this, &Session::statusTimerEvent);
+
+    this->init_vars();
+}
+
+void Session::statusTimerEvent() {
+    this->refresh_status(this->getId());
+}
+
+void Session::init_vars() {
+    setPin("-----");
+    setId(-1);
+    setURL(tr("Not available yet"));
+    setStatus("unknown");
+
+    _main_gui->setConnectButtonChecked(false);
+    _main_gui->setConnectButtonEnabled(true);
+    _main_gui->setStatusIndicator(false);
+
+    this->statusTimer->stop();
+}
+
+QString Session::getStatus() {
+    return _status;
+}
+
+QString Session::getURL() {
+    return _url;
+}
+
+int Session::getId() {
+    return _id;
+}
+
+QString Session::getPin() {
+    return _pin;
+}
+
+void Session::setStatus(QString status) {
+    _status = status;
+
+    QString guiString = tr("Unknown state of service");
+    _main_gui->setStatusIndicator(false);
+
+    if (status == "running") {
+        guiString = tr("Remote Support session is ready to be connected to");
+        _main_gui->setStatusIndicator(true, QColor(255, 255, 0, 127));
+    } else if (status == "dead") {
+        guiString = tr("Remote Support session was stopped ungracefully");
+
+        // Clear current variables
+        this->init_vars();
+        _main_gui->setStatusIndicator(true, QColor(255, 0, 0, 127));
+    } else if (status == "stopped") {
+        guiString = tr("Remote Support session was stopped");
+
+        // Clear current variables
+        this->init_vars();
+    } else if (status == "active") {
+        guiString = tr("Your partner is connected to the Remote Support session");
+        _main_gui->setStatusIndicator(true, QColor(0, 255, 0, 127));
+    } else if (status == "waiting_start_request_answer") {
+        guiString = tr("Trying to reach session service...");
+    }
+
+    _main_gui->setStatus(guiString);
+
+    emit statusChanged(_status);
+}
+
+void Session::setURL(QString url) {
+    _url = url;
+    emit urlChanged(url);
+}
+
+void Session::setId(int id) {
+    _id = id;
+    emit idChanged(id);
+}
+
+void Session::setPin(QString pin) {
+    _pin = pin;
+    emit pinChanged(pin);
+}
+
+void Session::handleConnectButtonClick(bool checked) {
+    qDebug() << "-----Connect button handler-----\nCurrent session #" << this->getId();
+
+    // Stopping even if nothing is running
+    this->stop(this->getId());
+
+    if (checked) {
+        // Start the Session again
+        this->start();
+    }
+    qDebug() << "-----\\Connect button handler-----";
+}
+
+void Session::_initDBus() {
+    if (!QDBusConnection::sessionBus().isConnected()) {
+        qCritical() << "Cannot connect to the D-Bus session bus.";
+    }
+
+    // Create DBus object
+    _dbus_rwa = new OrgArcticaProjectRWAInterface("org.ArcticaProject.RWA", "/RWA",
+                               QDBusConnection::sessionBus(), this->parent());
+
+    qDebug("Initialized DBus object!");
+}
+
+void Session::start() {
+    qDebug() << "Requesting D-Bus session service to start a new session";
+
+    // Make an asynchrous 'start' call (Response will be sent to 'start_dbus_replied')
+    QDBusPendingCall async = _dbus_rwa->asyncCall("start");
+    QDBusPendingCallWatcher *watcher = new QDBusPendingCallWatcher(async, this);
+
+    QObject::connect(watcher, SIGNAL(finished(QDBusPendingCallWatcher*)),
+                     this, SLOT(start_dbus_replied(QDBusPendingCallWatcher*)));
+
+    // Disable connect button to reduce spam.
+    // And don't enable it as long there is no status update.
+    // (Or something fails)
+    _main_gui->setConnectButtonEnabled(false);
+    this->setStatus("waiting_start_request_answer");
+}
+
+void Session::start_dbus_replied(QDBusPendingCallWatcher *call) {
+    QString result = "";
+
+    QDBusPendingReply<QString> reply = *call;
+    if (reply.isError()) {
+        qCritical("Error: cannot parse 'start' reply from D-Bus service");
+        qDebug() << "The reply" << reply.error();
+
+        // The user should have the oportunity to try again.
+        this->init_vars();
+
+        _main_gui->openMessageDialog(tr("Remote Support for your Desktop"),
+                                     tr("D-Bus response was invalid.\nFunction call: '%1'\nMaybe service isn't listening?").arg("start"),
+                                     QMessageBox::Icon::Critical);
+
+        return;
+    } else {
+        result = reply.argumentAt<0>();
+    }
+    call->deleteLater();
+
+    qDebug() << "Raw JSON from starting session is:" << result;
+    QJsonDocument doc = QJsonDocument::fromJson(result.toUtf8());
+
+    // Get the QJsonObject
+    QJsonObject jObject = doc.object();
+    QVariantMap mainMap = jObject.toVariantMap();
+
+    // Session ID == PID
+    int sessionid = mainMap["id"].toInt();
+    this->setId(sessionid);
+
+    // URL of remote web app frontend
+    QString url = mainMap["url"].toString();
+    this->setURL(url);
+
+    // PIN
+    QString pin = mainMap["pin"].toString();
+    this->setPin(pin);
+
+    qDebug() << "Got session id:" << sessionid;
+    qDebug() << "Got url:" << url;
+    qDebug() << "Got pin:" << pin;
+
+    this->getStatus();
+    emit pinChanged(pin);
+    emit urlChanged(url);
+    emit idChanged(sessionid);
+
+    statusTimer->start(1000);
+}
+
+void Session::stop(int pid) {
+    if (pid <= 0 ){
+        qDebug() << "Won't send a request to D-Bus service to stop a session when session ID <= 0";
+        return;
+    }
+
+    // Stopping now.
+    qDebug() << "Requesting D-Bus session service to stop session #" << pid;
+
+    // Make an asynchrous 'stop' call (Response will be sent to 'stop_dbus_replied')
+    QDBusPendingCall async = _dbus_rwa->asyncCall("stop", pid);
+    QDBusPendingCallWatcher *watcher = new QDBusPendingCallWatcher(async, this);
+
+    QObject::connect(watcher, SIGNAL(finished(QDBusPendingCallWatcher*)),
+                     this, SLOT(stop_dbus_replied(QDBusPendingCallWatcher*)));
+
+    // Clear current variables
+    this->init_vars();
+
+    // Disable connect button to reduce spam.
+    // And don't enable it as long there is no status update.
+    // (Or something fails)
+    _main_gui->setConnectButtonEnabled(false);
+}
+
+void Session::stop_dbus_replied(QDBusPendingCallWatcher *call) {
+    QString result = "";
+
+    QDBusPendingReply<QString> reply = *call;
+    if (reply.isError()) {
+        qCritical("Error: cannot parse 'stop' reply from D-Bus service");
+        qDebug() << "The reply" << reply.error();
+
+        this->init_vars();
+
+        _main_gui->openMessageDialog(tr("Remote Support for your Desktop"),
+                                     tr("D-Bus response was invalid.\nFunction call: '%1'\nMaybe service isn't listening?").arg("stop"),
+                                     QMessageBox::Icon::Critical);
+
+        return;
+    } else {
+        result = reply.argumentAt<0>();
+    }
+    call->deleteLater();
+
+    // Get the QJsonObject
+    QJsonDocument doc = QJsonDocument::fromJson(result.toUtf8());
+    QJsonObject jObject = doc.object();
+    QVariantMap mainMap = jObject.toVariantMap();
+    qDebug() << "Refreshed status:" << mainMap["status"].toString();
+    this->setStatus(mainMap["status"].toString());
+
+    // Clear current variables
+    this->init_vars();
+}
+
+void Session::status(int pid) {
+    // Requesting status now.
+    qDebug() << "Requesting status for session #" << pid;
+
+    // Make an asynchrous 'status' call (Response will be sent to 'status_dbus_replied')
+    QDBusPendingCall async = _dbus_rwa->asyncCall("status", pid);
+    QDBusPendingCallWatcher *watcher = new QDBusPendingCallWatcher(async, this);
+
+    QObject::connect(watcher, SIGNAL(finished(QDBusPendingCallWatcher*)),
+                     this, SLOT(stop_dbus_replied(QDBusPendingCallWatcher*)));
+}
+
+void Session::refresh_status(int pid) {
+    // Refreshing status
+    qDebug() << "Requesting status refresh for session #" << pid;
+
+    // Make an asynchrous 'refresh_status' call (Response will be sent to 'status_dbus_replied')
+    QDBusPendingCall async = _dbus_rwa->asyncCall("refresh_status", pid);
+    QDBusPendingCallWatcher *watcher = new QDBusPendingCallWatcher(async, this);
+
+    QObject::connect(watcher, SIGNAL(finished(QDBusPendingCallWatcher*)),
+                     this, SLOT(status_dbus_replied(QDBusPendingCallWatcher*)));
+}
+
+void Session::status_dbus_replied(QDBusPendingCallWatcher *call) {
+    QString result = "";
+
+    QDBusPendingReply<QString> reply = *call;
+    if (reply.isError()) {
+        qCritical("Error: cannot parse 'status' reply from D-Bus service");
+        qDebug() << "The reply" << reply.error();
+
+        _main_gui->openMessageDialog(tr("Remote Support for your Desktop"),
+                                     tr("D-Bus response was invalid.\nFunction call: '%1'\nMaybe service isn't listening?").arg("status"),
+                                     QMessageBox::Icon::Critical);
+
+        // Includes stopping status-refreshing timer
+        this->init_vars();
+
+        return;
+    } else {
+        result = reply.argumentAt<0>();
+    }
+    call->deleteLater();
+
+    // Get the QJsonObject
+    QJsonDocument doc = QJsonDocument::fromJson(result.toUtf8());
+    QJsonObject jObject = doc.object();
+    QVariantMap mainMap = jObject.toVariantMap();
+    qDebug() << "Refreshed status:" << mainMap["status"].toString();
+    this->setStatus(mainMap["status"].toString());
+
+    // Enable (dis)connect button
+    _main_gui->setConnectButtonEnabled(true);
+}
+
+//void Session::onCloseHandler() {
+//    qDebug() << "Inside Session::onCloseHandler()";
+
+//    this->stop(this->getId());
+//}