[Friends] Add basic friends support using steam

This commit is contained in:
momo5502 2017-01-25 22:39:00 +01:00
parent c8a41139cf
commit 9c823e14b2
5 changed files with 277 additions and 0 deletions

View File

@ -42,6 +42,7 @@ namespace Components
Loader::Register(new Window());
Loader::Register(new Command());
Loader::Register(new Console());
Loader::Register(new Friends());
Loader::Register(new IPCPipe());
Loader::Register(new ModList());
Loader::Register(new Network());

View File

@ -68,6 +68,7 @@ namespace Components
#include "Modules/RCon.hpp"
#include "Modules/Party.hpp" // Destroys the order, but requires network classes :D
#include "Modules/Logger.hpp"
#include "Modules/Friends.hpp"
#include "Modules/Download.hpp"
#include "Modules/Playlist.hpp"
#include "Modules/RawFiles.hpp"

View File

@ -0,0 +1,224 @@
#include "STDInclude.hpp"
namespace Components
{
unsigned int Friends::CurrentFriend;
std::recursive_mutex Friends::Mutex;
std::vector<Friends::Friend> Friends::FriendsList;
void Friends::UpdateUserInfo(SteamID user)
{
if (!Steam::Proxy::SteamFriends) return;
std::lock_guard<std::recursive_mutex> _(Friends::Mutex);
Friends::Friend userInfo;
auto i = std::find_if(Friends::FriendsList.begin(), Friends::FriendsList.end(), [user] (Friends::Friend entry)
{
return (entry.userId.Bits == user.Bits);
});
if(i != Friends::FriendsList.end())
{
userInfo = *i;
}
userInfo.userId = user;
userInfo.online = Steam::Proxy::SteamFriends->GetFriendPersonaState(user) != 0;
userInfo.name = Steam::Proxy::SteamFriends->GetFriendPersonaName(user);
userInfo.statusName = Steam::Proxy::SteamFriends->GetFriendRichPresence(user, "iw4x_status");
//if (!userInfo.online) return;
if (i != Friends::FriendsList.end())
{
*i = userInfo;
}
else
{
Friends::FriendsList.push_back(userInfo);
}
qsort(Friends::FriendsList.data(), Friends::FriendsList.size(), sizeof(Friends::Friend), [](const void* first, const void* second)
{
const Friends::Friend* friend1 = static_cast<const Friends::Friend*>(first);
const Friends::Friend* friend2 = static_cast<const Friends::Friend*>(second);
std::string name1 = Utils::String::ToLower(Colors::Strip(friend1->name));
std::string name2 = Utils::String::ToLower(Colors::Strip(friend2->name));
return name1.compare(name2);
});
}
void Friends::UpdateFriends()
{
if (!Steam::Proxy::SteamFriends) return;
std::lock_guard<std::recursive_mutex> _(Friends::Mutex);
auto listCopy = Friends::FriendsList;
Friends::FriendsList.clear();
int count = Steam::Proxy::SteamFriends->GetFriendCount(4);
Friends::FriendsList.reserve(count);
for(int i = 0; i < count; ++i)
{
SteamID friendId = Steam::Proxy::SteamFriends->GetFriendByIndex(i, 4);
//if(!Steam::Proxy::SteamFriends->GetFriendPersonaState(friendId)) continue; // Offline
auto entry = std::find_if(listCopy.begin(), listCopy.end(), [friendId](Friends::Friend entry)
{
return (entry.userId.Bits == friendId.Bits);
});
if (entry != listCopy.end())
{
Friends::FriendsList.push_back(*entry);
}
Friends::UpdateUserInfo(friendId);
Steam::Proxy::SteamFriends->RequestFriendRichPresence(friendId);
}
}
unsigned int Friends::GetFriendCount()
{
std::lock_guard<std::recursive_mutex> _(Friends::Mutex);
return Friends::FriendsList.size();
}
const char* Friends::GetFriendText(unsigned int index, int column)
{
std::lock_guard<std::recursive_mutex> _(Friends::Mutex);
if (index >= Friends::FriendsList.size()) return "";
auto user = Friends::FriendsList[index];
switch(column)
{
case 0:
{
static char buffer[0x100];
std::string rankIcon = "rank_prestige9";
std::string result = "^\x02";
char size = 0x30;
// Icon size
result.append(&size, 1);
result.append(&size, 1);
// Icon name length
size = static_cast<char>(rankIcon.size());
result.append(&size, 1);
result.append(rankIcon);
result.append(" 70");
std::memcpy(buffer, result.data(), std::min(result.size() + 1, sizeof(buffer)));
buffer[sizeof(buffer) - 1] = 0;
return buffer;
}
case 1:
return Utils::String::VA("%s", user.name.data());
case 2:
return "Trickshot Isnipe server";
default:
break;
}
return "";
}
void Friends::SelectFriend(unsigned int index)
{
std::lock_guard<std::recursive_mutex> _(Friends::Mutex);
if (index >= Friends::FriendsList.size()) return;
Friends::CurrentFriend = index;
}
Friends::Friends()
{
Command::Add("setpres", [](Command::Params* params)
{
if (Steam::Proxy::SteamFriends && params->length() >= 3)
{
Logger::Print("%d\n", Steam::Proxy::SteamFriends->SetRichPresence(params->get(1), params->get(2)));
}
});
Command::Add("getpres", [](Command::Params* params)
{
if (Steam::Proxy::SteamFriends && params->length() >= 3)
{
std::string player = params->get(1);
SteamID playerId;
playerId.Bits = 0;
int count = Steam::Proxy::SteamFriends->GetFriendCount(0x04);
for (int i = 0; i < count; ++i)
{
SteamID _friend = Steam::Proxy::SteamFriends->GetFriendByIndex(i, 0x04);
std::string _name = Steam::Proxy::SteamFriends->GetFriendPersonaName(_friend);
//Logger::Print("%d: %s\n", i, _name.data());
if (_name == player)
{
playerId = _friend;
break;
}
}
if (playerId.Bits)
{
Steam::Proxy::SteamFriends->RequestFriendRichPresence(playerId);
const char* pres = Steam::Proxy::SteamFriends->GetFriendRichPresence(playerId, params->get(2));
Logger::Print("%s: %s\n", player.data(), pres);
}
}
});
// Callback to update user information
Steam::Proxy::RegisterCallback(336, [](void* data)
{
Friends::FriendRichPresenceUpdate* update = static_cast<Friends::FriendRichPresenceUpdate*>(data);
Friends::UpdateUserInfo(update->m_steamIDFriend);
});
// Persona state has changed
Steam::Proxy::RegisterCallback(304, [](void* data)
{
Friends::PersonaStateChange* state = static_cast<Friends::PersonaStateChange*>(data);
if(Steam::Proxy::SteamFriends) Steam::Proxy::SteamFriends->RequestFriendRichPresence(state->m_ulSteamID);
});
UIScript::Add("LoadFriends", [](UIScript::Token)
{
Friends::UpdateFriends();
});
UIFeeder::Add(39.0f, Friends::GetFriendCount, Friends::GetFriendText, Friends::SelectFriend);
}
Friends::~Friends()
{
Steam::Proxy::UnregisterCallback(304);
Steam::Proxy::UnregisterCallback(336);
if (Steam::Proxy::SteamFriends)
{
Steam::Proxy::SteamFriends->ClearRichPresence();
}
{
std::lock_guard<std::recursive_mutex> _(Friends::Mutex);
Friends::FriendsList.clear();
}
}
}

View File

@ -0,0 +1,50 @@
#pragma once
namespace Components
{
class Friends : public Component
{
public:
Friends();
~Friends();
#if defined(DEBUG) || defined(FORCE_UNIT_TESTS)
const char* getName() override { return "Friends"; };
#endif
static void UpdateFriends();
private:
struct FriendRichPresenceUpdate
{
SteamID m_steamIDFriend; // friend who's rich presence has changed
int32_t m_nAppID; // the appID of the game (should always be the current game)
};
struct PersonaStateChange
{
SteamID m_ulSteamID; // steamID of the friend who changed
int m_nChangeFlags; // what's changed
};
class Friend
{
public:
SteamID userId;
std::string name;
Network::Address server;
std::string statusName;
bool online;
};
static unsigned int CurrentFriend;
static std::recursive_mutex Mutex;
static std::vector<Friend> FriendsList;
static void UpdateUserInfo(SteamID user);
static unsigned int GetFriendCount();
static const char* GetFriendText(unsigned int index, int column);
static void SelectFriend(unsigned int index);
};
}

View File

@ -695,6 +695,7 @@ namespace Components
Menus::Add("ui_mp/stats_unlock.menu");
Menus::Add("ui_mp/security_increase_popmenu.menu");
Menus::Add("ui_mp/mod_download_popmenu.menu");
Menus::Add("ui_mp/popup_friends.menu");
}
Menus::~Menus()