First Commit

For Azeroth!

src/server/game/Entities/Player/SocialMgr.cpp

@@ -0,0 +1,343 @@
+/*
+ * Copyright (C) 
+ * Copyright (C) 
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "SocialMgr.h"
+
+#include "DatabaseEnv.h"
+#include "Opcodes.h"
+#include "WorldPacket.h"
+#include "Player.h"
+#include "ObjectMgr.h"
+#include "World.h"
+#include "Util.h"
+#include "AccountMgr.h"
+
+PlayerSocial::PlayerSocial()
+{
+    m_playerGUID = 0;
+}
+
+PlayerSocial::~PlayerSocial()
+{
+    m_playerSocialMap.clear();
+}
+
+uint32 PlayerSocial::GetNumberOfSocialsWithFlag(SocialFlag flag) const
+{
+    uint32 counter = 0;
+    for (PlayerSocialMap::const_iterator itr = m_playerSocialMap.begin(); itr != m_playerSocialMap.end(); ++itr)
+        if (itr->second.Flags & flag)
+            ++counter;
+
+    return counter;
+}
+
+bool PlayerSocial::AddToSocialList(uint32 friendGuid, bool ignore)
+{
+    // check client limits
+    if (ignore)
+    {
+        if (GetNumberOfSocialsWithFlag(SOCIAL_FLAG_IGNORED) >= SOCIALMGR_IGNORE_LIMIT)
+            return false;
+    }
+    else
+    {
+        if (GetNumberOfSocialsWithFlag(SOCIAL_FLAG_FRIEND) >= SOCIALMGR_FRIEND_LIMIT)
+            return false;
+    }
+
+    uint8 flag = SOCIAL_FLAG_FRIEND;
+    if (ignore)
+        flag = SOCIAL_FLAG_IGNORED;
+
+    PlayerSocialMap::const_iterator itr = m_playerSocialMap.find(friendGuid);
+    if (itr != m_playerSocialMap.end())
+    {
+        PreparedStatement* stmt = CharacterDatabase.GetPreparedStatement(CHAR_UPD_ADD_CHARACTER_SOCIAL_FLAGS);
+
+        stmt->setUInt8(0, flag);
+        stmt->setUInt32(1, GetPlayerGUID());
+        stmt->setUInt32(2, friendGuid);
+
+        CharacterDatabase.Execute(stmt);
+
+        m_playerSocialMap[friendGuid].Flags |= flag;
+    }
+    else
+    {
+        PreparedStatement* stmt = CharacterDatabase.GetPreparedStatement(CHAR_INS_CHARACTER_SOCIAL);
+
+        stmt->setUInt32(0, GetPlayerGUID());
+        stmt->setUInt32(1, friendGuid);
+        stmt->setUInt8(2, flag);
+
+        CharacterDatabase.Execute(stmt);
+
+        FriendInfo fi;
+        fi.Flags |= flag;
+        m_playerSocialMap[friendGuid] = fi;
+    }
+    return true;
+}
+
+void PlayerSocial::RemoveFromSocialList(uint32 friendGuid, bool ignore)
+{
+    PlayerSocialMap::iterator itr = m_playerSocialMap.find(friendGuid);
+    if (itr == m_playerSocialMap.end())                     // not exist
+        return;
+
+    uint8 flag = SOCIAL_FLAG_FRIEND;
+    if (ignore)
+        flag = SOCIAL_FLAG_IGNORED;
+
+    itr->second.Flags &= ~flag;
+    if (itr->second.Flags == 0)
+    {
+        PreparedStatement* stmt = CharacterDatabase.GetPreparedStatement(CHAR_DEL_CHARACTER_SOCIAL);
+
+        stmt->setUInt32(0, GetPlayerGUID());
+        stmt->setUInt32(1, friendGuid);
+
+        CharacterDatabase.Execute(stmt);
+
+        m_playerSocialMap.erase(itr);
+    }
+    else
+    {
+        PreparedStatement* stmt = CharacterDatabase.GetPreparedStatement(CHAR_UPD_REM_CHARACTER_SOCIAL_FLAGS);
+
+        stmt->setUInt8(0, flag);
+        stmt->setUInt32(1, GetPlayerGUID());
+        stmt->setUInt32(2, friendGuid);
+
+        CharacterDatabase.Execute(stmt);
+    }
+}
+
+void PlayerSocial::SetFriendNote(uint32 friendGuid, std::string note)
+{
+    PlayerSocialMap::const_iterator itr = m_playerSocialMap.find(friendGuid);
+    if (itr == m_playerSocialMap.end())                     // not exist
+        return;
+
+    utf8truncate(note, 48);                                  // DB and client size limitation
+
+    PreparedStatement* stmt = CharacterDatabase.GetPreparedStatement(CHAR_UPD_CHARACTER_SOCIAL_NOTE);
+
+    stmt->setString(0, note);
+    stmt->setUInt32(1, GetPlayerGUID());
+    stmt->setUInt32(2, friendGuid);
+
+    CharacterDatabase.Execute(stmt);
+
+    m_playerSocialMap[friendGuid].Note = note;
+}
+
+void PlayerSocial::SendSocialList(Player* player)
+{
+    if (!player)
+        return;
+
+    uint32 size = m_playerSocialMap.size();
+
+    WorldPacket data(SMSG_CONTACT_LIST, (4+4+size*25));     // just can guess size
+    data << uint32(7);                                      // 0x1 = Friendlist update. 0x2 = Ignorelist update. 0x4 = Mutelist update.
+    data << uint32(size);                                   // friends count
+
+    for (PlayerSocialMap::iterator itr = m_playerSocialMap.begin(); itr != m_playerSocialMap.end(); ++itr)
+    {
+        sSocialMgr->GetFriendInfo(player, itr->first, itr->second);
+
+        data << uint64(itr->first);                         // player guid
+        data << uint32(itr->second.Flags);                  // player flag (0x1 = Friend, 0x2 = Ignored, 0x4 = Muted)
+        data << itr->second.Note;                           // string note
+        if (itr->second.Flags & SOCIAL_FLAG_FRIEND)         // if IsFriend()
+        {
+            data << uint8(itr->second.Status);              // online/offline/etc?
+            if (itr->second.Status)                         // if online
+            {
+                data << uint32(itr->second.Area);           // player area
+                data << uint32(itr->second.Level);          // player level
+                data << uint32(itr->second.Class);          // player class
+            }
+        }
+    }
+
+    player->GetSession()->SendPacket(&data);
+    ;//sLog->outDebug(LOG_FILTER_NETWORKIO, "WORLD: Sent SMSG_CONTACT_LIST");
+}
+
+bool PlayerSocial::HasFriend(uint32 friend_guid) const
+{
+    PlayerSocialMap::const_iterator itr = m_playerSocialMap.find(friend_guid);
+    if (itr != m_playerSocialMap.end())
+        return itr->second.Flags & SOCIAL_FLAG_FRIEND;
+    return false;
+}
+
+bool PlayerSocial::HasIgnore(uint32 ignore_guid) const
+{
+    PlayerSocialMap::const_iterator itr = m_playerSocialMap.find(ignore_guid);
+    if (itr != m_playerSocialMap.end())
+        return itr->second.Flags & SOCIAL_FLAG_IGNORED;
+    return false;
+}
+
+SocialMgr::SocialMgr()
+{
+}
+
+SocialMgr::~SocialMgr()
+{
+}
+
+void SocialMgr::GetFriendInfo(Player* player, uint32 friendGUID, FriendInfo &friendInfo)
+{
+    if (!player)
+        return;
+
+    friendInfo.Status = FRIEND_STATUS_OFFLINE;
+    friendInfo.Area = 0;
+    friendInfo.Level = 0;
+    friendInfo.Class = 0;
+
+    Player* pFriend = ObjectAccessor::FindPlayerInOrOutOfWorld(friendGUID);
+    if (!pFriend || AccountMgr::IsGMAccount(pFriend->GetSession()->GetSecurity()))
+        return;
+
+    TeamId teamId = player->GetTeamId();
+    AccountTypes security = player->GetSession()->GetSecurity();
+    AccountTypes gmLevelInWhoList = AccountTypes(sWorld->getIntConfig(CONFIG_GM_LEVEL_IN_WHO_LIST));
+
+    PlayerSocialMap::iterator itr = player->GetSocial()->m_playerSocialMap.find(friendGUID);
+    if (itr != player->GetSocial()->m_playerSocialMap.end())
+        friendInfo.Note = itr->second.Note;
+
+    // PLAYER see his team only and PLAYER can't see MODERATOR, GAME MASTER, ADMINISTRATOR characters
+    // MODERATOR, GAME MASTER, ADMINISTRATOR can see all
+    if (pFriend && (!AccountMgr::IsPlayerAccount(security) || (pFriend->GetTeamId() == teamId && pFriend->GetSession()->GetSecurity() <= gmLevelInWhoList)) && pFriend->IsVisibleGloballyFor(player))
+    {
+        friendInfo.Status = FRIEND_STATUS_ONLINE;
+        if (pFriend->isAFK())
+            friendInfo.Status = FRIEND_STATUS_AFK;
+        if (pFriend->isDND())
+            friendInfo.Status = FRIEND_STATUS_DND;
+        friendInfo.Area = pFriend->GetZoneId();
+        friendInfo.Level = pFriend->getLevel();
+        friendInfo.Class = pFriend->getClass();
+    }
+}
+
+void SocialMgr::MakeFriendStatusPacket(FriendsResult result, uint32 guid, WorldPacket* data)
+{
+    data->Initialize(SMSG_FRIEND_STATUS, 9);
+    *data << uint8(result);
+    *data << uint64(guid);
+}
+
+void SocialMgr::SendFriendStatus(Player* player, FriendsResult result, uint32 friendGuid, bool broadcast)
+{
+    FriendInfo fi;
+
+    WorldPacket data;
+    MakeFriendStatusPacket(result, friendGuid, &data);
+    GetFriendInfo(player, friendGuid, fi);
+    switch (result)
+    {
+        case FRIEND_ADDED_OFFLINE:
+        case FRIEND_ADDED_ONLINE:
+            data << fi.Note;
+            break;
+        default:
+            break;
+    }
+
+    switch (result)
+    {
+        case FRIEND_ADDED_ONLINE:
+        case FRIEND_ONLINE:
+            data << uint8(fi.Status);
+            data << uint32(fi.Area);
+            data << uint32(fi.Level);
+            data << uint32(fi.Class);
+            break;
+        default:
+            break;
+    }
+
+    if (broadcast)
+        BroadcastToFriendListers(player, &data);
+    else
+        player->GetSession()->SendPacket(&data);
+}
+
+void SocialMgr::BroadcastToFriendListers(Player* player, WorldPacket* packet)
+{
+    if (!player)
+        return;
+
+    TeamId teamId = player->GetTeamId();
+    AccountTypes security = player->GetSession()->GetSecurity();
+    uint32 guid = player->GetGUIDLow();
+    AccountTypes gmLevelInWhoList = AccountTypes(sWorld->getIntConfig(CONFIG_GM_LEVEL_IN_WHO_LIST));
+
+    for (SocialMap::const_iterator itr = m_socialMap.begin(); itr != m_socialMap.end(); ++itr)
+    {
+        PlayerSocialMap::const_iterator itr2 = itr->second.m_playerSocialMap.find(guid);
+        if (itr2 != itr->second.m_playerSocialMap.end() && (itr2->second.Flags & SOCIAL_FLAG_FRIEND))
+        {
+            Player* pFriend = ObjectAccessor::FindPlayer(MAKE_NEW_GUID(itr->first, 0, HIGHGUID_PLAYER));
+
+            // PLAYER see his team only and PLAYER can't see MODERATOR, GAME MASTER, ADMINISTRATOR characters
+            // MODERATOR, GAME MASTER, ADMINISTRATOR can see all
+            if (pFriend && (!AccountMgr::IsPlayerAccount(pFriend->GetSession()->GetSecurity()) || (pFriend->GetTeamId() == teamId && security <= gmLevelInWhoList)) && player->IsVisibleGloballyFor(pFriend))
+                pFriend->GetSession()->SendPacket(packet);
+        }
+    }
+}
+
+PlayerSocial* SocialMgr::LoadFromDB(PreparedQueryResult result, uint32 guid)
+{
+    PlayerSocial *social = &m_socialMap[guid];
+    social->SetPlayerGUID(guid);
+
+    if (!result)
+        return social;
+
+    uint32 friendGuid = 0;
+    uint8 flags = 0;
+    std::string note = "";
+
+    do
+    {
+        Field* fields = result->Fetch();
+
+        friendGuid = fields[0].GetUInt32();
+        flags = fields[1].GetUInt8();
+        note = fields[2].GetString();
+
+        social->m_playerSocialMap[friendGuid] = FriendInfo(flags, note);
+
+        // client's friends list and ignore list limit
+        if (social->m_playerSocialMap.size() >= (SOCIALMGR_FRIEND_LIMIT + SOCIALMGR_IGNORE_LIMIT))
+            break;
+    }
+    while (result->NextRow());
+
+    return social;
+}

src/server/game/Entities/Player/SocialMgr.h

@@ -0,0 +1,150 @@
+/*
+ * Copyright (C) 
+ * Copyright (C) 
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef __TRINITY_SOCIALMGR_H
+#define __TRINITY_SOCIALMGR_H
+
+#include <ace/Singleton.h>
+#include "DatabaseEnv.h"
+#include "Common.h"
+
+class SocialMgr;
+class PlayerSocial;
+class Player;
+class WorldPacket;
+
+enum FriendStatus
+{
+    FRIEND_STATUS_OFFLINE   = 0x00,
+    FRIEND_STATUS_ONLINE    = 0x01,
+    FRIEND_STATUS_AFK       = 0x02,
+    FRIEND_STATUS_DND       = 0x04,
+    FRIEND_STATUS_RAF       = 0x08
+};
+
+enum SocialFlag
+{
+    SOCIAL_FLAG_FRIEND      = 0x01,
+    SOCIAL_FLAG_IGNORED     = 0x02,
+    SOCIAL_FLAG_MUTED       = 0x04,                          // guessed
+    SOCIAL_FLAG_UNK         = 0x08                           // Unknown - does not appear to be RaF
+};
+
+struct FriendInfo
+{
+    FriendStatus Status;
+    uint8 Flags;
+    uint32 Area;
+    uint8 Level;
+    uint8 Class;
+    std::string Note;
+
+    FriendInfo() : Status(FRIEND_STATUS_OFFLINE), Flags(0), Area(0), Level(0), Class(0), Note()
+    { }
+
+    FriendInfo(uint8 flags, std::string const& note) : Status(FRIEND_STATUS_OFFLINE), Flags(flags), Area(0), Level(0), Class(0), Note(note)
+    { }
+};
+
+typedef std::map<uint32, FriendInfo> PlayerSocialMap;
+typedef std::map<uint32, PlayerSocial> SocialMap;
+
+/// Results of friend related commands
+enum FriendsResult
+{
+    FRIEND_DB_ERROR         = 0x00,
+    FRIEND_LIST_FULL        = 0x01,
+    FRIEND_ONLINE           = 0x02,
+    FRIEND_OFFLINE          = 0x03,
+    FRIEND_NOT_FOUND        = 0x04,
+    FRIEND_REMOVED          = 0x05,
+    FRIEND_ADDED_ONLINE     = 0x06,
+    FRIEND_ADDED_OFFLINE    = 0x07,
+    FRIEND_ALREADY          = 0x08,
+    FRIEND_SELF             = 0x09,
+    FRIEND_ENEMY            = 0x0A,
+    FRIEND_IGNORE_FULL      = 0x0B,
+    FRIEND_IGNORE_SELF      = 0x0C,
+    FRIEND_IGNORE_NOT_FOUND = 0x0D,
+    FRIEND_IGNORE_ALREADY   = 0x0E,
+    FRIEND_IGNORE_ADDED     = 0x0F,
+    FRIEND_IGNORE_REMOVED   = 0x10,
+    FRIEND_IGNORE_AMBIGUOUS = 0x11,                         // That name is ambiguous, type more of the player's server name
+    FRIEND_MUTE_FULL        = 0x12,
+    FRIEND_MUTE_SELF        = 0x13,
+    FRIEND_MUTE_NOT_FOUND   = 0x14,
+    FRIEND_MUTE_ALREADY     = 0x15,
+    FRIEND_MUTE_ADDED       = 0x16,
+    FRIEND_MUTE_REMOVED     = 0x17,
+    FRIEND_MUTE_AMBIGUOUS   = 0x18,                         // That name is ambiguous, type more of the player's server name
+    FRIEND_UNK7             = 0x19,                         // no message at client
+    FRIEND_UNKNOWN          = 0x1A                          // Unknown friend response from server
+};
+
+#define SOCIALMGR_FRIEND_LIMIT  50
+#define SOCIALMGR_IGNORE_LIMIT  50
+
+class PlayerSocial
+{
+    friend class SocialMgr;
+    public:
+        PlayerSocial();
+        ~PlayerSocial();
+        // adding/removing
+        bool AddToSocialList(uint32 friend_guid, bool ignore);
+        void RemoveFromSocialList(uint32 friend_guid, bool ignore);
+        void SetFriendNote(uint32 friendGuid, std::string note);
+        // Packet send's
+        void SendSocialList(Player* player);
+        // Misc
+        bool HasFriend(uint32 friend_guid) const;
+        bool HasIgnore(uint32 ignore_guid) const;
+        uint32 GetPlayerGUID() const { return m_playerGUID; }
+        void SetPlayerGUID(uint32 guid) { m_playerGUID = guid; }
+        uint32 GetNumberOfSocialsWithFlag(SocialFlag flag) const;
+    private:
+        PlayerSocialMap m_playerSocialMap;
+        uint32 m_playerGUID;
+};
+
+class SocialMgr
+{
+    friend class ACE_Singleton<SocialMgr, ACE_Null_Mutex>;
+
+    private:
+        SocialMgr();
+        ~SocialMgr();
+
+    public:
+        // Misc
+        void RemovePlayerSocial(uint32 guid) { m_socialMap.erase(guid); }
+
+        void GetFriendInfo(Player* player, uint32 friendGUID, FriendInfo &friendInfo);
+        // Packet management
+        void MakeFriendStatusPacket(FriendsResult result, uint32 friend_guid, WorldPacket* data);
+        void SendFriendStatus(Player* player, FriendsResult result, uint32 friend_guid, bool broadcast);
+        void BroadcastToFriendListers(Player* player, WorldPacket* packet);
+        // Loading
+        PlayerSocial *LoadFromDB(PreparedQueryResult result, uint32 guid);
+    private:
+        SocialMap m_socialMap;
+};
+
+#define sSocialMgr ACE_Singleton<SocialMgr, ACE_Null_Mutex>::instance()
+#endif
+