1 files changed, 72 insertions, 0 deletions
diff --git a/src/networkmanager.h b/src/networkmanager.h
new file mode 100644
index 00000000..984f30f8
--- /dev/null
+++ b/src/networkmanager.h
@@ -0,0 +1,72 @@
+#pragma once
+
+#include "aoapplication.h"
+#include "aopacket.h"
+
+#include <QDnsLookup>
+#include <QNetworkAccessManager>
+#include <QTime>
+#include <QTimer>
+#include <QtWebSockets/QWebSocket>
+
+#include <cstring>
+
+enum MSDocumentType
+{
+  PrivacyPolicy,
+  Motd,
+  ClientVersion
+};
+
+class NetworkManager : public QObject
+{
+  Q_OBJECT
+
+public:
+  explicit NetworkManager(AOApplication *parent);
+  ~NetworkManager() = default;
+
+  void connect_to_server(server_type p_server);
+  void disconnect_from_server();
+
+public Q_SLOTS:
+  void get_server_list();
+  void ship_server_packet(AOPacket p_packet);
+  void join_to_server();
+  void handle_server_packet(const QString &p_data);
+
+  void request_document(MSDocumentType document_type, const std::function<void(QString)> &cb);
+  void send_heartbeat();
+
+Q_SIGNALS:
+  void server_connected(bool state);
+
+private Q_SLOTS:
+  void ms_request_finished(QNetworkReply *reply);
+
+private:
+  AOApplication *ao_app;
+  QNetworkAccessManager *http;
+
+  union
+  {
+    QWebSocket *ws;
+    QTcpSocket *tcp;
+  } server_socket;
+  connection_type active_connection_type;
+  bool connected = false;
+
+  QTimer *heartbeat_timer;
+
+  const QString DEFAULT_MS_BASEURL = "http://servers.aceattorneyonline.com";
+  QString ms_baseurl = DEFAULT_MS_BASEURL;
+
+  const int heartbeat_interval = 60 * 5 * 1000;
+
+  bool partial_packet = false;
+  QString temp_packet;
+
+  unsigned int s_decryptor = 5;
+
+  QString get_user_agent() const { return QStringLiteral("AttorneyOnline/%1 (Desktop)").arg(ao_app->get_version_string()); }
+};