Initial Commit

Source/DiscordRpc/Private/DiscordRpc.cpp

@@ -0,0 +1,77 @@
+// Copyright 1998-2017 Epic Games, Inc. All Rights Reserved.
+
+#include "DiscordRpc.h"
+#include "DiscordRpcPrivatePCH.h"
+#include "Interfaces/IPluginManager.h"
+#include "Modules/ModuleManager.h"
+
+#define LOCTEXT_NAMESPACE "FDiscordRpcModule"
+
+void FDiscordRpcModule::StartupModule()
+{
+#if !PLATFORM_LINUX
+#if defined(DISCORD_DYNAMIC_LIB)
+    // Get the base directory of this plugin
+    FString BaseDir = IPluginManager::Get().FindPlugin("DiscordRpc")->GetBaseDir();
+    const FString SDKDir =
+      FPaths::Combine(*BaseDir, TEXT("Source"), TEXT("ThirdParty"), TEXT("DiscordRpcLibrary"));
+#if PLATFORM_WINDOWS
+    const FString LibName = TEXT("discord-rpc");
+    const FString LibDir = FPaths::Combine(*SDKDir, TEXT("Win64"));
+    if (!LoadDependency(LibDir, LibName, DiscordRpcLibraryHandle)) {
+        FMessageDialog::Open(
+          EAppMsgType::Ok,
+          LOCTEXT(LOCTEXT_NAMESPACE,
+                  "Failed to load DiscordRpc plugin. Plug-in will not be functional."));
+        FreeDependency(DiscordRpcLibraryHandle);
+    }
+#elif PLATFORM_MAC
+    const FString LibName = TEXT("libdiscord-rpc");
+    const FString LibDir = FPaths::Combine(*SDKDir, TEXT("Mac"));
+    if (!LoadDependency(LibDir, LibName, DiscordRpcLibraryHandle)) {
+        FMessageDialog::Open(
+          EAppMsgType::Ok,
+          LOCTEXT(LOCTEXT_NAMESPACE,
+                  "Failed to load DiscordRpc plugin. Plug-in will not be functional."));
+        FreeDependency(DiscordRpcLibraryHandle);
+    }
+#endif
+#endif
+#endif
+}
+
+void FDiscordRpcModule::ShutdownModule()
+{
+    // Free the dll handle
+#if !PLATFORM_LINUX
+#if defined(DISCORD_DYNAMIC_LIB)
+    FreeDependency(DiscordRpcLibraryHandle);
+#endif
+#endif
+}
+
+bool FDiscordRpcModule::LoadDependency(const FString& Dir, const FString& Name, void*& Handle)
+{
+    FString Lib = Name + TEXT(".") + FPlatformProcess::GetModuleExtension();
+    FString Path = Dir.IsEmpty() ? *Lib : FPaths::Combine(*Dir, *Lib);
+
+    Handle = FPlatformProcess::GetDllHandle(*Path);
+
+    if (Handle == nullptr) {
+        return false;
+    }
+
+    return true;
+}
+
+void FDiscordRpcModule::FreeDependency(void*& Handle)
+{
+    if (Handle != nullptr) {
+        FPlatformProcess::FreeDllHandle(Handle);
+        Handle = nullptr;
+    }
+}
+
+#undef LOCTEXT_NAMESPACE
+
+IMPLEMENT_MODULE(FDiscordRpcModule, DiscordRpc)

Source/DiscordRpc/Private/DiscordRpcBlueprint.cpp

@@ -0,0 +1,172 @@
+#include "DiscordRpcBlueprint.h"
+#include "DiscordRpcPrivatePCH.h"
+#include "DiscordRpcBlueprint.h"
+#include "discord_rpc.h"
+
+DEFINE_LOG_CATEGORY(Discord)
+
+static UDiscordRpc* self = nullptr;
+
+static void ReadyHandler(const DiscordUser* connectedUser)
+{
+    FDiscordUserData ud;
+    ud.userId = ANSI_TO_TCHAR(connectedUser->userId);
+    ud.username = ANSI_TO_TCHAR(connectedUser->username);
+    ud.discriminator = ANSI_TO_TCHAR(connectedUser->discriminator);
+    ud.avatar = ANSI_TO_TCHAR(connectedUser->avatar);
+    UE_LOG(Discord,
+           Log,
+           TEXT("Discord connected to %s - %s#%s"),
+           *ud.userId,
+           *ud.username,
+           *ud.discriminator);
+    if (self) {
+        self->IsConnected = true;
+        self->OnConnected.Broadcast(ud);
+    }
+}
+
+static void DisconnectHandler(int errorCode, const char* message)
+{
+    auto msg = FString(message);
+    UE_LOG(Discord, Log, TEXT("Discord disconnected (%d): %s"), errorCode, *msg);
+    if (self) {
+        self->IsConnected = false;
+        self->OnDisconnected.Broadcast(errorCode, msg);
+    }
+}
+
+static void ErroredHandler(int errorCode, const char* message)
+{
+    auto msg = FString(message);
+    UE_LOG(Discord, Log, TEXT("Discord error (%d): %s"), errorCode, *msg);
+    if (self) {
+        self->OnErrored.Broadcast(errorCode, msg);
+    }
+}
+
+static void JoinGameHandler(const char* joinSecret)
+{
+    auto secret = FString(joinSecret);
+    UE_LOG(Discord, Log, TEXT("Discord join %s"), *secret);
+    if (self) {
+        self->OnJoin.Broadcast(secret);
+    }
+}
+
+static void SpectateGameHandler(const char* spectateSecret)
+{
+    auto secret = FString(spectateSecret);
+    UE_LOG(Discord, Log, TEXT("Discord spectate %s"), *secret);
+    if (self) {
+        self->OnSpectate.Broadcast(secret);
+    }
+}
+
+static void JoinRequestHandler(const DiscordUser* request)
+{
+    FDiscordUserData ud;
+    ud.userId = ANSI_TO_TCHAR(request->userId);
+    ud.username = ANSI_TO_TCHAR(request->username);
+    ud.discriminator = ANSI_TO_TCHAR(request->discriminator);
+    ud.avatar = ANSI_TO_TCHAR(request->avatar);
+    UE_LOG(Discord,
+           Log,
+           TEXT("Discord join request from %s - %s#%s"),
+           *ud.userId,
+           *ud.username,
+           *ud.discriminator);
+    if (self) {
+        self->OnJoinRequest.Broadcast(ud);
+    }
+}
+
+void UDiscordRpc::Initialize(const FString& applicationId,
+                             bool autoRegister,
+                             const FString& optionalSteamId)
+{
+    self = this;
+    IsConnected = false;
+    DiscordEventHandlers handlers{};
+    handlers.ready = ReadyHandler;
+    handlers.disconnected = DisconnectHandler;
+    handlers.errored = ErroredHandler;
+    if (OnJoin.IsBound()) {
+        handlers.joinGame = JoinGameHandler;
+    }
+    if (OnSpectate.IsBound()) {
+        handlers.spectateGame = SpectateGameHandler;
+    }
+    if (OnJoinRequest.IsBound()) {
+        handlers.joinRequest = JoinRequestHandler;
+    }
+    auto appId = StringCast<ANSICHAR>(*applicationId);
+    auto steamId = StringCast<ANSICHAR>(*optionalSteamId);
+    Discord_Initialize(
+      (const char*)appId.Get(), &handlers, autoRegister, (const char*)steamId.Get());
+}
+
+void UDiscordRpc::Shutdown()
+{
+    Discord_Shutdown();
+    self = nullptr;
+}
+
+void UDiscordRpc::RunCallbacks()
+{
+    Discord_RunCallbacks();
+}
+
+void UDiscordRpc::UpdatePresence()
+{
+    DiscordRichPresence rp{};
+
+    auto state = StringCast<ANSICHAR>(*RichPresence.state);
+    rp.state = state.Get();
+
+    auto details = StringCast<ANSICHAR>(*RichPresence.details);
+    rp.details = details.Get();
+
+    auto largeImageKey = StringCast<ANSICHAR>(*RichPresence.largeImageKey);
+    rp.largeImageKey = largeImageKey.Get();
+
+    auto largeImageText = StringCast<ANSICHAR>(*RichPresence.largeImageText);
+    rp.largeImageText = largeImageText.Get();
+
+    auto smallImageKey = StringCast<ANSICHAR>(*RichPresence.smallImageKey);
+    rp.smallImageKey = smallImageKey.Get();
+
+    auto smallImageText = StringCast<ANSICHAR>(*RichPresence.smallImageText);
+    rp.smallImageText = smallImageText.Get();
+
+    auto partyId = StringCast<ANSICHAR>(*RichPresence.partyId);
+    rp.partyId = partyId.Get();
+
+    auto matchSecret = StringCast<ANSICHAR>(*RichPresence.matchSecret);
+    rp.matchSecret = matchSecret.Get();
+
+    auto joinSecret = StringCast<ANSICHAR>(*RichPresence.joinSecret);
+    rp.joinSecret = joinSecret.Get();
+
+    auto spectateSecret = StringCast<ANSICHAR>(*RichPresence.spectateSecret);
+    rp.spectateSecret = spectateSecret.Get();
+    rp.startTimestamp = RichPresence.startTimestamp;
+    rp.endTimestamp = RichPresence.endTimestamp;
+    rp.partySize = RichPresence.partySize;
+    rp.partyMax = RichPresence.partyMax;
+    rp.instance = RichPresence.instance;
+
+    Discord_UpdatePresence(&rp);
+}
+
+void UDiscordRpc::ClearPresence()
+{
+    Discord_ClearPresence();
+}
+
+void UDiscordRpc::Respond(const FString& userId, int reply)
+{
+    UE_LOG(Discord, Log, TEXT("Responding %d to join request from %s"), reply, *userId);
+    FTCHARToUTF8 utf8_userid(*userId);
+    Discord_Respond(utf8_userid.Get(), reply);
+}

Source/DiscordRpc/Private/DiscordRpcPrivatePCH.h

@@ -0,0 +1,2 @@
+#include "Core.h"
+#include "DiscordRpc.h"