1 files changed, 162 insertions, 0 deletions

@@ -0,0 +1,162 @@
+#include "ap_state.h"
+
+#include <hkutil/string.h>
+
+#include <apuuid.hpp>
+#include <chrono>
+#include <exception>
+#include <list>
+#include <thread>
+
+constexpr int AP_MAJOR = 0;
+constexpr int AP_MINOR = 4;
+constexpr int AP_REVISION = 0;
+
+constexpr int ITEM_HANDLING = 7;  // <- all
+
+APState::APState() {
+  std::thread([this]() {
+    for (;;) {
+      {
+        std::lock_guard client_guard(client_mutex_);
+        if (apclient_) {
+          apclient_->poll();
+        }
+      }
+
+      std::this_thread::sleep_for(std::chrono::milliseconds(100));
+    }
+  }).detach();
+}
+
+void APState::Connect(std::string server, std::string player,
+                      std::string password) {
+  tracker_frame_->SetStatusMessage("Connecting to Archipelago server....");
+
+  {
+    std::lock_guard client_guard(client_mutex_);
+
+    if (apclient_) {
+      apclient_->reset();
+    }
+
+    apclient_ = std::make_unique<APClient>(ap_get_uuid(""), "Lingo", server);
+  }
+
+  inventory_.clear();
+  checked_locations_.clear();
+
+  bool connected = false;
+  bool has_connection_result = false;
+
+  apclient_->set_room_info_handler([&]() {
+    tracker_frame_->SetStatusMessage(
+        "Connected to Archipelago server. Authenticating...");
+
+    apclient_->ConnectSlot(player, password, ITEM_HANDLING, {"Tracker"},
+                           {AP_MAJOR, AP_MINOR, AP_REVISION});
+  });
+
+  apclient_->set_location_checked_handler(
+      [&](const std::list<int64_t>& locations) {
+        for (const int64_t location_id : locations) {
+          checked_locations_.insert(location_id);
+          std::cout << "Location: " << location_id << std::endl;
+        }
+
+        tracker_frame_->Refresh();
+      });
+
+  apclient_->set_slot_disconnected_handler([&]() {
+    tracker_frame_->SetStatusMessage("Disconnected from Archipelago.");
+  });
+
+  apclient_->set_socket_disconnected_handler([&]() {
+    tracker_frame_->SetStatusMessage("Disconnected from Archipelago.");
+  });
+
+  apclient_->set_items_received_handler(
+      [&](const std::list<APClient::NetworkItem>& items) {
+        for (const APClient::NetworkItem& item : items) {
+          // TODO: Progressive items.
+
+          inventory_.insert(item.item);
+          std::cout << "Item: " << item.item << std::endl;
+        }
+
+        tracker_frame_->Refresh();
+      });
+
+  apclient_->set_slot_connected_handler([&](const nlohmann::json& slot_data) {
+    tracker_frame_->SetStatusMessage("Connected to Archipelago!");
+
+    connected = true;
+    has_connection_result = true;
+  });
+
+  apclient_->set_slot_refused_handler(
+      [&](const std::list<std::string>& errors) {
+        connected = false;
+        has_connection_result = true;
+
+        tracker_frame_->SetStatusMessage("Disconnected from Archipelago.");
+
+        std::vector<std::string> error_messages;
+        error_messages.push_back("Could not connect to Archipelago.");
+
+        for (const std::string& error : errors) {
+          if (error == "InvalidSlot") {
+            error_messages.push_back("Invalid player name.");
+          } else if (error == "InvalidGame") {
+            error_messages.push_back(
+                "The specified player is not playing Lingo.");
+          } else if (error == "IncompatibleVersion") {
+            error_messages.push_back(
+                "The Archipelago server is not the correct version for this "
+                "client.");
+          } else if (error == "InvalidPassword") {
+            error_messages.push_back("Incorrect password.");
+          } else if (error == "InvalidItemsHandling") {
+            error_messages.push_back(
+                "Invalid item handling flag. This is a bug with the tracker. "
+                "Please report it to the lingo-ap-tracker GitHub.");
+          } else {
+            error_messages.push_back("Unknown error.");
+          }
+        }
+
+        std::string full_message = hatkirby::implode(error_messages, " ");
+
+        wxMessageBox(full_message, "Connection failed", wxOK | wxICON_ERROR);
+      });
+
+  client_active_ = true;
+
+  int timeout = 5000;  // 5 seconds
+  int interval = 100;
+  int remaining_loops = timeout / interval;
+  while (!has_connection_result) {
+    if (interval == 0) {
+      connected = false;
+      has_connection_result = true;
+
+      tracker_frame_->SetStatusMessage("Disconnected from Archipelago.");
+
+      wxMessageBox("Timeout while connecting to Archipelago server.",
+                   "Connection failed", wxOK | wxICON_ERROR);
+    }
+
+    std::this_thread::sleep_for(std::chrono::milliseconds(100));
+
+    interval--;
+  }
+
+  if (!connected) {
+    client_active_ = false;
+  }
+}
+
+APState& GetAPState() {
+  static APState* instance = new APState();
+  return *instance;
+}