src/game/SocialMgr.cpp

   1 /*
   2  * Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #include "SocialMgr.h"
  20 #include "Policies/SingletonImp.h"
  21 #include "Database/DatabaseEnv.h"
  22 #include "Opcodes.h"
  23 #include "WorldPacket.h"
  24 #include "WorldSession.h"
  25 #include "Player.h"
  26 #include "ObjectMgr.h"
  27 #include "World.h"
  28 #include "Util.h"
  29
  30 INSTANTIATE_SINGLETON_1( SocialMgr );
  31
  32 PlayerSocial::PlayerSocial()
  33 {
  34     m_playerGUID = 0;
  35 }
  36
  37 PlayerSocial::~PlayerSocial()
  38 {
  39     m_playerSocialMap.clear();
  40 }
  41
  42 uint32 PlayerSocial::GetNumberOfSocialsWithFlag(SocialFlag flag)
  43 {
  44     uint32 counter = 0;
  45     for(PlayerSocialMap::iterator itr = m_playerSocialMap.begin(); itr != m_playerSocialMap.end(); ++itr)
  46     {
  47         if(itr->second.Flags & flag)
  48             counter++;
  49     }
  50     return counter;
  51 }
  52
  53 bool PlayerSocial::AddToSocialList(uint32 friend_guid, bool ignore)
  54 {
  55     // check client limits
  56     if(ignore)
  57     {
  58         if(GetNumberOfSocialsWithFlag(SOCIAL_FLAG_IGNORED) >= SOCIALMGR_IGNORE_LIMIT)
  59             return false;
  60     }
  61     else
  62     {
  63         if(GetNumberOfSocialsWithFlag(SOCIAL_FLAG_FRIEND) >= SOCIALMGR_FRIEND_LIMIT)
  64             return false;
  65     }
  66
  67     uint32 flag = SOCIAL_FLAG_FRIEND;
  68     if(ignore)
  69         flag = SOCIAL_FLAG_IGNORED;
  70
  71     PlayerSocialMap::iterator itr = m_playerSocialMap.find(friend_guid);
  72     if(itr != m_playerSocialMap.end())
  73     {
  74         CharacterDatabase.PExecute("UPDATE character_social SET flags = (flags | %u) WHERE guid = '%u' AND friend = '%u'", flag, GetPlayerGUID(), friend_guid);
  75         m_playerSocialMap[friend_guid].Flags |= flag;
  76     }
  77     else
  78     {
  79         CharacterDatabase.PExecute("INSERT INTO character_social (guid, friend, flags) VALUES ('%u', '%u', '%u')", GetPlayerGUID(), friend_guid, flag);
  80         FriendInfo fi;
  81         fi.Flags |= flag;
  82         m_playerSocialMap[friend_guid] = fi;
  83     }
  84     return true;
  85 }
  86
  87 void PlayerSocial::RemoveFromSocialList(uint32 friend_guid, bool ignore)
  88 {
  89     PlayerSocialMap::iterator itr = m_playerSocialMap.find(friend_guid);
  90     if(itr == m_playerSocialMap.end())                      // not exist
  91         return;
  92
  93     uint32 flag = SOCIAL_FLAG_FRIEND;
  94     if(ignore)
  95         flag = SOCIAL_FLAG_IGNORED;
  96
  97     itr->second.Flags &= ~flag;
  98     if(itr->second.Flags == 0)
  99     {
 100         CharacterDatabase.PExecute("DELETE FROM character_social WHERE guid = '%u' AND friend = '%u'", GetPlayerGUID(), friend_guid);
 101         m_playerSocialMap.erase(itr);
 102     }
 103     else
 104     {
 105         CharacterDatabase.PExecute("UPDATE character_social SET flags = (flags & ~%u) WHERE guid = '%u' AND friend = '%u'", flag, GetPlayerGUID(), friend_guid);
 106     }
 107 }
 108
 109 void PlayerSocial::SetFriendNote(uint32 friend_guid, std::string note)
 110 {
 111     PlayerSocialMap::iterator itr = m_playerSocialMap.find(friend_guid);
 112     if(itr == m_playerSocialMap.end())                      // not exist
 113         return;
 114
 115     utf8truncate(note,48);                                  // DB and client size limitation
 116
 117     CharacterDatabase.escape_string(note);
 118     CharacterDatabase.PExecute("UPDATE character_social SET note = '%s' WHERE guid = '%u' AND friend = '%u'", note.c_str(), GetPlayerGUID(), friend_guid);
 119     m_playerSocialMap[friend_guid].Note = note;
 120 }
 121
 122 void PlayerSocial::SendSocialList()
 123 {
 124     Player *plr = objmgr.GetPlayer(GetPlayerGUID());
 125     if(!plr)
 126         return;
 127
 128     uint32 size = m_playerSocialMap.size();
 129
 130     WorldPacket data(SMSG_CONTACT_LIST, (4+4+size*25));     // just can guess size
 131     data << uint32(7);                                      // unk flag (0x1, 0x2, 0x4), 0x7 if it include ignore list
 132     data << uint32(size);                                   // friends count
 133
 134     for(PlayerSocialMap::iterator itr = m_playerSocialMap.begin(); itr != m_playerSocialMap.end(); ++itr)
 135     {
 136         sSocialMgr.GetFriendInfo(plr, itr->first, itr->second);
 137
 138         data << uint64(itr->first);                         // player guid
 139         data << uint32(itr->second.Flags);                  // player flag (0x1-friend?, 0x2-ignored?, 0x4-muted?)
 140         data << itr->second.Note;                           // string note
 141         if(itr->second.Flags & SOCIAL_FLAG_FRIEND)          // if IsFriend()
 142         {
 143             data << uint8(itr->second.Status);              // online/offline/etc?
 144             if(itr->second.Status)                          // if online
 145             {
 146                 data << uint32(itr->second.Area);           // player area
 147                 data << uint32(itr->second.Level);          // player level
 148                 data << uint32(itr->second.Class);          // player class
 149             }
 150         }
 151     }
 152
 153     plr->GetSession()->SendPacket(&data);
 154     sLog.outDebug("WORLD: Sent SMSG_CONTACT_LIST");
 155 }
 156
 157 bool PlayerSocial::HasFriend(uint32 friend_guid)
 158 {
 159     PlayerSocialMap::iterator itr = m_playerSocialMap.find(friend_guid);
 160     if(itr != m_playerSocialMap.end())
 161         return itr->second.Flags & SOCIAL_FLAG_FRIEND;
 162     return false;
 163 }
 164
 165 bool PlayerSocial::HasIgnore(uint32 ignore_guid)
 166 {
 167     PlayerSocialMap::iterator itr = m_playerSocialMap.find(ignore_guid);
 168     if(itr != m_playerSocialMap.end())
 169         return itr->second.Flags & SOCIAL_FLAG_IGNORED;
 170     return false;
 171 }
 172
 173 SocialMgr::SocialMgr()
 174 {
 175
 176 }
 177
 178 SocialMgr::~SocialMgr()
 179 {
 180
 181 }
 182
 183 void SocialMgr::GetFriendInfo(Player *player, uint32 friendGUID, FriendInfo &friendInfo)
 184 {
 185     if(!player)
 186         return;
 187
 188     Player *pFriend = ObjectAccessor::FindPlayer(friendGUID);
 189
 190     uint32 team = player->GetTeam();
 191     uint32 security = player->GetSession()->GetSecurity();
 192     bool allowTwoSideWhoList = sWorld.getConfig(CONFIG_ALLOW_TWO_SIDE_WHO_LIST);
 193     bool gmInWhoList = sWorld.getConfig(CONFIG_GM_IN_WHO_LIST) || security > SEC_PLAYER;
 194
 195     PlayerSocialMap::iterator itr = player->GetSocial()->m_playerSocialMap.find(friendGUID);
 196     if(itr != player->GetSocial()->m_playerSocialMap.end())
 197         friendInfo.Note = itr->second.Note;
 198
 199     // PLAYER see his team only and PLAYER can't see MODERATOR, GAME MASTER, ADMINISTRATOR characters
 200     // MODERATOR, GAME MASTER, ADMINISTRATOR can see all
 201     if( pFriend && pFriend->GetName() &&
 202         ( security > SEC_PLAYER ||
 203         ( pFriend->GetTeam() == team || allowTwoSideWhoList ) &&
 204         ( pFriend->GetSession()->GetSecurity() == SEC_PLAYER || gmInWhoList && pFriend->IsVisibleGloballyFor(player) )))
 205     {
 206         friendInfo.Status = FRIEND_STATUS_ONLINE;
 207         if(pFriend->isAFK())
 208             friendInfo.Status = FRIEND_STATUS_AFK;
 209         if(pFriend->isDND())
 210             friendInfo.Status = FRIEND_STATUS_DND;
 211         friendInfo.Area = pFriend->GetZoneId();
 212         friendInfo.Level = pFriend->getLevel();
 213         friendInfo.Class = pFriend->getClass();
 214     }
 215     else
 216     {
 217         friendInfo.Status = FRIEND_STATUS_OFFLINE;
 218         friendInfo.Area = 0;
 219         friendInfo.Level = 0;
 220         friendInfo.Class = 0;
 221     }
 222 }
 223
 224 void SocialMgr::MakeFriendStatusPacket(FriendsResult result, uint32 guid, WorldPacket *data)
 225 {
 226     data->Initialize(SMSG_FRIEND_STATUS, 5);
 227     *data << uint8(result);
 228     *data << uint64(guid);
 229 }
 230
 231 void SocialMgr::SendFriendStatus(Player *player, FriendsResult result, uint32 friend_guid, bool broadcast)
 232 {
 233     FriendInfo fi;
 234
 235     WorldPacket data;
 236     MakeFriendStatusPacket(result, friend_guid, &data);
 237     GetFriendInfo(player, friend_guid, fi);
 238     switch(result)
 239     {
 240         case FRIEND_ADDED_OFFLINE:
 241         case FRIEND_ADDED_ONLINE:
 242             data << fi.Note;
 243             break;
 244     }
 245
 246     switch(result)
 247     {
 248         case FRIEND_ADDED_ONLINE:
 249         case FRIEND_ONLINE:
 250             data << uint8(fi.Status);
 251             data << uint32(fi.Area);
 252             data << uint32(fi.Level);
 253             data << uint32(fi.Class);
 254             break;
 255     }
 256
 257     if(broadcast)
 258         BroadcastToFriendListers(player, &data);
 259     else
 260         player->GetSession()->SendPacket(&data);
 261 }
 262
 263 void SocialMgr::BroadcastToFriendListers(Player *player, WorldPacket *packet)
 264 {
 265     if(!player)
 266         return;
 267
 268     uint32 team     = player->GetTeam();
 269     uint32 security = player->GetSession()->GetSecurity();
 270     uint32 guid     = player->GetGUIDLow();
 271     bool gmInWhoList = sWorld.getConfig(CONFIG_GM_IN_WHO_LIST);
 272     bool allowTwoSideWhoList = sWorld.getConfig(CONFIG_ALLOW_TWO_SIDE_WHO_LIST);
 273
 274     for(SocialMap::iterator itr = m_socialMap.begin(); itr != m_socialMap.end(); ++itr)
 275     {
 276         PlayerSocialMap::const_iterator itr2 = itr->second.m_playerSocialMap.find(guid);
 277         if(itr2 != itr->second.m_playerSocialMap.end() && (itr2->second.Flags & SOCIAL_FLAG_FRIEND))
 278         {
 279             Player *pFriend = ObjectAccessor::FindPlayer(MAKE_NEW_GUID(itr->first, 0, HIGHGUID_PLAYER));
 280
 281             // PLAYER see his team only and PLAYER can't see MODERATOR, GAME MASTER, ADMINISTRATOR characters
 282             // MODERATOR, GAME MASTER, ADMINISTRATOR can see all
 283             if( pFriend && pFriend->IsInWorld() &&
 284                 ( pFriend->GetSession()->GetSecurity() > SEC_PLAYER ||
 285                 ( pFriend->GetTeam() == team || allowTwoSideWhoList ) &&
 286                 (security == SEC_PLAYER || gmInWhoList && player->IsVisibleGloballyFor(pFriend) )))
 287             {
 288                 pFriend->GetSession()->SendPacket(packet);
 289             }
 290         }
 291     }
 292 }
 293
 294 PlayerSocial *SocialMgr::LoadFromDB(QueryResult *result, uint32 guid)
 295 {
 296     PlayerSocial *social = &m_socialMap[guid];
 297     social->SetPlayerGUID(guid);
 298
 299     if(!result)
 300         return social;
 301
 302     uint32 friend_guid = 0;
 303     uint32 flags = 0;
 304     std::string note = "";
 305
 306     // used to speed up check below. Using GetNumberOfSocialsWithFlag will cause unneeded iteration
 307     uint32 friendCounter=0, ignoreCounter=0;
 308
 309     do
 310     {
 311         Field *fields  = result->Fetch();
 312
 313         friend_guid = fields[0].GetUInt32();
 314         flags = fields[1].GetUInt32();
 315         note = fields[2].GetCppString();
 316
 317         if((flags & SOCIAL_FLAG_IGNORED) && ignoreCounter >= SOCIALMGR_IGNORE_LIMIT)
 318             continue;
 319         if((flags & SOCIAL_FLAG_FRIEND) && friendCounter >= SOCIALMGR_FRIEND_LIMIT)
 320             continue;
 321
 322         social->m_playerSocialMap[friend_guid] = FriendInfo(flags, note);
 323
 324         if(flags & SOCIAL_FLAG_IGNORED)
 325             ignoreCounter++;
 326         else
 327             friendCounter++;
 328     }
 329     while( result->NextRow() );
 330     delete result;
 331     return social;
 332 }