DanhengServer-OpenSource/GameServer/Game/Lineup/LineupManager.cs

using EggLink.DanhengServer.Data;
using EggLink.DanhengServer.Database;
using EggLink.DanhengServer.Database.Lineup;
using EggLink.DanhengServer.Game.Player;
using EggLink.DanhengServer.Game.Scene;
using EggLink.DanhengServer.Server.Packet.Send.Lineup;
using EggLink.DanhengServer.Util;

namespace EggLink.DanhengServer.Game.Lineup
{
    public class LineupManager : BasePlayerManager
    {
        public LineupData LineupData { get; private set; }

        public LineupManager(PlayerInstance player) : base(player)
        {
            LineupData = DatabaseHelper.Instance!.GetInstanceOrCreateNew<LineupData>(player.Uid);
            foreach (var lineupInfo in LineupData.Lineups.Values)
            {
                lineupInfo.LineupData = LineupData;
                lineupInfo.AvatarData = player.AvatarManager!.AvatarData;
            }
        }

        #region Detail

        public LineupInfo? GetLineup(int lineupIndex)
        {
            LineupData.Lineups.TryGetValue(lineupIndex, out var lineup);
            return lineup;
        }

        public LineupInfo? GetExtraLineup(Proto.ExtraLineupType type)
        {
            var index = (int)type + 10;
            LineupData.Lineups.TryGetValue(index, out var lineup);
            return lineup;
        }

        public LineupInfo? GetCurLineup()
        {
            return GetLineup(LineupData.GetCurLineupIndex());
        }

        public List<AvatarSceneInfo> GetAvatarsFromTeam(int index)
        {
            var lineup = GetLineup(index);
            if (lineup == null)
            {
                return [];
            }

            var avatarList = new List<AvatarSceneInfo>();
            foreach (var avatar in lineup.BaseAvatars!)
            {
                Proto.AvatarType avatarType = Proto.AvatarType.AvatarFormalType;
                Database.Avatar.AvatarInfo? avatarInfo = null;
                if (avatar.SpecialAvatarId > 0)
                {
                    GameData.SpecialAvatarData.TryGetValue(avatar.SpecialAvatarId, out var specialAvatar);
                    if (specialAvatar == null) continue;
                    avatarType = Proto.AvatarType.AvatarTrialType;
                    avatarInfo = specialAvatar.ToAvatarData(Player.Uid);
                }
                else if (avatar.AssistUid > 0)
                {
                    var avatarStorage = DatabaseHelper.Instance?.GetInstance<Database.Avatar.AvatarData>(avatar.AssistUid);
                    avatarType = Proto.AvatarType.AvatarAssistType;
                    if (avatarStorage == null) continue;
                    foreach (var avatarData in avatarStorage.Avatars!)
                    {
                        if (avatarData.AvatarId == avatar.BaseAvatarId)
                        {
                            avatarInfo = avatarData;
                            break;
                        }
                    }
                }
                else
                {
                    avatarInfo = Player.AvatarManager!.GetAvatar(avatar.BaseAvatarId);
                }
                if (avatarInfo == null) continue;
                avatarList.Add(new AvatarSceneInfo(avatarInfo, avatarType, Player));
            }

            return avatarList;
        }

        public List<AvatarSceneInfo> GetAvatarsFromCurTeam()
        {
            return GetAvatarsFromTeam(LineupData.GetCurLineupIndex());
        }

        public List<LineupInfo> GetAllLineup()
        {
            var lineupList = new List<LineupInfo>();
            foreach (var lineupInfo in LineupData.Lineups.Values)
            {
                lineupList.Add(lineupInfo);
            }
            if (lineupList.Count < GameConstants.MAX_LINEUP_COUNT)
            {
                for (int i = lineupList.Count; i < GameConstants.MAX_LINEUP_COUNT; i++)
                {
                    var lineup = new LineupInfo()
                    {
                        Name = "",
                        LineupType = 0,
                        BaseAvatars = [],
                        LineupData = LineupData,
                        AvatarData = Player.AvatarManager!.AvatarData,
                    };
                    lineupList.Add(lineup);
                    LineupData.Lineups.Add(i, lineup);
                }
            }
            return lineupList;
        }

        #endregion

        #region Management

        public bool SetCurLineup(int lineupIndex)
        {
            if (lineupIndex < 0 || !LineupData.Lineups.ContainsKey(lineupIndex))
            {
                return false;
            }
            if (GetLineup(lineupIndex)!.BaseAvatars!.Count == 0)
            {
                return false;
            }
            LineupData.CurLineup = lineupIndex;
            LineupData.CurExtraLineup = -1;
            DatabaseHelper.Instance?.UpdateInstance(LineupData);

            Player.SceneInstance?.SyncLineup();
            Player.SendPacket(new PacketSyncLineupNotify(GetCurLineup()!));

            return true;
        }

        public void SetExtraLineup(Proto.ExtraLineupType type, List<int> baseAvatarIds)
        {
            if (type == Proto.ExtraLineupType.LineupNone)
            {
                // reset lineup
                LineupData.CurExtraLineup = -1;
                DatabaseHelper.Instance?.UpdateInstance(LineupData);
                return;
            }
            var index = (int)type + 10;

            // destroy old lineup
            LineupData.Lineups.Remove(index);

            // create new lineup
            var lineup = new LineupInfo()
            {
                Name = "",
                LineupType = (int)type,
                BaseAvatars = [],
                LineupData = LineupData,
                AvatarData = Player.AvatarManager!.AvatarData,
            };

            foreach (var avatarId in baseAvatarIds)
            {
                lineup.BaseAvatars!.Add(new() { BaseAvatarId = avatarId });
            }

            LineupData.Lineups.Add(index, lineup);
            LineupData.CurExtraLineup = index;
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
        }

        public void AddAvatar(int lineupIndex, int avatarId, bool sendPacket = true)
        {
            if (lineupIndex < 0)
            {
                return;
            }
            LineupData.Lineups.TryGetValue(lineupIndex, out LineupInfo? lineup);
            if (lineup == null)
            {
                lineup = new()
                {
                    Name = "",
                    LineupType = 0,
                    BaseAvatars = [new() { BaseAvatarId = avatarId }],
                    LineupData = LineupData,
                    AvatarData = Player.AvatarManager!.AvatarData,
                };
                LineupData.Lineups.Add(lineupIndex, lineup);
            } else
            {
                if (lineup.BaseAvatars!.Count >= 4)
                {
                    return;
                }
                lineup.BaseAvatars?.Add(new() { BaseAvatarId = avatarId });
                LineupData.Lineups[lineupIndex] = lineup;
            }
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
            if (sendPacket)
            {
                if (lineupIndex == LineupData.GetCurLineupIndex())
                {
                    Player.SceneInstance?.SyncLineup();
                }
                Player.SendPacket(new PacketSyncLineupNotify(lineup));
            }
        }

        public void AddAvatarToCurTeam(int avatarId, bool sendPacket = true)
        {
            AddAvatar(LineupData.GetCurLineupIndex(), avatarId, sendPacket);
        }

        public void AddSpecialAvatarToCurTeam(int specialAvatarId, bool sendPacket = true)
        {
            LineupData.Lineups.TryGetValue(LineupData.GetCurLineupIndex(), out LineupInfo? lineup);
            GameData.SpecialAvatarData.TryGetValue(specialAvatarId, out var specialAvatar);
            if (specialAvatar == null)
            {
                return;
            }
            if (lineup == null)
            {
                lineup = new()
                {
                    Name = "",
                    LineupType = 0,
                    BaseAvatars = [new() { BaseAvatarId = specialAvatar.AvatarID, SpecialAvatarId = specialAvatarId }],
                    LineupData = LineupData,
                    AvatarData = Player.AvatarManager!.AvatarData,
                };
                LineupData.Lineups.Add(LineupData.GetCurLineupIndex(), lineup);
            } else
            {
                lineup.BaseAvatars?.Add(new() { BaseAvatarId = specialAvatar.AvatarID, SpecialAvatarId = specialAvatarId });
                LineupData.Lineups[LineupData.GetCurLineupIndex()] = lineup;
            }
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
            if (sendPacket)
            {
                Player.SceneInstance?.SyncLineup();
                Player.SendPacket(new PacketSyncLineupNotify(lineup));
            }
        }

        public void RemoveAvatar(int lineupIndex, int avatarId)
        {
            if (lineupIndex < 0)
            {
                return;
            }
            LineupData.Lineups.TryGetValue(lineupIndex, out LineupInfo? lineup);
            if (lineup == null)
            {
                return;
            }
            lineup.BaseAvatars?.RemoveAll(avatar => avatar.BaseAvatarId == avatarId);
            LineupData.Lineups[lineupIndex] = lineup;
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
            if (lineupIndex == LineupData.GetCurLineupIndex())
            {
                Player.SceneInstance?.SyncLineup();
            }
            Player.SendPacket(new PacketSyncLineupNotify(lineup));
        }

        public void RemoveAvatarFromCurTeam(int avatarId)
        {
            RemoveAvatar(LineupData.GetCurLineupIndex(), avatarId);
        }

        public void RemoveSpecialAvatarFromCurTeam(int specialAvatarId)
        {
            LineupData.Lineups.TryGetValue(LineupData.GetCurLineupIndex(), out LineupInfo? lineup);
            if (lineup == null)
            {
                return;
            }
            lineup.BaseAvatars?.RemoveAll(avatar => avatar.SpecialAvatarId == specialAvatarId);
            LineupData.Lineups[LineupData.GetCurLineupIndex()] = lineup;
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
            Player.SceneInstance?.SyncLineup();
            Player.SendPacket(new PacketSyncLineupNotify(lineup));
        }

        public void ReplaceLineup(Proto.ReplaceLineupCsReq req)
        {
            if (req.ExtraLineupType != Proto.ExtraLineupType.LineupNone)
            {
                LineupData.CurExtraLineup = (int)req.ExtraLineupType + 10;
                if (!LineupData.Lineups.ContainsKey(LineupData.CurExtraLineup))
                {
                    SetExtraLineup(req.ExtraLineupType, []);
                }
            }

            LineupInfo lineup;
            if (LineupData.CurExtraLineup != -1)
            {
                lineup = LineupData.Lineups[LineupData.CurExtraLineup];  // Extra lineup
            } else if (req.Index < 0 || !LineupData.Lineups.ContainsKey((int)req.Index))
            {
                return;
            } else
            {
                lineup = LineupData.Lineups[(int)req.Index];
            }
            lineup.BaseAvatars = [];
            var index = lineup.LineupType == 0 ? (int)req.Index : LineupData.GetCurLineupIndex();
            foreach (var avatar in req.LineupSlotList)
            {
                AddAvatar(index, (int)avatar.Id, false);
            }
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
            if (index == LineupData.GetCurLineupIndex())
            {
                Player.SceneInstance?.SyncLineup();
            }
            Player.SendPacket(new PacketSyncLineupNotify(lineup));
        }

        public void CostMp(int count)
        {
            var curLineup = GetCurLineup()!;
            curLineup.Mp -= count;
            curLineup.Mp = Math.Min(Math.Max(0, curLineup.Mp), 5);
            DatabaseHelper.Instance?.UpdateInstance(LineupData);

            Player.SendPacket(new PacketSceneCastSkillMpUpdateScNotify(1, curLineup.Mp));
        }

        public void GainMp(int count, bool sendPacket = true)
        {
            count = Math.Min(Math.Max(0, count), 2);
            var curLineup = GetCurLineup()!;
            curLineup.Mp += count;
            curLineup.Mp = Math.Min(Math.Max(0, curLineup.Mp), 5);
            DatabaseHelper.Instance?.UpdateInstance(LineupData);
            if (sendPacket)
            {
                Player.SendPacket(new PacketSyncLineupNotify(GetCurLineup()!, Proto.SyncLineupReason.SyncReasonMpAddPropHit));
            }
        }

        #endregion
    }
}