1 files changed, 101 insertions, 0 deletions
diff --git a/src/widgets/direct_connect_dialog.cpp b/src/widgets/direct_connect_dialog.cpp
new file mode 100644
index 00000000..82eaa23a
--- /dev/null
+++ b/src/widgets/direct_connect_dialog.cpp
@@ -0,0 +1,101 @@
+#include "widgets/direct_connect_dialog.h"
+
+#include "networkmanager.h"
+#include "options.h"
+#include "debug_functions.h"
+
+#include <QComboBox>
+#include <QLabel>
+#include <QLineEdit>
+#include <QPushButton>
+#include <QSpinBox>
+#include <QUiLoader>
+#include <QVBoxLayout>
+#include <QRegularExpressionMatch>
+#include <QStringBuilder>
+#include <QUrl>
+
+#define FROM_UI(type, name)                                                    \
+  ;                                                                            \
+  ui_##name = findChild<type *>(#name);
+
+DirectConnectDialog::DirectConnectDialog(NetworkManager *p_net_manager) :
+    net_manager(p_net_manager)
+{
+    QUiLoader l_loader(this);
+    QFile l_uiFile(Options::getInstance().getUIAsset(DEFAULT_UI));
+
+    if (!l_uiFile.open(QFile::ReadOnly)) {
+      qCritical() << "Unable to open file " << l_uiFile.fileName();
+      return;
+    }
+    ui_widget = l_loader.load(&l_uiFile, this);
+
+    auto l_layout = new QVBoxLayout(this);
+    l_layout->addWidget(ui_widget);
+
+    FROM_UI(QLineEdit, direct_hostname_edit)
+
+    FROM_UI(QLabel, direct_connection_status_lbl)
+
+    FROM_UI(QPushButton, direct_connect_button);
+    connect(ui_direct_connect_button, &QPushButton::pressed,
+            this, &DirectConnectDialog::onConnectPressed);
+    FROM_UI(QPushButton, direct_cancel_button);
+    connect(ui_direct_cancel_button, &QPushButton::pressed,
+            this, &DirectConnectDialog::close);
+
+    connect(net_manager, &NetworkManager::server_connected,
+            this, &DirectConnectDialog::onServerConnected);
+
+    connect(&connect_timeout, &QTimer::timeout, this,
+            &DirectConnectDialog::onConnectTimeout);
+    connect_timeout.setSingleShot(true);
+}
+
+void DirectConnectDialog::onConnectPressed()
+{
+  QString l_hostname = ui_direct_hostname_edit->text();
+  if (!SCHEME_PATTERN.match(l_hostname).hasMatch()) {
+    l_hostname = "tcp://" % l_hostname;
+  }
+  QUrl l_url(l_hostname);
+  if (!l_url.isValid()) {
+    call_error(tr("Invalid URL."));
+    return;
+  }
+  if (!to_connection_type.contains(l_url.scheme())) {
+    call_error(tr("Scheme not recognized. Must be either of the following: ") % QStringList::fromVector(to_connection_type.keys().toVector()).join(", "));
+    return;
+  }
+  if (l_url.port() == -1) {
+    call_error(tr("Invalid server port."));
+    return;
+  }
+  server_type l_server;
+  l_server.socket_type = to_connection_type[l_url.scheme()];
+  l_server.ip = l_url.host();
+  l_server.port = l_url.port();
+  l_server.name = "Direct Connection";
+
+  net_manager->connect_to_server(l_server);
+  ui_direct_connect_button->setEnabled(false);
+  ui_direct_connection_status_lbl->setText("Connecting...");
+  ui_direct_connection_status_lbl->setStyleSheet("color : rgb(0,64,156)");
+  connect_timeout.start(CONNECT_TIMEOUT);
+}
+
+void DirectConnectDialog::onServerConnected()
+{
+  net_manager->join_to_server();
+  ui_direct_connection_status_lbl->setText("Connected!");
+  ui_direct_connection_status_lbl->setStyleSheet("color: rgb(0,128,0)");
+  close();
+}
+
+void DirectConnectDialog::onConnectTimeout()
+{
+  ui_direct_connect_button->setEnabled(true);
+  ui_direct_connection_status_lbl->setText("Connection Timeout!");
+  ui_direct_connection_status_lbl->setStyleSheet("color: rgb(255,0,0)");
+}