1 files changed, 29 insertions, 9 deletions
diff --git a/src/server/game/DungeonFinding/LFGPlayerData.cpp b/src/server/game/DungeonFinding/LFGPlayerData.cpp
index 4a6e86ab966..6f030831342 100644
--- a/src/server/game/DungeonFinding/LFGPlayerData.cpp
+++ b/src/server/game/DungeonFinding/LFGPlayerData.cpp
@@ -17,8 +17,8 @@
 
 #include "LFGPlayerData.h"
 
-LfgPlayerData::LfgPlayerData():
-m_State(LFG_STATE_NONE), m_OldState(LFG_STATE_NONE), m_Roles(0), m_Comment("")
+LfgPlayerData::LfgPlayerData(): m_State(LFG_STATE_NONE), m_OldState(LFG_STATE_NONE),
+    m_Team(0), m_Group(0), m_Roles(0), m_Comment("")
 {}
 
 LfgPlayerData::~LfgPlayerData()
@@ -30,10 +30,10 @@ void LfgPlayerData::SetState(LfgState state)
     switch (state)
     {
         case LFG_STATE_NONE:
-        case LFG_STATE_DUNGEON:
         case LFG_STATE_FINISHED_DUNGEON:
+        case LFG_STATE_DUNGEON:
             m_OldState = state;
-                    // No break on purpose
+            // No break on purpose
         default:
             m_State = state;
     }
@@ -51,6 +51,16 @@ void LfgPlayerData::SetLockedDungeons(const LfgLockMap& lockStatus)
     m_LockedDungeons = lockStatus;
 }
 
+void LfgPlayerData::SetTeam(uint8 team)
+{
+    m_Team = team;
+}
+
+void LfgPlayerData::SetGroup(uint64 group)
+{
+    m_Group = group;
+}
+
 void LfgPlayerData::SetRoles(uint8 roles)
 {
     m_Roles = roles;
@@ -66,21 +76,31 @@ void LfgPlayerData::SetSelectedDungeons(const LfgDungeonSet& dungeons)
     m_SelectedDungeons = dungeons;
 }
 
-void LfgPlayerData::ClearSelectedDungeons()
+LfgState LfgPlayerData::GetState() const
 {
-    m_SelectedDungeons.clear();
+    return m_State;
 }
 
-LfgState LfgPlayerData::GetState() const
+LfgState LfgPlayerData::GetOldState() const
 {
-    return m_State;
+    return m_OldState;
 }
 
-const LfgLockMap & LfgPlayerData::GetLockedDungeons() const
+const LfgLockMap& LfgPlayerData::GetLockedDungeons() const
 {
     return m_LockedDungeons;
 }
 
+uint8 LfgPlayerData::GetTeam() const
+{
+    return m_Team;
+}
+
+uint64 LfgPlayerData::GetGroup() const
+{
+    return m_Group;
+}
+
 uint8 LfgPlayerData::GetRoles() const
 {
     return m_Roles;