1 files changed, 56 insertions, 7 deletions

diff --git a/discord_rich_presence.cpp b/discord_rich_presence.cpp
index f5f6c11..bcc0d2a 100644
--- a/discord_rich_presence.cpp
+++ b/discord_rich_presence.cpp
@@ -1,6 +1,7 @@
 #include "discord_rich_presence.h"
 
 #include <cstring>
+#include <ctime>
 
 #include <QDebug>
 
@@ -14,15 +15,13 @@ Discord::Discord()
   handlers.ready = [] {
     qInfo() << "Discord RPC ready";
   };
-  /*
   handlers.disconnected = [](int errorCode, const char* message) {
     qInfo() << "Discord RPC disconnected! " << message;
   };
   handlers.errored = [](int errorCode, const char* message) {
     qWarning() << "Discord RPC errored out! " << message;
   };
-  */
-  qInfo() << "Are things working out all right?";
+  qInfo() << "Initializing Discord RPC";
   Discord_Initialize(APPLICATION_ID, &handlers, 1, nullptr);
 }
 
@@ -35,24 +34,74 @@ void Discord::state_lobby()
 {
   DiscordRichPresence presence;
   std::memset(&presence, 0, sizeof(presence));
+  presence.largeImageKey = "ao2-logo";
+  presence.largeImageText = "Objection!";
+  presence.instance = 1;
+
   presence.state = "In Lobby";
   presence.details = "Idle";
+  Discord_UpdatePresence(&presence);
+}
+
+void Discord::state_server(std::string name, std::string server_id)
+{
+  qDebug() << "Discord RPC: Setting server state";
+
+  DiscordRichPresence presence;
+  std::memset(&presence, 0, sizeof(presence));
   presence.largeImageKey = "ao2-logo";
   presence.largeImageText = "Objection!";
   presence.instance = 1;
+
+  auto timestamp = static_cast<int64_t>(std::time(nullptr));
+
+  presence.state = "In a Server";
+  presence.details = name.c_str();
+  presence.matchSecret = server_id.c_str();
+  presence.startTimestamp = this->timestamp;
+
+  this->server_id = server_id;
+  this->server_name = name;
+  this->timestamp = timestamp;
   Discord_UpdatePresence(&presence);
 }
 
-void Discord::state_server(const char* name, const char* server_id)
+void Discord::state_character(std::string name)
 {
+  auto name_internal = QString(name.c_str()).toLower().replace(' ', '_').toStdString();
+  auto name_friendly = QString(name.c_str()).replace('_', ' ').toStdString();
+  const std::string playing_as = "Playing as " + name_friendly;
+  qDebug() << "Discord RPC: Setting character state (" << playing_as.c_str() << ")";
+
+  DiscordRichPresence presence;
+  std::memset(&presence, 0, sizeof(presence));
+  presence.largeImageKey = "ao2-logo";
+  presence.largeImageText = "Objection!";
+  presence.instance = 1;
+  presence.details = this->server_name.c_str();
+  presence.matchSecret = this->server_id.c_str();
+  presence.startTimestamp = this->timestamp;
+
+  presence.state = playing_as.c_str();
+  presence.smallImageKey = name_internal.c_str();
+  // presence.smallImageText = name_internal.c_str();
+  Discord_UpdatePresence(&presence);
+}
+
+void Discord::state_spectate()
+{
+  qDebug() << "Discord RPC: Setting specator state";
+
   DiscordRichPresence presence;
   std::memset(&presence, 0, sizeof(presence));
-  presence.state = "In a Server";
-  presence.details = name;
   presence.largeImageKey = "ao2-logo";
   presence.largeImageText = "Objection!";
   presence.instance = 1;
-  presence.matchSecret = server_id;
+  presence.details = this->server_name.c_str();
+  presence.matchSecret = this->server_id.c_str();
+  presence.startTimestamp = this->timestamp;
+
+  presence.state = "Spectating";
   Discord_UpdatePresence(&presence);
 }